mirror of
https://github.com/mattermost/focalboard.git
synced 2024-12-12 09:04:14 +02:00
190 lines
6.0 KiB
Swift
190 lines
6.0 KiB
Swift
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
import Cocoa
|
|
import WebKit
|
|
|
|
class ViewController:
|
|
NSViewController,
|
|
WKUIDelegate,
|
|
WKNavigationDelegate {
|
|
@IBOutlet var webView: WKWebView!
|
|
|
|
override func viewDidLoad() {
|
|
super.viewDidLoad()
|
|
|
|
NSLog("viewDidLoad")
|
|
|
|
webView.navigationDelegate = self
|
|
webView.uiDelegate = self
|
|
webView.isHidden = true
|
|
|
|
clearWebViewCache()
|
|
|
|
// Load the home page if the server was started, otherwise wait until it has
|
|
let appDelegate = NSApplication.shared.delegate as! AppDelegate
|
|
if (appDelegate.isServerStarted) {
|
|
self.updateSessionToken()
|
|
self.loadHomepage()
|
|
}
|
|
|
|
// Do any additional setup after loading the view.
|
|
NotificationCenter.default.addObserver(self, selector: #selector(onServerStarted), name: AppDelegate.serverStartedNotification, object: nil)
|
|
}
|
|
|
|
override var representedObject: Any? {
|
|
didSet {
|
|
// Update the view, if already loaded.
|
|
}
|
|
}
|
|
|
|
private func clearWebViewCache() {
|
|
let websiteDataTypes = NSSet(array: [WKWebsiteDataTypeDiskCache, WKWebsiteDataTypeMemoryCache])
|
|
let date = Date(timeIntervalSince1970: 0)
|
|
WKWebsiteDataStore.default().removeData(ofTypes: websiteDataTypes as! Set<String>, modifiedSince: date, completionHandler:{ })
|
|
}
|
|
|
|
@objc func onServerStarted() {
|
|
NSLog("onServerStarted")
|
|
DispatchQueue.main.asyncAfter(deadline: .now() + 0.5) {
|
|
self.updateSessionToken()
|
|
self.loadHomepage()
|
|
}
|
|
}
|
|
|
|
private func updateSessionToken() {
|
|
let appDelegate = NSApplication.shared.delegate as! AppDelegate
|
|
let script = WKUserScript(
|
|
source: "localStorage.setItem('sessionId', '\(appDelegate.sessionToken)');",
|
|
injectionTime: .atDocumentStart,
|
|
forMainFrameOnly: true
|
|
)
|
|
webView.configuration.userContentController.removeAllUserScripts()
|
|
webView.configuration.userContentController.addUserScript(script)
|
|
}
|
|
|
|
private func loadHomepage() {
|
|
let appDelegate = NSApplication.shared.delegate as! AppDelegate
|
|
let port = appDelegate.serverPort
|
|
let url = URL(string: "http://localhost:\(port)/")!
|
|
let request = URLRequest(url: url)
|
|
webView.load(request)
|
|
}
|
|
|
|
private func downloadJsonUrl(_ url: URL) {
|
|
NSLog("downloadJsonUrl")
|
|
let prefix = "data:text/json,"
|
|
let urlString = url.absoluteString
|
|
let encodedJson = String(urlString[urlString.index(urlString.startIndex, offsetBy: prefix.lengthOfBytes(using: .utf8))...])
|
|
guard let json = encodedJson.removingPercentEncoding else {
|
|
return
|
|
}
|
|
|
|
// Form file name
|
|
let dateFormatter = DateFormatter()
|
|
dateFormatter.dateFormat = "yyyy-M-d"
|
|
let dateString = dateFormatter.string(from: Date())
|
|
let filename = "archive-\(dateString).focalboard"
|
|
|
|
// Save file
|
|
let savePanel = NSSavePanel()
|
|
savePanel.canCreateDirectories = true
|
|
savePanel.nameFieldStringValue = filename
|
|
// BUGBUG: Specifying the allowedFileTypes causes Catalina to hang / error out
|
|
//savePanel.allowedFileTypes = [".focalboard"]
|
|
savePanel.begin { (result) in
|
|
if result.rawValue == NSApplication.ModalResponse.OK.rawValue,
|
|
let fileUrl = savePanel.url {
|
|
try? json.write(to: fileUrl, atomically: true, encoding: .utf8)
|
|
}
|
|
}
|
|
}
|
|
|
|
private func downloadCsvUrl(_ url: URL) {
|
|
NSLog("downloadCsvUrl")
|
|
let prefix = "data:text/csv;charset=utf-8,"
|
|
let urlString = url.absoluteString
|
|
let encodedContents = String(urlString[urlString.index(urlString.startIndex, offsetBy: prefix.lengthOfBytes(using: .utf8))...])
|
|
guard let contents = encodedContents.removingPercentEncoding else {
|
|
return
|
|
}
|
|
|
|
let filename = "data.csv"
|
|
|
|
// Save file
|
|
let savePanel = NSSavePanel()
|
|
savePanel.canCreateDirectories = true
|
|
savePanel.nameFieldStringValue = filename
|
|
// BUGBUG: Specifying the allowedFileTypes causes Catalina to hang / error out
|
|
//savePanel.allowedFileTypes = [".focalboard"]
|
|
savePanel.begin { (result) in
|
|
if result.rawValue == NSApplication.ModalResponse.OK.rawValue,
|
|
let fileUrl = savePanel.url {
|
|
try? contents.write(to: fileUrl, atomically: true, encoding: .utf8)
|
|
}
|
|
}
|
|
}
|
|
|
|
func webView(_ webView: WKWebView, runOpenPanelWith parameters: WKOpenPanelParameters, initiatedByFrame frame: WKFrameInfo, completionHandler: @escaping ([URL]?) -> Void) {
|
|
NSLog("webView runOpenPanel")
|
|
let openPanel = NSOpenPanel()
|
|
openPanel.canChooseFiles = true
|
|
openPanel.begin { (result) in
|
|
if result == NSApplication.ModalResponse.OK {
|
|
if let url = openPanel.url {
|
|
completionHandler([url])
|
|
}
|
|
} else if result == NSApplication.ModalResponse.cancel {
|
|
completionHandler(nil)
|
|
}
|
|
}
|
|
}
|
|
|
|
func webView(_ webView: WKWebView, decidePolicyFor navigationAction: WKNavigationAction, decisionHandler: @escaping (WKNavigationActionPolicy) -> Void) {
|
|
if let url = navigationAction.request.url {
|
|
// Intercept archive downloads, and present native UI
|
|
if (url.absoluteString.hasPrefix("data:text/json,")) {
|
|
decisionHandler(.cancel)
|
|
downloadJsonUrl(url)
|
|
return
|
|
}
|
|
if (url.absoluteString.hasPrefix("data:text/csv;charset=utf-8,")) {
|
|
decisionHandler(.cancel)
|
|
downloadCsvUrl(url)
|
|
return
|
|
}
|
|
NSLog("decidePolicyFor navigationAction: \(url.absoluteString)")
|
|
}
|
|
decisionHandler(.allow)
|
|
}
|
|
|
|
func webView(_ webView: WKWebView, didFinish navigation: WKNavigation!) {
|
|
NSLog("webView didFinish navigation: \(webView.url?.absoluteString ?? "")")
|
|
// Disable right-click menu
|
|
webView.evaluateJavaScript("document.body.setAttribute('oncontextmenu', 'event.preventDefault();');", completionHandler: nil)
|
|
webView.isHidden = false
|
|
}
|
|
|
|
func webView(_ webView: WKWebView, didFail navigation: WKNavigation!, withError error: Error) {
|
|
webView.isHidden = false
|
|
}
|
|
|
|
|
|
func webView(_ webView: WKWebView, createWebViewWith configuration: WKWebViewConfiguration, for navigationAction: WKNavigationAction, windowFeatures: WKWindowFeatures) -> WKWebView? {
|
|
if let frame = navigationAction.targetFrame,
|
|
frame.isMainFrame {
|
|
return nil
|
|
}
|
|
// for _blank target or non-mainFrame target, open in default browser
|
|
if let url = navigationAction.request.url {
|
|
NSWorkspace.shared.open(url)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
@IBAction func navigateToHome(_ sender: NSObject) {
|
|
loadHomepage()
|
|
}
|
|
}
|
|
|