1 files changed, 73 insertions, 25 deletions

diff --git a/controllers/archive_page.php b/controllers/archive_page.php
index 593076f..ff766dd 100644
--- a/controllers/archive_page.php
+++ b/controllers/archive_page.php
@@ -2,40 +2,88 @@
 
 function on_post() {
     $WEBSITE_CATEGORY = 'page_url';
+    $DOWNLOADS_FOLDER = getenv('ARCHIVES_DIR');
     $website_url = $_POST[$WEBSITE_CATEGORY];
-    $website_exists = does_website_exist($website_url) ? "true" : "false";
-    echo "Website exists: $website_exists" . "<br/>";
+    $currentPage = new DownloadPage($website_url, "test2.zip", $DOWNLOADS_FOLDER);
 }
 
-function apply_correct_protocol($url, $protocol) {
-    if (str_contains($url, $protocol)) {
-        return $url;
-    }
+class DownloadPage {
+    private $zip_location;
+    private $zip_name;
+    private $page_url;
+    private $page_contents;
 
-    return $protocol . $url;
-}
+    function __construct($page_url, $zip_name, $zip_location) {
+        $this->zip_location = $zip_location;
+        $this->zip_name = $zip_name;
+        $this->page_url = $page_url;
+        list($website_exists, $this->page_url) = $this->does_website_exist($this->page_url);
+        if ($website_exists) {
+            $this->page_contents = file_get_contents($this->page_url);
+            $zip = $this->create_zip_archive();
+        } else {
+            echo "Website does not exist";
+        }
+    }
 
-function does_website_exist($url) {
-    $result = false;
+    function set_zip_location($zip_location) {
+        $this->zip_location = $zip_location;
+    }
+    function set_zip_name($zip_name) {
+        $this->zip_name = $zip_name;
+    }
+    function set_page_url($page_url) {
+        $this->page_url = $page_url;
+    }
+    function apply_correct_protocol($url, $protocol) {
+        if (str_contains($url, $protocol)) {
+            return $url;
+        }
 
-    // Check if the site exists with https
-    $https_url = apply_correct_protocol($url, "https://");
-    if ($https_url != $url) {
-        $url_headers = @get_headers($https_url);
-        $result |= $url_headers && $url_headers[0] != 'HTTP/1.1 404 Not Found';
+        return $protocol . $url;
     }
 
-    // Check if the site exists with http
-    $http_url = apply_correct_protocol($url, "http://");
-    if ($http_url != $url) {
-        $url_headers = @get_headers($http_url);
-        $result |= $url_headers && $url_headers[0] != 'HTTP/1.1 404 Not Found';
+    function does_website_exist($url) {
+
+        // Check if the site exists with https
+        $https_url = $this->apply_correct_protocol($url, "https://");
+        if ($https_url != $url) {
+            $url_headers = @get_headers($https_url);
+            if ($url_headers && $url_headers[0] != 'HTTP/1.1 404 Not Found') {
+                return array(true, $https_url);
+            }
+        }
+
+        // Check if the site exists with http
+        $http_url = $this->apply_correct_protocol($url, "http://");
+        if ($http_url != $url) {
+            $url_headers = @get_headers($http_url);
+            if ($url_headers && $url_headers[0] != 'HTTP/1.1 404 Not Found') {
+                return array(true, $http_url);
+            }
+        }
+
+        // Check if the site exists as is
+        // Will take effect when the user has entered the https/http protocol with the site
+        $url_headers = @get_headers($url);
+        if ($url_headers && $url_headers[0] != 'HTTP/1.1 404 Not Found') {
+            return array(true, $url);
+        }
+
+        return array(false, $url);
     }
 
-    // Check if the site exists as is
-    // Will take effect when the user has entered the https/http protocol with the site
-    $url_headers = @get_headers($url);
-    $result |= $url_headers && $url_headers[0] != 'HTTP/1.1 404 Not Found';
+    function create_zip_archive() {
+        // Creates and returns a zip object resulted from zipping the page that was downloaded
+        $zip = new ZipArchive();
+        if ($zip->open($this->zip_location . '/' . $this->zip_name, ZipArchive::CREATE | ZipArchive::OVERWRITE) == TRUE) {
+            $zip->addFromString('index.html', $this->page_contents);
+            $zip->close();
+            echo "Archived {$this->page_url}";
+        } else {
+            echo "Zip archive could not be open";
+        }
 
-    return $result;
+        return $zip;
+    }
 }