abstractcommand.cpp

// SPDX-FileCopyrightText: 2022-2026 Paul Colby <git@colby.id.au>
// SPDX-License-Identifier: LGPL-3.0-or-later
 
#include "abstractcommand.h"
#include "../stringliterals_p.h"
 
#include <qtpokit/pokitdevice.h>
#include <qtpokit/pokitdiscoveryagent.h>
 
#include <QLocale>
#include <QTimer>
#include <QtMath>
 
#include <cmath>
#include <ratio>
 
DOKIT_USE_STRINGLITERALS
 
/*!
 * \class AbstractCommand
 *
 * The AbstractCommand class provides a consistent base for the classes that implement CLI commands.
 */
 
/*!
 * Constructs a new command with \a parent.
 */
AbstractCommand::AbstractCommand(QObject * const parent) : QObject(parent),
    discoveryAgent(new PokitDiscoveryAgent(this))
{
    connect(discoveryAgent, &PokitDiscoveryAgent::pokitDeviceDiscovered,
            this, &AbstractCommand::deviceDiscovered);
    connect(discoveryAgent, &PokitDiscoveryAgent::finished,
            this, &AbstractCommand::deviceDiscoveryFinished);
    connect(discoveryAgent,
        #if (QT_VERSION < QT_VERSION_CHECK(6, 2, 0))
        QOverload<PokitDiscoveryAgent::Error>::of(&PokitDiscoveryAgent::error),
        #else
        &PokitDiscoveryAgent::errorOccurred,
        #endif
        this, [](const PokitDiscoveryAgent::Error &error) {
        qCWarning(lc).noquote() << tr("Bluetooth discovery error:") << error;
        QTimer::singleShot(0, QCoreApplication::instance(), [](){
            QCoreApplication::exit(EXIT_FAILURE);
        });
    });
}
 
/*!
 * Returns a list of CLI option names required by this command. The main console appication may
 * use this list to output an error (and exit) if any of the returned names are not found in the
 * parsed CLI options.
 *
 * The (already parsed) \a parser may be used adjust the returned required options depending on the
 * value of other options. For example, the `logger` command only requires the `--mode` option if
 * the `--command` option is `start`.
 *
 * This base implementation simply returns an empty list. Derived classes should override this
 * function to include any required options.
 */
QStringList AbstractCommand::requiredOptions(const QCommandLineParser &parser) const
{
    Q_UNUSED(parser)
    return QStringList();
}
 
/*!
 * Returns a list of CLI option names supported by this command. The main console appication may
 * use this list to output a warning for any parsed CLI options not included in the returned list.
 *
 * The (already parsed) \a parser may be used adjust the returned supported options depending on the
 * value of other options. For example, the `logger` command only supported the `--timestamp` option
 * if the `--command` option is `start`.
 *
 * This base implementation simply returns requiredOptions(). Derived classes should override this
 * function to include optional options, such as:
 *
 * ```
 * QStringList Derived::supportedOptions(const QCommandLineParser &parser) const
 * {
 *     const QStringList list = AbstractCommand::supportedOptions(parser) + QStringList{ ... };
 *     list.sort();
 *     list.removeDuplicates(); // Optional, recommended.
 *     return list;
 * }
 * ```
 */
QStringList AbstractCommand::supportedOptions(const QCommandLineParser &parser) const
{
    return requiredOptions(parser) + QStringList{
        u"debug"_s,
        u"device"_s, u"d"_s,
        u"output"_s,
        u"timeout"_s,
    };
}
 
/*!
 * Returns a human-readable version of \a value, with up-to \a precision decimal digits, and SI unit prefix when
 * appropiate.
 *
 * \note Only supports positive values. Tip: use appendSiPrefix(qAbs(...)) for now.
 */
