From f1cb6b1304d2f2e71876113b0c66274e57299273 Mon Sep 17 00:00:00 2001
From: Alec Murphy <alec@checksum.fail>
Date: Mon, 17 Feb 2025 08:17:51 -0500
Subject: [PATCH] Slon/Modules/Db: Add following to database

---
 Slon/Modules/Db.HC | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/Slon/Modules/Db.HC b/Slon/Modules/Db.HC
index f8c29bb..2388a9d 100644
--- a/Slon/Modules/Db.HC
+++ b/Slon/Modules/Db.HC
@@ -64,6 +64,26 @@ U0 @slon_db_load_followers_from_disk()
     db->set("followers", followers, JSON_OBJECT);
 }
 
+U0 @slon_db_load_following_from_disk()
+{
+    JsonObject* following = Json.CreateObject();
+    U8 scratch_buffer[256];
+    StrPrint(scratch_buffer, "%s/following/*.json", SLON_DB_PATH);
+    CDirEntry* files = FilesFind(scratch_buffer);
+    CDirEntry* de = files;
+    JsonArray* following_array = NULL;
+    while (de) {
+        following_array = Json.ParseFile(de->full_name);
+        if (following_array) {
+            StrFind(".json", de->name)[0] = NULL;
+            following->set(de->name, following_array, JSON_ARRAY);
+        }
+        de = de->next;
+    }
+    DirTreeDel(files);
+    db->set("following", following, JSON_OBJECT);
+}
+
 U0 @slon_db_load_statuses_from_disk()
 {
     JsonObject* statuses = Json.CreateObject();
@@ -161,6 +181,17 @@ U0 @slon_db_save_followers_to_disk()
     }
 }
 
+U0 @slon_db_save_following_to_disk()
+{
+    U8 scratch_buffer[256];
+    JsonKey* key = db->o("following")->keys;
+    while (key) {
+        StrPrint(scratch_buffer, "%s/following/%s.json", SLON_DB_PATH, key->name);
+        Json.DumpToFile(scratch_buffer, key->value);
+        key = key->next;
+    }
+}
+
 U0 @slon_db_save_statuses_to_disk()
 {
     U8 scratch_buffer[256];
@@ -189,6 +220,7 @@ U0 @slon_db_save_to_disk()
     @slon_db_save_actors_to_disk();
     @slon_db_save_apps_to_disk();
     @slon_db_save_followers_to_disk();
+    @slon_db_save_following_to_disk();
     @slon_db_save_instance_to_disk();
     @slon_db_save_oauth_to_disk();
     @slon_db_save_private_keys_to_disk();
@@ -206,6 +238,7 @@ U0 @slon_db_load_from_defaults()
     db->set("private_keys_binary", Json.CreateObject(), JSON_OBJECT);
     db->set("public_keys", Json.CreateObject(), JSON_OBJECT);
     db->set("followers", Json.CreateObject(), JSON_OBJECT);
+    db->set("following", Json.CreateObject(), JSON_OBJECT);
     db->set("instance", Json.ParseFile("M:/Slon/Static/defaults/instance.json"), JSON_OBJECT);
     db->set("statuses", Json.CreateObject(), JSON_OBJECT);
     db->set("timelines", Json.CreateObject(), JSON_OBJECT);
@@ -229,6 +262,7 @@ U0 @slon_db_load_from_disk()
     db->set("private_keys_binary", Json.CreateObject(), JSON_OBJECT);
     db->set("public_keys", Json.CreateObject(), JSON_OBJECT);
     @slon_db_load_followers_from_disk();
+    @slon_db_load_following_from_disk();
     @slon_db_load_instance_from_disk();
     @slon_db_load_oauth_from_disk();
     @slon_db_load_statuses_from_disk();