From 0b742829d07f4cb8ef08580ef8212bca1f72b06f Mon Sep 17 00:00:00 2001
From: Faisal Shahzad <84210709+seowings@users.noreply.github.com>
Date: Wed, 6 Sep 2023 19:55:18 +0200
Subject: [PATCH] fetch Redirects via API

---
 src/main.py | 38 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 37 insertions(+), 1 deletion(-)

diff --git a/src/main.py b/src/main.py
index f5b8e2c..7d8bbca 100644
--- a/src/main.py
+++ b/src/main.py
@@ -42,6 +42,7 @@
 import base64
 import json
 import re
+from urllib import parse
 
 # +++++++++++++++++++++++++++++++++++++++++++++++++++++
 # IMPORTS (3rd Party)
@@ -52,7 +53,6 @@
 import requests
 from requests.structures import CaseInsensitiveDict
 
-
 # +++++++++++++++++++++++++++++++++++++++++++++++++++++
 # Local Imports
 # +++++++++++++++++++++++++++++++++++++++++++++++++++++
@@ -104,6 +104,7 @@ def __init__(self, config_=None):
             self.redirect_page = Path(
                 self.output_folder, self.config["pages"]["redirect"]
             )
+            self.all_redirects = self.config["pages"]["all-redirects"]
             self.robots_txt_page = Path(
                 self.output_folder, self.config["pages"]["robots"]
             )
@@ -412,6 +413,18 @@ def create_redirect_toml_file(self):
         else:
             helpers.log_to_console("WARNING", "No Redirect File found")
 
+        for redirect in self.all_redirects:
+            rules.append(
+                [
+                    f"[[redirects]]\n",
+                    f'from = "{redirect["from"]}"\n',
+                    f'to = "{redirect["to"]}"\n',
+                    f'status = {redirect["status"]}\n',
+                    "force = true\n",
+                    "\n",
+                ]
+            )
+
         netlify_toml_file = Path(self.output_folder, "netlify.toml")
 
         with open(netlify_toml_file, "w", encoding="utf-8") as f:
@@ -460,6 +473,28 @@ def create_redirect_toml_file(self):
     except:
         pass
 
+    # redirects
+    all_redirects = []
+    try:
+        redirects_source = src_url + "/wp-json/redirection/v1/redirect"
+        red_response = requests.get(
+            redirects_source, headers={"Authorization": "Basic " + wp_token}
+        )
+
+        redirects_dict = json.loads(red_response.content)
+
+        all_redirects = [
+            {
+                "from": red["url"],
+                "to": red["action_data"]["url"],
+                "status": red["action_code"],
+                "force": True,
+            }
+            for red in redirects_dict["items"]
+        ]
+    except:
+        pass
+
     if wordpress_simply_static_zip_url:
         configurations = {
             "root": "",
@@ -473,6 +508,7 @@ def create_redirect_toml_file(self):
                 "redirect": page_redirects,
                 "robots": page_robots,
                 "search": page_search,
+                "all-redirects": all_redirects,
             },
             "search": {
                 "title": "true",