CLIArgs.h

#ifndef HL_CLI_ARGS_H
#define HL_CLI_ARGS_H

#include <cstdio>
#include <QCommandLineParser>

#include <hlaudio/hlaudio.h>
#include <hlcontrol/hlcontrol.h>

#include "CLICommon.h"

using namespace hula;

#define HL_OUT_FILE_SO        "f"
#define HL_OUT_FILE_LO        "out-file"
#define HL_DELAY_TIME_SO      "d"
#define HL_DELAY_TIME_LO      "delay"
#define HL_RECORD_TIME_SO     "t"
#define HL_RECORD_TIME_LO     "time"
#define HL_TRIGGER_RECORD_SO  "r"
#define HL_TRIGGER_RECORD_LO  "record"
#define HL_SAMPLE_RATE_SO     "s"
#define HL_SAMPLE_RATE_LO     "sample-rate"
#define HL_ENCODING_SO        "e"
#define HL_ENCODING_LO        "encoding"
#define HL_INPUT_DEVICE_SO    "i"
#define HL_INPUT_DEVICE_LO    "input-device"
#define HL_OUTPUT_DEVICE_SO   "o"
#define HL_OUTPUT_DEVICE_LO   "output-device"
#define HL_LIST_DEVICES_SO    "l"
#define HL_LIST_DEVICES_LO    "list"

typedef struct HulaImmediateArgs
{
    bool startRecord = false;
    bool exit = false;
} HulaImmediateArgs;

void invalidArg(std::string name, QString arg, std::string message = "")
{
    fprintf(stderr, "Invalid argument '%s' provided to option '%s'.\n", arg.toStdString().c_str(), name.c_str());
    if (message.size() > 0)
    {
        fprintf(stderr, "%s\n", message.c_str());
    }
}

bool parseArgsQt(QCoreApplication &app, HulaImmediateArgs &extraArgs)
{
    QCommandLineParser parser;
    parser.setApplicationDescription("Simple cross-platform audio loopback and recording.");

    parser.addHelpOption();
    parser.addVersionOption();

    parser.addOptions(
    {
        {{HL_OUT_FILE_SO, HL_OUT_FILE_LO}, QCoreApplication::translate("main", "Path to audio output file."), QCoreApplication::translate("main", "out-file-path")},
        {{HL_DELAY_TIME_SO, HL_DELAY_TIME_LO}, QCoreApplication::translate("main", "Duration, in seconds, of the countdown timer before record."), QCoreApplication::translate("main", "delay")},
        {{HL_RECORD_TIME_SO, HL_RECORD_TIME_LO}, QCoreApplication::translate("main", "Duration, in seconds, of the record."), QCoreApplication::translate("main", "record duration")},
        {{HL_TRIGGER_RECORD_SO, HL_TRIGGER_RECORD_LO}, QCoreApplication::translate("main", "Start the countdown/record immediately.")},
        {{HL_SAMPLE_RATE_SO, HL_SAMPLE_RATE_LO}, QCoreApplication::translate("main", "Desired sample rate of the output file."), QCoreApplication::translate("main", "sample rate")},
        //{{"b", "bit-depth"}, QCoreApplication::translate("main", "Sample format for the output file. Valid options are 8, 8u, 16, 16u, 32, 32u, 32f. This will default to 32f.")},
        {{HL_ENCODING_SO, HL_ENCODING_LO}, QCoreApplication::translate("main", "Encoding format for the output file. Valid options are WAV and MP3. This will default to WAV."), QCoreApplication::translate("main", "encoding")},
        {{HL_INPUT_DEVICE_SO, HL_INPUT_DEVICE_LO}, QCoreApplication::translate("main", "System name of the input device. This will default if not provided."), QCoreApplication::translate("main", "input-device-name")},
        {{HL_OUTPUT_DEVICE_SO, HL_OUTPUT_DEVICE_LO}, QCoreApplication::translate("main", "System name of the output device. This will default if not provided."), QCoreApplication::translate("main", "output-device-name")},
        {{HL_LIST_DEVICES_SO, HL_LIST_DEVICES_LO}, QCoreApplication::translate("main", "List available input and output devices.")}

        // -d --delay  Countdown timer before record.
        // -l --length Record duration.
        // -r --record Start recording/countdown timer immediately
        // -s --sample-rate Sample rate
        // -e --encoding Encoding format: valid formats WAV, MP3, guess from file, default to WAV
        // -i --input-device System name of input device, will default
        // -o --output-device System name of output device, will default
    });

    // This will exit if any of the args are incorrect
    parser.process(app);

    // Setup the settings module
    HulaSettings *settings = HulaSettings::getInstance();

    if (parser.isSet(HL_OUT_FILE_LO))
    {
        settings->setOutputFilePath(parser.value(HL_OUT_FILE_LO).toStdString());
    }

    if (parser.isSet(HL_DELAY_TIME_LO))
    {
        bool ok = false;
        double delay = parser.value(HL_DELAY_TIME_LO).toDouble(&ok);
        if (!ok)
        {
            invalidArg(HL_DELAY_TIME_LO, parser.value(HL_DELAY_TIME_LO));
            return false;
        }
        settings->setDelayTimer(delay);
    }

    if (parser.isSet(HL_RECORD_TIME_LO))
    {
        bool ok = false;
        double duration = parser.value(HL_RECORD_TIME_LO).toDouble(&ok);
        if (!ok)
        {
            invalidArg(HL_RECORD_TIME_LO, parser.value(HL_RECORD_TIME_LO));
            return false;
        }
        settings->setRecordDuration(duration);
    }

    if (parser.isSet(HL_TRIGGER_RECORD_LO))
    {
        extraArgs.startRecord = true;
    }

    if (parser.isSet(HL_SAMPLE_RATE_LO))
    {
        bool ok = false;
        int rate = parser.value(HL_SAMPLE_RATE_LO).toInt(&ok);
        if (!ok)
        {
            invalidArg(HL_SAMPLE_RATE_LO, parser.value(HL_SAMPLE_RATE_LO));
            return false;
        }
        settings->setSampleRate(rate);
    }

    if (parser.isSet(HL_ENCODING_LO))
    {
        std::string encoding = parser.value(HL_ENCODING_LO).toStdString();
        if (encoding == "WAV")
        {
            encoding = WAV;
        }
        else
        {
            invalidArg(HL_ENCODING_LO, parser.value(HL_ENCODING_LO), "Only WAV format is supported at this time.");
            return false;
        }
    }

    if (parser.isSet(HL_INPUT_DEVICE_LO))
    {
        std::string device = parser.value(HL_INPUT_DEVICE_LO).toStdString();
        // TODO: See if we can check device list here
        settings->setDefaultInputDeviceName(device);
    }

    if (parser.isSet(HL_OUTPUT_DEVICE_LO))
    {
        std::string device = parser.value(HL_OUTPUT_DEVICE_LO).toStdString();
        // TODO: See if we can check device list here
        settings->setDefaultOutputDeviceName(device);
    }

    if (parser.isSet(HL_LIST_DEVICES_LO))
    {
        Transport t;

        printf("\n-------- Device List --------\n");
        printDeviceList(&t);

        extraArgs.exit = true;
    }

    return true;
}

#endif // END HL_CLI_ARGS_H