136 lines
4.4 KiB
Swift
136 lines
4.4 KiB
Swift
//
|
|
// BrowserViewController.swift
|
|
// SBrowser
|
|
//
|
|
// Created by James Magahern on 7/21/20.
|
|
//
|
|
|
|
import UIKit
|
|
|
|
class BrowserViewController: UIViewController,
|
|
SBRProcessBundleBridgeDelegate, WKNavigationDelegate,
|
|
UITextFieldDelegate, ScriptPolicyViewControllerDelegate
|
|
{
|
|
let bridge = SBRProcessBundleBridge()
|
|
let browserView = BrowserView()
|
|
|
|
private let policyManager = ResourcePolicyManager()
|
|
private let toolbarController = ToolbarViewController()
|
|
private var blockedScriptOrigins = Set<String>()
|
|
override var canBecomeFirstResponder: Bool { true }
|
|
|
|
private var loadingObservation: NSKeyValueObservation?
|
|
|
|
init() {
|
|
super.init(nibName: nil, bundle: nil)
|
|
}
|
|
|
|
required init?(coder: NSCoder) { fatalError("init(coder:) has not been implemented") }
|
|
|
|
override func loadView() {
|
|
bridge.delegate = self
|
|
bridge.policyDataSource = policyManager
|
|
|
|
let webView = bridge.webView
|
|
webView.allowsBackForwardNavigationGestures = true
|
|
webView.navigationDelegate = self
|
|
|
|
browserView.webView = webView
|
|
browserView.toolbarView = toolbarController.toolbarView
|
|
|
|
// Refresh button
|
|
toolbarController.urlBar.refreshButton.addAction(UIAction(handler: { action in
|
|
if webView.isLoading {
|
|
webView.stopLoading()
|
|
} else {
|
|
webView.reload()
|
|
}
|
|
}), for: .touchUpInside)
|
|
|
|
// Script button
|
|
toolbarController.scriptControllerIconView.addAction(UIAction(handler: { action in
|
|
let scriptViewController = ScriptPolicyViewController(policyManager: self.policyManager, blockedScripts: self.blockedScriptOrigins)
|
|
scriptViewController.delegate = self
|
|
|
|
let navController = UINavigationController(rootViewController: scriptViewController)
|
|
self.present(navController, animated: true, completion: nil)
|
|
}), for: .touchUpInside)
|
|
|
|
// TextField delegate
|
|
toolbarController.urlBar.textField.delegate = self
|
|
|
|
// Load progress
|
|
loadingObservation = webView.observe(\.estimatedProgress) { (webView, observedChange) in
|
|
self.toolbarController.urlBar.loadProgress = .loading(progress: webView.estimatedProgress)
|
|
}
|
|
|
|
self.view = browserView
|
|
}
|
|
|
|
override func viewDidLoad() {
|
|
beginLoadingURL(URL(string: "https://reddit.com")!)
|
|
}
|
|
|
|
override func viewWillAppear(_ animated: Bool) {
|
|
super.viewWillAppear(animated)
|
|
becomeFirstResponder()
|
|
}
|
|
|
|
private func updateScriptBlockerButton() {
|
|
toolbarController.scriptControllerIconView.setBlockedScriptsNumber(blockedScriptOrigins.count)
|
|
}
|
|
|
|
func beginLoadingURL(_ url: URL) {
|
|
let request = URLRequest(url: url)
|
|
bridge.webView.load(request)
|
|
}
|
|
|
|
// MARK: SBRProcessBundleBridgeDelegate
|
|
|
|
func webProcess(_ bridge: SBRProcessBundleBridge, didBlockScriptResourceFromOrigin origin: String) {
|
|
print("Blocked script resource from origin: \(origin)")
|
|
|
|
blockedScriptOrigins.formUnion([ origin ])
|
|
updateScriptBlockerButton()
|
|
}
|
|
|
|
// MARK: Navigation Delegate
|
|
|
|
func webView(_ webView: WKWebView, didStartProvisionalNavigation navigation: WKNavigation!) {
|
|
// Reset tracking this
|
|
blockedScriptOrigins.removeAll()
|
|
|
|
if let urlString = webView.url?.absoluteString {
|
|
toolbarController.urlBar.textField.text = urlString
|
|
}
|
|
}
|
|
|
|
func webView(_ webView: WKWebView, didFinish navigation: WKNavigation!) {
|
|
toolbarController.urlBar.loadProgress = .complete
|
|
}
|
|
|
|
// MARK: UITextField Delegate
|
|
|
|
func textFieldShouldReturn(_ textField: UITextField) -> Bool {
|
|
if let text = textField.text, let url = URL(string: text) {
|
|
if url.scheme == nil {
|
|
let urlString = "https://\(text)"
|
|
if let url = URL(string: urlString) {
|
|
beginLoadingURL(url)
|
|
}
|
|
}
|
|
}
|
|
|
|
textField.resignFirstResponder()
|
|
return false
|
|
}
|
|
|
|
// MARK: Script Policy View Controller Delegate
|
|
|
|
func didChangeScriptPolicy() {
|
|
bridge.policyDataSourceDidChange()
|
|
bridge.webView.reload()
|
|
}
|
|
|
|
}
|