From bf43a198c79ef23067763b834ab1c9d046d3040d Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Mon, 19 Aug 2019 11:53:14 -0400
Subject: [PATCH] response error and clipboard messages

---
 .../SafariExtensionViewController.swift       | 22 +++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/src/safari/app/desktop/safari/SafariExtensionViewController.swift b/src/safari/app/desktop/safari/SafariExtensionViewController.swift
index f8a86094ad..cdfb9aa2b8 100644
--- a/src/safari/app/desktop/safari/SafariExtensionViewController.swift
+++ b/src/safari/app/desktop/safari/SafariExtensionViewController.swift
@@ -115,7 +115,8 @@ class SafariExtensionViewController: SFSafariExtensionViewController, WKScriptMe
                             winIndex = winIndex + 1
                         }
                         if theWin == nil {
-                            // TODO: error
+                            m!.responseError = true
+                            self.replyMessage(message: m!)
                         } else {
                             var theTab: SFSafariTab?
                             theWin!.getAllTabs { tabs in
@@ -128,11 +129,16 @@ class SafariExtensionViewController: SFSafariExtensionViewController, WKScriptMe
                                     tabIndex = tabIndex + 1
                                 }
                                 if theTab == nil {
-                                    // TODO: error
+                                    m!.responseError = true
+                                    self.replyMessage(message: m!)
                                 } else {
                                     theTab!.getActivePage { activePage in
                                         if activePage != nil {
                                             activePage?.dispatchMessageToScript(withName: "bitwarden", userInfo: ["msg": tabMsg!.obj])
+                                            self.replyMessage(message: m!)
+                                        } else {
+                                            m!.responseError = true
+                                            self.replyMessage(message: m!)
                                         }
                                     }
                                 }
@@ -141,6 +147,16 @@ class SafariExtensionViewController: SFSafariExtensionViewController, WKScriptMe
                     }
                 } else if command == "hideWindow" {
                     dismissPopover()
+                    replyMessage(message: m!)
+                } else if command == "copyToClipboard" {
+                    let pasteboard = NSPasteboard.general
+                    pasteboard.declareTypes([NSPasteboard.PasteboardType.string], owner: nil)
+                    pasteboard.setString(m!.data ?? "", forType: NSPasteboard.PasteboardType.string)
+                    replyMessage(message: m!)
+                } else if command == "readFromClipboard" {
+                    let pasteboard = NSPasteboard.general
+                    m?.responseData = pasteboard.pasteboardItems?.first?.string(forType: .string)
+                    replyMessage(message: m!)
                 }
             }
         }
@@ -265,12 +281,14 @@ class AppMessage: Decodable, Encodable {
         command = ""
         data = nil
         responseData = nil
+        responseError = nil
     }
 
     var id: String
     var command: String
     var data: String?
     var responseData: String?
+    var responseError: Bool?
 }
 
 class StorageData: Decodable, Encodable {