profxadke
/
OpenAsar
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Registry] Rewrite to be more like old, integrate installProtocol into FirstRun win32

This commit is contained in:
Ducko 2022-04-12 18:12:42 +01:00
parent 76f391e7c7
commit 4e9c629714
2 changed files with 47 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
src/firstRun/win32.js
View File

@ -1,20 +1,19 @@
const fs = require('fs');
const { join, resolve, basename } = require('path');
const registry = require('../utils/registry');
const Constants = require('../Constants');
const appPath = resolve(process.execPath, '..');
const rootPath = resolve(appPath, '..');
const app = resolve(process.execPath, '..');
const root = resolve(app, '..');
const updateShortcuts = (updater) => {
try {
const file = Constants.APP_NAME_FOR_HUMANS + '.lnk';
const icon_Path = join(rootPath, 'app.ico');
const icon_Path = join(root, 'app.ico');
try {
fs.copyFileSync(join(appPath, 'app.ico'), icon_path);
fs.copyFileSync(join(app, 'app.ico'), icon_path);
} catch { }
for (const shortcut_path of [
@ -24,14 +23,14 @@ const updateShortcuts = (updater) => {
if (!fs.existsSync(shortcut_path)) continue; // Don't update already deleted paths
updater.createShortcut({
target_path: join(rootPath, 'Update.exe'),
target_path: join(root, 'Update.exe'),
shortcut_path,
arguments: '--processStart ' + basename(process.execPath),
icon_path,
icon_index: 0,
description: Constants.APP_DESCRIPTION,
app_user_model_id: Constants.APP_ID,
working_directory: appPath
working_directory: app
});
}
@ -43,11 +42,41 @@ const updateShortcuts = (updater) => {
exports.do = (updater) => {
const flag = join(appPath, '.first-run');
const flag = join(app, '.first-run');
if (fs.existsSync(flag)) return; // Already done, skip
registry.installProtocol(Constants.APP_PROTOCOL, () => {
if (!updateShortcuts(updater)) return;
const proto = Constants.APP_PROTOCOL;
const base = 'HKCU\\Software\\Classes\\' + proto;
require('../utils/registry').add([[base, '/ve', '/d', `URL:${proto} Protocol`], [base, '/v', 'URL Protocol'], [base + '\\DefaultIcon', '/ve', '/d', `"${process.execPath}",-1`], [base + '\\shell\\open\\command', '/ve', '/d', `"${process.execPath}" --url -- "%1"`]], () => { // Make protocol
try { // Make shortcuts
const file = Constants.APP_NAME_FOR_HUMANS + '.lnk';
const icon_Path = join(root, 'app.ico');
try {
fs.copyFileSync(join(app, 'app.ico'), icon_path);
} catch { }
for (const shortcut_path of [
join(updater.getKnownFolder('desktop'), file),
join(updater.getKnownFolder('programs'), Constants.APP_COMPANY, file)
]) {
if (!fs.existsSync(shortcut_path)) continue; // Don't update already deleted paths
updater.createShortcut({
target_path: join(root, 'Update.exe'),
shortcut_path,
arguments: '--processStart ' + basename(process.execPath),
icon_path,
icon_index: 0,
description: Constants.APP_DESCRIPTION,
app_user_model_id: Constants.APP_ID,
working_directory: app
});
}
} catch (e) {
return log('FirstRun', e);
}
try {
fs.writeFileSync(flag, 'true');

32
src/utils/registry.js
View File

@ -1,35 +1,19 @@
const CP = require('child_process');
const spawn = (args, callback) => {
let process, stdout = '';
exports.spawn = (args, callback) => {
const process = CP.spawn('reg.exe', args);
let out = '';
try {
process = CP.spawn('reg.exe', args);
} catch (e) {
callback(e, stdout);
}
process.stdout.on('data', data => out += data);
process.stdout.on('data', data => stdout += data);
process.on('error', err => callback(err, stdout));
process.on('exit', (code, signal) => callback(code !== 0 ? new Error('Spawn: ' + signal ?? code) : null, stdout));
process.on('error', e => callback(e, out));
process.on('exit', (c, s) => callback(c !== 0 ? (s ?? c) : null, out));
};
const add = (todo, callback) => {
exports.add = (todo, callback) => {
const x = todo.shift();
if (!x) return callback();
spawn([ 'add', ...x, '/f' ], () => add(todo, callback));
};
module.exports = {
spawn,
add,
installProtocol: (protocol, callback) => {
const base = 'HKCU\\Software\\Classes\\' + protocol;
add([[base, '/ve', '/d', `URL:${protocol} Protocol`], [base, '/v', 'URL Protocol'], [base + '\\DefaultIcon', '/ve', '/d', `"${process.execPath}",-1`], [base + '\\shell\\open\\command', '/ve', '/d', `"${process.execPath}" --url -- "%1"`]], callback);
}
exports.spawn([ 'add', ...x, '/f' ], () => exports.add(todo, callback));
};