mirror of
https://codeberg.org/scip/epuppy.git
synced 2025-12-16 20:11:00 +01:00
Add dump feature, enhance code quality (#1)
This commit is contained in:
@@ -5,6 +5,7 @@ import (
|
||||
"encoding/xml"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"path"
|
||||
)
|
||||
|
||||
@@ -34,8 +35,8 @@ func (p *Book) Files() []string {
|
||||
}
|
||||
|
||||
// Close close file reader
|
||||
func (p *Book) Close() {
|
||||
p.fd.Close()
|
||||
func (p *Book) Close() error {
|
||||
return p.fd.Close()
|
||||
}
|
||||
|
||||
// -----------------------------------------------------------------------------
|
||||
@@ -48,7 +49,13 @@ func (p *Book) readXML(n string, v interface{}) error {
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
defer fd.Close()
|
||||
|
||||
defer func() {
|
||||
if err := fd.Close(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
dec := xml.NewDecoder(fd)
|
||||
|
||||
if err := dec.Decode(v); err != nil {
|
||||
@@ -63,7 +70,11 @@ func (p *Book) readBytes(n string) ([]byte, error) {
|
||||
if err != nil {
|
||||
return nil, nil
|
||||
}
|
||||
defer fd.Close()
|
||||
defer func() {
|
||||
if err := fd.Close(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
data, err := io.ReadAll(fd)
|
||||
if err != nil {
|
||||
|
||||
@@ -42,7 +42,6 @@ func (c *Content) String(content []byte) error {
|
||||
txt = cleanmarkup.ReplaceAllString(txt, "")
|
||||
txt = cleanentities.ReplaceAllString(txt, " ")
|
||||
txt = cleancomments.ReplaceAllString(txt, "")
|
||||
|
||||
txt = strings.TrimSpace(txt)
|
||||
|
||||
c.Body = cleanspace.ReplaceAllString(txt, "")
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
package epub
|
||||
|
||||
import (
|
||||
"log"
|
||||
"testing"
|
||||
)
|
||||
|
||||
@@ -9,7 +10,13 @@ func TestEpub(t *testing.T) {
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer bk.Close()
|
||||
|
||||
defer func() {
|
||||
if err := bk.Close(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
}
|
||||
|
||||
func open(t *testing.T, f string) (*Book, error) {
|
||||
@@ -17,7 +24,12 @@ func open(t *testing.T, f string) (*Book, error) {
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer bk.Close()
|
||||
|
||||
defer func() {
|
||||
if err := bk.Close(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
t.Logf("files: %+v", bk.Files())
|
||||
t.Logf("book: %+v", bk)
|
||||
|
||||
@@ -2,6 +2,7 @@ package epub
|
||||
|
||||
import (
|
||||
"archive/zip"
|
||||
"log"
|
||||
"strings"
|
||||
)
|
||||
|
||||
@@ -12,7 +13,11 @@ func Open(fn string) (*Book, error) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
defer fd.Close()
|
||||
defer func() {
|
||||
if err := fd.Close(); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
bk := Book{fd: fd}
|
||||
mt, err := bk.readBytes("mimetype")
|
||||
|
||||
Reference in New Issue
Block a user