arg.inl

#include <stdbool.h>
#include <tchar.h>

bool* g_argb = NULL;
int g_argc = 0;
const TCHAR* const* g_argv = NULL;
const TCHAR* g_argdescription = NULL;
bool g_argshowUsage = false;
typedef struct TagArgArgDescription
{
    const TCHAR*  arg;
    const TCHAR*  value;
    const TCHAR*  desc;
} ArgArgDescription;
ArgArgDescription g_argargdescription[20];
int g_argargdescriptioncount = 0;
typedef struct TagArgArgNumDescription
{
    const TCHAR*  arg;
    const TCHAR*  desc;
    bool used;
} ArgArgNumDescription;
ArgArgNumDescription g_argargnumdescription[10];
int g_argargnumdescriptioncount = 0;
int g_argargnumoptional = 65536;

#ifdef __cplusplus
#define ARG_OPTIONAL(x) = x
#else
#define ARG_OPTIONAL(x)
#endif
#define ARG_COLOR(n, s) "\x1B[" #n "m" s "\x1B[0m"
#ifdef _WINDOWS
#define ARG_COMMAND(s) s
#define ARG_OPTION(s) s
#define ARG_VALUE(s) s
#define ARG_ERROR(s) s
#else
#define ARG_COMMAND(s) ARG_COLOR(37, s)
#define ARG_OPTION(s) ARG_COLOR(33, s)
#define ARG_VALUE(s) ARG_COLOR(36, s)
#define ARG_ERROR(s) ARG_COLOR(31, s)
#endif

bool argswitch(const TCHAR* argf, const TCHAR* desc);

void arginit(int argc, const TCHAR* const argv[], const TCHAR* argdescription ARG_OPTIONAL(NULL))
{
    g_argc = argc;
    g_argv = argv;
    g_argb = (bool*) malloc(argc * sizeof(bool));
    for (int argi = 1; argi < g_argc; ++argi)
    {
        g_argb[argi] = false;
    }

    g_argdescription = argdescription;
}

const TCHAR* argapp()
{
    const TCHAR* app = _tcsrchr(g_argv[0], _T('\\'));
    return app == NULL ? g_argv[0] : app + 1;
}

bool argcleanup()
{
#ifdef _WINDOWS
    TCHAR msg[1024] = TEXT("");
    int msgoffset = 0;
#define msg(...) msgoffset += _stprintf_s(msg + msgoffset, ARRAYSIZE(msg) - msgoffset, __VA_ARGS__)
#else
#define msg(...) _ftprintf(stderr, __VA_ARGS__)
#endif

    g_argshowUsage = argswitch(_T("/?"), _T("Show usage"));
    bool ret = true;

    for (int argi = 1; argi < g_argc; ++argi)
    {
        if (!g_argb[argi])
        {
            const TCHAR* arg = g_argv[argi];
            if (arg[0] == '/')
            {
                msg(_T(ARG_ERROR("Unknown option:") " \"%s\".\n"), arg);
                ret = false;
            }
            else
            {
                msg(_T(ARG_ERROR("Unknown argument:") " \"%s\".\n"), arg);
                ret = false;
            }
        }
    }

    if (!g_argshowUsage)
    {
        for (int i = 0; i < g_argargnumdescriptioncount && i < g_argargnumoptional; ++i)
        {
            const ArgArgNumDescription* argargnumdescription = &g_argargnumdescription[i];
            if (!argargnumdescription->used)
            {
                msg(_T(ARG_ERROR("Missing argument:") " %s \"%s\".\n"), argargnumdescription->arg, argargnumdescription->desc);
                ret = false;
            }
        }
    }

    free(g_argb);
    g_argb = NULL;

#ifdef _WINDOWS
    if (msg[0] != TEXT('\0'))
        MessageBox(NULL, msg, argapp(), MB_OK | MB_ICONERROR);
#endif
#undef msg
    return ret;
}

