From f9f23f2d3f4625c1f293d10d7f53342016f66a34 Mon Sep 17 00:00:00 2001 From: Evan Reichard Date: Wed, 12 Nov 2025 19:13:04 -0500 Subject: [PATCH] fix: word count calculation --- metadata/epub.go | 6 ++++-- metadata/metadata.go | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/metadata/epub.go b/metadata/epub.go index 19cda2e..ed0eeb7 100644 --- a/metadata/epub.go +++ b/metadata/epub.go @@ -53,10 +53,12 @@ func countEPUBWords(filepath string) (int64, error) { rf := rc.Rootfiles[0] var completeCount int64 - for _, item := range rf.Spine.Itemrefs { + for _, item := range rf.Itemrefs { f, _ := item.Open() doc, _ := goquery.NewDocumentFromReader(f) - completeCount = completeCount + int64(len(strings.Fields(doc.Text()))) + doc.Find("script, style, noscript, iframe").Remove() + words := len(strings.Fields(doc.Text())) + completeCount = completeCount + int64(words) } return completeCount, nil diff --git a/metadata/metadata.go b/metadata/metadata.go index d3444c2..5ed7d1a 100644 --- a/metadata/metadata.go +++ b/metadata/metadata.go @@ -87,7 +87,7 @@ func GetWordCount(filepath string) (*int64, error) { } return &totalWords, nil } else { - return nil, fmt.Errorf("Invalid extension: %s", fileExtension) + return nil, fmt.Errorf("invalid extension: %s", fileExtension) } }