prey/prey-node-client

const shared = require('./shared');

const log = (str) => shared.log(str);

const verify_data = (hash) => {
  const obj = {};

  for (const key in hash) {
    if (typeof hash[key] !== 'undefined') obj[key] = hash[key];
  }

  return obj;
};

// set api key and call setup so device gets linked
// called from cli.js after signing up or signing in
const setApiKeyAndRegister = (key, cb) => {
  // ensure the plugin is loaded if registration succeeds
  const success = () => cb();

  // set API key and clear Device key
  shared.keys.set_api_key(key, (err) => {
    if (err) return cb(err);

    log('Linking device...');
    shared.panel.link((err) => {
      if (!err) return success();

      // ok, we got an error. clear the API key before returning.
      // call back the original error as well.
      shared.keys.set_api_key('', (e) => {
        cb(e);
      });
    });
  });
};

exports.authorize = function (values, cb) {
  const opts = {};
  const args = {
    api_key: values['-a'],
    email: values['-e'],
    password: values['-p'],
  };

  if (args.email && !args.password) return cb(new Error('Password required.'));

  opts.username = args.email || args.api_key;
  opts.password = args.password || 'x';

  shared.panel.authorize(opts, (err, key) => {
    if (err || !key) return cb(err || new Error("Couldn't verify credentials."));

    setApiKeyAndRegister(key, cb);
  });
};

exports.verify = function (values, cb) {
  let obj = {};
  const current = values['-c'] === true;
  const update = values['-u'] === true;

  if (current) {
    obj = shared.keys.get();
  } else {
    obj.api = values['-a'];
    obj.device = values['-d'];
  }

  shared.panel.verify_keys(obj, (err) => {
    if (!err) log('Keys are A-OK.');

    // if error or just checking current (no need to update)
    // or no update requested, then just return
    if (err || current || !update) return cb(err);

    shared.keys.set(obj, cb);
  });
};

exports.signup = (values, cb) => {
  if (shared.keys.is_api_key_set()) return cb(new Error('Account already set up!'));

  const accept_opts = ['yes', 'y', 'true'];
  const terms_check = values['-t'] ? (accept_opts.indexOf(values['-t'].toLowerCase()) != -1) : false;
  const age_check = values['-a'] ? (accept_opts.indexOf(values['-a'].toLowerCase()) != -1) : false;

  const data = verify_data({
    name: values['-n'],
    email: values['-e'],
    password: values['-p'],
    password_confirmation: values['-p'],
    policy_rule_privacy_terms: terms_check,
    policy_rule_age: age_check,
    country: values['-c'],
  });

  shared.panel.signup(data, (err, key) => {
    if (err || !key) return cb(err || new Error('No API Key received.'));

    log('Account created!');
    setApiKeyAndRegister(key, cb);
  });
};

exports.setup = (values, cb) => {
  const run_again = values['-f'] === true;

  if (shared.keys.is_api_key_set() && !run_again) return cb(new Error('Account already set up! Run with -f/--force to continue anyway.'));

  shared.panel.setup('control-panel', (err) => {
    if (err) return cb(err);

    log('Credentials verified.');
    shared.keys.set_device_key('', (err) => {
      if (err) return cb(err);

      log('Linking device...');
      shared.panel.link(cb);
    });
  });
};