QString AbstractCommand::appendSiPrefix(const double value, int precision) {
    // Scale the number to an appropriate SI prefix.
    QStringList siPrefixes{ u""_s, u"m"_s, u"μ"_s };
    int index = 0;
    if (value != 0.)
        for (index=0; (index < siPrefixes.length()-1) && (qAbs(value) * qPow(1000.0, (double)index) < 1); ++index);
    QString number = QString::number(value * qPow(1000.0, (double)index), 'f', precision);
 
    // Trim trailing zeros, and decimal indicator if no decimals remain.
    if (const DOKIT_STRING_INDEX_TYPE decimalPos = number.indexOf(u'.'); decimalPos > 0) {
        DOKIT_STRING_INDEX_TYPE nonZeroPos;
        for (nonZeroPos = number.length()-1; (nonZeroPos > decimalPos) && (number.at(nonZeroPos) == u'0'); --nonZeroPos);
        number.truncate((nonZeroPos == decimalPos) ? nonZeroPos : nonZeroPos+1);
    }
    return QString(u"%1%2"_s).arg(number, siPrefixes.at(index));
}
 
/*!
 * Returns an RFC 4180 compliant version of \a field. That is, if \a field contains any of the
 * the below four characters, than any double quotes are escaped (by addition double-quotes), and
 * the string itself surrounded in double-quotes. Otherwise, \a field is returned verbatim.
 *
 * Some examples:
 * ```
 * QCOMPARE(escapeCsvField("abc"), "abc");           // Returned unchanged.
 * QCOMPARE(escapeCsvField("a,c"), R"("a,c")");      // Wrapped in double-quotes.
 * QCOMPARE(escapeCsvField(R"(a"c)"), R("("a""c")"); // Existing double-quotes doubled, then wrapped.
 * ```
 */
QString AbstractCommand::escapeCsvField(const QString &field)
{
    if (field.contains(','_L1) || field.contains('\r'_L1) || field.contains('"'_L1) || field.contains('\n'_L1)) {
        return uR"("%1")"_s.arg(QString(field).replace('"'_L1, uR"("")"_s));
    } else return field;
}
 
/*!
 * \internal
 * A (run-time) class approximately equivalent to the compile-time std::ratio template.
 */
struct Ratio {
    std::intmax_t num { 0 }; ///< Numerator.
    std::intmax_t den { 0 }; ///< Denominator.
    //! Returns \a true if both #num and #den are non-zero.
    bool isValid() const { return (num != 0) && (den != 0); }
};
 
/*!
 * \internal
 * Returns a (run-time) Ratio representation of (compile-time) ratio \a R.
 */
template<typename R> constexpr Ratio makeRatio() { return Ratio{ R::num, R::den }; }
 
/*!
 * Returns \a value as an integer multiple of the ratio \a R, as number of type \a T. The string \a value
 * may end with the optional \a unit, such as `V` or `s`, which may also be preceded with a SI unit
 * prefix such as `m` for `milli`. If \a value contains no SI unit prefix, then the result will be
 * multiplied by 1,000 enough times to be greater than \a sensibleMinimum. This allows for
 * convenient use like:
 *
 * ```
 * const quint32 timeout = parseNumber<std::milli>(parser.value("window"), 's', (quint32)500'000);
 * ```
 *
 * So that an unqualified period like "300" will be assumed to be 300 milliseconds, and not 300
 * microseconds, while a period like "1000" will be assume to be 1 second.
 *
 * If conversion fails for any reason, quiet-NaN (if supported by \a T) or 0 is returned.
 */
