#include <string>
#include <sodium.h>
#include <sstream>
#include <rapidjson/document.h>

namespace util
{

/**
 * Encodes provided bytes to base64 string.
 * 
 * @param encoded_string String reference to assign the base64 encoded output. 
 * @param bin Bytes to encode.
 * @param bin_len Bytes length.
 */
int base64_encode(std::string &encoded_string, const unsigned char *bin, size_t bin_len)
{
    // Get length of encoded result from sodium.
    const size_t base64_len = sodium_base64_encoded_len(bin_len, sodium_base64_VARIANT_ORIGINAL);
    char base64chars[base64_len];

    // Get encoded string.
    const char *encoded_str_char = sodium_bin2base64(
        base64chars, base64_len,
        bin, bin_len,
        sodium_base64_VARIANT_ORIGINAL);

    if (encoded_str_char == NULL)
        return -1;

    // Assign the encoded char* onto the provided string reference.
    // "base64_len - 1" because sodium include '\0' in the calculated base64 length.
    //      Therefore we need to omit it when initializing the std::string.
    encoded_string = std::string(base64chars, base64_len - 1);
    return 0;
}

/**
 * Decodes provided base64 string into bytes.
 * 
 * @param decodedbuf Buffer to assign decoded bytes.
 * @param decodedbuf_len Decoded buffer size.
 * @param base64_str Base64 string to decode.
 */
int base64_decode(unsigned char *decodedbuf, size_t decodedbuf_len, std::string_view base64_str)
{
    const char *b64_end;
    size_t bin_len;
    if (sodium_base642bin(
            decodedbuf, decodedbuf_len,
            base64_str.data(), base64_str.size() + 1,
            "", &bin_len, &b64_end,
            sodium_base64_VARIANT_ORIGINAL))
    {
        return -1;
    }

    return 0;
}

/**
 * Compare two version strings in the format of "1.12.3".
 * v1 <  v2  -> returns -1
 * v1 == v2  -> returns  0
 * v1 >  v2  -> returns +1
 * Error     -> returns -2
 * 
 * Remark on string_view: In other places of the code-base we utilize string_view
 * to pass immutable string references around. However in this function we keep the 'const string&'
 * syntax because istringstream doesn't support string_view. It's not worth optmising
 * this code as it's not being used in high-scale processing.
 */
int version_compare(const std::string &x, const std::string &y)
{
    std::istringstream ix(x), iy(y);
    while (ix.good() || iy.good())
    {
        int cx = 0, cy = 0;
        ix >> cx;
        iy >> cy;

        if ((!ix.eof() && !ix.good()) || (!iy.eof() && !iy.good()))
            return -2;

        if (cx > cy)
            return 1;
        if (cx < cy)
            return -1;

        ix.ignore();
        iy.ignore();
    }

    return 0;
}

/**
 * Returns a std::string_view pointing to the rapidjson Value which is assumed
 * to be a string. We use this function because rapidjson does not have build-in string_view
 * support. Passing a non-string 'v' is not supported.
 */
std::string_view getsv(const rapidjson::Value &v)
{
    return std::string_view(v.GetString(), v.GetStringLength());
}

} // namespace util