serverless/lib/utils/autocomplete.js

'use strict';

const Serverless = require('../Serverless');
const crypto = require('crypto');

const getCacheFile = require('./getCacheFile');
const getServerlessConfigFile = require('./getServerlessConfigFile').getServerlessConfigFile;

const getSuggestions = (commands, env) => {
  if (!env.complete) return;
  const tokens = env.line.split(/\s+/);
  const { log } = require('tabtab');
  switch (tokens.length) {
    case 1:
    case 2:
      log(Object.keys(commands).filter((commandName) => !commands[commandName].isHidden));
      return;
    case 3:
      if (commands[env.prev]) log(commands[env.prev]);
      return;
    default:
      return;
  }
};

const cacheFileValid = (serverlessConfigFile, validationHash) => {
  const serverlessConfigFileHash = crypto
    .createHash('sha256')
    .update(JSON.stringify(serverlessConfigFile))
    .digest('hex');

  return validationHash === serverlessConfigFileHash;
};

const autocomplete = async () => {
  const servicePath = process.cwd();
  const serverlessConfigFile = await getServerlessConfigFile({
    processedInput: { options: {} },
    config: { servicePath },
  });
  let cacheFile = await getCacheFile(servicePath);
  if (!cacheFile || !cacheFileValid(serverlessConfigFile, cacheFile.validationHash)) {
    const serverless = new Serverless();
    await serverless.init();
    cacheFile = await getCacheFile(servicePath);
  }
  if (!cacheFile || !cacheFileValid(serverlessConfigFile, cacheFile.validationHash)) {
    return null;
  }
  return getSuggestions(cacheFile.commands, require('tabtab').parseEnv(process.env));
};

module.exports = autocomplete;