///////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2018 Edouard Griffiths, F4EXB.                                  //
//                                                                               //
// SDRdaemon command line parser                                                 //
//                                                                               //
// SDRdaemon is a detached SDR front end that handles the interface with a       //
// physical device and sends or receives the I/Q samples stream to or from a     //
// SDRangel instance via UDP. It is controlled via a Web REST API.               //
//                                                                               //
// This program is free software; you can redistribute it and/or modify          //
// it under the terms of the GNU General Public License as published by          //
// the Free Software Foundation as version 3 of the License, or                  //
//                                                                               //
// This program is distributed in the hope that it will be useful,               //
// but WITHOUT ANY WARRANTY; without even the implied warranty of                //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the                  //
// GNU General Public License V3 for more details.                               //
//                                                                               //
// You should have received a copy of the GNU General Public License             //
// along with this program. If not, see <http://www.gnu.org/licenses/>.          //
///////////////////////////////////////////////////////////////////////////////////

#include "sdrdaemonparser.h"

#include <QCommandLineOption>
#include <QRegExpValidator>
#include <QDebug>

SDRDaemonParser::SDRDaemonParser() :
    m_serverAddressOption(QStringList() << "a" << "api-address",
        "API server and data (Tx) address.",
        "localAddress",
        "127.0.0.1"),
    m_serverPortOption(QStringList() << "p" << "api-port",
        "Web API server port.",
        "apiPort",
        "9091"),
    m_dataAddressOption(QStringList() << "A" << "data-address",
        "Remote data address (Rx).",
        "remoteAddress",
        "127.0.0.1"),
    m_dataPortOption(QStringList() << "D" << "data-port",
        "UDP stream data port.",
        "dataPort",
        "9090"),
    m_deviceTypeOption(QStringList() << "T" << "device-type",
        "Device type.",
        "deviceType",
        "TestSource"),
    m_txOption(QStringList() << "t" << "tx",
        "Tx indicator."),
    m_serialOption(QStringList() << "s" << "serial",
        "Device serial number.",
        "serial"),
    m_sequenceOption(QStringList() << "i" << "sequence",
        "Device sequence index in enumeration for the same device type.",
        "sequence"),
    m_txDelayOption(QStringList() << "d" << "tx-delay",
        "delay between transmission of UDP blocks (ms).",
        "txDelay",
        "100"),
    m_nbBlocksFECOption(QStringList() << "f" << "fec-blocks",
        "Number of FEC blocks per frame.",
        "nbBlocksFEC",
        "8")
{
    m_serverAddress = "127.0.0.1";
    m_serverPort = 9091;
    m_dataAddress = "127.0.0.1";
    m_dataPort = 9090;
    m_deviceType = "TestSource";
    m_tx = false;
    m_sequence = 0;
    m_txDelay = 100;
    m_nbBlocksFEC = 8;
    m_hasSequence = false;
    m_hasSerial = false;

    m_parser.setApplicationDescription("Software Defined Radio RF header server");
    m_parser.addHelpOption();
    m_parser.addVersionOption();

    m_parser.addOption(m_serverAddressOption);
    m_parser.addOption(m_serverPortOption);
    m_parser.addOption(m_dataAddressOption);
    m_parser.addOption(m_dataPortOption);
    m_parser.addOption(m_deviceTypeOption);
    m_parser.addOption(m_txOption);
    m_parser.addOption(m_serialOption);
    m_parser.addOption(m_sequenceOption);
    m_parser.addOption(m_txDelayOption);
    m_parser.addOption(m_nbBlocksFECOption);
}

SDRDaemonParser::~SDRDaemonParser()
{ }

