utils.cpp

/*
 * SPDX-FileCopyrightText: (C) 2015-2022 Daniel Nicoletti <dantti12@gmail.com>
 * SPDX-License-Identifier: BSD-3-Clause
 */
#include "utils.h"

#include <algorithm>

#include <QTextStream>
#include <QVector>

using namespace Cutelyst;

namespace {

QByteArray buildTableDivision(const QVector<int> &columnsSize)
{
    QByteArray buffer;
    QTextStream out(&buffer, QTextStream::WriteOnly);
    for (int i = 0; i < columnsSize.size(); ++i) {
        if (i) {
            out << '+';
        } else {
            out << '.';
        }
        out << QByteArray().fill('-', columnsSize[i] + 2).data();
    }
    out << '.';

    return buffer;
}

} // namespace

QByteArray Utils::buildTable(const QVector<QStringList> &table,
                             const QStringList &headers,
                             const QString &title)
{
    QByteArray buffer;
    QVector<int> columnsSize;

    if (!headers.isEmpty()) {
        std::ranges::transform(headers, std::back_inserter(columnsSize), [](const QString &header) {
            return header.size();
        });
    } else {
        for (const QStringList &rows : table) {
            if (columnsSize.empty()) {
                std::ranges::transform(rows,
                                       std::back_inserter(columnsSize),
                                       [](const QString &row) { return row.size(); });
            } else if (rows.size() != columnsSize.size()) {
                qFatal("Incomplete table");
            }
        }
    }

    for (const QStringList &row : table) {
        if (row.size() > columnsSize.size()) {
            qFatal("Incomplete table");
        }

        for (int i = 0; i < row.size(); ++i) {
            columnsSize[i] = qMax(columnsSize[i], row[i].size());
        }
    }

    // printing
    QTextStream out(&buffer, QTextStream::WriteOnly);

    out.setFieldAlignment(QTextStream::AlignLeft);
    QByteArray div = buildTableDivision(columnsSize);

    if (!title.isEmpty()) {
        out << title << '\n';
    }

    // Top line
    out << div << '\n';

    if (!headers.isEmpty()) {
        // header titles
        for (int i = 0; i < headers.size(); ++i) {
            out << "| ";

            out.setFieldWidth(columnsSize[i]);
            out << headers[i];

            out.setFieldWidth(0);
            out << ' ';
        }
        out << '|' << '\n';

        // header bottom line
        out << div << '\n';
    }

    for (const QStringList &row : table) {
        // content table
        for (int i = 0; i < row.size(); ++i) {
            out << "| ";

            out.setFieldWidth(columnsSize[i]);
            out << row[i];

            out.setFieldWidth(0);
            out << ' ';
        }
        out << '|' << '\n';
    }

    // table bottom line
    out << div;

    return buffer;
}

QString Utils::decodePercentEncoding(QString *s)
{
    if (s->isEmpty()) {
        return *s;
    }

    QByteArray ba = s->toLatin1();

    char *data           = ba.data();
    const char *inputPtr = data;

    const int len = ba.length();
    bool skipUtf8 = true;
    int outlen    = 0;
    for (int i = 0; i < len; ++i, ++outlen) {
        const char c = inputPtr[i];
        if (c == '%' && i + 2 < len) {
            int a = static_cast<int>(static_cast<unsigned char>(inputPtr[++i]));
            int b = static_cast<int>(static_cast<unsigned char>(inputPtr[++i]));

            if (a >= '0' && a <= '9') {
                a -= '0';
            } else if (a >= 'a' && a <= 'f') {
                a = a - 'a' + 10;
            } else if (a >= 'A' && a <= 'F') {
                a = a - 'A' + 10;
            }

            if (b >= '0' && b <= '9') {
                b -= '0';
            } else if (b >= 'a' && b <= 'f') {
                b = b - 'a' + 10;
            } else if (b >= 'A' && b <= 'F') {
                b = b - 'A' + 10;
            }

            *data++  = (char) ((a << 4) | b);
            skipUtf8 = false;
        } else if (c == '+') {
            *data++ = ' ';
        } else {
            *data++ = c;
        }
    }

    if (skipUtf8) {
        return *s;
    }

    return QString::fromUtf8(ba.data(), outlen);
}

