fastfetch/src/modules/localip.c

#include "fastfetch.h"
#include "common/printing.h"
#include "detection/localip/localip.h"

#define FF_LOCALIP_MODULE_NAME "Local IP"
#define FF_LOCALIP_NUM_FORMAT_ARGS 2

static int sortIpsV4First(const FFLocalIpResult* left, const FFLocalIpResult* right)
{
    int name = ffStrbufComp(&left->name, &right->name);
    if (name != 0)
        return name;

    if (left->ipv6 != right->ipv6)
        return left->ipv6 == false ? -1 : 1;

    return ffStrbufComp(&left->addr, &right->addr);
}

static int sortIpsV6First(const FFLocalIpResult* left, const FFLocalIpResult* right)
{
    int name = ffStrbufComp(&left->name, &right->name);
    if (name != 0)
        return name;

    if (left->ipv6 != right->ipv6)
        return left->ipv6 == true ? -1 : 1;

    return ffStrbufComp(&left->addr, &right->addr);
}

static void formatKey(const FFinstance* instance, const FFLocalIpResult* ip, FFstrbuf* key)
{
    if(instance->config.localIP.key.length == 0)
    {
        if(ip->name.length)
            ffStrbufSetF(key, FF_LOCALIP_MODULE_NAME " (%*s)", ip->name.length, ip->name.chars);
        else
            ffStrbufSetS(key, FF_LOCALIP_MODULE_NAME);
    }
    else
    {
        ffStrbufClear(key);
        ffParseFormatString(key, &instance->config.localIP.key, 1, (FFformatarg[]){
            {FF_FORMAT_ARG_TYPE_STRBUF, &ip->name}
        });
    }
}

static void printIpsOneline(FFlist* ips)
{
    uint32_t index = 0;
    FF_LIST_FOR_EACH(FFLocalIpResult, ip, *ips)
    {
        if (index++ > 0)
            putchar(' ');
        ffStrbufWriteTo(&ip->addr, stdout);
    }
}

static void printIpsWithKey(FFinstance* instance, FFlist* ips)
{
    FFLocalIpResult* ip = (FFLocalIpResult*) ffListGet(ips, 0);
    if (instance->config.localIpCompactType == FF_LOCALIP_COMPACT_TYPE_MULTILINE)
    {
        FF_STRBUF_AUTO_DESTROY key;
        ffStrbufInit(&key);
        formatKey(instance, ip, &key);
        ffPrintLogoAndKey(instance, key.chars, 0, NULL);
    }
    printIpsOneline(ips);
    if (instance->config.localIpCompactType == FF_LOCALIP_COMPACT_TYPE_ONELINE)
        printf(" (%s) ", ip->name.chars);
    else
        putchar('\n');
}

void ffPrintLocalIp(FFinstance* instance)
{
    FF_LIST_AUTO_DESTROY results;
    ffListInit(&results, sizeof(FFLocalIpResult));

    const char* error = ffDetectLocalIps(instance, &results);

    if(error)
    {
        ffPrintError(instance, FF_LOCALIP_MODULE_NAME, 0, &instance->config.localIP, "%s", error);
        return;
    }

    if(results.length == 0)
    {
        ffPrintError(instance, FF_LOCALIP_MODULE_NAME, 0, &instance->config.localIP, "Failed to detect any IPs");
        return;
    }

    ffListSort(&results, instance->config.localIpV6First ? (void*) sortIpsV6First : (void*) sortIpsV4First);

    if (instance->config.localIpCompactType != FF_LOCALIP_COMPACT_TYPE_NONE)
    {
        if (instance->config.localIpCompactType == FF_LOCALIP_COMPACT_TYPE_ONELINE)
            ffPrintLogoAndKey(instance, FF_LOCALIP_MODULE_NAME, 0, &instance->config.localIP.key);

        FF_LIST_AUTO_DESTROY ips;
        ffListInit(&ips, sizeof(FFLocalIpResult)); // weak references

        FF_LIST_FOR_EACH(FFLocalIpResult, ip, results)
        {
            if (ips.length > 0 && !ffStrbufEqual(&ip->name, &ip[-1].name))
            {
                printIpsWithKey(instance, &ips);
                ips.length = 0;
            }
            *(FFLocalIpResult*)ffListAdd(&ips) = *ip;
        }
        printIpsWithKey(instance, &ips);
    }
    else
    {
        FF_STRBUF_AUTO_DESTROY key;
        ffStrbufInit(&key);

        FF_LIST_FOR_EACH(FFLocalIpResult, ip, results)
        {
            formatKey(instance, ip, &key);
            if(instance->config.localIP.outputFormat.length == 0)
            {
                ffPrintLogoAndKey(instance, key.chars, 0, NULL);
                ffStrbufPutTo(&ip->addr, stdout);
            }
            else
            {
                ffPrintFormatString(instance, key.chars, 0, NULL, &instance->config.localIP.outputFormat, FF_LOCALIP_NUM_FORMAT_ARGS, (FFformatarg[]){
                    {FF_FORMAT_ARG_TYPE_STRBUF, &ip->addr},
                    {FF_FORMAT_ARG_TYPE_STRING, ip->ipv6 ? "IPv6" : "IPv4"}
                });
            }

            ffStrbufDestroy(&ip->name);
            ffStrbufDestroy(&ip->addr);
        }
    }
}