refactor: split reader architecture and chrome handling

2026-05-31 21:47:54 +08:00
parent ea21c6a831
commit 44202357c0
80 changed files with 10635 additions and 8522 deletions
@@ -0,0 +1,84 @@
+import Foundation
+
+extension RDEPUBReaderController {
+    func resolvedCurrentTableOfContentsItem() -> RDEPUBReaderTableOfContentsItem? {
+        let items = flattenedTableOfContents
+        guard !items.isEmpty else { return nil }
+
+        let currentPageNumber = max(readerView.currentPage + 1, 1)
+        let pageAnchoredMatch = items.last { item in
+            guard let pageNumber = item.pageNumber else { return false }
+            return pageNumber <= currentPageNumber
+        }
+        if let pageAnchoredMatch {
+            return pageAnchoredMatch
+        }
+
+        guard let publication,
+              let currentLocation = currentVisibleLocation(),
+              let normalizedCurrentHref = publication.resourceResolver.normalizedHref(currentLocation.href) else {
+            return nil
+        }
+
+        if let textBook,
+           let chapterData = textBook.chapterData(
+            for: currentLocation,
+            resolver: publication.resourceResolver,
+            bookIdentifier: currentBookIdentifier
+           ),
+           let tocItem = chapterData.primaryTableOfContentsItem(
+            from: publication.tableOfContents,
+            normalizer: { publication.resourceResolver.normalizedHref($0) }
+           ) {
+            return items.last { $0.href == tocItem.href } ?? items.last {
+                guard let normalizedItemHref = publication.resourceResolver.normalizedHref($0.href.components(separatedBy: "#").first ?? $0.href) else {
+                    return false
+                }
+                return normalizedItemHref == normalizedCurrentHref
+            }
+        }
+
+        return items.last { item in
+            guard let normalizedItemHref = publication.resourceResolver.normalizedHref(item.href.components(separatedBy: "#").first ?? item.href) else {
+                return false
+            }
+            return normalizedItemHref == normalizedCurrentHref
+        }
+    }
+
+    func flattenedTableOfContentsItems(
+        from items: [EPUBTableOfContentsItem],
+        depth: Int = 0
+    ) -> [RDEPUBReaderTableOfContentsItem] {
+        items.flatMap { item in
+            let location = RDEPUBLocation(bookIdentifier: currentBookIdentifier, href: item.href, progression: 0)
+            let pageNumber: Int?
+            if let textBook, let publication,
+               let chapterData = textBook.chapterData(for: location, resolver: publication.resourceResolver, bookIdentifier: currentBookIdentifier) {
+                let normalizedLocation = publication.resourceResolver.normalizedLocation(
+                    location,
+                    bookIdentifier: currentBookIdentifier
+                ) ?? location
+                pageNumber = chapterData.pageNumber(for: normalizedLocation)
+                    ?? textBook.pageNumber(
+                        for: location,
+                        resolver: publication.resourceResolver,
+                        bookIdentifier: currentBookIdentifier
+                    )
+            } else if let readingSession {
+                pageNumber = readingSession.pageIndex(for: location, bookIdentifier: currentBookIdentifier).map { $0 + 1 }
+            } else {
+                pageNumber = nil
+            }
+
+            let current = RDEPUBReaderTableOfContentsItem(
+                title: item.title,
+                href: item.href,
+                depth: depth,
+                pageNumber: pageNumber
+            )
+            return [current] + flattenedTableOfContentsItems(from: item.children, depth: depth + 1)
+        }
+    }
+}
+