ParamsMultiMap Utils::decodePercentEncoding(char *data, int len)
{
    ParamsMultiMap ret;
    if (len <= 0) {
        return ret;
    }

    QString key;

    const char *inputPtr = data;

    bool hasKey   = false;
    bool skipUtf8 = true;
    char *from    = data;
    int outlen    = 0;

    auto processKeyPair = [&] {
        if (hasKey) {
            if ((data - from) == 0) {
                if (!key.isEmpty()) {
                    ret.insert(key, {});
                }
            } else {
                ret.insert(key,
                           skipUtf8 ? QString::fromLatin1(from, data - from)
                                    : QString::fromUtf8(from, data - from));
            }
        } else if ((data - from) > 0) {
            ret.insert(skipUtf8 ? QString::fromLatin1(from, data - from)
                                : QString::fromUtf8(from, data - from),
                       {});
        }
    };

    for (int i = 0; i < len; ++i, ++outlen) {
        const char c = inputPtr[i];
        if (c == '%' && i + 2 < len) {
            int a = static_cast<int>(static_cast<unsigned char>(inputPtr[++i]));
            int b = static_cast<int>(static_cast<unsigned char>(inputPtr[++i]));

            if (a >= '0' && a <= '9') {
                a -= '0';
            } else if (a >= 'a' && a <= 'f') {
                a = a - 'a' + 10;
            } else if (a >= 'A' && a <= 'F') {
                a = a - 'A' + 10;
            }

            if (b >= '0' && b <= '9') {
                b -= '0';
            } else if (b >= 'a' && b <= 'f') {
                b = b - 'a' + 10;
            } else if (b >= 'A' && b <= 'F') {
                b = b - 'A' + 10;
            }

            *data++  = (char) ((a << 4) | b);
            skipUtf8 = false;
        } else if (c == '+') {
            *data++ = ' ';
        } else if (c == '=') {
            key      = skipUtf8 ? QString::fromLatin1(from, data - from)
                                : QString::fromUtf8(from, data - from);
            from     = data;
            hasKey   = true;
            skipUtf8 = true; // reset
        } else if (c == '&') {
            processKeyPair();
            key.clear();
            hasKey   = false;
            from     = data;
            skipUtf8 = true; // reset
        } else {
            *data++ = c;
        }
    }

    processKeyPair();

    return ret;
}

QString Utils::decodePercentEncoding(QByteArray *ba)
{
    if (ba->isEmpty()) {
        return {};
    }

    char *data           = ba->data();
    const char *inputPtr = data;

    int len       = ba->length();
    bool skipUtf8 = true;
    int outlen    = 0;
    for (int i = 0; i < len; ++i, ++outlen) {
        const char c = inputPtr[i];
        if (c == '%' && i + 2 < len) {
            int a = static_cast<int>(static_cast<unsigned char>(inputPtr[++i]));
            int b = static_cast<int>(static_cast<unsigned char>(inputPtr[++i]));

            if (a >= '0' && a <= '9') {
                a -= '0';
            } else if (a >= 'a' && a <= 'f') {
                a = a - 'a' + 10;
            } else if (a >= 'A' && a <= 'F') {
                a = a - 'A' + 10;
            }

            if (b >= '0' && b <= '9') {
                b -= '0';
            } else if (b >= 'a' && b <= 'f') {
                b = b - 'a' + 10;
            } else if (b >= 'A' && b <= 'F') {
                b = b - 'A' + 10;
            }

            *data++  = (char) ((a << 4) | b);
            skipUtf8 = false;
        } else if (c == '+') {
            *data++ = ' ';
        } else {
            *data++ = c;
        }
    }

    if (skipUtf8) {
        return QString::fromLatin1(ba->data(), outlen);
    } else {
        return QString::fromUtf8(ba->data(), outlen);
    }
}

