SomeBot/util.cpp

#include "util.hpp"

#include <dpp/dpp.h>



namespace Util {
uint32_t get_color_of_user(const dpp::user& user) {
    uint32_t colors[] = {
        0x424cb6,
        0x585f68,
        0x2c7c45,
        0xbc7d14,
        0xb23234
    };
    return colors[user.discriminator % 5];
}

std::vector<std::string_view> split_str(std::string_view s, char delimiter, size_t times) {
    std::vector<std::string_view> to_return;
    decltype(s.size()) start = 0, finish = 0;
    while ((finish = s.find_first_of(delimiter, start)) != std::string_view::npos) {
        to_return.emplace_back(s.substr(start, finish - start));
        start = finish + 1;
        if (to_return.size() == times) { break; }
    }
    to_return.emplace_back(s.substr(start));
    return to_return;
}

static const char* B64chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
static const int B64index[256] = {
    0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
    0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,
    0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  0,  62, 63, 62, 62, 63,
    52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 0,  0,  0,  0,  0,  0,
    0,  0,  1,  2,  3,  4,  5,  6,  7,  8,  9,  10, 11, 12, 13, 14,
    15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 0,  0,  0,  0,  63,
    0,  26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
    41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51
};

std::string b64encode(const void* data, const size_t &len) {
    std::string result((len + 2) / 3 * 4, '=');
    unsigned char *p = (unsigned  char*) data;
    char *str = &result[0];
    size_t j = 0, pad = len % 3;
    const size_t last = len - pad;

    for (size_t i = 0; i < last; i += 3) {
        int n = int(p[i]) << 16 | int(p[i + 1]) << 8 | p[i + 2];
        str[j++] = B64chars[n >> 18];
        str[j++] = B64chars[n >> 12 & 0x3F];
        str[j++] = B64chars[n >> 6 & 0x3F];
        str[j++] = B64chars[n & 0x3F];
    }
    if (pad) {
        int n = --pad ? int(p[last]) << 8 | p[last + 1] : p[last];
        str[j++] = B64chars[pad ? n >> 10 & 0x3F : n >> 2];
        str[j++] = B64chars[pad ? n >> 4 & 0x03F : n << 4 & 0x3F];
        str[j++] = pad ? B64chars[n << 2 & 0x3F] : '=';
    }
    return result;
}

std::string b64decode(const void* data, const size_t &len) {
    if (len == 0) return "";

    unsigned char *p = (unsigned char*) data;
    size_t j = 0,
        pad1 = len % 4 || p[len - 1] == '=',
        pad2 = pad1 && (len % 4 > 2 || p[len - 2] != '=');
    const size_t last = (len - pad1) / 4 << 2;
    std::string result(last / 4 * 3 + pad1 + pad2, '\0');
    unsigned char *str = (unsigned char*) &result[0];

    for (size_t i = 0; i < last; i += 4) {
        int n = B64index[p[i]] << 18 | B64index[p[i + 1]] << 12 | B64index[p[i + 2]] << 6 | B64index[p[i + 3]];
        str[j++] = n >> 16;
        str[j++] = n >> 8 & 0xFF;
        str[j++] = n & 0xFF;
    }
    if (pad1) {
        int n = B64index[p[last]] << 18 | B64index[p[last + 1]] << 12;
        str[j++] = n >> 16;
        if (pad2) {
            n |= B64index[p[last + 2]] << 6;
            str[j++] = n >> 8 & 0xFF;
        }
    }
    return result;
}

void point_to_comma(std::string& s) {
    for (auto& c : s) {
        if (c == '.') c = ',';
    }
}

unsigned str_to_color(std::string color_str) {
    // Remove leading #
    if (color_str[0] == '#') {
        color_str.erase(0, 1);
    }
    // Lowercase string
    for (auto& c : color_str) {
        c = tolower(c);
    }
    // Get color code from remaining string
    try {
        return std::stoul(color_str, nullptr, 16);
    } catch (...) {
        if (color_str == "green") {
            return 0x00ff00;
        } else if (color_str == "red") {
            return 0xff0000;
        } else if (color_str == "blue") {
            return 0x0000ff;
        } else if (color_str == "yellow") {
            return 0xffff00;
        } else if (color_str == "red") {
            return 0xff0000;
        } else if (color_str == "magenta") {
            return 0xff00ff;
        } else if (color_str == "aqua") {
            return 0x00ffff;
        } else if (color_str == "white") {
            return 0xffffff;
        } else if (color_str == "black") {
            return 0x000000;
        } else if (color_str == "grey" || color_str == "gray") {
            return 0x888888;
        }
    }
    return -1;
}
}