DNS: add new module

src/modules/dns/dns.c

@@ -0,0 +1,141 @@
+#include "common/printing.h"
+#include "common/jsonconfig.h"
+#include "detection/dns/dns.h"
+#include "modules/dns/dns.h"
+#include "util/stringUtils.h"
+
+#define FF_DNS_NUM_FORMAT_ARGS 1
+
+void ffPrintDNS(FFDNSOptions* options)
+{
+    FF_LIST_AUTO_DESTROY result = ffListCreate(sizeof(FFstrbuf));
+
+    const char* error = ffDetectDNS(&result);
+
+    if (error)
+    {
+        ffPrintError(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "%s", error);
+        return;
+    }
+
+    if (result.length == 0)
+    {
+        ffPrintError(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "NO DNS detected");
+        return;
+    }
+
+    FF_STRBUF_AUTO_DESTROY buf = ffStrbufCreate();
+    FF_LIST_FOR_EACH(FFstrbuf, item, result)
+    {
+        if (buf.length)
+            ffStrbufAppendC(&buf, ' ');
+        ffStrbufAppend(&buf, item);
+    }
+
+    if(options->moduleArgs.outputFormat.length == 0)
+    {
+        ffPrintLogoAndKey(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT);
+
+        ffStrbufPutTo(&buf, stdout);
+    }
+    else
+    {
+        FF_PRINT_FORMAT_CHECKED(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, FF_DNS_NUM_FORMAT_ARGS, ((FFformatarg[]) {
+            {FF_FORMAT_ARG_TYPE_STRBUF, &buf, "result"},
+        }));
+    }
+
+    FF_LIST_FOR_EACH(FFstrbuf, item, result)
+    {
+        ffStrbufDestroy(item);
+    }
+}
+
+bool ffParseDNSCommandOptions(FFDNSOptions* options, const char* key, const char* value)
+{
+    const char* subKey = ffOptionTestPrefix(key, FF_DNS_MODULE_NAME);
+    if (!subKey) return false;
+    if (ffOptionParseModuleArgs(key, subKey, value, &options->moduleArgs))
+        return true;
+
+    return false;
+}
+
+void ffParseDNSJsonObject(FFDNSOptions* options, yyjson_val* module)
+{
+    yyjson_val *key_, *val;
+    size_t idx, max;
+    yyjson_obj_foreach(module, idx, max, key_, val)
+    {
+        const char* key = yyjson_get_str(key_);
+        if(ffStrEqualsIgnCase(key, "type"))
+            continue;
+
+        if (ffJsonConfigParseModuleArgs(key, val, &options->moduleArgs))
+            continue;
+
+        ffPrintError(FF_DNS_MODULE_NAME, 0, &options->moduleArgs, FF_PRINT_TYPE_DEFAULT, "Unknown JSON key %s", key);
+    }
+}
+
+void ffGenerateDNSJsonConfig(FFDNSOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module)
+{
+    __attribute__((__cleanup__(ffDestroyDNSOptions))) FFDNSOptions defaultOptions;
+    ffInitDNSOptions(&defaultOptions);
+
+    ffJsonConfigGenerateModuleArgsConfig(doc, module, &defaultOptions.moduleArgs, &options->moduleArgs);
+}
+
+void ffGenerateDNSJsonResult(FF_MAYBE_UNUSED FFDNSOptions* options, yyjson_mut_doc* doc, yyjson_mut_val* module)
+{
+    FF_LIST_AUTO_DESTROY result = ffListCreate(sizeof(FFstrbuf));
+
+    const char* error = ffDetectDNS(&result);
+
+    if (error)
+    {
+        yyjson_mut_obj_add_str(doc, module, "error", error);
+        goto exit;
+    }
+
+    yyjson_mut_val* arr = yyjson_mut_obj_add_arr(doc, module, "result");
+
+    FF_LIST_FOR_EACH(FFstrbuf, item, result)
+    {
+        yyjson_mut_arr_add_strbuf(doc, arr, item);
+    }
+
+exit:
+    FF_LIST_FOR_EACH(FFstrbuf, item, result)
+    {
+        ffStrbufDestroy(item);
+    }
+}
+
+void ffPrintDNSHelpFormat(void)
+{
+    FF_PRINT_MODULE_FORMAT_HELP_CHECKED(FF_DNS_MODULE_NAME, "{1}", FF_DNS_NUM_FORMAT_ARGS, ((const char* []) {
+        "DNS result - result",
+    }));
+}
+
+void ffInitDNSOptions(FFDNSOptions* options)
+{
+    ffOptionInitModuleBaseInfo(
+        &options->moduleInfo,
+        FF_DNS_MODULE_NAME,
+        "Print configured DNS servers",
+        ffParseDNSCommandOptions,
+        ffParseDNSJsonObject,
+        ffPrintDNS,
+        ffGenerateDNSJsonResult,
+        ffPrintDNSHelpFormat,
+        ffGenerateDNSJsonConfig
+    );
+    ffOptionInitModuleArg(&options->moduleArgs);
+}
+
+void ffDestroyDNSOptions(FFDNSOptions* options)
+{
+    ffOptionDestroyModuleArg(&options->moduleArgs);
+}

src/modules/dns/dns.h

@@ -0,0 +1,9 @@
+#pragma once
+
+#include "fastfetch.h"
+
+#define FF_DNS_MODULE_NAME "DNS"
+
+void ffPrintDNS(FFDNSOptions* options);
+void ffInitDNSOptions(FFDNSOptions* options);
+void ffDestroyDNSOptions(FFDNSOptions* options);

src/modules/dns/option.h

@@ -0,0 +1,11 @@
+#pragma once
+
+// This file will be included in "fastfetch.h", do NOT put unnecessary things here
+
+#include "common/option.h"
+
+typedef struct FFDNSOptions
+{
+    FFModuleBaseInfo moduleInfo;
+    FFModuleArgs moduleArgs;
+} FFDNSOptions;

src/modules/modules.h

@@ -23,6 +23,7 @@
 #include "modules/diskio/diskio.h"
 #include "modules/display/display.h"
 #include "modules/de/de.h"
+#include "modules/dns/dns.h"
 #include "modules/editor/editor.h"
 #include "modules/font/font.h"
 #include "modules/gamepad/gamepad.h"

src/modules/options.h

@@ -23,6 +23,7 @@
 #include "modules/disk/option.h"
 #include "modules/diskio/option.h"
 #include "modules/display/option.h"
+#include "modules/dns/option.h"
 #include "modules/editor/option.h"
 #include "modules/font/option.h"
 #include "modules/host/option.h"