template<typename R, typename T>
T AbstractCommand::parseNumber(const QString &value, const QString &unit, const T sensibleMinimum)
{
    static const QMap<QChar, Ratio> unitPrefixScaleMap {
        { 'E'_L1,        makeRatio<std::exa>()   },
        { 'P'_L1,        makeRatio<std::peta>()  },
        { 'T'_L1,        makeRatio<std::tera>()  },
        { 'G'_L1,        makeRatio<std::giga>()  },
        { 'M'_L1,        makeRatio<std::mega>()  },
        { 'K'_L1,        makeRatio<std::kilo>()  }, // Not official SI unit prefix, but commonly used.
        { 'k'_L1,        makeRatio<std::kilo>()  },
        { 'h'_L1,        makeRatio<std::hecto>() },
        { 'd'_L1,        makeRatio<std::deci>()  },
        { 'c'_L1,        makeRatio<std::centi>() },
        { 'm'_L1,        makeRatio<std::milli>() },
        { 'u'_L1,        makeRatio<std::micro>() }, // Not official SI unit prefix, but commonly used.
        { QChar(0x00B5), makeRatio<std::micro>() }, // Unicode micro symbol (µ).
        { QChar(0x03BC), makeRatio<std::micro>() }, // Unicode lower mu (μ).
        { 'n'_L1,        makeRatio<std::nano>()  },
        { 'p'_L1,        makeRatio<std::pico>()  },
        { 'f'_L1,        makeRatio<std::femto>() },
        { 'a'_L1,        makeRatio<std::atto>()  },
    };
 
    // Remove the optional (whole) unit suffix.
    Ratio ratio;
    QString number = value.trimmed();
    if ((!unit.isEmpty()) && (number.endsWith(unit, Qt::CaseInsensitive))) {
        number.chop(unit.length());
        ratio = makeRatio<std::ratio<1>>();
    }
 
    // Parse, and remove, the optional SI unit prefix.
    if (!number.isEmpty()) {
        #if (QT_VERSION >= QT_VERSION_CHECK(5, 10, 0))
        const QChar siPrefix = number.back(); // QString::back() introduced in Qt 5.10.
        #else
        const QChar siPrefix = number.at(number.size() - 1);
        #endif
        const auto iter = unitPrefixScaleMap.constFind(siPrefix);
        if (iter != unitPrefixScaleMap.constEnd()) {
            Q_ASSERT(iter->isValid());
            ratio = *iter;
            number.chop(1);
        }
    }
 
    // Parse the number as an (unsigned) integer.
    QLocale locale; bool ok;
    if (const qulonglong integer = locale.toULongLong(number, &ok); (ok)) {
        if (integer == 0) {
            return static_cast<T>(integer); // Otherwise the next for loop would be be infinite.
        }
        #define DOKIT_RESULT(var) (static_cast<T>(var) * ratio.num * R::den / ratio.den / R::num)
        if (!ratio.isValid()) {
            for (ratio = makeRatio<R>(); DOKIT_RESULT(integer) < sensibleMinimum; ratio.num *= 1000);
        }
        return DOKIT_RESULT(integer);
        #undef DOKIT_RESULT
    }
 
    // Parse the number as a (double) floating point number, and check that it is positive.
    if (const double dbl = locale.toDouble(number, &ok); (ok) && (dbl > 0.0)) {
        if (dbl == 0.) {
            return static_cast<T>(dbl); // Otherwise the next for loop would be be infinite.
        }
        #define DOKIT_RESULT(var) (var * ratio.num * R::den / ratio.den / R::num)
        if (!ratio.isValid()) {
            for (ratio = makeRatio<R>(); DOKIT_RESULT(dbl) < sensibleMinimum; ratio.num *= 1000);
        }
        return (std::is_integral_v<T>) ? static_cast<T>(std::llround(DOKIT_RESULT(dbl))) : DOKIT_RESULT(dbl);
        #undef DOKIT_RESULT
    }
 
    // Failed to parse as either integer, or float.
    return (std::numeric_limits<T>::has_quiet_NaN) ? std::numeric_limits<T>::quiet_NaN() : 0;
}
 
#define DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(Ratio, Type) template \
Type AbstractCommand::parseNumber<Ratio>(const QString &value, const QString &unit, const Type sensibleMinimum)
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::exa,      quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::peta,     quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::tera,     quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::giga,     quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::mega,     quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::kilo,     quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::hecto,    quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::deca,     quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::ratio<1>, quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::ratio<1>, float);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::deci,     quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::centi,    quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::milli,    quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::micro,    quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::nano,     quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::pico,     quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::femto,    quint32);
DOKIT_INSTANTIATE_TEMPLATE_FUNCTION(std::atto,     quint32);
#undef DOKIT_INSTANTIATE_TEMPLATE_FUNCTION
 
