embark/lib/modules/console/suggestions.js

let utils = require('../../utils/utils');

class Suggestions {
  constructor(embark, options) {
    this.events = embark.events;
    this.plugins = options.plugins;
    this.registerApi();
    this.listenToEvents();
    this.contracts = {}
  }

  registerApi() {
    let plugin = this.plugins.createPlugin('consoleApi', {});
    plugin.registerAPICall('post', '/embark-api/suggestions', (req, res) => {
      let suggestions = this.getSuggestions(req.body.command)
      res.send({result: suggestions})
    });
  }

  listenToEvents() {
    this.events.on("deploy:contract:deployed", (contract) => {
      this.contracts[contract.className] = contract
    })
  }

  getSuggestions(cmd) {
    cmd = cmd.toLowerCase()

    if (cmd === 'web3' || cmd === 'web3.') {
      return [
        {value: 'web3.eth', command_type: "web3 object", description: "module for interacting with the Ethereum network"},
        {value: 'web3.net', command_type: "web3 object", description: "module for interacting with network properties"},
        {value: 'web3.shh', command_type: "web3 object", description: "module for interacting with the whisper protocol"},
        {value: 'web3.bzz', command_type: "web3 object", description: "module for interacting with the swarm network"},
        {value: 'web3.eth.getAccounts()', command_type: "web3 object", description: "get list of accounts"}
      ]
    }

    let suggestions = []
    for (let contractName in this.contracts) {
      let contract = this.contracts[contractName]
      if (contract.className.toLowerCase().indexOf(cmd) >= 0 || contract.deployedAddress.indexOf(cmd) >= 0 || cmd.indexOf('contract') >= 0) {
        suggestions.push({value: contract.className, command_type: "web3 object", description: "contract deployed at " + contract.deployedAddress});
      }

      if ('profile'.indexOf(cmd) >= 0) {
        suggestions.push({value: "profile " + contract.className, command_type: "embark command", description: "profile " + contract.className + " contract"});
      }
    }

    // TODO: make this registered through an API instead

    if ('help'.indexOf(cmd) >= 0) {
      suggestions.push({value: 'help', command_type: "embark command", description: "displays quick list of some of the available embark commands"});
    }
    if ('versions'.indexOf(cmd) >= 0) {
      suggestions.push({value: 'versions', command_type: "embark command", description: "display versions in use for libraries and tools like web3 and solc"});
    }
    if ('ipfs'.indexOf(cmd) >= 0) {
      suggestions.push({value: 'ipfs', command_type: "javascript object", description: "instantiated js-ipfs object configured to the current environment (available if ipfs is enabled)"});
    }
    if ('swarm'.indexOf(cmd) >= 0) {
      suggestions.push({value: 'swarm', command_type: "javascript object", description: "instantiated swarm-api object configured to the current environment (available if swarm is enabled)"});
    }
    if (cmd == 'w' || cmd === 'we' || cmd === 'web') {
      suggestions.push({value: 'web3', command_type: "javascript object", description: "instantiated web3.js object configured to the current environment"});
    }
    if ('embarkjs'.indexOf(cmd) >= 0) {
      suggestions.push({value: 'EmbarkJS', command_type: "javascript object", description: "EmbarkJS static functions for Storage, Messages, Names, etc."});
    }

    // sort first the ones that match the command at the beginning of the string, then prefer smaller commands first
    return suggestions.sort((x,y) => {
      let diff = x.value.indexOf(cmd) - y.value.indexOf(cmd)
      if (diff !== 0) return diff;
      return x.value.length - y.value.length;
    });
  }
}

module.exports = Suggestions;