WiiUPluginLoaderBackend/source/utils/storage/StorageItem.cpp

#include "StorageItem.h"

#include "utils/base64.h"
#include "utils/logger.h"

StorageItem::StorageItem(const std::string_view key) : mKey(key) {
}

uint32_t StorageItem::getHandle() const {
    return reinterpret_cast<uint32_t>(this);
}

void StorageItem::setValue(const std::string &value) {
    mData                 = value;
    mType                 = StorageItemType::String;
    mBinaryConversionDone = false;
}

void StorageItem::setValue(bool value) {
    mData                 = value;
    mType                 = StorageItemType::Boolean;
    mBinaryConversionDone = true;
}

void StorageItem::setValue(const int32_t value) {
    mData                 = static_cast<int64_t>(value);
    mType                 = StorageItemType::S64;
    mBinaryConversionDone = true;
}

void StorageItem::setValue(int64_t value) {
    mData                 = value;
    mType                 = StorageItemType::S64;
    mBinaryConversionDone = true;
}

void StorageItem::setValue(uint64_t value) {
    mData                 = value;
    mType                 = StorageItemType::U64;
    mBinaryConversionDone = true;
}

void StorageItem::setValue(const uint32_t value) {
    mData                 = static_cast<uint64_t>(value);
    mType                 = StorageItemType::U64;
    mBinaryConversionDone = true;
}

void StorageItem::setValue(const float value) {
    mData                 = static_cast<double>(value);
    mType                 = StorageItemType::Double;
    mBinaryConversionDone = true;
}

void StorageItem::setValue(double value) {
    mData                 = value;
    mType                 = StorageItemType::Double;
    mBinaryConversionDone = true;
}

void StorageItem::setValue(const std::vector<uint8_t> &data) {
    mData                 = data;
    mType                 = StorageItemType::Binary;
    mBinaryConversionDone = true;
}

void StorageItem::setValue(const uint8_t *data, const size_t size) {
    setValue(std::vector(data, data + size));
}

bool StorageItem::getValue(bool &result) const {
    if (mType == StorageItemType::Boolean) {
        result = std::get<bool>(mData);
        return true;
    } else if (mType == StorageItemType::S64) {
        result = !!(std::get<int64_t>(mData));
        return true;
    } else if (mType == StorageItemType::U64) {
        result = !!(std::get<uint64_t>(mData));
        return true;
    }
    return false;
}

bool StorageItem::getValue(int32_t &result) const {
    if (mType == StorageItemType::S64) {
        result = static_cast<int32_t>(std::get<int64_t>(mData));
        return true;
    } else if (mType == StorageItemType::U64) {
        result = static_cast<int32_t>(std::get<uint64_t>(mData));
        return true;
    }
    return false;
}

bool StorageItem::getValue(std::vector<uint8_t> &result) const {
    if (mType == StorageItemType::Binary) {
        result = std::get<std::vector<uint8_t>>(mData);
        return true;
    }
    return false;
}

bool StorageItem::getValue(std::string &result) const {
    if (mType == StorageItemType::String) {
        result = std::get<std::string>(mData);
        return true;
    }
    return false;
}

bool StorageItem::getValue(double &result) const {
    if (mType == StorageItemType::Double) {
        result = std::get<double>(mData);
        return true;
    }
    return false;
}

bool StorageItem::getValue(float &result) const {
    if (mType == StorageItemType::Double) {
        result = static_cast<float>(std::get<double>(mData));
        return true;
    }
    return false;
}

bool StorageItem::getValue(uint64_t &result) const {
    if (mType == StorageItemType::U64) {
        result = std::get<uint64_t>(mData);
        return true;
    } else if (mType == StorageItemType::S64) {
        result = static_cast<uint64_t>(std::get<int64_t>(mData));
        return true;
    }
    return false;
}

bool StorageItem::getValue(uint32_t &result) const {
    if (mType == StorageItemType::U64) {
        result = static_cast<uint32_t>(std::get<uint64_t>(mData));
        return true;
    } else if (mType == StorageItemType::S64) {
        result = static_cast<uint32_t>(std::get<int64_t>(mData));
        return true;
    }
    return false;
}

bool StorageItem::getValue(int64_t &result) const {
    if (mType == StorageItemType::S64) {
        result = std::get<int64_t>(mData);
        return true;
    } else if (mType == StorageItemType::U64) {
        result = static_cast<int64_t>(std::get<uint64_t>(mData));
        return true;
    }
    return false;
}

bool StorageItem::getItemSizeString(uint32_t &outSize) const {
    if (mType == StorageItemType::String) {
        outSize = (std::get<std::string>(mData).length() + 1);
        return true;
    }
    return false;
}

bool StorageItem::getItemSizeBinary(uint32_t &outSize) const {
    if (mType == StorageItemType::Binary) {
        outSize = std::get<std::vector<uint8_t>>(mData).size();
        return true;
    }
    return false;
}

bool StorageItem::attemptBinaryConversion() {
    if (mBinaryConversionDone) {
        return true;
    }
    if (mType == StorageItemType::String) {
        const auto &tmp = std::get<std::string>(mData);
        if (const auto dec_size = b64_decoded_size(tmp.c_str()); dec_size > 0) {
            if (auto *dec = static_cast<uint8_t *>(malloc(dec_size))) {
                if (b64_decode(tmp.c_str(), dec, dec_size)) {
                    setValue(dec, dec_size);
                }
                free(dec);
            } else {
                DEBUG_FUNCTION_LINE_WARN("Malloc failed for string->binary parsing");
                return false;
            }
        }
    }
    mBinaryConversionDone = true;
    return true;
}