353 lines
14 KiB
JavaScript
353 lines
14 KiB
JavaScript
const { app, BrowserWindow, ipcMain, dialog } = require('electron');
|
|
const { setupTitlebar, attachTitlebarToWindow } = require('custom-electron-titlebar/main');
|
|
const windowManager = require('./ui/windowManager');
|
|
const osuUtil = require('./osuUtil');
|
|
const ezppUtil = require('./ezppUtil');
|
|
const config = require('./config');
|
|
const fs = require('fs');
|
|
const rpc = require('./discordPresence');
|
|
const windowName = require('get-window-by-name');
|
|
const terminalUtil = require('./terminalUtil');
|
|
const osUtil = require('./osUtil');
|
|
|
|
let tempOsuPath;
|
|
let osuWindowInfo;
|
|
let isIngame;
|
|
const platform = process.platform;
|
|
let linuxWMCtrlFound = false;
|
|
let osuLoaded = false;
|
|
|
|
const run = () => {
|
|
const gotTheLock = app.requestSingleInstanceLock()
|
|
if (!gotTheLock) {
|
|
app.quit();
|
|
return;
|
|
}
|
|
|
|
setInterval(async () => {
|
|
if (platform == "win32") {
|
|
osuWindowInfo = windowName.getWindowText("osu!.exe");
|
|
const firstInstance = osuWindowInfo[0];
|
|
if (firstInstance) {
|
|
if (firstInstance.processTitle && firstInstance.processTitle.length > 0) {
|
|
if (!osuLoaded) {
|
|
osuLoaded = true;
|
|
//TODO: do patch
|
|
}
|
|
const windowTitle = firstInstance.processTitle;
|
|
let rpcOsuVersion = "";
|
|
let currentMap = undefined;
|
|
if (!windowTitle.includes("-")) {
|
|
rpcOsuVersion = windowTitle;
|
|
rpc.updateState("Idle...");
|
|
} else {
|
|
var string = windowTitle;
|
|
var components = string.split(' - ');
|
|
const splitArray = [components.shift(), components.join(' - ')];
|
|
rpcOsuVersion = splitArray[0];
|
|
currentMap = splitArray[1];
|
|
if (currentMap.endsWith(".osu")) {
|
|
rpc.updateState("Editing...");
|
|
currentMap = currentMap.substring(0, currentMap.length - 4);
|
|
} else rpc.updateState("Playing...");
|
|
}
|
|
|
|
rpc.updateStatus(currentMap, rpcOsuVersion);
|
|
} else {
|
|
if (osuLoaded) osuLoaded = false;
|
|
rpc.updateState("Idle in Launcher...");
|
|
rpc.updateStatus(undefined, undefined);
|
|
}
|
|
} else {
|
|
if (osuLoaded) osuLoaded = false;
|
|
rpc.updateState("Idle in Launcher...");
|
|
rpc.updateStatus(undefined, undefined);
|
|
}
|
|
} else if (platform == "linux") {
|
|
if (!linuxWMCtrlFound) {
|
|
if (isIngame) {
|
|
rpc.updateState("Playing...");
|
|
rpc.updateStatus("Clicking circles!", "runningunderwine");
|
|
} else {
|
|
rpc.updateState("Idle in Launcher...");
|
|
rpc.updateStatus(undefined, undefined);
|
|
}
|
|
return;
|
|
}
|
|
const processesOutput = await terminalUtil.execCommand(`wmctrl -l|awk '{$3=""; $2=""; $1=""; print $0}'`);
|
|
const allLines = processesOutput.split("\n");
|
|
const filteredProcesses = allLines.filter((line) => line.trim().startsWith("osu!"));
|
|
if (filteredProcesses.length > 0) {
|
|
const windowTitle = filteredProcesses[0].trim();
|
|
let rpcOsuVersion = "";
|
|
let currentMap = undefined;
|
|
if (!windowTitle.includes("-")) {
|
|
rpcOsuVersion = windowTitle;
|
|
rpc.updateState("Idle...");
|
|
} else {
|
|
var string = windowTitle;
|
|
var components = string.split(' - ');
|
|
const splitArray = [components.shift(), components.join(' - ')];
|
|
rpcOsuVersion = splitArray[0];
|
|
currentMap = splitArray[1];
|
|
if (currentMap.endsWith(".osu")) {
|
|
rpc.updateState("Editing/Modding...");
|
|
currentMap = currentMap.substring(0, currentMap.length - 4);
|
|
}
|
|
else rpc.updateState("Playing...");
|
|
}
|
|
|
|
rpc.updateStatus(currentMap, rpcOsuVersion);
|
|
} else {
|
|
rpc.updateState("Idle in Launcher...");
|
|
rpc.updateStatus(undefined, undefined);
|
|
}
|
|
} else {
|
|
if (isIngame) {
|
|
rpc.updateState("Playing...");
|
|
rpc.updateStatus("Clicking circles!", "runningunderwine");
|
|
} else {
|
|
rpc.updateState("Idle in Launcher...");
|
|
rpc.updateStatus(undefined, undefined);
|
|
}
|
|
}
|
|
}, 2000);
|
|
|
|
setupTitlebar();
|
|
|
|
rpc.connect();
|
|
|
|
|
|
let mainWindow;
|
|
app.whenReady().then(() => {
|
|
|
|
mainWindow = createWindow();
|
|
mainWindow.on('show', async () => {
|
|
await updateConfigVars(mainWindow);
|
|
await tryLogin(mainWindow);
|
|
await doUpdateCheck(mainWindow);
|
|
if (platform === "linux") {
|
|
const linuxDistroInfo = await osUtil.getLinuxDistroInfo();
|
|
if (linuxDistroInfo?.id != "arch") {
|
|
if (linuxDistroInfo?.id_like != "arch") {
|
|
mainWindow.webContents.send('status_update', {
|
|
type: "info",
|
|
message: "We detected that you are running the Launcher under Linux. It's currently just compatible with Arch like distributions!"
|
|
});
|
|
}
|
|
}
|
|
try {
|
|
await terminalUtil.execCommand(`osu-stable -h`);
|
|
} catch (err) {
|
|
mainWindow.webContents.send('status_update', {
|
|
type: "package-issue",
|
|
message: "Seems like you dont have the osu AUR Package installed, please install it."
|
|
});
|
|
return;
|
|
}
|
|
/* mainWindow.webContents.send('status_update', {
|
|
type: "info",
|
|
message: "We detected that you are running the Launcher under Linux. It's currently just compatible with Arch and the osu AUR package!"
|
|
}); */
|
|
const terminalTest = await terminalUtil.execCommand(`wmctrl -l|awk '{$3=""; $2=""; $1=""; print $0}'`);
|
|
const isFailed = terminalTest.trim() == "";
|
|
if (isFailed) {
|
|
mainWindow.webContents.send('status_update', {
|
|
type: "info",
|
|
message: "Seems like you are missing the wmctrl package, please install it for the RPC to work!"
|
|
});
|
|
} else linuxWMCtrlFound = true;
|
|
} else {
|
|
const osuFolder = await config.get("osuPath");
|
|
if (!osuFolder || osuFolder == "") {
|
|
const foundFolder = await osuUtil.findOsuInstallation();
|
|
console.log("osu! Installation located at: ",foundFolder);
|
|
}
|
|
}
|
|
})
|
|
app.on('activate', function () {
|
|
if (BrowserWindow.getAllWindows().length === 0) mainWindow = createWindow();
|
|
})
|
|
app.on('window-all-closed', () => {
|
|
app.quit()
|
|
})
|
|
ipcMain.handle('launch', async () => {
|
|
const osuConfig = await osuUtil.getLatestConfig(tempOsuPath);
|
|
const username = await config.get('username');
|
|
const password = await config.get('password');
|
|
if (password && username) {
|
|
await osuUtil.setConfigValue(osuConfig.path, "SaveUsername", "1");
|
|
await osuUtil.setConfigValue(osuConfig.path, "SavePassword", "1");
|
|
await osuUtil.setConfigValue(osuConfig.path, "Username", username);
|
|
await osuUtil.setConfigValue(osuConfig.path, "Password", password);
|
|
await osuUtil.setConfigValue(osuConfig.path, "CredentialEndpoint", "ez-pp.farm");
|
|
} else {
|
|
await osuUtil.setConfigValue(osuConfig.path, "Username", "");
|
|
await osuUtil.setConfigValue(osuConfig.path, "Password", "");
|
|
}
|
|
rpc.updateState("Launching osu!...");
|
|
isIngame = true;
|
|
mainWindow.hide();
|
|
const result = await osuUtil.startOsuWithDevServer(tempOsuPath, "ez-pp.farm", async () => {
|
|
isIngame = false;
|
|
mainWindow.show();
|
|
await doUpdateCheck(mainWindow);
|
|
});
|
|
return result;
|
|
})
|
|
ipcMain.on('do-update-check', async () => {
|
|
await doUpdateCheck(mainWindow);
|
|
})
|
|
ipcMain.on('do-update', async () => {
|
|
const osuPath = await config.get("osuPath", "");
|
|
const isValid = await osuUtil.isValidOsuFolder(osuPath);
|
|
if (osuPath.trim == "" || !isValid) {
|
|
mainWindow.webContents.send('status_update', {
|
|
type: "error",
|
|
message: "Invalid osu! folder"
|
|
});
|
|
return;
|
|
}
|
|
if (fs.existsSync(osuPath)) {
|
|
tempOsuPath = osuPath;
|
|
const osuConfig = await osuUtil.getLatestConfig(tempOsuPath);
|
|
const lastVersion = await osuConfig.get("LastVersion");
|
|
let releaseStream = "stable40";
|
|
if (lastVersion.endsWith("cuttingedge"))
|
|
releaseStream = "cuttingedge"
|
|
else if (lastVersion.endsWith("beta"))
|
|
releaseStream = "beta";
|
|
|
|
const releaseFiles = await osuUtil.getUpdateFiles(releaseStream);
|
|
const filesToDownload = await osuUtil.filesThatNeedUpdate(tempOsuPath, releaseFiles);
|
|
const downloadTask = await osuUtil.downloadUpdateFiles(osuPath, filesToDownload);
|
|
downloadTask.on('completed', () => {
|
|
mainWindow.webContents.send('status_update', {
|
|
type: "update-complete"
|
|
})
|
|
});
|
|
downloadTask.on('error', () => {
|
|
mainWindow.webContents.send('status_update', {
|
|
type: "error",
|
|
message: "An error occured while updating."
|
|
});
|
|
});
|
|
|
|
} else
|
|
mainWindow.webContents.send('status_update', {
|
|
type: "error",
|
|
message: "Invalid osu! folder"
|
|
});
|
|
})
|
|
ipcMain.handle('set-osu-dir', async (event) => {
|
|
const yes = await dialog.showOpenDialog({
|
|
properties: ['openDirectory']
|
|
})
|
|
if (yes.filePaths.length <= 0)
|
|
return undefined;
|
|
const folderPath = yes.filePaths[0];
|
|
const validOsuDir = await osuUtil.isValidOsuFolder(folderPath);
|
|
|
|
if (validOsuDir) await config.set("osuPath", folderPath);
|
|
|
|
return { validOsuDir, folderPath };
|
|
})
|
|
ipcMain.handle('perform-login', async (event, data) => {
|
|
const { username, password } = data;
|
|
const loginData = await ezppUtil.performLogin(username, password);
|
|
if (loginData && loginData.code === 200) {
|
|
await config.set("username", username);
|
|
await config.set("password", password);
|
|
}
|
|
return loginData;
|
|
})
|
|
ipcMain.on('perform-logout', async (event) => {
|
|
await config.remove("username");
|
|
await config.remove("password");
|
|
})
|
|
})
|
|
}
|
|
|
|
async function updateConfigVars(window) {
|
|
const osuPath = await config.get("osuPath", "");
|
|
window.webContents.send('config_update', {
|
|
osuPath: osuPath
|
|
})
|
|
}
|
|
|
|
async function tryLogin(window) {
|
|
const username = await config.get("username", "");
|
|
const password = await config.get("password", "");
|
|
if ((username && username.length > 0) && (password && password.length > 0)) {
|
|
const passwordPlain = password;
|
|
const loginResponse = await ezppUtil.performLogin(username, passwordPlain);
|
|
if (loginResponse && loginResponse.code === 200) {
|
|
window.webContents.send('account_update', {
|
|
type: "loggedin",
|
|
user: loginResponse.user
|
|
})
|
|
} else {
|
|
window.webContents.send('account_update', {
|
|
type: "login-failed"
|
|
})
|
|
}
|
|
} else {
|
|
window.webContents.send('account_update', {
|
|
type: "not-loggedin"
|
|
})
|
|
}
|
|
}
|
|
|
|
async function doUpdateCheck(window) {
|
|
const osuPath = await config.get("osuPath");
|
|
if (!osuPath || osuPath.trim == "") {
|
|
window.webContents.send('status_update', {
|
|
type: "missing-folder"
|
|
})
|
|
return;
|
|
}
|
|
const isValid = await osuUtil.isValidOsuFolder(osuPath);
|
|
if (!isValid) {
|
|
window.webContents.send('status_update', {
|
|
type: "missing-folder"
|
|
})
|
|
return;
|
|
}
|
|
if (fs.existsSync(osuPath)) {
|
|
tempOsuPath = osuPath;
|
|
const osuConfig = await osuUtil.getLatestConfig(tempOsuPath);
|
|
const lastVersion = await osuConfig.get("LastVersion");
|
|
let releaseStream = "stable40";
|
|
if (lastVersion.endsWith("cuttingedge"))
|
|
releaseStream = "cuttingedge"
|
|
else if (lastVersion.endsWith("beta"))
|
|
releaseStream = "beta";
|
|
|
|
const releaseFiles = await osuUtil.getUpdateFiles(releaseStream);
|
|
const filesToDownload = await osuUtil.filesThatNeedUpdate(tempOsuPath, releaseFiles);
|
|
window.webContents.send('status_update', {
|
|
type: filesToDownload.length > 0 ? "update-available" : "up-to-date"
|
|
})
|
|
} else
|
|
window.webContents.send('status_update', {
|
|
type: "missing-folder"
|
|
})
|
|
}
|
|
|
|
function createWindow() {
|
|
// Create the browser window.
|
|
const win = windowManager.createWindow(700, 460);
|
|
|
|
win.loadFile('./html/index.html');
|
|
|
|
attachTitlebarToWindow(win);
|
|
win.webContents.setWindowOpenHandler(() => "deny");
|
|
win.webContents.on('did-finish-load', function () {
|
|
if (win.webContents.getZoomFactor() != 0.9)
|
|
win.webContents.setZoomFactor(0.9)
|
|
});
|
|
|
|
return win;
|
|
}
|
|
|
|
run(); |