From 48a7c075511b8996b61bc3dd80aaebd1f911da51 Mon Sep 17 00:00:00 2001
From: James Magahern <jmagahern@apple.com>
Date: Tue, 23 Feb 2021 17:08:35 -0800
Subject: [PATCH] DocumentControls: Adds archive.today shortcut

---
 App/Browser View/BrowserViewController.swift             | 9 +++++++++
 .../DocumentControlViewController.swift                  | 5 +++++
 2 files changed, 14 insertions(+)

diff --git a/App/Browser View/BrowserViewController.swift b/App/Browser View/BrowserViewController.swift
index ca68e35..b3163af 100644
--- a/App/Browser View/BrowserViewController.swift	
+++ b/App/Browser View/BrowserViewController.swift	
@@ -271,6 +271,15 @@ class BrowserViewController: UIViewController
                 }
             }, for: .touchUpInside)
             
+            // Archive
+            documentControls.archiveView.addAction(UIAction { [unowned self] _ in
+                guard let currentURL = webView.url else { return }
+                guard let archiveURL = URL(string: "https://archive.today/\(currentURL.absoluteString)") else { return }
+                
+                tab.beginLoadingURL(archiveURL)
+                documentControls.dismiss(animated: true, completion: nil)
+            }, for: .touchUpInside)
+            
             // Dark mode
             documentControls.darkModeView.addAction(UIAction { [unowned self] _ in
                 self.darkModeEnabled = !self.darkModeEnabled
diff --git a/App/Document Controls UI/DocumentControlViewController.swift b/App/Document Controls UI/DocumentControlViewController.swift
index 5e7b2ad..29a44b2 100644
--- a/App/Document Controls UI/DocumentControlViewController.swift	
+++ b/App/Document Controls UI/DocumentControlViewController.swift	
@@ -16,6 +16,7 @@ class DocumentControlViewController: UIViewController
     let settingsView = DocumentControlView()
     let readabilityView = DocumentControlView()
     let darkModeView = DocumentControlView()
+    let archiveView = DocumentControlView()
     
     var observations: [NSKeyValueObservation] = []
     
@@ -33,6 +34,9 @@ class DocumentControlViewController: UIViewController
         readabilityView.label.text = "Reader Mode"
         readabilityView.imageView.image = UIImage(systemName: "doc.richtext")
         
+        archiveView.label.text = "Archive.today"
+        archiveView.imageView.image = UIImage(systemName: "shippingbox")
+        
         if darkModeEnabled {
             darkModeView.label.text = "Disable Dark Mode"
         } else {
@@ -46,6 +50,7 @@ class DocumentControlViewController: UIViewController
         documentControlView.addArrangedSubview(findOnPageControlView)
         documentControlView.addArrangedSubview(darkModeView)
         documentControlView.addArrangedSubview(readabilityView)
+        documentControlView.addArrangedSubview(archiveView)
         
         documentControlView.addArrangedSubview(settingsView)