From d5ca6aa462c3b131c980055142045f3cbdb3b47c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E9=80=9A=E6=B4=B2?= <zhangsongcui3371@sina.com>
Date: Thu, 31 Aug 2023 14:13:14 +0800
Subject: [PATCH] Disk: add JSON printing support

---
 src/modules/disk/disk.c | 46 ++++++++++++++++++++++++++++++++++++++++-
 src/modules/disk/disk.h |  1 +
 2 files changed, 46 insertions(+), 1 deletion(-)

diff --git a/src/modules/disk/disk.c b/src/modules/disk/disk.c
index f5d2bae98..4f74a380a 100644
--- a/src/modules/disk/disk.c
+++ b/src/modules/disk/disk.c
@@ -202,7 +202,7 @@ void ffPrintDisk(FFDiskOptions* options)
 
 void ffInitDiskOptions(FFDiskOptions* options)
 {
-    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_DISK_MODULE_NAME, ffParseDiskCommandOptions, ffParseDiskJsonObject, ffPrintDisk, NULL);
+    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_DISK_MODULE_NAME, ffParseDiskCommandOptions, ffParseDiskJsonObject, ffPrintDisk, ffGenerateDiskJson);
     ffOptionInitModuleArg(&options->moduleArgs);
 
     ffStrbufInit(&options->folders);
@@ -370,3 +370,47 @@ void ffParseDiskJsonObject(FFDiskOptions* options, yyjson_val* module)
         ffPrintError(FF_DISK_MODULE_NAME, 0, &options->moduleArgs, "Unknown JSON key %s", key);
     }
 }
+
+void ffGenerateDiskJson(FFDiskOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module)
+{
+    FF_LIST_AUTO_DESTROY disks = ffListCreate(sizeof (FFDisk));
+    const char* error = ffDetectDisks(options, &disks);
+
+    if(error)
+    {
+        yyjson_mut_obj_add_str(doc, module, "result", error);
+        return;
+    }
+
+    yyjson_mut_val* arr = yyjson_mut_obj_add_arr(doc, module, "result");
+
+    FF_LIST_FOR_EACH(FFDisk, item, disks)
+    {
+        yyjson_mut_val* obj = yyjson_mut_arr_add_obj(doc, arr);
+        yyjson_mut_obj_add_uint(doc, obj, "bytesAvailable", item->bytesAvailable);
+        yyjson_mut_obj_add_uint(doc, obj, "bytesFree", item->bytesFree);
+        yyjson_mut_obj_add_uint(doc, obj, "bytesTotal", item->bytesTotal);
+        yyjson_mut_obj_add_uint(doc, obj, "bytesUsed", item->bytesUsed);
+        yyjson_mut_obj_add_uint(doc, obj, "filesTotal", item->filesTotal);
+        yyjson_mut_obj_add_uint(doc, obj, "filesUsed", item->filesUsed);
+        yyjson_mut_obj_add_strbuf(doc, obj, "filesystem", &item->filesystem);
+        yyjson_mut_obj_add_strbuf(doc, obj, "mountpoint", &item->mountpoint);
+        yyjson_mut_obj_add_strbuf(doc, obj, "name", &item->name);
+        yyjson_mut_val* typeArr = yyjson_mut_obj_add_arr(doc, obj, "type");
+        if(item->type & FF_DISK_VOLUME_TYPE_EXTERNAL_BIT)
+            yyjson_mut_arr_add_str(doc, typeArr, "External");
+        if(item->type & FF_DISK_VOLUME_TYPE_SUBVOLUME_BIT)
+            yyjson_mut_arr_add_str(doc, typeArr, "Subvolume");
+        if(item->type & FF_DISK_VOLUME_TYPE_HIDDEN_BIT)
+            yyjson_mut_arr_add_str(doc, typeArr, "Hidden");
+        if(item->type & FF_DISK_VOLUME_TYPE_READONLY_BIT)
+            yyjson_mut_arr_add_str(doc, typeArr, "Read-only");
+    }
+
+    FF_LIST_FOR_EACH(FFDisk, item, disks)
+    {
+        ffStrbufDestroy(&item->mountpoint);
+        ffStrbufDestroy(&item->filesystem);
+        ffStrbufDestroy(&item->name);
+    }
+}
diff --git a/src/modules/disk/disk.h b/src/modules/disk/disk.h
index 1f71f766b..eb4520b35 100644
--- a/src/modules/disk/disk.h
+++ b/src/modules/disk/disk.h
@@ -9,3 +9,4 @@ void ffInitDiskOptions(FFDiskOptions* options);
 bool ffParseDiskCommandOptions(FFDiskOptions* options, const char* key, const char* value);
 void ffDestroyDiskOptions(FFDiskOptions* options);
 void ffParseDiskJsonObject(FFDiskOptions* options, yyjson_val* module);
+void ffGenerateDiskJson(FFDiskOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module);