[Registry] Rewrite to be more like old, integrate installProtocol into FirstRun win32
This commit is contained in:
parent
76f391e7c7
commit
4e9c629714
2 changed files with 47 additions and 34 deletions
|
@ -1,20 +1,19 @@
|
||||||
const fs = require('fs');
|
const fs = require('fs');
|
||||||
const { join, resolve, basename } = require('path');
|
const { join, resolve, basename } = require('path');
|
||||||
|
|
||||||
const registry = require('../utils/registry');
|
|
||||||
const Constants = require('../Constants');
|
const Constants = require('../Constants');
|
||||||
|
|
||||||
const appPath = resolve(process.execPath, '..');
|
const app = resolve(process.execPath, '..');
|
||||||
const rootPath = resolve(appPath, '..');
|
const root = resolve(app, '..');
|
||||||
|
|
||||||
|
|
||||||
const updateShortcuts = (updater) => {
|
const updateShortcuts = (updater) => {
|
||||||
try {
|
try {
|
||||||
const file = Constants.APP_NAME_FOR_HUMANS + '.lnk';
|
const file = Constants.APP_NAME_FOR_HUMANS + '.lnk';
|
||||||
const icon_Path = join(rootPath, 'app.ico');
|
const icon_Path = join(root, 'app.ico');
|
||||||
|
|
||||||
try {
|
try {
|
||||||
fs.copyFileSync(join(appPath, 'app.ico'), icon_path);
|
fs.copyFileSync(join(app, 'app.ico'), icon_path);
|
||||||
} catch { }
|
} catch { }
|
||||||
|
|
||||||
for (const shortcut_path of [
|
for (const shortcut_path of [
|
||||||
|
@ -24,14 +23,14 @@ const updateShortcuts = (updater) => {
|
||||||
if (!fs.existsSync(shortcut_path)) continue; // Don't update already deleted paths
|
if (!fs.existsSync(shortcut_path)) continue; // Don't update already deleted paths
|
||||||
|
|
||||||
updater.createShortcut({
|
updater.createShortcut({
|
||||||
target_path: join(rootPath, 'Update.exe'),
|
target_path: join(root, 'Update.exe'),
|
||||||
shortcut_path,
|
shortcut_path,
|
||||||
arguments: '--processStart ' + basename(process.execPath),
|
arguments: '--processStart ' + basename(process.execPath),
|
||||||
icon_path,
|
icon_path,
|
||||||
icon_index: 0,
|
icon_index: 0,
|
||||||
description: Constants.APP_DESCRIPTION,
|
description: Constants.APP_DESCRIPTION,
|
||||||
app_user_model_id: Constants.APP_ID,
|
app_user_model_id: Constants.APP_ID,
|
||||||
working_directory: appPath
|
working_directory: app
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -43,11 +42,41 @@ const updateShortcuts = (updater) => {
|
||||||
|
|
||||||
|
|
||||||
exports.do = (updater) => {
|
exports.do = (updater) => {
|
||||||
const flag = join(appPath, '.first-run');
|
const flag = join(app, '.first-run');
|
||||||
if (fs.existsSync(flag)) return; // Already done, skip
|
if (fs.existsSync(flag)) return; // Already done, skip
|
||||||
|
|
||||||
registry.installProtocol(Constants.APP_PROTOCOL, () => {
|
const proto = Constants.APP_PROTOCOL;
|
||||||
if (!updateShortcuts(updater)) return;
|
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 {
|
try {
|
||||||
fs.writeFileSync(flag, 'true');
|
fs.writeFileSync(flag, 'true');
|
||||||
|
|
|
@ -1,35 +1,19 @@
|
||||||
const CP = require('child_process');
|
const CP = require('child_process');
|
||||||
|
|
||||||
|
|
||||||
const spawn = (args, callback) => {
|
exports.spawn = (args, callback) => {
|
||||||
let process, stdout = '';
|
const process = CP.spawn('reg.exe', args);
|
||||||
|
let out = '';
|
||||||
|
|
||||||
try {
|
process.stdout.on('data', data => out += data);
|
||||||
process = CP.spawn('reg.exe', args);
|
|
||||||
} catch (e) {
|
|
||||||
callback(e, stdout);
|
|
||||||
}
|
|
||||||
|
|
||||||
process.stdout.on('data', data => stdout += data);
|
process.on('error', e => callback(e, out));
|
||||||
|
process.on('exit', (c, s) => callback(c !== 0 ? (s ?? c) : null, out));
|
||||||
process.on('error', err => callback(err, stdout));
|
|
||||||
|
|
||||||
process.on('exit', (code, signal) => callback(code !== 0 ? new Error('Spawn: ' + signal ?? code) : null, stdout));
|
|
||||||
};
|
};
|
||||||
|
|
||||||
const add = (todo, callback) => {
|
exports.add = (todo, callback) => {
|
||||||
const x = todo.shift();
|
const x = todo.shift();
|
||||||
if (!x) return callback();
|
if (!x) return callback();
|
||||||
|
|
||||||
spawn([ 'add', ...x, '/f' ], () => add(todo, callback));
|
exports.spawn([ 'add', ...x, '/f' ], () => exports.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);
|
|
||||||
}
|
|
||||||
};
|
};
|
Loading…
Reference in a new issue