feat(archive): Set the user when someone logged-in archives

2 files changed, 21 insertions, 3 deletions

diff --git a/controllers/archive.php b/controllers/archive.php
index 0d21a1e..6057a4a 100644
--- a/controllers/archive.php
+++ b/controllers/archive.php
@@ -2,12 +2,20 @@
 namespace Controller;
 use Database;
 use DOMDocument;
+use Exception;
 
 function on_post() {
     $WEBSITE_CATEGORY = 'page_url';
     $DOWNLOADS_FOLDER = getenv('ARCHIVES_DIR');
     $website_url = $_POST[$WEBSITE_CATEGORY];
-    $currentPage = new DownloadPage($website_url, $DOWNLOADS_FOLDER);
+    $uid = 1;
+    if (array_key_exists('token', $_POST) && strlen($_POST['token']) === 36) {
+        try {
+            $uid = Database\Cookie::fromDB($_POST['token'])->UID;
+        }
+        catch (Exception $e) {}
+    }
+    $currentPage = new DownloadPage($website_url, $DOWNLOADS_FOLDER, $uid);
 }
 
 class DownloadPage {
@@ -18,7 +26,7 @@ class DownloadPage {
     private $favicon_path;
     private $page_title;
 
-    function __construct($page_url, $folder_location) {
+    function __construct($page_url, $folder_location, $requester_uid) {
         $this->folder_location = $folder_location;
         $this->page_url = $page_url;
         list($website_exists, $this->page_url) = $this->doesWebsiteExist($this->page_url);
@@ -34,7 +42,7 @@ class DownloadPage {
                 // Fallback and try to download them from the server directly
                 $this->tryDownloadFavicon();
             }
-            Database\Webpage::create($folder_location, $page_url, 1, $this->favicon_path, $this->page_title);
+            Database\Webpage::create($folder_location, $page_url, $requester_uid, $this->favicon_path, $this->page_title);
         } else {
             echo "Website does not exist";
         }
diff --git a/views/archive/index.php b/views/archive/index.php
index 2c784de..7d41890 100644
--- a/views/archive/index.php
+++ b/views/archive/index.php
@@ -16,8 +16,13 @@
 
     <form action="#" method="POST">
         <input type="hidden" name="page_url" value="<?= $url ?>">
+        <input id="token" type="hidden" name="token" value="">
         <input type="submit" value="Archive Now!">
     </form>
+    <script type="text/javascript">
+        const tokenInput = document.getElementById('token');
+        tokenInput.value = sessionStorage.getItem('token');
+    </script>
     <!-- Button to add to list -->
     <!-- Button to delete -->
 
@@ -54,7 +59,12 @@
     <h2>"<?= $url ?>" hasn't been archived yet!</h2>
     <form action="#" method="POST">
         <input type="hidden" name="page_url" value="<?= $url ?>">
+        <input id="token" type="hidden" name="token" value="">
         <input type="submit" value="Archive Now!">
     </form>
+    <script type="text/javascript">
+        const tokenInput = document.getElementById('token');
+        tokenInput.value = sessionStorage.getItem('token');
+    </script>
 
 <?php endif; ?>