All files / lib touch-enroll.js

69.57% Statements 16/23
0% Branches 0/4
33.33% Functions 3/9
68.42% Lines 13/19
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108    1x 1x 1x 1x                                                         8x     2x       2x     2x             4x 4x 4x 4x       3x                                                                                                  
import { exec } from 'teen_process';
 
let touchEnrollMenuKeys = ['Toggle Enrolled State', 'Touch ID Enrolled'];
const touchEnrollBackupsStack = [];
const NS_USER_KEY_EQUIVALENTS = 'NSUserKeyEquivalents';
const TOUCH_ENROLL_KEY_CODE = '@~$^t';
 
async function setTouchEnrollKey () {
  await backupTouchEnrollShortcuts();
  for (let key of touchEnrollMenuKeys) {
    await setUserDefault(NS_USER_KEY_EQUIVALENTS, key, TOUCH_ENROLL_KEY_CODE);
  }
}
 
async function setTouchEnrollKeys (pairs) {
  for (let [key, value] of pairs) {
    await setUserDefault(NS_USER_KEY_EQUIVALENTS, key, value);
  }
}
 
async function getTouchEnrollKeys () {
  let backups = [];
  for (let key of touchEnrollMenuKeys) {
    backups.push([key, await getUserDefault(NS_USER_KEY_EQUIVALENTS, key)]);
  }
  return backups;
}
 
/**
 * Get MacOS User Defaults by domain and key (for reference: `man defaults`)
 * @param {*} domain {string}
 * @param {*} key {string|number|boolean}
 */
async function getUserDefault (domain, key) {
  let res;
  try {
    // If it doesn't find anything for this domain, it will throw an error so return undefined
    res = await exec('defaults', ['read', 'Apple Global Domain', domain]);
  } catch (stderr) {
    return;
  }
  let stdout = res.stdout;
 
  // Parse the result into a Javascript array
  let nsUserKeyArr = stdout.trim()
    .replace(/^{/, '') // Remove leading {
    .replace(/}$/, '') // Remove trailing }
    .trim()
    .replace(/;$/, '') // Remove trailing semicolon
    .split(';') // Break up expressions by semicolon
    .map((expr) => {
      let [key, value] = expr.split('=');
      key = key.trim().replace(/^"/, '').replace(/"$/, '');
      value = value.trim().replace(/^"/, '').replace(/"$/, '');
      return [key, value];
    });
 
 
  for (let [testKey, value] of nsUserKeyArr) {
    if (testKey === key) {
      return value.replace(/\\\\/g, '\\');
    }
  }
}
 
/**
 * Sets a MacOS User Default value on a domain
 * @param {*} domain
 * @param {*} key
 * @param {*} value
 */
async function setUserDefault (domain, key, value) {
  await exec('defaults', ['write', 'Apple Global Domain', domain, '-dict-add', key, typeof value === 'undefined' ? 'nil' : value]);
}
 
/**
 * Backup current values of user defaults and store them into an internal variable.
 * Only one backup can be stored. All the further calls to this method will be ignored unless
 * {#restoreTouchEnrollShortcuts} is called.
 */
async function backupTouchEnrollShortcuts () {
  if (!touchEnrollBackupsStack.length) {
    touchEnrollBackupsStack.push(await getTouchEnrollKeys());
  }
}
 
/**
 * Restore the user defaults previously preserved by {#backupTouchEnrollShortcuts}.
 * Only one backup can be preserved. Multiple calls to this method
 * will have no effect if the backup has already been restored or hasn't been made.
 */
async function restoreTouchEnrollShortcuts () {
  if (touchEnrollBackupsStack.length) {
    const preservedKeys = touchEnrollBackupsStack.pop();
    if (preservedKeys) {
      await setTouchEnrollKeys(preservedKeys);
    }
  }
}
 
function getTouchEnrollBackups () {
  return touchEnrollBackupsStack.length ? touchEnrollBackupsStack[touchEnrollBackupsStack.length - 1] : undefined;
}
 
export { setTouchEnrollKey, getTouchEnrollKeys, setTouchEnrollKeys, getUserDefault, setUserDefault, getTouchEnrollBackups,
  touchEnrollMenuKeys, backupTouchEnrollShortcuts, restoreTouchEnrollShortcuts,
  NS_USER_KEY_EQUIVALENTS, TOUCH_ENROLL_KEY_CODE};