From a7c6e1165c1a43522dc3238b19a0f6803b486900 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E9=80=9A=E6=B4=B2?= <zhangsongcui3371@sina.com>
Date: Wed, 6 Sep 2023 16:01:10 +0800
Subject: [PATCH] Swap: JSON format support

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

diff --git a/src/modules/swap/swap.c b/src/modules/swap/swap.c
index 1361009f9..331cbf26d 100644
--- a/src/modules/swap/swap.c
+++ b/src/modules/swap/swap.c
@@ -75,7 +75,7 @@ void ffPrintSwap(FFSwapOptions* options)
 
 void ffInitSwapOptions(FFSwapOptions* options)
 {
-    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_SWAP_MODULE_NAME, ffParseSwapCommandOptions, ffParseSwapJsonObject, ffPrintSwap, NULL);
+    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_SWAP_MODULE_NAME, ffParseSwapCommandOptions, ffParseSwapJsonObject, ffPrintSwap, ffGenerateSwapJson);
     ffOptionInitModuleArg(&options->moduleArgs);
 }
 
@@ -110,3 +110,19 @@ void ffParseSwapJsonObject(FFSwapOptions* options, yyjson_val* module)
         ffPrintError(FF_SWAP_MODULE_NAME, 0, &options->moduleArgs, "Unknown JSON key %s", key);
     }
 }
+
+void ffGenerateSwapJson(FF_MAYBE_UNUSED FFSwapOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module)
+{
+    FFSwapResult storage;
+    const char* error = ffDetectSwap(&storage);
+
+    if(error)
+    {
+        yyjson_mut_obj_add_str(doc, module, "error", error);
+        return;
+    }
+
+    yyjson_mut_val* obj = yyjson_mut_obj_add_obj(doc, module, "result");
+    yyjson_mut_obj_add_uint(doc, obj, "total", storage.bytesTotal);
+    yyjson_mut_obj_add_uint(doc, obj, "used", storage.bytesUsed);
+}
diff --git a/src/modules/swap/swap.h b/src/modules/swap/swap.h
index 5921ad9c5..fbb125a44 100644
--- a/src/modules/swap/swap.h
+++ b/src/modules/swap/swap.h
@@ -9,3 +9,4 @@ void ffInitSwapOptions(FFSwapOptions* options);
 bool ffParseSwapCommandOptions(FFSwapOptions* options, const char* key, const char* value);
 void ffDestroySwapOptions(FFSwapOptions* options);
 void ffParseSwapJsonObject(FFSwapOptions* options, yyjson_val* module);
+void ffGenerateSwapJson(FFSwapOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module);