forked from Mirrors/OpenAsar
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
425 lines
14 KiB
425 lines
14 KiB
"use strict";
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
exports.initSplash = initSplash;
|
|
exports.focusWindow = focusWindow;
|
|
exports.pageReady = pageReady;
|
|
exports.events = exports.APP_SHOULD_SHOW = exports.APP_SHOULD_LAUNCH = void 0;
|
|
var _electron = require("electron");
|
|
var _events = require("events");
|
|
var _fs = _interopRequireDefault(require("fs"));
|
|
var _path = _interopRequireDefault(require("path"));
|
|
var _url = _interopRequireDefault(require("url"));
|
|
var _Backoff = _interopRequireDefault(require("../utils/Backoff"));
|
|
var moduleUpdater = _interopRequireWildcard(require("../updater/moduleUpdater"));
|
|
var paths = _interopRequireWildcard(require("../paths"));
|
|
var _securityUtils = require("../utils/securityUtils");
|
|
var _updater = require("../updater/updater");
|
|
const ipcMain = _electron.ipcMain;
|
|
function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
|
|
function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
|
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
|
const UPDATE_TIMEOUT_WAIT = 10000;
|
|
const RETRY_CAP_SECONDS = 60;
|
|
// TODO: see if we can eliminate fudge by using useContentSize BrowserWindow option
|
|
const LOADING_WINDOW_WIDTH = 300;
|
|
const LOADING_WINDOW_HEIGHT = process.platform === 'darwin' ? 300 : 350;
|
|
const CHECKING_FOR_UPDATES = 'checking-for-updates';
|
|
const UPDATE_CHECK_FINISHED = 'update-check-finished';
|
|
const UPDATE_FAILURE = 'update-failure';
|
|
const LAUNCHING = 'launching';
|
|
const DOWNLOADING_MODULE = 'downloading-module';
|
|
const DOWNLOADING_UPDATES = 'downloading-updates';
|
|
const DOWNLOADING_MODULES_FINISHED = 'downloading-modules-finished';
|
|
const DOWNLOADING_MODULE_PROGRESS = 'downloading-module-progress';
|
|
const DOWNLOADED_MODULE = 'downloaded-module';
|
|
const NO_PENDING_UPDATES = 'no-pending-updates';
|
|
const INSTALLING_MODULE = 'installing-module';
|
|
const INSTALLING_UPDATES = 'installing-updates';
|
|
const INSTALLED_MODULE = 'installed-module';
|
|
const INSTALLING_MODULE_PROGRESS = 'installing-module-progress';
|
|
const INSTALLING_MODULES_FINISHED = 'installing-modules-finished';
|
|
const UPDATE_MANUALLY = 'update-manually';
|
|
const APP_SHOULD_LAUNCH = 'APP_SHOULD_LAUNCH';
|
|
exports.APP_SHOULD_LAUNCH = APP_SHOULD_LAUNCH;
|
|
const APP_SHOULD_SHOW = 'APP_SHOULD_SHOW';
|
|
exports.APP_SHOULD_SHOW = APP_SHOULD_SHOW;
|
|
const events = new _events.EventEmitter();
|
|
exports.events = events;
|
|
function webContentsSend(win, event, ...args) {
|
|
if (splashWindow != null && !splashWindow.isDestroyed() && !splashWindow.webContents.isDestroyed()) {
|
|
try {
|
|
win.webContents.send(`DISCORD_${event}`, ...args);
|
|
} catch (e) {
|
|
log('Splash', 'Failed to send to webcontents');
|
|
}
|
|
}
|
|
}
|
|
let splashWindow;
|
|
let modulesListeners;
|
|
let updateTimeout;
|
|
let updateAttempt;
|
|
let splashState;
|
|
let launchedMainWindow;
|
|
let restartRequired = false;
|
|
let newUpdater;
|
|
const updateBackoff = new _Backoff.default(1000, 30000);
|
|
//
|
|
// Disabled because Rust interop stuff is going on in here.
|
|
class TaskProgress {
|
|
constructor() {
|
|
this.inProgress = new Map();
|
|
this.finished = new Set();
|
|
this.allTasks = new Set();
|
|
}
|
|
recordProgress(progress, task) {
|
|
this.allTasks.add(task.package_sha256);
|
|
if (progress.state !== _updater.TASK_STATE_WAITING) {
|
|
this.inProgress.set(task.package_sha256, progress.percent);
|
|
if (progress.state === _updater.TASK_STATE_COMPLETE) {
|
|
this.finished.add(task.package_sha256);
|
|
}
|
|
}
|
|
}
|
|
updateSplashState(newState) {
|
|
if (this.inProgress.size > 0 && this.inProgress.size > this.finished.size) {
|
|
let totalPercent = 0;
|
|
for (const item of this.inProgress.values()) {
|
|
totalPercent += item;
|
|
}
|
|
totalPercent /= this.allTasks.size;
|
|
splashState = {
|
|
current: this.finished.size + 1,
|
|
total: this.allTasks.size,
|
|
progress: totalPercent
|
|
};
|
|
updateSplashState(newState);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
}
|
|
async function updateUntilCurrent() {
|
|
const retryOptions = {
|
|
skip_host_delta: false,
|
|
skip_module_delta: {}
|
|
};
|
|
while (true) {
|
|
updateSplashState(CHECKING_FOR_UPDATES);
|
|
try {
|
|
let installedAnything = false;
|
|
const downloads = new TaskProgress();
|
|
const installs = new TaskProgress();
|
|
await newUpdater.updateToLatestWithOptions(retryOptions, progress => {
|
|
const task = progress.task;
|
|
const downloadTask = task.HostDownload || task.ModuleDownload;
|
|
const installTask = task.HostInstall || task.ModuleInstall;
|
|
installedAnything = true;
|
|
if (downloadTask != null) {
|
|
downloads.recordProgress(progress, downloadTask);
|
|
}
|
|
if (installTask != null) {
|
|
installs.recordProgress(progress, installTask);
|
|
if (progress.state.Failed != null) {
|
|
if (task.HostInstall != null) {
|
|
retryOptions.skip_host_delta = true;
|
|
} else if (task.ModuleInstall != null) {
|
|
retryOptions.skip_module_delta[installTask.version.module.name] = true;
|
|
}
|
|
}
|
|
}
|
|
if (!downloads.updateSplashState(DOWNLOADING_UPDATES)) {
|
|
installs.updateSplashState(INSTALLING_UPDATES);
|
|
}
|
|
});
|
|
if (!installedAnything) {
|
|
await newUpdater.startCurrentVersion();
|
|
newUpdater.setRunningInBackground();
|
|
newUpdater.collectGarbage();
|
|
launchMainWindow();
|
|
updateBackoff.succeed();
|
|
updateSplashState(LAUNCHING);
|
|
return;
|
|
}
|
|
} catch (e) {
|
|
console.error('Update failed', e);
|
|
await new Promise(resolve => {
|
|
const delayMs = updateBackoff.fail(resolve);
|
|
splashState.seconds = Math.round(delayMs / 1000);
|
|
updateSplashState(UPDATE_FAILURE);
|
|
});
|
|
}
|
|
}
|
|
}
|
|
const oldCheckForUpdates = () => {
|
|
if (oaConfig.skipStartupUpdateChecks !== true) {
|
|
moduleUpdater.checkForUpdates();
|
|
} else {
|
|
log('Splash', 'Skipping startup update checking (enabled)');
|
|
modulesListeners[UPDATE_CHECK_FINISHED]({
|
|
succeeded: true,
|
|
updateCount: 0
|
|
});
|
|
}
|
|
};
|
|
function initOldUpdater() {
|
|
modulesListeners = {};
|
|
addModulesListener(CHECKING_FOR_UPDATES, () => {
|
|
startUpdateTimeout();
|
|
updateSplashState(CHECKING_FOR_UPDATES);
|
|
});
|
|
addModulesListener(UPDATE_CHECK_FINISHED, ({
|
|
succeeded,
|
|
updateCount,
|
|
manualRequired
|
|
}) => {
|
|
stopUpdateTimeout();
|
|
if (!succeeded) {
|
|
scheduleUpdateCheck();
|
|
updateSplashState(UPDATE_FAILURE);
|
|
} else if (updateCount === 0) {
|
|
moduleUpdater.setInBackground();
|
|
launchMainWindow();
|
|
updateSplashState(LAUNCHING);
|
|
}
|
|
});
|
|
addModulesListener(DOWNLOADING_MODULE, ({
|
|
name,
|
|
current,
|
|
total
|
|
}) => {
|
|
stopUpdateTimeout();
|
|
splashState = {
|
|
current,
|
|
total
|
|
};
|
|
updateSplashState(DOWNLOADING_UPDATES);
|
|
});
|
|
addModulesListener(DOWNLOADING_MODULE_PROGRESS, ({
|
|
name,
|
|
progress
|
|
}) => {
|
|
splashState.progress = progress;
|
|
updateSplashState(DOWNLOADING_UPDATES);
|
|
});
|
|
addModulesListener(DOWNLOADED_MODULE, ({
|
|
name,
|
|
current,
|
|
total,
|
|
succeeded
|
|
}) => {
|
|
delete splashState.progress;
|
|
if (name === 'host') {
|
|
restartRequired = true;
|
|
}
|
|
});
|
|
addModulesListener(DOWNLOADING_MODULES_FINISHED, ({
|
|
succeeded,
|
|
failed
|
|
}) => {
|
|
if (failed > 0) {
|
|
scheduleUpdateCheck();
|
|
updateSplashState(UPDATE_FAILURE);
|
|
} else {
|
|
process.nextTick(() => {
|
|
if (restartRequired) {
|
|
moduleUpdater.quitAndInstallUpdates();
|
|
} else {
|
|
moduleUpdater.installPendingUpdates();
|
|
}
|
|
});
|
|
}
|
|
});
|
|
addModulesListener(NO_PENDING_UPDATES, () => oldCheckForUpdates());
|
|
addModulesListener(INSTALLING_MODULE, ({
|
|
name,
|
|
current,
|
|
total
|
|
}) => {
|
|
splashState = {
|
|
current,
|
|
total
|
|
};
|
|
updateSplashState(INSTALLING_UPDATES);
|
|
});
|
|
addModulesListener(INSTALLED_MODULE, ({
|
|
name,
|
|
current,
|
|
total,
|
|
succeeded
|
|
}) => delete splashState.progress);
|
|
addModulesListener(INSTALLING_MODULE_PROGRESS, ({
|
|
name,
|
|
progress
|
|
}) => {
|
|
splashState.progress = progress;
|
|
updateSplashState(INSTALLING_UPDATES);
|
|
});
|
|
addModulesListener(INSTALLING_MODULES_FINISHED, ({
|
|
succeeded,
|
|
failed
|
|
}) => oldCheckForUpdates());
|
|
addModulesListener(UPDATE_MANUALLY, ({
|
|
newVersion
|
|
}) => {
|
|
splashState.newVersion = newVersion;
|
|
updateSplashState(UPDATE_MANUALLY);
|
|
});
|
|
}
|
|
function initSplash(startMinimized = false) {
|
|
log('Splash', `Initing splash`);
|
|
splashState = {};
|
|
launchedMainWindow = false;
|
|
updateAttempt = 0;
|
|
newUpdater = (0, _updater.getUpdater)();
|
|
if (newUpdater == null) {
|
|
initOldUpdater();
|
|
}
|
|
launchSplashWindow(startMinimized);
|
|
log('Splash', 'Quickstart config:', process.env.OPENASAR_QUICKSTART || oaConfig.quickstart, '-', process.env.OPENASAR_QUICKSTART, oaConfig.quickstart);
|
|
if (newUpdater != null) {
|
|
updateUntilCurrent();
|
|
} else {
|
|
moduleUpdater.installPendingUpdates();
|
|
}
|
|
if (process.env.OPENASAR_QUICKSTART || oaConfig.quickstart) setTimeout(() => {
|
|
destroySplash();
|
|
if (newUpdater != null) {
|
|
require('../utils/u2LoadModulePath')();
|
|
}
|
|
/* if (newUpdater != null) {
|
|
updateUntilCurrent();
|
|
} else {
|
|
moduleUpdater.installPendingUpdates();
|
|
moduleUpdater.setInBackground();
|
|
} */
|
|
launchMainWindow();
|
|
setTimeout(() => {
|
|
events.emit(APP_SHOULD_SHOW);
|
|
}, 100);
|
|
}, 50);
|
|
}
|
|
function destroySplash() {
|
|
log('Splash', `Destroying splash`);
|
|
stopUpdateTimeout();
|
|
if (splashWindow) {
|
|
splashWindow.setSkipTaskbar(true);
|
|
const _nukeWindow = () => {
|
|
if (splashWindow != null) {
|
|
splashWindow.hide();
|
|
splashWindow.close();
|
|
splashWindow = null;
|
|
}
|
|
};
|
|
setTimeout(_nukeWindow, 100);
|
|
}
|
|
}
|
|
function addModulesListener(event, listener) {
|
|
if (newUpdater != null) return;
|
|
modulesListeners[event] = listener;
|
|
moduleUpdater.events.addListener(event, listener);
|
|
}
|
|
function removeModulesListeners() {
|
|
if (newUpdater != null) return;
|
|
for (const event of Object.keys(modulesListeners)) {
|
|
moduleUpdater.events.removeListener(event, modulesListeners[event]);
|
|
}
|
|
}
|
|
function startUpdateTimeout() {
|
|
if (!updateTimeout) {
|
|
updateTimeout = setTimeout(() => scheduleUpdateCheck(), UPDATE_TIMEOUT_WAIT);
|
|
}
|
|
}
|
|
function stopUpdateTimeout() {
|
|
if (updateTimeout) {
|
|
clearTimeout(updateTimeout);
|
|
updateTimeout = null;
|
|
}
|
|
}
|
|
function updateSplashState(event) {
|
|
webContentsSend(splashWindow, 'SPLASH_UPDATE_STATE', {
|
|
status: event,
|
|
...splashState
|
|
});
|
|
}
|
|
function launchSplashWindow(startMinimized) {
|
|
const windowConfig = {
|
|
width: LOADING_WINDOW_WIDTH,
|
|
height: LOADING_WINDOW_HEIGHT,
|
|
transparent: false,
|
|
frame: false,
|
|
resizable: false,
|
|
center: true,
|
|
show: false,
|
|
webPreferences: {
|
|
nodeIntegration: false,
|
|
enableRemoteModule: false,
|
|
contextIsolation: true,
|
|
preload: _path.default.join(__dirname, 'preload.js')
|
|
}
|
|
};
|
|
splashWindow = new _electron.BrowserWindow(windowConfig);
|
|
log('Splash', 'Created BrowserWindow');
|
|
splashWindow.webContents.on('will-navigate', e => e.preventDefault());
|
|
splashWindow.webContents.on('new-window', (e, windowURL) => {
|
|
e.preventDefault();
|
|
(0, _securityUtils.saferShellOpenExternal)(windowURL);
|
|
setTimeout(_electron.app.quit, 500);
|
|
});
|
|
if (process.platform !== 'darwin') {
|
|
splashWindow.on('closed', () => {
|
|
splashWindow = null;
|
|
if (!launchedMainWindow) {
|
|
_electron.app.quit();
|
|
}
|
|
});
|
|
}
|
|
ipcMain.on('DISCORD_SPLASH_SCREEN_READY', () => {
|
|
log('Splash', 'Window declared ready, showing and starting update process');
|
|
if (oaConfig.themeSync !== false) try {
|
|
splashWindow.webContents.insertCSS(JSON.parse(_fs.default.readFileSync(_path.default.join(paths.getUserData(), 'userDataCache.json'), 'utf8')).openasarSplashCSS);
|
|
} catch (e) { }
|
|
if (oaConfig.splashText === true) try {
|
|
const buildInfo = require('../utils/buildInfo.js');
|
|
splashWindow.webContents.executeJavaScript(`debug.textContent = '${buildInfo.releaseChannel} ${buildInfo.version}\\nOpenAsar ${oaVersion}'`);
|
|
} catch (e) { }
|
|
if (splashWindow && !startMinimized) {
|
|
splashWindow.show();
|
|
}
|
|
});
|
|
ipcMain.on('DISCORD_SPLASH_SCREEN_QUIT', () => {
|
|
_electron.app.quit();
|
|
});
|
|
const splashUrl = _url.default.format({
|
|
protocol: 'file',
|
|
slashes: true,
|
|
pathname: _path.default.join(__dirname, 'index.html')
|
|
});
|
|
splashWindow.loadURL(splashUrl);
|
|
log('Splash', `Loading window (with url ${splashUrl})`);
|
|
}
|
|
function launchMainWindow() {
|
|
removeModulesListeners();
|
|
if (!launchedMainWindow && splashWindow != null) {
|
|
launchedMainWindow = true;
|
|
events.emit(APP_SHOULD_LAUNCH);
|
|
}
|
|
}
|
|
function scheduleUpdateCheck() {
|
|
updateAttempt += 1;
|
|
const retryInSeconds = Math.min(updateAttempt * 10, RETRY_CAP_SECONDS);
|
|
splashState.seconds = retryInSeconds;
|
|
setTimeout(() => moduleUpdater.checkForUpdates(), retryInSeconds * 1000);
|
|
}
|
|
function focusWindow() {
|
|
log('Splash', `Told to focus splash window`);
|
|
if (splashWindow != null) {
|
|
splashWindow.focus();
|
|
}
|
|
}
|
|
function pageReady() {
|
|
log('Splash', `Page ready called, destroying splash and marking app to show`);
|
|
destroySplash();
|
|
process.nextTick(() => events.emit(APP_SHOULD_SHOW));
|
|
} |