book.go

package epub

import (
	"archive/zip"
	"encoding/xml"
	"fmt"
	"io"
	"io/ioutil"
	"os"
	"path"
	"path/filepath"
)

// Book epub book
type Book struct {
	Ncx        Ncx        `json:"ncx"`
	Opf        Opf        `json:"opf"`
	Container  Container  `json:"-"`
	Mimetype   string     `json:"-"`
	Encryption Encryption `json:"-"`

	fd        *zip.ReadCloser
	directory string
}

//Open open resource file
func (p *Book) Open(n string) (io.ReadCloser, error) {
	return p.open(p.filename(n))
}

//Files list resource files
func (p *Book) Files() []string {
	var fns []string
	for _, f := range p.fd.File {
		fns = append(fns, f.Name)
	}
	return fns
}

//Close close file reader
func (p *Book) Close() {
	p.fd.Close()
}

//-----------------------------------------------------------------------------
func (p *Book) filename(n string) string {
	return path.Join(path.Dir(p.Container.Rootfile.Path), n)
}

func (p *Book) readXML(n string, v interface{}) error {
	fd, err := p.open(n)
	if err != nil {
		return nil
	}
	defer fd.Close()
	dec := xml.NewDecoder(fd)
	return dec.Decode(v)
}

func (p *Book) readBytes(n string) ([]byte, error) {
	fd, err := p.open(n)
	if err != nil {
		return nil, nil
	}
	defer fd.Close()

	return ioutil.ReadAll(fd)

}

func (p *Book) open(n string) (io.ReadCloser, error) {
	if p.directory != "" {
		filename := path.Join(path.Dir(p.directory+string(filepath.Separator)), n)
		return os.Open(filename)
	}

	for _, f := range p.fd.File {
		if f.Name == n {
			return f.Open()
		}
	}
	return nil, fmt.Errorf("file %s not exist", n)
}

// ZipReader return the internal file descriptor
func (p *Book) ZipReader() *zip.ReadCloser {
	return p.fd
}

// GetSMIL parse and return SMIL structure
func (p *Book) GetSMIL(ressouce string) SMIL {
	var smil SMIL

	p.readXML(ressouce, &smil)

	return smil
}