void SDRDaemonParser::parse(const QCoreApplication& app)
{
    m_parser.process(app);

    int pos;
    bool ok;

    // server address

    QString serverAddress = m_parser.value(m_serverAddressOption);

    QString ipRange = "(?:[0-1]?[0-9]?[0-9]|2[0-4][0-9]|25[0-5])";
    QRegExp ipRegex ("^" + ipRange
                     + "\\." + ipRange
                     + "\\." + ipRange
                     + "\\." + ipRange + "$");
    QRegExpValidator ipValidator(ipRegex);

    if (ipValidator.validate(serverAddress, pos) == QValidator::Acceptable)
    {
        m_serverAddress = serverAddress;
        qDebug() << "SDRDaemonParser::parse: server address: " << m_serverAddress;
    }
    else
    {
        qWarning() << "SDRDaemonParser::parse: server address invalid. Defaulting to " << m_serverAddress;
    }

    // server port

    QString serverPortStr = m_parser.value(m_serverPortOption);
    int serverPort = serverPortStr.toInt(&ok);

    if (ok && (serverPort > 1023) && (serverPort < 65536))
    {
        m_serverPort = serverPort;
        qDebug() << "SDRDaemonParser::parse: server port: " << m_serverPort;
    }
    else
    {
        qWarning() << "SDRDaemonParser::parse: server port invalid. Defaulting to " << m_serverPort;
    }

    // data address

    QString dataAddress = m_parser.value(m_dataAddressOption);

    if (ipValidator.validate(dataAddress, pos) == QValidator::Acceptable)
    {
        m_dataAddress = dataAddress;
        qDebug() << "SDRDaemonParser::parse: data address: " << m_dataAddress;
    }
    else
    {
        qWarning() << "SDRDaemonParser::parse: data address invalid. Defaulting to " << m_dataAddress;
    }

    // data port

    QString dataPortStr = m_parser.value(m_dataPortOption);
    serverPort = dataPortStr.toInt(&ok);

    if (ok && (serverPort > 1023) && (serverPort < 65536))
    {
        m_dataPort = serverPort;
        qDebug() << "SDRDaemonParser::parse: data port: " << m_dataPort;
    }
    else
    {
        qWarning() << "SDRDaemonParser::parse: data port invalid. Defaulting to " << m_dataPort;
    }

    // tx
    m_tx = m_parser.isSet(m_txOption);
    qDebug() << "SDRDaemonParser::parse: tx: " << m_tx;

    // device type

    if (m_parser.isSet(m_deviceTypeOption))
    {
        QString deviceType = m_parser.value(m_deviceTypeOption);

        QRegExp deviceTypeRegex("^[A-Z][A-Za-z0-9]+$");
        QRegExpValidator deviceTypeValidator(deviceTypeRegex);

        if (deviceTypeValidator.validate(deviceType, pos) == QValidator::Acceptable)
        {
            m_deviceType = deviceType;
            qDebug() << "SDRDaemonParser::parse: device type: " << m_deviceType;
        }
        else
        {
            m_deviceType = m_tx ? "FileSink" : "TestSource";
            qWarning() << "SDRDaemonParser::parse: device type invalid. Defaulting to " << m_deviceType;
        }
    }
    else
    {
        m_deviceType = m_tx ? "FileSink" : "TestSource";
        qInfo() << "SDRDaemonParser::parse: device type not specified. defaulting to " << m_deviceType;
    }


    // serial
    m_hasSerial = m_parser.isSet(m_serialOption);

    if (m_hasSerial)
    {
        m_serial = m_parser.value(m_serialOption);
        qDebug() << "SDRDaemonParser::parse: serial: " << m_serial;
    }

    // sequence
    m_hasSequence = m_parser.isSet(m_sequenceOption);

    if (m_hasSequence)
    {
        QString sequenceStr = m_parser.value(m_sequenceOption);
        int sequence = sequenceStr.toInt(&ok);

        if (ok && (sequence >= 0) && (sequence < 65536)) {
            m_sequence = sequence;
            qDebug() << "SDRDaemonParser::parse: sequence: " << m_sequence;
        } else {
            qWarning() << "SDRDaemonParser::parse: sequence invalid. Defaulting to " << m_sequence;
        }
    }

    // Tx delay
    if (m_parser.isSet(m_txDelayOption))
    {
        QString txDelayStr = m_parser.value(m_txDelayOption);
        int txDelay = txDelayStr.toInt(&ok);

        if (ok && (txDelay >= 0))
        {
            m_txDelay = txDelay;
            qDebug() << "SDRDaemonParser::parse: Tx delay: " << m_txDelay;
        }
        else
        {
            qWarning() << "SDRDaemonParser::parse: Tx delay invalid. Defaulting to " << m_txDelay;
        }
    }

    // nb FEC blocks
    if (m_parser.isSet(m_nbBlocksFECOption))
    {
        QString nbBlocksFECStr = m_parser.value(m_nbBlocksFECOption);
        int nbBlocksFEC = nbBlocksFECStr.toInt(&ok);

        if (ok && (nbBlocksFEC >= 0) && (nbBlocksFEC < 128))
        {
            m_nbBlocksFEC = nbBlocksFEC;
            qDebug() << "SDRDaemonParser::parse: number of FEC blocks: " << m_nbBlocksFEC;
        }
        else
        {
            qWarning() << "SDRDaemonParser::parse: number of FEC blocks invalid. Defaulting to " << m_nbBlocksFEC;
        }
    }
}