Merge pull request #29 from Derpy-Leggies/unstable

Unstable

onlylegs/utils/colour.py

@@ -58,7 +58,7 @@ class Colour:
             s = 0.0
         else:
             d = high - low
-            s = d / (2 - high - low) if l > 0.5 else d / (high + low)
+            s = d / (2 - high - low) if low > 0.5 else d / (high + low)
             h = {
                 r: (g - b) / d + (6 if g < b else 0),
                 g: (b - r) / d + 2,

onlylegs/utils/startup.py

@@ -0,0 +1,133 @@
+"""
+OnlyLegs - Setup
+Runs when the app detects that there is no user directory
+"""
+import os
+import re
+import platformdirs
+import yaml
+
+
+APPLICATION_ROOT = platformdirs.user_config_dir("onlylegs")
+REQUIRED_DIRS = {
+    "root": APPLICATION_ROOT,
+    "instance": os.path.join(APPLICATION_ROOT, "instance"),
+    "media": os.path.join(APPLICATION_ROOT, "media"),
+    "uploads": os.path.join(APPLICATION_ROOT, "media", "uploads"),
+    "cache": os.path.join(APPLICATION_ROOT, "media", "cache"),
+    "pfp": os.path.join(APPLICATION_ROOT, "media", "pfp"),
+    "banner": os.path.join(APPLICATION_ROOT, "media", "banner"),
+}
+
+EMAIL_REGEX = re.compile(r"\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b")
+USERNAME_REGEX = re.compile(r"\b[A-Za-z0-9._%+-]+\b")
+
+config = {
+    # Version of the config file
+    "version": "0.1.7",
+    # Not really used much, but good to have for future use
+    "admin": {
+        "username": "admin",
+        "email": "admin@example.com",
+    },
+    "upload": {
+        "allowed-extensions": {
+            "jpg": "jpeg",
+            "jpeg": "jpeg",
+            "png": "png",
+            "webp": "webp",
+        },
+        # Max size in MB
+        "max-size": 69,
+        # Max images to load per page
+        "max-load": 50,
+    },
+    "website": {
+        # Website name and motto
+        # Also CSS styling, hue is the color offset for hsl
+        "name": "OnlyLegs",
+        "motto": "A gallery built for fast and simple image management!",
+        "styling": {
+            "force": False,
+            "hue": "69",
+            "saturation": "25%",
+            "rad": "0.4rem",
+        },
+    },
+}
+
+
+def check_dirs():
+    for directory in REQUIRED_DIRS.values():
+        if not os.path.exists(directory):
+            os.makedirs(directory)
+            print("Created directory at:", directory)
+        print("User directory already exists at:", directory)
+
+
+def check_env():
+    if os.path.exists(os.path.join(APPLICATION_ROOT, ".env")):
+        print("Environment file already exists at:", APPLICATION_ROOT)
+        return
+
+    env_conf = {
+        "FLASK_SECRET": os.urandom(32).hex(),
+    }
+
+    with open(
+        os.path.join(APPLICATION_ROOT, ".env"), encoding="utf-8", mode="w+"
+    ) as file:
+        for key, value in env_conf.items():
+            file.write(key + "=" + value + "\n")
+
+    print(
+        "####################################################",
+        "# A NEW KEY WAS GENERATED FOR YOU! PLEASE NOTE     #",
+        "# DOWN THE FLASK_SECRET KEY LOCATED IN YOUR        #",
+        "# ~/.config/onlylegs/.env FOLDER! LOOSING THIS KEY #",
+        "# WILL RESULT IN YOU BEING UNABLE TO LOG IN!       #",
+        "####################################################",
+        sep="\n",
+    )
+
+
+def check_conf():
+    config_file = os.path.join(APPLICATION_ROOT, "conf.yml")
+    if os.path.exists(config_file):
+        print("Config file already exists at:", APPLICATION_ROOT)
+        return
+
+    cant_continue = True
+    username = "admin"
+    email = "admin@example.com"
+
+    print("No config file found, please enter the following information:")
+    while cant_continue:
+        username = input("Admin username: ").strip()
+        email = input("Admin email: ").strip()
+
+        if not username or not USERNAME_REGEX.match(username):
+            print("Username is invalid!")
+            continue
+        if not email or not EMAIL_REGEX.match(email):
+            print("Email is invalid!")
+            continue
+
+        # Check if user is happy with the values
+        is_correct = input("Is this correct? (Y/n): ").lower().strip()
+        if is_correct == "y" or not is_correct:
+            cant_continue = False
+
+    config["admin"]["username"] = username
+    config["admin"]["email"] = email
+
+    with open(config_file, encoding="utf-8", mode="w+") as file:
+        yaml.dump(config, file, default_flow_style=False)
+
+    print(
+        "####################################################",
+        "# A NEW CONFIG HAS BEEN GENERATED AT:              #",
+        "# ~/.config/onlylegs/conf.yml                      #",
+        "####################################################",
+        sep="\n",
+    )