From acd353b8b3c95db2672a18d67ada2e642afa074d 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:26 +0800
Subject: [PATCH] Chassis: add Json printing support

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

diff --git a/src/modules/chassis/chassis.c b/src/modules/chassis/chassis.c
index d685ceb0b..93c91c942 100644
--- a/src/modules/chassis/chassis.c
+++ b/src/modules/chassis/chassis.c
@@ -52,7 +52,7 @@ exit:
 
 void ffInitChassisOptions(FFChassisOptions* options)
 {
-    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_CHASSIS_MODULE_NAME, ffParseChassisCommandOptions, ffParseChassisJsonObject, ffPrintChassis, NULL);
+    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_CHASSIS_MODULE_NAME, ffParseChassisCommandOptions, ffParseChassisJsonObject, ffPrintChassis, ffGenerateChassisJson);
     ffOptionInitModuleArg(&options->moduleArgs);
 }
 
@@ -87,3 +87,36 @@ void ffParseChassisJsonObject(FFChassisOptions* options, yyjson_val* module)
         ffPrintError(FF_CHASSIS_MODULE_NAME, 0, &options->moduleArgs, "Unknown JSON key %s", key);
     }
 }
+
+void ffGenerateChassisJson(FF_MAYBE_UNUSED FFChassisOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module)
+{
+    FFChassisResult result;
+    ffStrbufInit(&result.type);
+    ffStrbufInit(&result.vendor);
+    ffStrbufInit(&result.version);
+
+    const char* error = ffDetectChassis(&result);
+
+    if (error)
+    {
+        yyjson_mut_obj_add_str(doc, module, "error", error);
+        goto exit;
+    }
+
+    if(result.type.length == 0)
+    {
+        yyjson_mut_obj_add_str(doc, module, "error", "chassis_type is not set by O.E.M.");
+        goto exit;
+    }
+
+    yyjson_mut_val* obj = yyjson_mut_obj(doc);
+    yyjson_mut_obj_add_val(doc, module, "result", obj);
+    yyjson_mut_obj_add_strbuf(doc, obj, "type", &result.type);
+    yyjson_mut_obj_add_strbuf(doc, obj, "vendor", &result.vendor);
+    yyjson_mut_obj_add_strbuf(doc, obj, "version", &result.version);
+
+exit:
+    ffStrbufDestroy(&result.type);
+    ffStrbufDestroy(&result.vendor);
+    ffStrbufDestroy(&result.version);
+}
diff --git a/src/modules/chassis/chassis.h b/src/modules/chassis/chassis.h
index 5fe4bb62f..c5553f772 100644
--- a/src/modules/chassis/chassis.h
+++ b/src/modules/chassis/chassis.h
@@ -9,3 +9,4 @@ void ffInitChassisOptions(FFChassisOptions* options);
 bool ffParseChassisCommandOptions(FFChassisOptions* options, const char* key, const char* value);
 void ffDestroyChassisOptions(FFChassisOptions* options);
 void ffParseChassisJsonObject(FFChassisOptions* options, yyjson_val* module);
+void ffGenerateChassisJson(FFChassisOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module);