From 971a69ad1b009feaabda4b7deb39d10040cb3729 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E9=80=9A=E6=B4=B2?= <zhangsongcui3371@sina.com>
Date: Tue, 5 Sep 2023 15:57:18 +0800
Subject: [PATCH] Host: add JSON formatting support

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

diff --git a/src/modules/host/host.c b/src/modules/host/host.c
index 4fa03cb64..30df24f12 100644
--- a/src/modules/host/host.c
+++ b/src/modules/host/host.c
@@ -67,7 +67,7 @@ exit:
 
 void ffInitHostOptions(FFHostOptions* options)
 {
-    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_HOST_MODULE_NAME, ffParseHostCommandOptions, ffParseHostJsonObject, ffPrintHost, NULL);
+    ffOptionInitModuleBaseInfo(&options->moduleInfo, FF_HOST_MODULE_NAME, ffParseHostCommandOptions, ffParseHostJsonObject, ffPrintHost, ffGenerateHostJson);
     ffOptionInitModuleArg(&options->moduleArgs);
 }
 
@@ -102,3 +102,40 @@ void ffParseHostJsonObject(FFHostOptions* options, yyjson_val* module)
         ffPrintError(FF_HOST_MODULE_NAME, 0, &options->moduleArgs, "Unknown JSON key %s", key);
     }
 }
+
+void ffGenerateHostJson(FF_MAYBE_UNUSED FFHostOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module)
+{
+    FFHostResult host;
+    ffStrbufInit(&host.productFamily);
+    ffStrbufInit(&host.productName);
+    ffStrbufInit(&host.productVersion);
+    ffStrbufInit(&host.productSku);
+    ffStrbufInit(&host.sysVendor);
+    const char* error = ffDetectHost(&host);
+
+    if (error)
+    {
+        yyjson_mut_obj_add_str(doc, module, "error", error);
+        goto exit;
+    }
+
+    if (host.productFamily.length == 0 && host.productName.length == 0)
+    {
+        yyjson_mut_obj_add_str(doc, module, "error", "neither product_family nor product_name is set by O.E.M.");
+        goto exit;
+    }
+
+    yyjson_mut_val* obj = yyjson_mut_obj_add_obj(doc, module, "result");
+    yyjson_mut_obj_add_strbuf(doc, obj, "family", &host.productFamily);
+    yyjson_mut_obj_add_strbuf(doc, obj, "name", &host.productName);
+    yyjson_mut_obj_add_strbuf(doc, obj, "version", &host.productVersion);
+    yyjson_mut_obj_add_strbuf(doc, obj, "sku", &host.productSku);
+    yyjson_mut_obj_add_strbuf(doc, obj, "sysVender", &host.sysVendor);
+
+exit:
+    ffStrbufDestroy(&host.productFamily);
+    ffStrbufDestroy(&host.productName);
+    ffStrbufDestroy(&host.productVersion);
+    ffStrbufDestroy(&host.productSku);
+    ffStrbufDestroy(&host.sysVendor);
+}
diff --git a/src/modules/host/host.h b/src/modules/host/host.h
index 5a689288a..4ed4575a1 100644
--- a/src/modules/host/host.h
+++ b/src/modules/host/host.h
@@ -9,3 +9,4 @@ void ffInitHostOptions(FFHostOptions* options);
 bool ffParseHostCommandOptions(FFHostOptions* options, const char* key, const char* value);
 void ffDestroyHostOptions(FFHostOptions* options);
 void ffParseHostJsonObject(FFHostOptions* options, yyjson_val* module);
+void ffGenerateHostJson(FFHostOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module);