From 7b8ce4ea830c73e33317c4c642324c553dd95b81 Mon Sep 17 00:00:00 2001
From: Brett Laptop <tri11paragon@tpgc.me>
Date: Sun, 9 Jun 2024 18:36:36 -0400
Subject: [PATCH] install script

---
 main.py | 95 +++++++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 86 insertions(+), 9 deletions(-)

diff --git a/main.py b/main.py
index 6dcf9e8..4cb15fa 100644
--- a/main.py
+++ b/main.py
@@ -1,6 +1,8 @@
-import requests
 import argparse
 import json
+import requests
+import time
+import sched
 from pathlib import Path
 
 
@@ -11,16 +13,26 @@ class Env:
 
 
 class Zone:
-    def __init__(self, zone_id, env, records):
-        self.zone_id = zone_id
+    def __init__(self, zone_name, env, records):
+        self.zone_name = zone_name
         self.env = env
         self.records = records
+        self.zone_id = None
+
+
+class Zones:
+    def __init__(self, ip_provider, r_zones):
+        self.ip_provider = ip_provider
+        self.zones = r_zones
 
 
 def parse_arguments():
     parser = argparse.ArgumentParser(description="Silly little DDNS client", prog="Brett's DDNS Client")
-    parser.add_argument("--env", "-e", default="/etc/bddns.env", required=False)
-    parser.add_argument("--config", "-c", default="/etc/bddns.conf", required=False)
+    parser.add_argument("--env", "-e", default="/etc/bddns/users.conf", required=False)
+    parser.add_argument("--config", "-c", default="/etc/bddns/sites.conf", required=False)
+    parser.add_argument("--time", "-u", default=300, required=False, help="Time between updates in seconds")
+    parser.add_argument("--install", action="store_true", default=False, required=False,
+                        help="Runs installer, will create config files")
 
     return parser.parse_args()
 
@@ -42,12 +54,77 @@ def get_conf(conf_path, envs):
             sites = []
             for site in zone["sites"]:
                 sites.append(site)
-            zones.append(Zone(zone["zone_id"], envs[zone["username"]], sites))
-        return zones
+            zones.append(Zone(zone["zone"], envs[zone["username"]], sites))
+        return Zones(json_env["ip_provider"], zones)
+
+
+def run_install():
+    config_folder = Path("/etc/bddns/")
+    users_path = Path(config_folder / "users.conf")
+    sites_path = Path(config_folder / "sites.conf")
+    print(f"Installing to {config_folder}")
+    print(f"\tUsers path: {users_path}")
+    print(f"\tSites path: {sites_path}")
+    if not config_folder.is_dir():
+        config_folder.mkdir()
+    if not users_path.exists():
+        users_path.touch()
+        with open(users_path, "w") as f:
+            json.dump({"users": [
+                {"username": "user@email.com", "api_key": "<KEY>"},
+                {"username": "another-user@email.com", "api_key": "<KEY>"},
+            ]}, f, indent=4)
+    else:
+        print(f"Warning {users_path} already exists, skipping!")
+    if not sites_path.exists():
+        sites_path.touch()
+        with open(sites_path, "w") as f:
+            json.dump({
+                "ip_provider": "https://api.ipify.org/",
+                "zones": [
+                    {
+                        "zone": "<zone_name.tld>",
+                        "username": "user@email.com",
+                        "sites": [
+                            "site.tld",
+                            "subdomain.site.tld"
+                        ]
+                    }
+                ]
+            }, f, indent=4)
+    else:
+        print(f"Warning {sites_path} already exists, skipping!")
+
+
+def fetch_ip(provider):
+    print(f"Fetching IP Address with '{provider}'")
+    r = requests.get(provider)
+    return r.text
+
+
+def fetch_zone_id(zone_name, key):
+    headers = "Authorization: Bearer"
+    return "hello"
+
+
+def run(scheduler, r_args, r_envs, r_zones):
+    scheduler.enter(r_args.time, 1, run, (scheduler, r_args, r_envs, r_zones))
+    address = fetch_ip(r_zones.ip_provider)
+
+    for zone in r_zones.zones:
+        user = zone.env
+        if zone.zone_id is None:
+            zone.zone_id = fetch_zone_id(zone.zone_name)
 
 
 if __name__ == '__main__':
     args = parse_arguments()
-    user_envs = get_env(args.env)
-
 
+    if args.install:
+        run_install()
+    else:
+        user_envs = get_env(args.env)
+        zones = get_conf(args.config, user_envs)
+        runner = sched.scheduler(time.time, time.sleep)
+        runner.enter(args.time, 1, run, (runner, args, user_envs, zones))
+        runner.run()