@zombiec0rn/zombie-swarm-cli/sub-commands/ls.js

import fs      from 'fs'
import Table   from 'cli-table'
import assign  from 'object.assign'
import values  from 'object.values'
import chalk   from 'chalk'
import request from 'request'
import * as utils from '../utils'
require('colors')

let cmd = {
  name: 'ls',
  usage: `Usage: zombie-swarm ls [OPTIONS]

List swarm nodes.

OPTIONS
`,
  options: utils.defaultOptions.concat([
    {
      name: 'out-file',
      help: 'File to dump the discovered swarm nodes (.json - can be used as plan input)'
    },
    {
      name: 'query',
      default: 1000,
      help: 'How long to query (ms)'
    }
  ]),
  command: function(args) {
    utils.assignZombieRC(args)
    utils.initCmd(args)
    utils.validateArgs(args)
    utils.querySwarmNodes((nodes) => {
      if (args['out-file']) fs.writeFileSync(args['out-file'], JSON.stringify(nodes, null, 2))
      let table = makeTable(nodes, args)
      console.log(table.toString())
      if (args['out-file']) console.log(`# Wrote config to file: ${args['out-file']}`)
      process.exit()
    }, args, args.query) 
  }
}

function makeTable(nodes, args) {
  let formatted = nodes.map((node) => {
    return {
      node: node.hostname,
      swarm: node.swarm, 
      tags: node.tags.join(','),
      ip: node.ip,
      engines: node.engines.join(','),
      memory: node.memory,
      cpus: `${node.cpus.length} x ${node.cpus[0].speed}`
    }
  })

  let table = new Table({
    head: Object.keys(formatted[0]).map(h => h.green),
    chars: { 'top': '' , 'top-mid': '' , 'top-left': '' , 'top-right': ''
          , 'bottom': '' , 'bottom-mid': '' , 'bottom-left': '' , 'bottom-right': ''
          , 'left': '' , 'left-mid': '' , 'mid': '' , 'mid-mid': ''
          , 'right': '' , 'right-mid': '' , 'middle': ' ' },
    style: { 'padding-left': 0, 'padding-right': 0 }
  })
  formatted.forEach(f => { table.push(values(f)) })
  return table
}

export { cmd as default }