/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4; fill-column: 100 -*- */
/*
 * Copyright the Collabora Online contributors.
 *
 * SPDX-License-Identifier: MPL-2.0
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */

/*
 * COOL protocol parsing and message formatting.
 * Functions: ParseVersion(), getAbbreviatedMessage(), tokenize()
 */

#include <config.h>

#include "Protocol.hpp"

#include <common/NumUtil.hpp>

#include <Poco/URI.h>

#include <cstring>
#include <map>
#include <string>
#include <string_view>

#include <COKit/COKit.hxx>

namespace COOLProtocol
{
    std::string buildErrorFrame(std::string_view cmd, std::string_view kind, std::string_view message)
    {
        std::string frame = "error: cmd=";
        frame.append(cmd);
        frame.append(" kind=");
        frame.append(kind);
        if (!message.empty())
        {
            // URL-encode so spaces and other special characters survive
            std::string encoded;
            Poco::URI::encode(std::string(message), " \t\r\n", encoded);
            frame.append(" errordetail=");
            frame.append(encoded);
        }
        return frame;
    }

    std::tuple<int, int, std::string> ParseVersion(const std::string& version)
    {
        int major = -1;
        int minor = -1;
        std::string patch;

        StringVector firstTokens(StringVector::tokenize(version, '.'));
        if (firstTokens.size() > 0)
        {
            major = NumUtil::stoi(firstTokens[0]);

            StringVector secondTokens;
            if (firstTokens.size() > 1)
            {
                secondTokens = StringVector::tokenize(firstTokens[1], '-');
            }
            if (secondTokens.size() > 0)
            {
                minor = NumUtil::stoi(secondTokens[0]);
            }

            if (secondTokens.size() > 1)
                patch = secondTokens[1];
        }
        return std::make_tuple(major, minor, patch);
    }

    bool getTokenInteger(const std::string_view token, const std::string_view name, int& value)
    {
        if (token.size() > (name.size() + 1) &&
            token.compare(0, name.size(), name) == 0 &&
            token[name.size()] == '=')
        {
            bool success;
            std::tie(value, success) = NumUtil::i32FromString(token.substr(name.size() + 1));
            return success;
        }

        return false;
    }

    bool getTokenUInt64(const std::string_view token, const std::string_view name, uint64_t& value)
    {
        if (token.size() > (name.size() + 1) &&
            token.compare(0, name.size(), name) == 0 &&
            token[name.size()] == '=')
        {
            const char* str = token.data() + name.size() + 1;
            char* endptr = nullptr;
            value = strtoull(str, &endptr, 10);
            return (endptr > str);
        }

        return false;
    }

    bool getTokenUInt32(const std::string_view token, const std::string_view name, uint32_t& value)
    {
        if (token.size() > (name.size() + 1) &&
            token.compare(0, name.size(), name) == 0 &&
            token[name.size()] == '=')
        {
            const char* str = token.data() + name.size() + 1;
            char* endptr = nullptr;
            value = strtoul(str, &endptr, 10);
            return (endptr > str);
        }

        return false;
    }

    bool getTokenString(const std::string_view token, const std::string_view name,
                        std::string& value)
    {
        if (token.size() >= (name.size() + 1) &&
            token.compare(0, name.size(), name) == 0 &&
            token[name.size()] == '=')
        {
            value = token.substr(name.size() + 1);
            return true;
        }

        return false;
    }

    bool getTokenStringFromMessage(const std::string_view message, const std::string_view name,
                                   std::string& value)
    {
        if (message.size() > name.size() + 1)
        {
            size_t pos = message.find(name);
            while (pos != std::string::npos)
            {
                bool spaceBefore = pos == 0 || message[pos-1] == ' ';
                const size_t beg = pos + name.size();
                if (spaceBefore && message[beg] == '=')
                {
                    const size_t end = message.find_first_of(" \n", beg);
                    value = message.substr(beg + 1, end - beg - 1);
                    return true;
                }

                pos = message.find(name, pos + name.size());
            }
        }

        return false;
    }
};

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */