download file via content script

src/browser/browserApi.ts

@@ -141,7 +141,8 @@ export class BrowserApi {
         if (BrowserApi.isChromeApi) {
             return chrome.extension.getURL(path);
         } else if (BrowserApi.isSafariApi) {
-            // TODO
+            // TODO: promisify
+            SafariApp.sendMessageToApp('getAppPath');
             return null;
         } else {
             return null;
@@ -188,9 +189,12 @@ export class BrowserApi {
                 data = Utils.fromBufferToB64(blobData);
             }
             SafariApp.sendMessageToApp('downloadFile', JSON.stringify({
-                data: data,
+                command: 'downloaderPageData',
-                type: type,
+                data: {
-                fileName: fileName,
+                    blobData: data,
+                    blobOptions: blobOptions,
+                    fileName: fileName,
+                },
             }), true);
         } else {
             const blob = new Blob([blobData], blobOptions);

src/content/downloader.ts

@@ -9,7 +9,16 @@ document.addEventListener('DOMContentLoaded', (event) => {
     safari.self.addEventListener('message', (msgEvent: any) => {
         const msg = JSON.parse(msgEvent.message.msg);
         if (msg.command === 'downloaderPageData') {
-            const blob = new Blob([msg.data.blobData], msg.data.blobOptions);
+            let data: any = msg.data.blobData;
+            if (msg.data.blobOptions == null || msg.data.blobOptions.type !== 'text/plain') {
+                const binaryString = window.atob(msg.data.blobData);
+                const bytes = new Uint8Array(binaryString.length);
+                for (let i = 0; i < binaryString.length; i++) {
+                    bytes[i] = binaryString.charCodeAt(i);
+                }
+                data = bytes.buffer;
+            }
+            const blob = new Blob([data], msg.data.blobOptions);
             if (navigator.msSaveOrOpenBlob) {
                 navigator.msSaveBlob(blob, msg.data.fileName);
             } else {
@@ -20,7 +29,6 @@ document.addEventListener('DOMContentLoaded', (event) => {
                 a.click();
                 document.body.removeChild(a);
             }
-            window.setTimeout(() => window.close(), 1500);
         }
     }, false);
 });

src/safari/app/desktop/safari/SafariExtensionViewController.swift

@@ -173,6 +173,21 @@ class SafariExtensionViewController: SFSafariExtensionViewController, WKScriptMe
             let pasteboard = NSPasteboard.general
             m?.responseData = pasteboard.pasteboardItems?.first?.string(forType: .string)
             replyMessage(message: m!)
+        } else if command == "downloadFile" {
+            SFSafariApplication.getActiveWindow { win in
+                win?.getActiveTab(completionHandler: { tab in
+                    tab?.getActivePage { activePage in
+                        activePage?.dispatchMessageToScript(withName: "bitwarden", userInfo: ["msg": m!.data])
+                    }
+                })
+            }
+        } else if command == "getAppPath" {
+            SFSafariExtension.getBaseURI(completionHandler: { uri in
+                if uri != nil {
+                    m!.responseData = uri!.absoluteString
+                    self.replyMessage(message: m!)
+                }
+            })
         }
     }
 
@@ -326,3 +341,9 @@ class TabMessage: Decodable, Encodable {
 class TabMessageOptions: Decodable, Encodable {
     var frameId: Int?
 }
+
+class DownloadFileMessage: Decodable, Encodable {
+    var fileName: String
+    var data: String?
+    var type: String?
+}