std::chrono::microseconds Utils::durationFromString(QStringView str, bool *ok)
{
    QList<std::pair<QString, QString>> parts;
    QString digitPart;
    QString unitPart;
    bool valid = true;
    // NOLINTBEGIN(bugprone-branch-clone)
    for (const QChar ch : str) {
        if (ch >= u'0' && ch <= u'9') {
            if (digitPart.isEmpty() && unitPart.isEmpty()) {
                // we are at the beginning of a new part
                digitPart.append(ch);
            } else if (digitPart.isEmpty() && !unitPart.isEmpty()) {
                // wrong order
                valid = false;
                break;
            } else if (!digitPart.isEmpty() && unitPart.isEmpty()) {
                // we are still in the digit part
                digitPart.append(ch);
            } else if (!digitPart.isEmpty() && !unitPart.isEmpty()) {
                // we start a new part
                parts.emplace_back(digitPart, unitPart);
                digitPart.clear();
                unitPart.clear();
                digitPart.append(ch);
            }
        } else if ((ch >= u'a' && ch <= u'z') || ch == u'M') {
            if (digitPart.isEmpty() && unitPart.isEmpty()) {
                // something is wrong with a digitless unit
                valid = false;
                break;
            } else if (digitPart.isEmpty() && !unitPart.isEmpty()) {
                // it should not be possible to be here
                valid = false;
                break;
            } else if (!digitPart.isEmpty() && unitPart.isEmpty()) {
                // we start adding the unit
                unitPart.append(ch);
            } else if (!digitPart.isEmpty() && !unitPart.isEmpty()) {
                // normal operation
                unitPart.append(ch);
            }
        }
    }
    // NOLINTEND(bugprone-branch-clone)

    if (!valid) {
        if (ok) {
            *ok = false;
        }
        return std::chrono::microseconds::zero();
    }

    if (!digitPart.isEmpty()) {
        parts.emplace_back(digitPart, unitPart);
    }

    if (parts.empty()) {
        if (ok) {
            *ok = false;
        }
        return std::chrono::microseconds::zero();
    }

    std::chrono::microseconds ms = std::chrono::microseconds::zero();

    for (const std::pair<QString, QString> &p : parts) {
        bool _ok             = false;
        const qulonglong dur = p.first.toULongLong(&_ok);
        if (!_ok) {
            valid = false;
            break;
        }

        if (p.second == u"usec" || p.second == u"us") {
            ms += std::chrono::microseconds{dur};
        } else if (p.second == u"msec" || p.second == u"ms") {
            ms += std::chrono::milliseconds{dur};
        } else if (p.second == u"seconds" || p.second == u"second" || p.second == u"sec" ||
                   p.second == u"s" || p.second.isEmpty()) {
            ms += std::chrono::seconds{dur};
        } else if (p.second == u"minutes" || p.second == u"minute" || p.second == u"min" ||
                   p.second == u"m") {
            ms += std::chrono::minutes{dur};
        } else if (p.second == u"hours" || p.second == u"hour" || p.second == u"hr" ||
                   p.second == u"h") {
            ms += std::chrono::hours{dur};
        } else if (p.second == u"days" || p.second == u"day" || p.second == u"d") {
            ms += std::chrono::days{dur};
        } else if (p.second == u"weeks" || p.second == u"week" || p.second == u"w") {
            ms += std::chrono::weeks{dur};
        } else if (p.second == u"months" || p.second == u"month" || p.second == u"M") {
            ms += std::chrono::months{dur};
        } else if (p.second == u"years" || p.second == u"year" || p.second == u"y") {
            ms += std::chrono::years{dur};
        } else {
            valid = false;
            break;
        }
    }

    if (!valid) {
        if (ok) {
            *ok = false;
        }
        return std::chrono::microseconds::zero();
    }

    if (ok) {
        *ok = true;
    }

    return ms;
}