From fbcb3c127f2f8c44792035fd6e6fcfb5d7bf63b8 Mon Sep 17 00:00:00 2001
From: Kyle Hickinson <git@kylehickinson.com>
Date: Mon, 22 Jul 2019 15:21:13 -0400
Subject: [PATCH] Fix #661: Open in Brave now opens the first URL found in
 shared content

---
 BraveShareTo/ShareToBraveViewController.swift | 24 +++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/BraveShareTo/ShareToBraveViewController.swift b/BraveShareTo/ShareToBraveViewController.swift
index aabd5f35e18..c9ea6e1a6e4 100644
--- a/BraveShareTo/ShareToBraveViewController.swift
+++ b/BraveShareTo/ShareToBraveViewController.swift
@@ -13,6 +13,19 @@ class ShareToBraveViewController: SLComposeServiceViewController {
         return URL(string: "brave://open-url?url=\(url)")
     }
     
+    private func firstURL(in string: String) -> URL? {
+        do {
+            let detector = try NSDataDetector(types: NSTextCheckingResult.CheckingType.link.rawValue)
+            if let match = detector.firstMatch(in: string, options: [], range: NSRange(location: 0, length: string.count)),
+                let range = Range(match.range, in: string) {
+                return URL(string: String(string[range]))
+            }
+        } catch {
+            return nil
+        }
+        return nil
+    }
+    
     override func configurationItems() -> [Any]! {
         guard let inputItems = extensionContext?.inputItems as? [NSExtensionItem] else {
             return []
@@ -33,19 +46,22 @@ class ShareToBraveViewController: SLComposeServiceViewController {
         }
         
         provider.loadItem(of: provider.isUrl ? kUTTypeURL : kUTTypeText) { item, error in
-            var urlItem: URL!
+            var urlItem: URL?
             
             // We can get urls from other apps as a kUTTypeText type, for example from Apple's mail.app.
             if let text = item as? String {
-                urlItem = URL(string: text)
+                urlItem = self.firstURL(in: text)
             } else if let url = item as? URL {
-                urlItem = url
+                urlItem = self.firstURL(in: url.absoluteString)
             } else {
                 self.cancel()
                 return
             }
             
-            if let braveUrl = urlItem.absoluteString.addingPercentEncoding(withAllowedCharacters: .alphanumerics).flatMap(self.urlScheme) {
+            // Just open the app if we don't find a url. In the future we could
+            // use this entry point to search instead of open a given URL
+            let urlString = urlItem?.absoluteString ?? ""
+            if let braveUrl = urlString.addingPercentEncoding(withAllowedCharacters: .alphanumerics).flatMap(self.urlScheme) {
                 self.handleUrl(braveUrl)
             }
         }