mirror of
https://github.com/HamletDuFromage/aio-switch-updater.git
synced 2024-11-08 19:41:49 +00:00
377 lines
No EOL
12 KiB
C++
377 lines
No EOL
12 KiB
C++
#include "color_swapper.hpp"
|
|
|
|
#include <filesystem>
|
|
#include <fstream>
|
|
#include <iomanip>
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
#include "constants.hpp"
|
|
#include "download.hpp"
|
|
#include "fs.hpp"
|
|
#include "progress_event.hpp"
|
|
#include "utils.hpp"
|
|
|
|
using json = nlohmann::json;
|
|
|
|
namespace {
|
|
|
|
constexpr const char BACKUP[] = "_backup";
|
|
|
|
int hexToBGR(const std::string& hex)
|
|
{
|
|
std::string R = hex.substr(0, 2);
|
|
std::string G = hex.substr(2, 2);
|
|
std::string B = hex.substr(4, 2);
|
|
return std::stoi(B + G + R, 0, 16);
|
|
}
|
|
|
|
std::string BGRToHex(int v)
|
|
{
|
|
std::stringstream ss;
|
|
v = ((v & 0xFF) << 16) + (v & 0xFF00) + (v >> 16) + 256;
|
|
ss << std::setfill('0') << std::setw(6) << std::right << std::hex << v;
|
|
return ss.str();
|
|
}
|
|
|
|
bool isHexaAnd3Bytes(const std::string& str)
|
|
{
|
|
if (str.size() != 6) return false;
|
|
for (char const& c : str) {
|
|
if (!isxdigit(c)) return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
namespace JC {
|
|
|
|
int setColor(const std::vector<int>& colors)
|
|
{
|
|
Result pads, ljc, rjc;
|
|
int res = 0;
|
|
s32 nbEntries;
|
|
HidsysUniquePadId UniquePadIds[2] = {};
|
|
|
|
pads = hidsysGetUniquePadsFromNpad(HidNpadIdType_Handheld, UniquePadIds, 2, &nbEntries);
|
|
if (R_SUCCEEDED(pads)) {
|
|
ljc = hiddbgUpdateControllerColor(colors[0], colors[1], UniquePadIds[0]);
|
|
if (R_FAILED(ljc)) res += 1;
|
|
rjc = hiddbgUpdateControllerColor(colors[2], colors[3], UniquePadIds[1]);
|
|
if (R_FAILED(rjc)) res += 2;
|
|
}
|
|
else {
|
|
res += 4;
|
|
}
|
|
return res;
|
|
}
|
|
|
|
int backupToJSON(json& profiles, const std::string& path)
|
|
{
|
|
HidNpadControllerColor color_left;
|
|
HidNpadControllerColor color_right;
|
|
Result res = hidGetNpadControllerColorSplit(HidNpadIdType_Handheld, &color_left, &color_right);
|
|
std::vector<int> oldBackups;
|
|
if (R_SUCCEEDED(res)) {
|
|
int i = 0;
|
|
for (const auto& x : profiles.items()) {
|
|
if (x.value()["name"] == BACKUP) {
|
|
oldBackups.push_back(i);
|
|
}
|
|
i++;
|
|
}
|
|
for (auto& k : oldBackups) {
|
|
profiles.erase(profiles.begin() + k);
|
|
}
|
|
json newBackup = json::object(
|
|
{{"name", BACKUP},
|
|
{"L_JC", BGRToHex(color_left.main)},
|
|
{"L_BTN", BGRToHex(color_left.sub)},
|
|
{"R_JC", BGRToHex(color_right.main)},
|
|
{"R_BTN", BGRToHex(color_right.sub)}});
|
|
profiles.push_back(newBackup);
|
|
fs::writeJsonToFile(profiles, path);
|
|
return 0;
|
|
}
|
|
else {
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
json backupProfile()
|
|
{
|
|
json newBackup;
|
|
HidNpadControllerColor color_left;
|
|
HidNpadControllerColor color_right;
|
|
Result res = hidGetNpadControllerColorSplit(HidNpadIdType_Handheld, &color_left, &color_right);
|
|
if (R_SUCCEEDED(res)) {
|
|
newBackup = json::object(
|
|
{{"name", BACKUP},
|
|
{"L_JC", BGRToHex(color_left.main)},
|
|
{"L_BTN", BGRToHex(color_left.sub)},
|
|
{"R_JC", BGRToHex(color_right.main)},
|
|
{"R_BTN", BGRToHex(color_right.sub)}});
|
|
}
|
|
return newBackup;
|
|
}
|
|
|
|
std::deque<std::pair<std::string, std::vector<int>>> getProfiles(const std::string& path)
|
|
{
|
|
std::deque<std::pair<std::string, std::vector<int>>> res;
|
|
bool properData;
|
|
std::fstream profilesFile;
|
|
nlohmann::ordered_json profilesJson;
|
|
download::getRequest(JC_COLOR_URL, profilesJson);
|
|
if (profilesJson.empty()) {
|
|
profilesJson = {{{"L_BTN", "0A1E0A"},
|
|
{"L_JC", "82FF96"},
|
|
{"R_BTN", "0A1E28"},
|
|
{"R_JC", "96F5F5"},
|
|
{"name", "Animal Crossing: New Horizons"}}};
|
|
}
|
|
for (const auto& profiles : {fs::parseJsonFile(path), profilesJson}) {
|
|
for (const auto& x : profiles.items()) {
|
|
std::string name = x.value()["name"];
|
|
brls::Logger::warning(name);
|
|
std::vector<std::string> values = {
|
|
std::string(x.value()["L_JC"]),
|
|
std::string(x.value()["L_BTN"]),
|
|
std::string(x.value()["R_JC"]),
|
|
std::string(x.value()["R_BTN"])};
|
|
properData = true;
|
|
for (auto& str : values) {
|
|
if (!isHexaAnd3Bytes(str)) {
|
|
properData = false;
|
|
}
|
|
}
|
|
if (properData) {
|
|
if (name == "") name = "Unamed";
|
|
auto profile = std::make_pair(name, (std::vector<int>){
|
|
hexToBGR(values[0]),
|
|
hexToBGR(values[1]),
|
|
hexToBGR(values[2]),
|
|
hexToBGR(values[3])});
|
|
if (name == BACKUP) {
|
|
res.push_front(profile);
|
|
}
|
|
else {
|
|
res.push_back(profile);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
void changeJCColor(const std::vector<int>& values)
|
|
{
|
|
hiddbgInitialize();
|
|
hidsysInitialize();
|
|
ProgressEvent::instance().setStep(1);
|
|
int res = setColor(values);
|
|
if (res != 0) {
|
|
util::showDialogBoxBlocking("Could not change the Joy-Cons color. Make sure they're docked and try again.\nError :" + std::to_string(res), "Ok");
|
|
}
|
|
hiddbgExit();
|
|
hidsysExit();
|
|
ProgressEvent::instance().setStep(ProgressEvent::instance().getMax());
|
|
}
|
|
|
|
void backupJCColor(const std::string& path)
|
|
{
|
|
hiddbgInitialize();
|
|
hidsysInitialize();
|
|
ProgressEvent::instance().setStep(1);
|
|
json backup;
|
|
json profiles;
|
|
std::fstream profilesFile;
|
|
if (std::filesystem::exists(path)) {
|
|
profilesFile.open(path, std::fstream::in);
|
|
profilesFile >> profiles;
|
|
profilesFile.close();
|
|
}
|
|
|
|
std::vector<int> oldBackups;
|
|
int i = 0;
|
|
for (const auto& x : profiles.items()) {
|
|
if (x.value()["name"] == BACKUP) {
|
|
oldBackups.push_back(i);
|
|
}
|
|
i++;
|
|
}
|
|
for (auto& k : oldBackups) {
|
|
profiles.erase(profiles.begin() + k);
|
|
}
|
|
|
|
while (backup.empty()) {
|
|
backup = backupProfile();
|
|
}
|
|
|
|
profiles.push_back(backup);
|
|
//backup.push_back(profiles);
|
|
fs::writeJsonToFile(profiles, path);
|
|
hiddbgExit();
|
|
hidsysExit();
|
|
ProgressEvent::instance().setStep(ProgressEvent::instance().getMax());
|
|
}
|
|
|
|
} // namespace JC
|
|
|
|
namespace PC {
|
|
|
|
int setColor(const std::vector<int>& colors)
|
|
{
|
|
Result pads, pc;
|
|
int res = 0;
|
|
s32 nbEntries;
|
|
HidsysUniquePadId UniquePadIds[1] = {};
|
|
|
|
pads = hidsysGetUniquePadsFromNpad(HidNpadIdType_No1, UniquePadIds, 1, &nbEntries);
|
|
if (R_SUCCEEDED(pads)) {
|
|
pc = hiddbgUpdateControllerColor(colors[0], colors[1], UniquePadIds[0]);
|
|
if (R_FAILED(pc)) res += 1;
|
|
}
|
|
else {
|
|
res += 4;
|
|
}
|
|
return res;
|
|
}
|
|
|
|
int backupToJSON(json& profiles, const std::string& path)
|
|
{
|
|
HidNpadControllerColor color;
|
|
Result res = hidGetNpadControllerColorSingle(HidNpadIdType_No1, &color);
|
|
std::vector<int> oldBackups;
|
|
if (R_SUCCEEDED(res)) {
|
|
int i = 0;
|
|
for (const auto& x : profiles.items()) {
|
|
if (x.value()["name"] == BACKUP) {
|
|
oldBackups.push_back(i);
|
|
}
|
|
i++;
|
|
}
|
|
for (auto& k : oldBackups) {
|
|
profiles.erase(profiles.begin() + k);
|
|
}
|
|
json newBackup = json::object(
|
|
{{"name", BACKUP},
|
|
{"BODY", BGRToHex(color.main)},
|
|
{"BTN", BGRToHex(color.sub)}});
|
|
profiles.push_back(newBackup);
|
|
fs::writeJsonToFile(profiles, path);
|
|
return 0;
|
|
}
|
|
else {
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
json backupProfile()
|
|
{
|
|
json newBackup;
|
|
HidNpadControllerColor color;
|
|
Result res = hidGetNpadControllerColorSingle(HidNpadIdType_No1, &color);
|
|
if (R_SUCCEEDED(res)) {
|
|
newBackup = json::object(
|
|
{{"name", BACKUP},
|
|
{"BODY", BGRToHex(color.main)},
|
|
{"BTN", BGRToHex(color.sub)}});
|
|
}
|
|
return newBackup;
|
|
}
|
|
|
|
std::deque<std::pair<std::string, std::vector<int>>> getProfiles(const std::string& path)
|
|
{
|
|
std::deque<std::pair<std::string, std::vector<int>>> res;
|
|
bool properData;
|
|
std::fstream profilesFile;
|
|
nlohmann::ordered_json profilesJson;
|
|
download::getRequest(PC_COLOR_URL, profilesJson);
|
|
if (profilesJson.empty()) {
|
|
profilesJson = {{{"BTN", "e6e6e6"},
|
|
{"BODY", "2d2d2d"},
|
|
{"name", "Default black"}}};
|
|
}
|
|
for (const auto& profiles : {fs::parseJsonFile(path), profilesJson}) {
|
|
for (const auto& x : profiles.items()) {
|
|
std::string name = x.value()["name"];
|
|
std::vector<std::string> values = {
|
|
std::string(x.value()["BODY"]),
|
|
std::string(x.value()["BTN"])};
|
|
properData = true;
|
|
for (auto& str : values) {
|
|
if (!isHexaAnd3Bytes(str)) {
|
|
properData = false;
|
|
}
|
|
}
|
|
if (properData) {
|
|
if (name == "") name = "Unamed";
|
|
auto profile = std::make_pair(name, (std::vector<int>){
|
|
hexToBGR(values[0]),
|
|
hexToBGR(values[1])});
|
|
if (name == BACKUP) {
|
|
res.push_front(profile);
|
|
}
|
|
else {
|
|
res.push_back(profile);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
void changePCColor(const std::vector<int>& values)
|
|
{
|
|
hiddbgInitialize();
|
|
hidsysInitialize();
|
|
ProgressEvent::instance().setStep(1);
|
|
int res = setColor(values);
|
|
if (res != 0) {
|
|
util::showDialogBoxBlocking("Could not change the Pro-Con color. Make they're connected to P1. This feature may not work on unofficial controllers. \nError :" + std::to_string(res), "Ok");
|
|
}
|
|
hiddbgExit();
|
|
hidsysExit();
|
|
ProgressEvent::instance().setStep(ProgressEvent::instance().getMax());
|
|
}
|
|
|
|
void backupPCColor(const std::string& path)
|
|
{
|
|
hiddbgInitialize();
|
|
hidsysInitialize();
|
|
ProgressEvent::instance().setStep(1);
|
|
json backup;
|
|
json profiles;
|
|
std::fstream profilesFile;
|
|
if (std::filesystem::exists(path)) {
|
|
profilesFile.open(path, std::fstream::in);
|
|
profilesFile >> profiles;
|
|
profilesFile.close();
|
|
}
|
|
|
|
std::vector<int> oldBackups;
|
|
int i = 0;
|
|
for (const auto& x : profiles.items()) {
|
|
if (x.value()["name"] == BACKUP) {
|
|
oldBackups.push_back(i);
|
|
}
|
|
i++;
|
|
}
|
|
for (auto& k : oldBackups) {
|
|
profiles.erase(profiles.begin() + k);
|
|
}
|
|
|
|
while (backup.empty()) {
|
|
backup = backupProfile();
|
|
}
|
|
|
|
profiles.push_back(backup);
|
|
//backup.push_back(profiles);
|
|
fs::writeJsonToFile(profiles, path);
|
|
hiddbgExit();
|
|
hidsysExit();
|
|
ProgressEvent::instance().setStep(ProgressEvent::instance().getMax());
|
|
}
|
|
|
|
} // namespace PC
|