fix: word count calculation

2025-11-12 19:13:04 -05:00
parent 3cff965393
commit f9f23f2d3f
2 changed files with 5 additions and 3 deletions
--- 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