mirror of
https://github.com/Alia5/GlosSI.git
synced 2024-10-30 15:20:38 +00:00
415 lines
15 KiB
C++
415 lines
15 KiB
C++
/*
|
|
Copyright 2021-2022 Peter Repukat - FlatspotSoftware
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
#include "UIModel.h"
|
|
|
|
#include <QDir>
|
|
#include <QGuiApplication>
|
|
#include <QJsonDocument>
|
|
|
|
#include <WinReg/WinReg.hpp>
|
|
|
|
#ifdef _WIN32
|
|
#include "UWPFetch.h"
|
|
#include <Windows.h>
|
|
#endif
|
|
|
|
UIModel::UIModel() : QObject(nullptr)
|
|
{
|
|
auto path = std::filesystem::temp_directory_path()
|
|
.parent_path()
|
|
.parent_path()
|
|
.parent_path();
|
|
|
|
path /= "Roaming";
|
|
path /= "GlosSI";
|
|
if (!std::filesystem::exists(path))
|
|
std::filesystem::create_directories(path);
|
|
|
|
config_path_ = path;
|
|
config_dir_name_ = (path /= "Targets").string().data();
|
|
|
|
if (!std::filesystem::exists(path))
|
|
std::filesystem::create_directories(path);
|
|
|
|
parseShortcutVDF();
|
|
readConfigs();
|
|
}
|
|
|
|
void UIModel::readConfigs()
|
|
{
|
|
QDir dir(config_dir_name_);
|
|
auto entries = dir.entryList(QDir::Files, QDir::SortFlag::Name);
|
|
entries.removeIf([](const auto& entry) {
|
|
return !entry.endsWith(".json");
|
|
});
|
|
|
|
std::for_each(entries.begin(), entries.end(), [this](const auto& name) {
|
|
auto path = config_path_;
|
|
path /= config_dir_name_.toStdString();
|
|
path /= name.toStdWString();
|
|
QFile file(path);
|
|
if (!file.open(QIODevice::Text | QIODevice::ReadOnly)) {
|
|
// meh
|
|
return;
|
|
}
|
|
const auto data = file.readAll();
|
|
file.close();
|
|
const auto jsondoc = QJsonDocument::fromJson(data);
|
|
const auto filejson = jsondoc.object();
|
|
|
|
QJsonObject json;
|
|
json["version"] = filejson["version"];
|
|
json["icon"] = filejson["icon"];
|
|
json["launch"] = filejson["launch"]["launch"];
|
|
json["launchPath"] = filejson["launch"]["launchPath"];
|
|
json["launchAppArgs"] = filejson["launch"]["launchAppArgs"];
|
|
json["closeOnExit"] = filejson["launch"]["closeOnExit"];
|
|
json["waitForChildProcs"] = filejson["launch"]["waitForChildProcs"];
|
|
json["hideDevices"] = filejson["devices"]["hideDevices"];
|
|
json["realDeviceIds"] = filejson["devices"]["realDeviceIds"];
|
|
json["windowMode"] = filejson["window"]["windowMode"];
|
|
json["maxFps"] = filejson["window"]["maxFps"];
|
|
json["scale"] = filejson["window"]["scale"];
|
|
json["disableOverlay"] = filejson["window"]["disableOverlay"];
|
|
json["maxControllers"] = filejson["controller"]["allowDesktopConfig"];
|
|
json["allowDesktopConfig"] = filejson["controller"]["allowDesktopConfig"];
|
|
|
|
json["name"] = filejson.contains("name") ? filejson["name"] : QString(name).replace(QRegularExpression("\\.json"), "");
|
|
|
|
targets_.append(json.toVariantMap());
|
|
});
|
|
|
|
emit targetListChanged();
|
|
}
|
|
|
|
QVariantList UIModel::getTargetList() const
|
|
{
|
|
return targets_;
|
|
}
|
|
|
|
void UIModel::addTarget(QVariant shortcut)
|
|
{
|
|
const auto map = shortcut.toMap();
|
|
const auto json = QJsonObject::fromVariantMap(map);
|
|
writeTarget(json, map["name"].toString());
|
|
targets_.append(QJsonDocument(json).toVariant());
|
|
emit targetListChanged();
|
|
}
|
|
|
|
void UIModel::updateTarget(int index, QVariant shortcut)
|
|
{
|
|
const auto map = shortcut.toMap();
|
|
const auto json = QJsonObject::fromVariantMap(map);
|
|
|
|
auto oldName = targets_[index].toMap()["name"].toString().replace(QRegularExpression("[\\\\/:*?\"<>|]"), "") + ".json";
|
|
auto path = config_path_;
|
|
path /= config_dir_name_.toStdString();
|
|
path /= (oldName).toStdString();
|
|
std::filesystem::remove(path);
|
|
|
|
writeTarget(json, map["name"].toString());
|
|
|
|
targets_.replace(index, QJsonDocument(json).toVariant());
|
|
emit targetListChanged();
|
|
}
|
|
|
|
void UIModel::deleteTarget(int index)
|
|
{
|
|
auto oldName = targets_[index].toMap()["name"].toString().replace(QRegularExpression("[\\\\/:*?\"<>|]"), "") + ".json";
|
|
auto path = config_path_;
|
|
path /= config_dir_name_.toStdString();
|
|
path /= (oldName).toStdString();
|
|
std::filesystem::remove(path);
|
|
targets_.remove(index);
|
|
emit targetListChanged();
|
|
}
|
|
|
|
bool UIModel::isInSteam(QVariant shortcut)
|
|
{
|
|
const auto map = shortcut.toMap();
|
|
for (auto& steam_shortcut : shortcuts_vdf_.shortcuts) {
|
|
if (map["name"].toString() == QString::fromStdString(steam_shortcut.appName.value)) {
|
|
if (QString::fromStdString(steam_shortcut.exe.value).toLower().contains("glossitarget.exe")) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool UIModel::addToSteam(QVariant shortcut, const QString& shortcutspath, bool from_cmd)
|
|
{
|
|
QDir appDir = QGuiApplication::applicationDirPath();
|
|
const auto map = shortcut.toMap();
|
|
const auto name = map["name"].toString();
|
|
const auto maybeLaunchPath = map["launchPath"].toString();
|
|
const auto launch = map["launch"].toBool();
|
|
|
|
VDFParser::Shortcut vdfshortcut;
|
|
vdfshortcut.idx = shortcuts_vdf_.shortcuts.size();
|
|
vdfshortcut.appName.value = name.toStdString();
|
|
vdfshortcut.exe.value = ("\"" + appDir.absolutePath() + "/GlosSITarget.exe" + "\"").toStdString();
|
|
vdfshortcut.StartDir.value = (launch && !maybeLaunchPath.isEmpty()
|
|
? (std::string("\"") + std::filesystem::path(maybeLaunchPath.toStdString()).parent_path().string() + "\"")
|
|
: ("\"" + appDir.absolutePath() + "\"").toStdString());
|
|
vdfshortcut.appId.value = VDFParser::Parser::calculateAppId(vdfshortcut);
|
|
// ShortcutPath; default
|
|
vdfshortcut.LaunchOptions.value = (name + ".json").toStdString();
|
|
// IsHidden; default
|
|
// AllowDesktopConfig; default
|
|
// AllowOverlay; default
|
|
// openvr; default
|
|
// Devkit; default
|
|
// DevkitGameID; default
|
|
// DevkitOverrideAppID; default
|
|
// LastPlayTime; default
|
|
auto maybeIcon = map["icon"].toString();
|
|
if (maybeIcon.isEmpty()) {
|
|
if (launch && !maybeLaunchPath.isEmpty())
|
|
vdfshortcut.icon.value =
|
|
"\"" + (is_windows_ ? QString(maybeLaunchPath).replace(QRegularExpression("\\/"), "\\").toStdString() : maybeLaunchPath.toStdString()) + "\"";
|
|
}
|
|
else {
|
|
vdfshortcut.icon.value =
|
|
"\"" + (is_windows_ ? QString(maybeIcon).replace(QRegularExpression("\\/"), "\\").toStdString() : maybeIcon.toStdString()) + "\"";
|
|
}
|
|
// Add installed locally and GlosSI tag
|
|
VDFParser::ShortcutTag locallyTag;
|
|
locallyTag.idx = 0;
|
|
locallyTag.value = "Installed locally";
|
|
vdfshortcut.tags.value.push_back(locallyTag);
|
|
|
|
VDFParser::ShortcutTag glossitag;
|
|
glossitag.idx = 1;
|
|
glossitag.value = "GlosSI";
|
|
vdfshortcut.tags.value.push_back(glossitag);
|
|
|
|
shortcuts_vdf_.shortcuts.push_back(vdfshortcut);
|
|
|
|
return writeShortcutsVDF(L"add", name.toStdWString(), shortcutspath.toStdWString(), from_cmd);
|
|
}
|
|
bool UIModel::addToSteam(const QString& name, const QString& shortcutspath, bool from_cmd)
|
|
{
|
|
qDebug() << "trying to add " << name << " to steam";
|
|
const auto target = std::find_if(targets_.begin(), targets_.end(), [&name](const auto& target) {
|
|
const auto map = target.toMap();
|
|
const auto target_name = map["name"].toString().replace(QRegularExpression("[\\\\/:*?\"<>|]"), "");
|
|
return name == target_name;
|
|
});
|
|
if (target != targets_.end()) {
|
|
return addToSteam(*target, shortcutspath, from_cmd);
|
|
}
|
|
qDebug() << name << " not found!";
|
|
return false;
|
|
}
|
|
bool UIModel::removeFromSteam(const QString& name, const QString& shortcutspath, bool from_cmd)
|
|
{
|
|
qDebug() << "trying to remove " << name << " from steam";
|
|
auto& scuts = shortcuts_vdf_.shortcuts;
|
|
scuts.erase(std::remove_if(scuts.begin(), scuts.end(), [&name](const auto& shortcut) {
|
|
return shortcut.appName.value == name.toStdString();
|
|
}),
|
|
scuts.end());
|
|
for (int i = 0; i < scuts.size(); i++) {
|
|
if (scuts[i].idx != i) {
|
|
scuts[i].idx = i;
|
|
}
|
|
}
|
|
return writeShortcutsVDF(L"remove", name.toStdWString(), shortcutspath.toStdWString(), from_cmd);
|
|
}
|
|
|
|
QVariantMap UIModel::manualProps(QVariant shortcut)
|
|
{
|
|
QDir appDir = QGuiApplication::applicationDirPath();
|
|
const auto map = shortcut.toMap();
|
|
const auto name = map["name"].toString().replace(QRegularExpression("[\\\\/:*?\"<>|]"), "");
|
|
const auto maybeLaunchPath = map["launchPath"].toString();
|
|
const auto launch = map["launch"].toBool();
|
|
|
|
QVariantMap res;
|
|
res.insert("name", name);
|
|
res.insert("config", name + ".json");
|
|
res.insert("launch", ("\"" + appDir.absolutePath() + "/GlosSITarget.exe" + "\""));
|
|
res.insert("launchDir", (
|
|
launch && !maybeLaunchPath.isEmpty()
|
|
? (QString("\"") + QString::fromStdString(std::filesystem::path(maybeLaunchPath.toStdString()).parent_path().string()) + "\"")
|
|
: ("\"" + appDir.absolutePath() + "\"")));
|
|
return res;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
QVariantList UIModel::uwpApps()
|
|
{
|
|
return UWPFetch::UWPAppList();
|
|
}
|
|
#endif
|
|
|
|
bool UIModel::writeShortcutsVDF(const std::wstring& mode, const std::wstring& name, const std::wstring& shortcutspath, bool is_admin_try) const
|
|
{
|
|
#ifdef _WIN32
|
|
const std::filesystem::path config_path = is_admin_try
|
|
? shortcutspath
|
|
: std::wstring(getSteamPath()) + user_data_path_.toStdWString() + getSteamUserId() + shortcutsfile_.toStdWString();
|
|
|
|
qDebug() << "Steam config Path: " << config_path;
|
|
qDebug() << "Trying to write config as admin: " << is_admin_try;
|
|
|
|
bool write_res;
|
|
try {
|
|
write_res = VDFParser::Parser::writeShortcuts(config_path, shortcuts_vdf_);
|
|
}
|
|
catch (const std::exception& e) {
|
|
qDebug() << "Couldn't backup shortcuts file: " << e.what();
|
|
}
|
|
|
|
if (!write_res && !is_admin_try) {
|
|
wchar_t szPath[MAX_PATH];
|
|
if (GetModuleFileName(NULL, szPath, ARRAYSIZE(szPath))) {
|
|
// Launch itself as admin
|
|
SHELLEXECUTEINFO sei = {sizeof(sei)};
|
|
sei.lpVerb = L"runas";
|
|
qDebug() << QString("exepath: %1").arg(szPath);
|
|
sei.lpFile = szPath;
|
|
const std::wstring paramstr = mode + L" " + name + L" \"" + config_path.wstring() + L"\"";
|
|
sei.lpParameters = paramstr.c_str();
|
|
sei.hwnd = NULL;
|
|
sei.nShow = SW_NORMAL;
|
|
sei.fMask = SEE_MASK_NOCLOSEPROCESS;
|
|
if (!ShellExecuteEx(&sei)) {
|
|
DWORD dwError = GetLastError();
|
|
if (dwError == ERROR_CANCELLED) {
|
|
qDebug() << "User cancelled UAC Prompt";
|
|
return false;
|
|
}
|
|
}
|
|
else {
|
|
qDebug() << QString("HProc: %1").arg((int)sei.hProcess);
|
|
|
|
if (sei.hProcess && WAIT_OBJECT_0 == WaitForSingleObject(sei.hProcess, INFINITE)) {
|
|
DWORD exitcode = 1;
|
|
GetExitCodeProcess(sei.hProcess, &exitcode);
|
|
qDebug() << QString("Exitcode: %1").arg((int)exitcode);
|
|
if (exitcode == 0) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
return write_res;
|
|
#else
|
|
return VDFParser::Parser::writeShortcuts(config_path, shortcuts_vdf_);
|
|
#endif
|
|
}
|
|
|
|
bool UIModel::getIsWindows() const
|
|
{
|
|
return is_windows_;
|
|
}
|
|
|
|
bool UIModel::hasAcrylicEffect() const
|
|
{
|
|
return has_acrylic_affect_;
|
|
}
|
|
|
|
void UIModel::setAcrylicEffect(bool has_acrylic_affect)
|
|
{
|
|
has_acrylic_affect_ = has_acrylic_affect;
|
|
emit acrylicChanged();
|
|
}
|
|
|
|
void UIModel::writeTarget(const QJsonObject& json, const QString& name)
|
|
{
|
|
auto path = config_path_;
|
|
path /= config_dir_name_.toStdWString();
|
|
path /= (QString(name).replace(QRegularExpression("[\\\\/:*?\"<>|]"), "") + ".json").toStdWString();
|
|
QFile file(path);
|
|
if (!file.open(QIODevice::Text | QIODevice::ReadWrite)) {
|
|
// meh
|
|
return;
|
|
}
|
|
QJsonObject fileJson;
|
|
fileJson["version"] = json["version"];
|
|
fileJson["icon"] = json["icon"];
|
|
fileJson["name"] = json["name"];
|
|
|
|
QJsonObject launchObject;
|
|
launchObject["launch"] = json["launch"];
|
|
launchObject["launchPath"] = json["launchPath"];
|
|
launchObject["launchAppArgs"] = json["launchAppArgs"];
|
|
launchObject["closeOnExit"] = json["closeOnExit"];
|
|
launchObject["waitForChildProcs"] = json["waitForChildProcs"];
|
|
fileJson["launch"] = launchObject;
|
|
|
|
QJsonObject devicesObject;
|
|
devicesObject["hideDevices"] = json["hideDevices"];
|
|
devicesObject["realDeviceIds"] = json["realDeviceIds"];
|
|
fileJson["devices"] = devicesObject;
|
|
|
|
QJsonObject windowObject;
|
|
windowObject["windowMode"] = json["windowMode"];
|
|
windowObject["maxFps"] = json["maxFps"];
|
|
windowObject["scale"] = json["scale"];
|
|
windowObject["disableOverlay"] = json["disableOverlay"];
|
|
fileJson["window"] = windowObject;
|
|
|
|
QJsonObject controllerObject;
|
|
controllerObject["maxControllers"] = json["maxControllers"];
|
|
controllerObject["allowDesktopConfig"] = json["allowDesktopConfig"];
|
|
fileJson["controller"] = controllerObject;
|
|
|
|
auto wtf = QString(QJsonDocument(fileJson).toJson(QJsonDocument::Indented)).toStdString();
|
|
file.write(wtf.data());
|
|
file.close();
|
|
}
|
|
|
|
std::filesystem::path UIModel::getSteamPath() const
|
|
{
|
|
#ifdef _WIN32
|
|
// TODO: check if keys/value exist
|
|
// steam should always be open and have written reg values...
|
|
winreg::RegKey key{HKEY_CURRENT_USER, L"SOFTWARE\\Valve\\Steam"};
|
|
const auto res = key.GetStringValue(L"SteamPath");
|
|
return res;
|
|
#else
|
|
return L""; // TODO LINUX
|
|
#endif
|
|
}
|
|
|
|
std::wstring UIModel::getSteamUserId() const
|
|
{
|
|
#ifdef _WIN32
|
|
// TODO: check if keys/value exist
|
|
// steam should always be open and have written reg values...
|
|
winreg::RegKey key{HKEY_CURRENT_USER, L"SOFTWARE\\Valve\\Steam\\ActiveProcess"};
|
|
const auto res = std::to_wstring(key.GetDwordValue(L"ActiveUser"));
|
|
if (res == L"0") {
|
|
qDebug() << "Steam not open?";
|
|
}
|
|
return res;
|
|
#else
|
|
return L""; // TODO LINUX
|
|
#endif
|
|
}
|
|
|
|
void UIModel::parseShortcutVDF()
|
|
{
|
|
const std::filesystem::path config_path = std::wstring(getSteamPath()) + user_data_path_.toStdWString() + getSteamUserId() + shortcutsfile_.toStdWString();
|
|
shortcuts_vdf_ = VDFParser::Parser::parseShortcuts(config_path);
|
|
}
|