bool argusage(bool bforce ARG_OPTIONAL(false))
{
#ifdef _WINDOWS
    TCHAR msg[1024] = TEXT("");
    int msgoffset = 0;
#define msg(...) msgoffset += _stprintf_s(msg + msgoffset, ARRAYSIZE(msg) - msgoffset, __VA_ARGS__)
#else
#define msg(...) _ftprintf(stdout, __VA_ARGS__)
#endif

    if (!bforce && !g_argshowUsage)
        return false;


    if (g_argdescription != NULL)
        msg(_T(ARG_COMMAND("%s") " - %s\n\n"), argapp(), g_argdescription);
    else if (g_argargdescriptioncount == 0)
        msg(_T(ARG_COMMAND("%s") "\n\n"), argapp());
    if (g_argargdescriptioncount > 0)
    {
        msg(_T(ARG_COMMAND("%s")), argapp());
        int i;
        for (i = 0; i < g_argargdescriptioncount; ++i)
        {
            const ArgArgDescription* argargdescription = &g_argargdescription[i];
            if (argargdescription->value != NULL)
                msg(_T(" [" ARG_OPTION("%s") "=" ARG_VALUE("%s") "]"), argargdescription->arg, argargdescription->value);
            else
                msg(_T(" [" ARG_OPTION("%s") "]"), argargdescription->arg);
        }
        for (i = 0; i < g_argargnumdescriptioncount; ++i)
        {
            const ArgArgNumDescription* argargnumdescription = &g_argargnumdescription[i];
            if (i >= g_argargnumoptional)
                msg(_T(" [" ARG_OPTION("%s") "]"), argargnumdescription->arg);
            else
                msg(_T(" <" ARG_OPTION("%s")) ">", argargnumdescription->arg);
        }
        msg(_T("\n"));
        for (i = 0; i < g_argargdescriptioncount; ++i)
        {
            const ArgArgDescription* argargdescription = &g_argargdescription[i];
            if (argargdescription->desc != NULL)
                msg(_T("   " ARG_OPTION("%s") "\t%s\n"), argargdescription->arg, argargdescription->desc);
        }
        for (i = 0; i < g_argargnumdescriptioncount; ++i)
        {
            const ArgArgNumDescription* argargnumdescription = &g_argargnumdescription[i];
            if (argargnumdescription->desc != NULL)
                msg(_T("   " ARG_OPTION("%s") "\t%s\n"), argargnumdescription->arg, argargnumdescription->desc);
        }
    }
#ifdef _WINDOWS
    MessageBox(NULL, msg, argapp(), MB_OK | MB_ICONINFORMATION);
#endif
#undef msg
    return true;
}

bool argswitch(const TCHAR* argf, const TCHAR* desc)
{
    ArgArgDescription* argargdescription = &g_argargdescription[g_argargdescriptioncount++];
    argargdescription->arg = argf;
    argargdescription->value = NULL;
    argargdescription->desc = desc;
    for (int argi = 1; argi < g_argc; ++argi)
    {
        const TCHAR* arg = g_argv[argi];
        if (_tcsicmp(arg, argf) == 0)
        {
            g_argb[argi] = true;
            return true;
        }
    }
    return false;
}

const TCHAR* argvalue(const TCHAR* argf, const TCHAR* def, const TCHAR* descvalue, const TCHAR* desc)
{
    ArgArgDescription* argargdescription = &g_argargdescription[g_argargdescriptioncount++];
    argargdescription->arg = argf;
    argargdescription->value = descvalue;
    argargdescription->desc = desc;
    const size_t len = _tcslen(argf);
    for (int argi = 1; argi < g_argc; ++argi)
    {
        const TCHAR* arg = g_argv[argi];
        if (_tcsnicmp(arg, argf, len) == 0 && arg[len] == _T('='))
        {
            g_argb[argi] = true;
            return arg + len + 1;
        }
    }
    return def;
}

int argvalueint(const TCHAR* argf, int def, const TCHAR* descvalue, const TCHAR* desc)
{
    const TCHAR* ret = argvalue(argf, NULL, descvalue, desc);
    return ret ? _ttoi(ret) : def;
}

const TCHAR* argnum(int i, const TCHAR* def, const TCHAR* descvalue, const TCHAR* desc)
{
    ArgArgNumDescription* argargnumdescription = &g_argargnumdescription[g_argargnumdescriptioncount++];
    argargnumdescription->arg = descvalue;
    argargnumdescription->desc = desc;
    argargnumdescription->used = false;
    for (int argi = 1; argi < g_argc && i >= 0; ++argi)
    {
        const TCHAR* arg = g_argv[argi];
        if (arg[0] != '/')
        {
            --i;
            if (i == 0)
            {
                g_argb[argi] = true;
                argargnumdescription->used = true;
                return arg;
            }
        }
    }
    // if g_argargnumoptional > g_argargnumdescriptioncount then def is useless
    if (i < 0)
#ifdef _WINDOWS
        MessageBox(NULL, _T("Error argument number\n"), argapp(), MB_OK | MB_ICONERROR);
#else
        _ftprintf(stderr, _T(ARG_ERROR("Error argument number") "\n"));
#endif
    return def;
}

void argoptional()
{
    g_argargnumoptional = g_argargnumdescriptioncount;
}

const TCHAR* argnext(const TCHAR* def, const TCHAR* descvalue, const TCHAR* desc)
{
    if (def != NULL && g_argargnumoptional == 65536)
        argoptional();

    ArgArgNumDescription* argargnumdescription = &g_argargnumdescription[g_argargnumdescriptioncount++];
    argargnumdescription->arg = descvalue;
    argargnumdescription->desc = desc;
    argargnumdescription->used = false;
    for (int argi = 1; argi < g_argc; ++argi)
    {
        const TCHAR* arg = g_argv[argi];
        if (arg[0] != '/' && !g_argb[argi])
        {
            g_argb[argi] = true;
            argargnumdescription->used = true;
            return arg;
        }
    }
    // if g_argargnumoptional > g_argargnumdescriptioncount then def is useless
    return def;
}