bitwarden/browser
1
0
Fork 0
mirror of https://github.com/bitwarden/browser synced 2025-12-12 06:13:38 +00:00
Code Issues Releases Wiki Activity

save and open files using panels

Browse Source
This commit is contained in:
Kyle Spearrin
2019-08-23 09:55:45 -04:00
parent 6fd71f9ebf
commit 0c76b0b8b5
1 changed files with 40 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
src/safari/safari/SafariExtensionViewController.swift
View File

@@ -33,6 +33,7 @@ class SafariExtensionViewController: SFSafariExtensionViewController, WKScriptMe
webView.allowsLinkPreview = false webView.allowsLinkPreview = false
webView.loadFileURL(url!, allowingReadAccessTo: bundleURL) webView.loadFileURL(url!, allowingReadAccessTo: bundleURL)
webView.alphaValue = 0.0 webView.alphaValue = 0.0
webView.uiDelegate = self
view.addSubview(webView) view.addSubview(webView)
} }
@@ -164,12 +165,35 @@ class SafariExtensionViewController: SFSafariExtensionViewController, WKScriptMe
m?.responseData = pasteboard.pasteboardItems?.first?.string(forType: .string) m?.responseData = pasteboard.pasteboardItems?.first?.string(forType: .string)
replyMessage(message: m!) replyMessage(message: m!)
} else if command == "downloadFile" { } else if command == "downloadFile" {
SFSafariApplication.getActiveWindow { win in if m!.data != nil {
win?.getActiveTab(completionHandler: { tab in if let dlMsg: DownloadFileMessage = jsonDeserialize(json: m!.data) {
tab?.getActivePage { activePage in var data: Data?
activePage?.dispatchMessageToScript(withName: "bitwarden", userInfo: ["msg": m!.data!]) if dlMsg.blobOptions?.type == "text/plain" {
data = dlMsg.blobData?.data(using: .utf8)
} else {
data = Data(base64Encoded: dlMsg.blobData!)
} }
}) if data != nil {
let panel = NSSavePanel()
panel.canCreateDirectories = true
panel.nameFieldStringValue = dlMsg.fileName
panel.begin { response in
if response == NSApplication.ModalResponse.OK {
if let url = panel.url {
do {
let fileManager = FileManager.default
if !fileManager.fileExists(atPath: url.absoluteString) {
fileManager.createFile(atPath: url.absoluteString, contents: Data(), attributes: nil)
}
try data!.write(to: url)
} catch {
print(error)
}
}
}
}
}
}
} }
} else if command == "getAppPath" { } else if command == "getAppPath" {
SFSafariExtension.getBaseURI(completionHandler: { uri in SFSafariExtension.getBaseURI(completionHandler: { uri in
@@ -191,7 +215,7 @@ class SafariExtensionViewController: SFSafariExtensionViewController, WKScriptMe
} }
extension SafariExtensionViewController: WKUIDelegate { extension SafariExtensionViewController: WKUIDelegate {
func webView(_ webView: WKWebView, runOpenPanelWith parameters: WKOpenPanelParameters, initiatedByFrame frame: WKFrameInfo, completionHandler: @escaping ([URL]?) -> Void) { func webView(_: WKWebView, runOpenPanelWith _: WKOpenPanelParameters, initiatedByFrame _: WKFrameInfo, completionHandler: @escaping ([URL]?) -> Void) {
let openPanel = NSOpenPanel() let openPanel = NSOpenPanel()
openPanel.canChooseFiles = true openPanel.canChooseFiles = true
openPanel.begin { result in openPanel.begin { result in
@@ -348,3 +372,13 @@ class TabMessage: Decodable, Encodable {
class TabMessageOptions: Decodable, Encodable { class TabMessageOptions: Decodable, Encodable {
var frameId: Int? var frameId: Int?
} }
class DownloadFileMessage: Decodable, Encodable {
var fileName: String
var blobData: String?
var blobOptions: DownloadFileMessageBlobOptions?
}
class DownloadFileMessageBlobOptions: Decodable, Encodable {
var type: String?
}