OpenAsar/src/firstRun/win32.js

const fs = require('fs');
const path = require('path');
const paths = require('../paths');
const windowsUtils = require('../utils/windowsUtils');
const Constants = require('../Constants');

const appPath = path.resolve(process.execPath, '..');
const rootPath = path.resolve(appPath, '..');
const exeFilename = path.basename(process.execPath);
const updateExe = path.join(rootPath, 'Update.exe');

const iconFile = 'app.ico';
const copyIconToRoot = () => {
  const currentPath = path.join(appPath, iconFile);
  const newPath = path.join(rootPath, iconFile);

  try {
    fs.copyFileSync(currentPath, newPath);
    return newPath;
  } catch (e) {
    log('FirstRun', 'Failed to copy icon to root', e);
    return currentPath;
  }
};

const updateShortcuts = (updater) => {
  const filename = Constants.APP_NAME_FOR_HUMANS + '.lnk';
  const paths = [
    path.join(updater.getKnownFolder('desktop'), filename),
    path.join(updater.getKnownFolder('programs'), Constants.APP_COMPANY, filename)
  ];

  const icon = copyIconToRoot();

  for (const path of paths) {
    if (!fs.existsSync(path)) continue; // Don't update already deleted paths

    updater.createShortcut({
      target_path: updateExe,
      shortcut_path: path,
      arguments: `--processStart ${exeFilename}`,
      icon_path: icon,
      icon_index: 0,
      description: Constants.APP_DESCRIPTION,
      app_user_model_id: Constants.APP_ID,
      working_directory: appPath
    });
  }
};

const installProtocol = (protocol, callback) => windowsUtils.addToRegistry([['HKCU\\Software\\Classes\\' + protocol, '/ve', '/d', `URL:${protocol} Protocol`], ['HKCU\\Software\\Classes\\' + protocol, '/v', 'URL Protocol'], ['HKCU\\Software\\Classes\\' + protocol + '\\DefaultIcon', '/ve', '/d', '"' + process.execPath + '",-1'], ['HKCU\\Software\\Classes\\' + protocol + '\\shell\\open\\command', '/ve', '/d', `"${process.execPath}" --url -- "%1"`]], callback);

exports.performFirstRunTasks = (updater) => {
  log('FirstRun', 'Perform');

  const flagPath = path.join(paths.getUserDataVersioned(), '.first-run');

  if (fs.existsSync(flagPath)) return; // Already ran first path, skip

  let shortcutSuccess = false;
  try {
    updateShortcuts(updater);
    shortcutSuccess = true;
  } catch (e) {
    log('FirstRun', 'Error updating shortcuts', e);
  }

  installProtocol(Constants.APP_PROTOCOL, () => {
    if (!shortcutSuccess) return;

    try {
      fs.writeFileSync(firstRunCompletePath, 'true');
    } catch (e) {
      log('FirstRun', 'Error writing .first-run', e);
    }
  });
};