/*!
 * Processes the relevant options from the command line \a parser.
 *
 * On success, returns an empty QStringList, otherwise returns a list of CLI errors that the caller
 * should report appropriately before exiting.
 *
 * This base implementations performs some common checks, such as ensuring that required options are
 * present. Derived classes should override this function to perform further processing, typically
 * invoking this base implementation as a first step, such as:
 *
 * ```
 * QStringList CustomCommand::processOptions(const QCommandLineParser &parser)
 * {
 *     QStringList errors = AbstractCommand::processOptions(parser);
 *     if (!errors.isEmpty()) {
 *         return errors;
 *     }
 *
 *     // Do further procession of options.
 *
 *     return errors;
 * }
 * ```
 */
QStringList AbstractCommand::processOptions(const QCommandLineParser &parser)
{
    // Report any supplied options that are not supported by this command.
    const QStringList suppliedOptionNames = parser.optionNames();
    const QStringList supportedOptionNames = supportedOptions(parser);
    for (const QString &option: suppliedOptionNames) {
        if (!supportedOptionNames.contains(option)) {
            qCInfo(lc).noquote() << tr("Ignoring option: %1").arg(option);
        }
    }
    QStringList errors;
 
    // Parse the device (name/addr/uuid) option.
    if (parser.isSet(u"device"_s)) {
        deviceToScanFor = parser.value(u"device"_s);
    }
 
    // Parse the output format options (if supported, and supplied).
    if ((supportedOptionNames.contains(u"output"_s)) && // Derived classes may have removed.
        (parser.isSet(u"output"_s)))
    {
        const QString output = parser.value(u"output"_s).toLower();
        if (output == u"csv"_s) {
            format = OutputFormat::Csv;
        } else if (output == u"json"_s) {
            format = OutputFormat::Json;
        } else if (output == u"text"_s) {
            format = OutputFormat::Text;
        } else {
            errors.append(tr("Unknown output format: %1").arg(output));
        }
    }
 
    // Parse the device scan timeout option.
    if (parser.isSet(u"timeout"_s)) {
        const quint32 timeout = parseNumber<std::milli>(parser.value(u"timeout"_s), u"s"_s, 500);
        if (timeout == 0) {
            errors.append(tr("Invalid timeout: %1").arg(parser.value(u"timeout"_s)));
        } else if (discoveryAgent->lowEnergyDiscoveryTimeout() == -1) {
            qCWarning(lc).noquote() << tr("Platform does not support Bluetooth scan timeout");
        } else {
            discoveryAgent->setLowEnergyDiscoveryTimeout(timeout);
            qCDebug(lc).noquote() << tr("Set scan timeout to %1").arg(
                discoveryAgent->lowEnergyDiscoveryTimeout());
        }
    }
 
    // Return errors for any required options that are absent.
    const QStringList requiredOptionNames = this->requiredOptions(parser);
    for (const QString &option: requiredOptionNames) {
        if (!parser.isSet(option)) {
            errors.append(tr("Missing required option: %1").arg(option));
        }
    }
    return errors;
}
 
/*!
 * \fn virtual bool AbstractCommand::start()
 *
 * Begins the functionality of this command, and returns `true` if begun successfully, `false`
 * otherwise.
 */
 
/*!
 * \fn virtual void AbstractCommand::deviceDiscovered(const QBluetoothDeviceInfo &info) = 0
 *
 * Handles PokitDiscoveryAgent::pokitDeviceDiscovered signal. Derived classes must
 * implement this slot to begin whatever actions are relevant when a Pokit device has been
 * discovered. For example, the 'scan' command would simply output the \a info details, whereas
 * most other commands would begin connecting if \a info is the device they're after.
 */
 
/*!
 * \fn virtual void AbstractCommand::deviceDiscoveryFinished() = 0
 *
 * Handles PokitDiscoveryAgent::deviceDiscoveryFinished signal. Derived classes must
 * implement this slot to perform whatever actions are appropriate when discovery is finished.
 * For example, the 'scan' command would simply exit, whereas most other commands would verify that
 * an appropriate device was found.
 */