1 files changed, 116 insertions, 0 deletions
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..35d51b2
--- /dev/null
+++ b/main.py
@@ -0,0 +1,116 @@
+from flask import Flask, request
+from flask import render_template
+from flask_compress import Compress
+import os
+from os.path import isfile
+from dataclasses import dataclass
+from typing import Literal
+
+app = Flask(__name__)
+Compress(app)
+
+### PAGES
+
+
+@app.route("/")
+def index():
+    files_expanded = request.args.get("show_files", "0") == "1"
+    dir = request.args.get("dir", "")
+    if not files_expanded:
+        return render_template("index.html", files_expanded=files_expanded)
+    else:
+        listing = get_file_listing_at_path(dir)
+        return render_template(
+            "index.html",
+            files_expanded=files_expanded,
+            listing=listing,
+        )
+
+
+### ACTIONS
+
+
+@app.post("/click_files_drawer_toggle")
+def click_files_drawer_toggle():
+    files_expanded = request.form.get("files_expanded") == "true"
+    listing = get_file_listing_at_path("")
+    return render_template(
+        "responses/click_files_drawer_toggle.html",
+        files_expanded=not files_expanded,
+        listing=listing,
+    )
+
+
+@app.post("/navigate_files/")
+@app.post("/navigate_files/<path:dir_path>")
+def navigate_files(dir_path=""):
+    listing = get_file_listing_at_path(dir_path)
+    return render_template(
+        "responses/navigate_files.html",
+        files_expanded=True,
+        listing=listing,
+    )
+
+
+### TYPES
+
+
+@dataclass
+class ListingItem:
+    kind: Literal["file", "dir"]
+    name: str
+    full_path: str
+    static_path: str
+
+
+# UTILS
+
+
+def get_file_listing_at_path(path) -> list[ListingItem]:
+    static_dir_segment = f"static/files/{path}"
+
+    listing = os.listdir(static_dir_segment)
+    listing = [to_listing_item(path, name) for name in listing]
+    listing = sorted(listing, key=sort_key)
+
+    if path != "":
+        path_parts = path.split("/")[:-1]
+        parent_dir_path = "/".join(path_parts)
+        listing.insert(
+            0,
+            ListingItem(
+                kind="dir",
+                name="..",
+                full_path=parent_dir_path,
+                static_path=f"static/files/{parent_dir_path}",
+            ),
+        )
+
+    return listing
+
+
+def to_listing_item(parent_dir: str, name: str) -> ListingItem:
+    if parent_dir != "":
+        static_path = f"static/files/{parent_dir}/{name}"
+        full_path = f"{parent_dir}/{name}"
+    else:
+        static_path = f"static/files/{name}"
+        full_path = name
+
+    kind = "file" if isfile(static_path) else "dir"
+    return ListingItem(
+        kind=kind,
+        name=name,
+        full_path=full_path,
+        static_path=static_path,
+    )
+
+
+def sort_key(item: ListingItem) -> str:
+    if item.kind == "file":
+        return item.name
+    return "0000" + item.name
+
+
+def get_app():
+    return app