feat: adds XML tree (#625) (#869)

Closes #625

.github/workflows/update_icons_and_resources.yml

@@ -65,7 +65,7 @@ jobs:
         run: |
           git config --global user.email "homarr-labs@proton.me"
           git config --global user.name "Dashboard Icons Bot"
-          git add tree.json
+          git add tree.json tree.xml
           git commit -m "ci: github-actions: generate file tree" || exit 0
           git pull --rebase origin ${{ github.ref_name }}
           git push origin HEAD:${{ github.ref_name }}

scripts/generate_file_tree.py

@@ -1,10 +1,13 @@
 import os
 import json
 import sys
+import base64
 from pathlib import Path
+import re
 
 def generate_folder_tree(paths):
     tree = {}
+    webp_files = []
     for path in paths:
         resolved_path = Path(path).resolve()
         base_folder = resolved_path.name or Path.cwd().name
@@ -13,7 +16,36 @@ def generate_folder_tree(paths):
             key = base_folder if relative_path == '.' else os.path.join(base_folder, relative_path)
             if files:
                 tree[key] = sorted(files)  # Sort the list of files alphabetically
-    return tree
+                # Collect WebP files for XML generation
+                for file in files:
+                    if file.lower().endswith('.webp'):
+                        webp_files.append(os.path.join(root, file))
+    return tree, webp_files
+
+def generate_webp_xml(webp_files):
+    json_obj_list = []
+    for path in webp_files:
+        with open(path, "rb") as image_file:
+            # Encode the WebP image to base64
+            data = base64.b64encode(image_file.read()).decode('ascii')
+
+            # Build the base64 string for data URI
+            data_string = f"data:image/webp;base64,{data}"
+
+            # Extract name from the path
+            name = Path(path).stem
+
+            # Append the base64 encoded WebP data into the list
+            json_obj_list.append({
+                "data": data_string,
+                "w": 48,
+                "h": 48,
+                "title": name,
+                "aspect": "fixed"
+            })
+
+    # Convert to JSON string with no extra spaces for XML output
+    return json.dumps(json_obj_list, separators=(',', ':'))
 
 if __name__ == "__main__":
     # Adjust paths to be one level up
@@ -24,14 +56,24 @@ if __name__ == "__main__":
         print("Please provide at least one folder path.")
         sys.exit(1)
 
-    # Generate the combined folder tree
-    folder_tree = generate_folder_tree(folder_paths)
+    # Generate the folder tree and get WebP files
+    folder_tree, webp_files = generate_folder_tree(folder_paths)
 
     # Write the JSON structure to 'tree.json' in the parent folder
     root_dir = Path(__file__).resolve().parent
     tree_json_path = root_dir.parent / 'tree.json'
-
     with open(tree_json_path, 'w') as f:
         json.dump(folder_tree, f, indent=4, sort_keys=True)  # Sort the keys in the JSON output
-
     print(f"Folder tree successfully written to '{tree_json_path}'.")
+
+    # Generate WebP XML
+    if webp_files:
+        webp_json = generate_webp_xml(webp_files)
+
+        # Write XML structure (with wrapped JSON) to 'tree.xml'
+        xml_output_path = root_dir.parent / 'tree.xml'
+        with open(xml_output_path, 'w') as xml_output:
+            print(f"<mxlibrary>{webp_json}</mxlibrary>", file=xml_output)
+        print(f"WebP assets successfully written to '{xml_output_path}'.")
+    else:
+        print("No WebP files found for XML generation.")