initial commit

pkg/epub/book.go

@@ -0,0 +1,73 @@
+package epub
+
+import (
+	"archive/zip"
+	"encoding/xml"
+	"fmt"
+	"io"
+	"path"
+)
+
+// Book epub book
+type Book struct {
+	Ncx       Ncx       `json:"ncx"`
+	Opf       Opf       `json:"opf"`
+	Container Container `json:"-"`
+	Mimetype  string    `json:"-"`
+
+	fd *zip.ReadCloser
+}
+
+// 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 io.ReadAll(fd)
+
+}
+
+func (p *Book) open(n string) (io.ReadCloser, error) {
+	for _, f := range p.fd.File {
+		if f.Name == n {
+			return f.Open()
+		}
+	}
+	return nil, fmt.Errorf("file %s not exist", n)
+}