1 files changed, 54 insertions, 0 deletions

diff --git a/vault.py b/vault.py
new file mode 100644
index 0000000..4696ffc
--- /dev/null
+++ b/vault.py
@@ -0,0 +1,54 @@
+import xml.etree.ElementTree as ET
+import requests
+
+url = "https://mitjafelicijan.fra1.digitaloceanspaces.com/"
+
+def truncate_filename(filename, max_length):
+    if len(filename) <= max_length:
+        return filename
+    file_extension = filename.split('.')[-1]
+    return f"{filename[:max_length - len(file_extension) - 5]}....{file_extension}"
+
+response = requests.get(url)
+if response.status_code == 200:
+    xml_data = response.text
+    root = ET.fromstring(xml_data)
+
+    # Handle namespace
+    ns = {'s3': 'http://s3.amazonaws.com/doc/2006-03-01/'}
+
+    # Create an empty dictionary to hold the tree structure
+    tree = {}
+
+    for content in root.findall(".//s3:Contents", ns):
+        key = content.find("s3:Key", ns).text
+        parts = key.split("/")
+        node = tree
+        for part in parts:
+            if part:
+                node = node.setdefault(part, {})
+
+    # Function to convert the tree structure to markdown list
+    def tree_to_md(tree, indent=0, path=""):
+        md = ""
+        for k, v in tree.items():
+            if v:  # If the node has children, it's a directory
+                md += "  " * indent + f"- {k}\n"
+                md += tree_to_md(v, indent + 1, path=f"{path}{k}/")
+            else:  # If the node is empty, it's a file
+                file_url = f"{url}{path}{k}"
+                file_name = truncate_filename(k, 80)
+                md += "  " * indent + f"- [{file_name}](<{file_url}>)\n"
+        return md
+
+    md = tree_to_md(tree)
+
+    with open("templates/vault.md", "r") as fp:
+        content = fp.read()
+
+    new_content = content.replace("{CONTENT}", md)
+
+    with open("content/pages/vault.md", "w") as fp:
+        fp.write(new_content)
+else:
+    print(f"Failed to fetch XML data. Status code: {response.status_code}")