Inject EPUB scripts with WKWebView (#204)

Former-commit-id: 06d877f

Author: mickael-menu

navigator/r2-navigator-swift/EPUB/EPUBFixedSpreadView.swift

@@ -7,6 +7,7 @@
 import Foundation
 import UIKit
 import WebKit
+import R2Shared
 
 
 /// A view rendering a spread of resources with a fixed layout.
@@ -17,6 +18,15 @@ final class EPUBFixedSpreadView: EPUBSpreadView {
     /// URL to load in the iframe once the wrapper page is loaded.
     private var urlToLoad: URL?
 
+    private static let fixedScript = loadScript(named: "readium-fixed")
+    
+    required init(publication: Publication, spread: EPUBSpread, resourcesURL: URL, readingProgression: ReadingProgression, userSettings: UserSettings, scripts: [WKUserScript], animatedLoad: Bool, editingActions: EditingActionsController, contentInset: [UIUserInterfaceSizeClass: EPUBContentInsets]) {
+        var scripts = scripts
+        scripts.append(WKUserScript(source: Self.fixedScript, injectionTime: .atDocumentStart, forMainFrameOnly: false))
+        
+        super.init(publication: publication, spread: spread, resourcesURL: resourcesURL, readingProgression: readingProgression, userSettings: userSettings, scripts: scripts, animatedLoad: animatedLoad, editingActions: editingActions, contentInset: contentInset)
+    }
+    
     override func setupWebView() {
         super.setupWebView()
 

navigator/r2-navigator-swift/EPUB/EPUBReflowableSpreadView.swift

@@ -16,14 +16,20 @@ final class EPUBReflowableSpreadView: EPUBSpreadView {
     private var topConstraint: NSLayoutConstraint!
     private var bottomConstraint: NSLayoutConstraint!
 
+    private static let reflowableScript = loadScript(named: "readium-reflowable")
+    
     required init(publication: Publication, spread: EPUBSpread, resourcesURL: URL, readingProgression: ReadingProgression, userSettings: UserSettings, scripts: [WKUserScript], animatedLoad: Bool, editingActions: EditingActionsController, contentInset: [UIUserInterfaceSizeClass: EPUBContentInsets]) {
         var scripts = scripts
+        
         let layout = ReadiumCSSLayout(languages: publication.metadata.languages, readingProgression: readingProgression)
         scripts.append(WKUserScript(
             source: "window.readiumCSSBaseURL = '\(resourcesURL.appendingPathComponent(layout.readiumCSSBasePath))'",
             injectionTime: .atDocumentStart,
             forMainFrameOnly: false
         ))
+        
+        scripts.append(WKUserScript(source: Self.reflowableScript, injectionTime: .atDocumentStart, forMainFrameOnly: false))
+        
         super.init(publication: publication, spread: spread, resourcesURL: resourcesURL, readingProgression: readingProgression, userSettings: userSettings, scripts: scripts, animatedLoad: animatedLoad, editingActions: editingActions, contentInset: contentInset)
     }
 

navigator/r2-navigator-swift/EPUB/EPUBSpreadView.swift

@@ -46,7 +46,6 @@ class EPUBSpreadView: UIView, Loggable, PageView {
 
     let readingProgression: ReadingProgression
     let userSettings: UserSettings
-    let scripts: [WKUserScript]
     let editingActions: EditingActionsController
 
     private var lastClick: ClickEvent? = nil
@@ -73,7 +72,6 @@ class EPUBSpreadView: UIView, Loggable, PageView {
         self.resourcesURL = resourcesURL
         self.readingProgression = readingProgression
         self.userSettings = userSettings
-        self.scripts = scripts
         self.editingActions = editingActions
         self.animatedLoad = animatedLoad
         self.webView = WebView(editingActions: editingActions)
@@ -404,6 +402,13 @@ class EPUBSpreadView: UIView, Loggable, PageView {
         applyUserSettingsStyle()
     }
 
+    
+    // MARK: - Scripts
+    
+    class func loadScript(named name: String) -> String {
+        return Bundle.module.url(forResource: "\(name)", withExtension: "js", subdirectory: "Assets/Static/scripts")
+            .flatMap { try? String(contentsOf: $0) }!
+    }
 }
 
 // MARK: - WKScriptMessageHandler for handling incoming message from the javascript layer.