From 4672a55a30e06b8b27e828803b3de192251dcd9d Mon Sep 17 00:00:00 2001
From: Sam Chau <chaucted@gmail.com>
Date: Tue, 21 Jan 2025 09:08:54 +1030
Subject: [PATCH] add(script): support for bundling markdown files with
 frontmatter and custom JSON encoder

---
 scripts/bundle.py | 82 +++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 68 insertions(+), 14 deletions(-)

diff --git a/scripts/bundle.py b/scripts/bundle.py
index 0f66927..e2e5b40 100644
--- a/scripts/bundle.py
+++ b/scripts/bundle.py
@@ -1,20 +1,72 @@
 import yaml
 import json
 from pathlib import Path
-from datetime import datetime, timezone
+from datetime import datetime, timezone, date
+
+
+# Add this class at the top with the imports
+class DateTimeEncoder(json.JSONEncoder):
+
+    def default(self, obj):
+        if isinstance(obj, (date, datetime)):
+            return obj.isoformat()
+        return super().default(obj)
+
+
+def parse_frontmatter(content):
+    """Parse YAML frontmatter from markdown"""
+    if content.startswith("---"):
+        try:
+            second_sep = content[3:].find("---")
+            if second_sep != -1:
+                yaml_text = content[3:second_sep + 3]
+                meta = yaml.safe_load(yaml_text)
+                content = content[second_sep + 6:].strip()
+                return meta, content
+        except yaml.YAMLError:
+            pass
+    return {}, content
+
+
+def bundle_markdown(folder_name):
+    """Bundle markdown files with frontmatter"""
+    data = []
+    folder_path = Path(folder_name)
+
+    if folder_path.exists():
+        for md_file in folder_path.glob("*.md"):
+            with open(md_file) as f:
+                content = f.read()
+
+            meta, content = parse_frontmatter(content)
+            item = {
+                "_id":
+                md_file.stem,
+                "content":
+                content,
+                "last_modified":
+                datetime.fromtimestamp(md_file.stat().st_mtime,
+                                       tz=timezone.utc).isoformat(),
+                **meta
+            }
+            data.append(item)
+    return data
 
 
 def bundle_folder(folder_name):
-    """Bundle all YML files from a folder into a list"""
-    data = []
-    folder_path = Path(folder_name)
-    if folder_path.exists():
-        for yml_file in folder_path.glob("*.yml"):
-            with open(yml_file) as f:
-                item = yaml.safe_load(f)
-                item["_id"] = yml_file.stem
-                data.append(item)
-    return data
+    """Bundle files based on type"""
+    if folder_name == "wiki":
+        return bundle_markdown(folder_name)
+    else:
+        data = []
+        folder_path = Path(folder_name)
+        if folder_path.exists():
+            for yml_file in folder_path.glob("*.yml"):
+                with open(yml_file) as f:
+                    item = yaml.safe_load(f)
+                    item["_id"] = yml_file.stem
+                    data.append(item)
+        return data
 
 
 # Create bundles directory
@@ -22,15 +74,17 @@ Path("bundles").mkdir(exist_ok=True)
 
 # Define folders to bundle
 folders = [
-    "custom_formats", "profiles", "regex_patterns", "group_tiers", "dev_logs"
+    "custom_formats", "profiles", "regex_patterns", "group_tiers", "dev_logs",
+    "wiki"
 ]
 
 # Bundle each folder
 for folder in folders:
     data = bundle_folder(folder)
     bundle_path = f"bundles/{folder}.json"
+    # Here's where we use the encoder
     with open(bundle_path, "w") as f:
-        json.dump(data, f, indent=2)
+        json.dump(data, f, indent=2, cls=DateTimeEncoder)
 
 # Create version file
 version = {
@@ -38,4 +92,4 @@ version = {
     "folders": folders
 }
 with open("bundles/version.json", "w") as f:
-    json.dump(version, f, indent=2)
+    json.dump(version, f, indent=2, cls=DateTimeEncoder)