From 01aa9b951661bcddd8885df52f4d9a86df0107e7 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, 30 Aug 2023 14:13:39 +0800
Subject: [PATCH] Command: add JSON printing support

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

diff --git a/src/modules/command/command.c b/src/modules/command/command.c
index ad429d201..393aab97b 100644
--- a/src/modules/command/command.c
+++ b/src/modules/command/command.c
@@ -36,7 +36,7 @@ void ffPrintCommand(FFCommandOptions* options)
 
 void ffInitCommandOptions(FFCommandOptions* options)
 {
-    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_COMMAND_MODULE_NAME, ffParseCommandCommandOptions, ffParseCommandJsonObject, ffPrintCommand, NULL);
+    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_COMMAND_MODULE_NAME, ffParseCommandCommandOptions, ffParseCommandJsonObject, ffPrintCommand, ffGenerateCommandJson);
     ffOptionInitModuleArg(&options->moduleArgs);
 
     ffStrbufInitStatic(&options->shell,
@@ -107,3 +107,32 @@ void ffParseCommandJsonObject(FFCommandOptions* options, yyjson_val* module)
         ffPrintError(FF_COMMAND_MODULE_NAME, 0, &options->moduleArgs, "Unknown JSON key %s", key);
     }
 }
+
+void ffGenerateCommandJson(FF_MAYBE_UNUSED FFCommandOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module)
+{
+    FF_STRBUF_AUTO_DESTROY result = ffStrbufCreate();
+    const char* error = ffProcessAppendStdOut(&result, (char* const[]){
+        options->shell.chars,
+        #ifdef _WIN32
+        "/c",
+        #else
+        "-c",
+        #endif
+        options->text.chars,
+        NULL
+    });
+
+    if(error)
+    {
+        yyjson_mut_obj_add_str(doc, module, "error", error);
+        return;
+    }
+
+    if(!result.length)
+    {
+        yyjson_mut_obj_add_str(doc, module, "error", "No result printed");
+        return;
+    }
+
+    yyjson_mut_obj_add_strbuf(doc, module, "result", &result);
+}
diff --git a/src/modules/command/command.h b/src/modules/command/command.h
index 8ee5cc82c..ec8e2bab3 100644
--- a/src/modules/command/command.h
+++ b/src/modules/command/command.h
@@ -9,3 +9,4 @@ void ffInitCommandOptions(FFCommandOptions* options);
 bool ffParseCommandCommandOptions(FFCommandOptions* options, const char* key, const char* value);
 void ffDestroyCommandOptions(FFCommandOptions* options);
 void ffParseCommandJsonObject(FFCommandOptions* options, yyjson_val* module);
+void ffGenerateCommandJson(FFCommandOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module);