central-cli 0.6.3

data/lib/central/cli/master/vagrant/terminate_command.rb

@@ -0,0 +1,13 @@
+module Central::Cli::Master::Vagrant
+  class TerminateCommand < Clamp::Command
+    include Central::Cli::Common
+
+    def execute
+      require_api_url
+
+      require 'central/machine/vagrant'
+      destroyer = Central::Machine::Vagrant::MasterDestroyer.new(client(require_token))
+      destroyer.run!
+    end
+  end
+end

data/lib/central/cli/master/vagrant_command.rb

@@ -0,0 +1,21 @@
+
+module Central::Cli::Master
+  require_relative 'vagrant/create_command'
+  require_relative 'vagrant/start_command'
+  require_relative 'vagrant/stop_command'
+  require_relative 'vagrant/restart_command'
+  require_relative 'vagrant/ssh_command'
+  require_relative 'vagrant/terminate_command'
+
+  class VagrantCommand < Clamp::Command
+    subcommand 'create', 'Create a new Vagrant master', Vagrant::CreateCommand
+    subcommand 'ssh', 'SSH into Vagrant master', Vagrant::SshCommand
+    subcommand 'start', 'Start Vagrant master', Vagrant::StartCommand
+    subcommand 'stop', 'Stop Vagrant master', Vagrant::StopCommand
+    subcommand 'restart', 'Restart Vagrant master', Vagrant::RestartCommand
+    subcommand 'terminate', 'Terminate Vagrant master', Vagrant::TerminateCommand
+
+    def execute
+    end
+  end
+end

data/lib/central/cli/master_command.rb

@@ -0,0 +1,19 @@
+require_relative 'master/vagrant_command'
+require_relative 'master/aws_command'
+require_relative 'master/digital_ocean_command'
+require_relative 'master/azure_command'
+require_relative 'master/use_command'
+require_relative 'master/list_command'
+require_relative 'master/users_command'
+
+class Central::Cli::MasterCommand < Clamp::Command
+  subcommand 'vagrant', 'Vagrant specific commands', Central::Cli::Master::VagrantCommand
+  subcommand 'aws', 'AWS specific commands', Central::Cli::Master::AwsCommand
+  subcommand 'digitalocean', 'DigitalOcean specific commands', Central::Cli::Master::DigitalOceanCommand
+  subcommand 'azure', 'Azure specific commands', Central::Cli::Master::AzureCommand
+  subcommand %w(list ls), 'List masters where client has logged in', Central::Cli::Master::ListCommand
+  subcommand 'use', 'Switch to use selected master', Central::Cli::Master::UseCommand
+  subcommand 'users', 'Users specific commands', Central::Cli::Master::UsersCommand
+  def execute
+  end
+end

data/lib/central/cli/node_command.rb

@@ -0,0 +1,30 @@
+require_relative 'nodes/list_command'
+require_relative 'nodes/remove_command'
+require_relative 'nodes/show_command'
+require_relative 'nodes/update_command'
+require_relative 'nodes/ssh_command'
+
+require_relative 'nodes/label_command'
+
+require_relative 'nodes/vagrant_command'
+require_relative 'nodes/digital_ocean_command'
+require_relative 'nodes/aws_command'
+require_relative 'nodes/azure_command'
+
+class Central::Cli::NodeCommand < Clamp::Command
+  subcommand %w(list ls), 'List stack nodes', Central::Cli::Nodes::ListCommand
+  subcommand 'show', 'Show node', Central::Cli::Nodes::ShowCommand
+  subcommand 'ssh', 'Ssh into node', Central::Cli::Nodes::SshCommand
+  subcommand 'update', 'Update node', Central::Cli::Nodes::UpdateCommand
+  subcommand %w(remove rm), 'Remove node', Central::Cli::Nodes::RemoveCommand
+
+  subcommand 'label', 'Node label specific commands', Central::Cli::Nodes::LabelCommand
+
+  subcommand 'vagrant', 'Vagrant specific commands', Central::Cli::Nodes::VagrantCommand
+  subcommand 'digitalocean', 'DigitalOcean specific commands', Central::Cli::Nodes::DigitalOceanCommand
+  subcommand 'aws', 'AWS specific commands', Central::Cli::Nodes::AwsCommand
+  subcommand 'azure', 'Azure specific commands', Central::Cli::Nodes::AzureCommand
+
+  def execute
+  end
+end

data/lib/central/cli/nodes/add_label_command.rb

@@ -0,0 +1,19 @@
+module Central::Cli::Nodes
+  class AddLabelCommand < Clamp::Command
+    include Central::Cli::Common
+
+    parameter 'NODE_ID', 'Node id'
+    parameter 'LABEL', 'Label'
+
+    def execute
+      require_api_url
+      require_current_stack
+      token = require_token
+
+      node = client(token).get("stacks/#{current_stack}/nodes/#{node_id}")
+      data = {}
+      data[:labels] = node['labels'].to_a | [label]
+      client.put("nodes/#{node['id']}", data, {}, 'Central-Stack-Token' => node['stack']['token'])
+    end
+  end
+end

data/lib/central/cli/nodes/aws/create_command.rb

@@ -0,0 +1,40 @@
+module Central::Cli::Nodes::Aws
+  class CreateCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+
+    parameter '[NAME]', 'Node name'
+    option '--access-key', 'ACCESS_KEY', 'AWS access key ID', environment_variable: 'AWS_ACCESS_KEY_ID', required: true
+    option '--secret-key', 'SECRET_KEY', 'AWS secret key', environment_variable: 'AWS_SECRET_ACCESS_KEY', required: true
+    option '--key-pair', 'KEY_PAIR', 'EC2 Key Pair', required: true
+    option '--region', 'REGION', 'EC2 Region', default: 'us-west-2'
+    option '--zone', 'ZONE', 'EC2 Availability Zone', default: 'a'
+    option '--vpc-id', 'VPC ID', 'Virtual Private Cloud (VPC) ID (default: default vpc)'
+    option '--subnet-id', 'SUBNET ID', 'VPC option to specify subnet to launch instance into (default: first subnet in vpc/az)'
+    option '--type', 'SIZE', 'Instance type', default: 't2.small'
+    option '--storage', 'STORAGE', 'Storage size (GiB)', default: '30'
+    option '--version', 'VERSION', 'Define installed Central version', default: 'latest'
+
+    def execute
+      require_api_url
+      require_current_stack
+
+      require 'central/machine/aws'
+      stack = client(require_token).get("stacks/#{current_stack}")
+      provisioner = Central::Machine::Aws::NodeProvisioner.new(client(require_token), access_key, secret_key, region)
+      provisioner.run!(
+        master_uri: api_url,
+        stack_token: stack['token'],
+        stack: current_stack,
+        name: name,
+        type: type,
+        vpc: vpc_id,
+        zone: zone,
+        subnet: subnet_id,
+        storage: storage,
+        version: version,
+        key_pair: key_pair
+      )
+    end
+  end
+end

data/lib/central/cli/nodes/aws/restart_command.rb

@@ -0,0 +1,29 @@
+module Central::Cli::Nodes::Aws
+  class RestartCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+
+    parameter 'NAME', 'Node name'
+    option '--access-key', 'ACCESS_KEY', 'AWS access key ID', environment_variable: 'AWS_ACCESS_KEY_ID', required: true
+    option '--secret-key', 'SECRET_KEY', 'AWS secret key', environment_variable: 'AWS_SECRET_ACCESS_KEY', required: true
+    option '--region', 'REGION', 'EC2 Region', required: true
+
+    def execute
+      require_api_url
+      require_current_stack
+
+      require 'central/machine/aws'
+
+      client = Fog::Compute.new(provider: 'AWS', aws_access_key_id: access_key, aws_secret_access_key: secret_key, region: region)
+      instance = client.servers.all('tag:central_name' => name).first
+      if instance
+        instance.reboot
+        ShellSpinner "Restarting AWS instance #{name.colorize(:cyan)} " do
+          instance.wait_for { ready? }
+        end
+      else
+        abort "Cannot find instance #{name.colorize(:cyan)} in AWS"
+      end
+    end
+  end
+end

data/lib/central/cli/nodes/aws/terminate_command.rb

@@ -0,0 +1,21 @@
+module Central::Cli::Nodes::Aws
+  class TerminateCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+
+    parameter 'NAME', 'Node name'
+    option '--access-key', 'ACCESS_KEY', 'AWS access key ID', environment_variable: 'AWS_ACCESS_KEY_ID', required: true
+    option '--secret-key', 'SECRET_KEY', 'AWS secret key', environment_variable: 'AWS_SECRET_ACCESS_KEY', required: true
+    option '--region', 'REGION', 'EC2 Region', default: 'us-west-2'
+
+    def execute
+      require_api_url
+      require_current_stack
+
+      require 'central/machine/aws'
+      stack = client(require_token).get("stacks/#{current_stack}")
+      destroyer = Central::Machine::Aws::NodeDestroyer.new(client(require_token), access_key, secret_key, region)
+      destroyer.run!(stack, name)
+    end
+  end
+end

data/lib/central/cli/nodes/aws_command.rb

@@ -0,0 +1,14 @@
+require_relative 'aws/create_command'
+require_relative 'aws/terminate_command'
+require_relative 'aws/restart_command'
+
+module Central::Cli::Nodes
+  class AwsCommand < Clamp::Command
+    subcommand 'create', 'Create a new AWS node', Aws::CreateCommand
+    subcommand 'terminate', 'Terminate AWS node', Aws::TerminateCommand
+    subcommand 'restart', 'Restart AWS node', Aws::RestartCommand
+
+    def execute
+    end
+  end
+end

data/lib/central/cli/nodes/azure/create_command.rb

@@ -0,0 +1,40 @@
+module Central::Cli::Nodes::Azure
+  class CreateCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+
+    option '--subscription-id', 'SUBSCRIPTION ID', 'Azure subscription id', required: true
+    option '--subscription-cert', 'CERTIFICATE', 'Path to Azure management certificate', attribute_name: :certificate, required: true
+    option '--size', 'SIZE', 'SIZE', default: 'Small'
+    option '--network', 'NETWORK', 'Virtual Network name'
+    option '--subnet', 'SUBNET', 'Subnet name'
+    option '--ssh-key', 'SSH KEY', 'SSH private key file', required: true
+    option '--password', 'PASSWORD', 'Password'
+    option '--location', 'LOCATION', 'Location', default: 'West Europe'
+    option '--version', 'VERSION', 'Define installed Central version', default: 'latest'
+
+    parameter '[NAME]', 'Node name'
+
+    def execute
+      require_api_url
+      require_current_stack
+
+      require 'central/machine/azure'
+      stack = client(require_token).get("stacks/#{current_stack}")
+      provisioner = Central::Machine::Azure::NodeProvisioner.new(client(require_token), subscription_id, certificate)
+      provisioner.run!(
+        master_uri: api_url,
+        stack_token: stack['token'],
+        stack: current_stack,
+        password: password,
+        ssh_key: ssh_key,
+        name: name,
+        size: size,
+        virtual_network: network,
+        subnet: subnet,
+        location: location,
+        version: version
+      )
+    end
+  end
+end

data/lib/central/cli/nodes/azure/restart_command.rb

@@ -0,0 +1,31 @@
+module Central::Cli::Nodes::Azure
+  class RestartCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+
+    parameter 'NAME', 'Node name'
+    option '--subscription-id', 'SUBSCRIPTION ID', 'Azure subscription id', required: true
+    option '--subscription-cert', 'CERTIFICATE', 'Path to Azure management certificate', attribute_name: :certificate, required: true
+
+    def execute
+      require_api_url
+      require_current_stack
+
+      require 'central/machine/azure'
+
+      client = ::Azure
+      client.management_certificate = certificate
+      client.subscription_id        = subscription_id
+
+      client.vm_management.initialize_external_logger(Central::Machine::Azure::Logger.new) # We don't want all the output
+      ShellSpinner "Restarting Azure VM #{name.colorize(:cyan)} " do
+        vm = client.vm_management.get_virtual_machine(name, "central-#{current_stack}-#{name}")
+        if vm
+          client.vm_management.restart_virtual_machine(name, "central-#{current_stack}-#{name}")
+        else
+          abort "\nCannot find Virtual Machine #{name.colorize(:cyan)} in Azure"
+        end
+      end
+    end
+  end
+end

data/lib/central/cli/nodes/azure/terminate_command.rb

@@ -0,0 +1,21 @@
+module Central::Cli::Nodes::Azure
+  class TerminateCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+
+    parameter 'NAME', 'Node name'
+    option '--subscription-id', 'SUBSCRIPTION ID', 'Azure subscription id', required: true
+    option '--subscription-cert', 'CERTIFICATE', 'Path to Azure management certificate', attribute_name: :certificate, required: true
+
+    def execute
+      require_api_url
+      require_current_stack
+
+      require 'central/machine/azure'
+
+      stack = client(require_token).get("stacks/#{current_stack}")
+      destroyer = Central::Machine::Azure::NodeDestroyer.new(client(require_token), subscription_id, certificate)
+      destroyer.run!(stack, name)
+    end
+  end
+end

data/lib/central/cli/nodes/azure_command.rb

@@ -0,0 +1,14 @@
+require_relative 'azure/create_command'
+require_relative 'azure/restart_command'
+require_relative 'azure/terminate_command'
+
+module Central::Cli::Nodes
+  class AzureCommand < Clamp::Command
+    subcommand 'create', 'Create a new Azure node', Azure::CreateCommand
+    subcommand 'restart', 'Restart Azure node', Azure::RestartCommand
+    subcommand 'terminate', 'Terminate Azure node', Azure::TerminateCommand
+
+    def execute
+    end
+  end
+end

data/lib/central/cli/nodes/digital_ocean/create_command.rb

@@ -0,0 +1,32 @@
+module Central::Cli::Nodes::DigitalOcean
+  class CreateCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+
+    parameter '[NAME]', 'Node name'
+    option '--token', 'TOKEN', 'DigitalOcean API token', required: true
+    option '--ssh-key', 'SSH_KEY', 'Path to ssh public key', required: true
+    option '--size', 'SIZE', 'Droplet size', default: '1gb'
+    option '--region', 'REGION', 'Region', default: 'ams2'
+    option '--version', 'VERSION', 'Define installed Central version', default: 'latest'
+
+    def execute
+      require_api_url
+      require_current_stack
+
+      require 'central/machine/digital_ocean'
+      stack = client(require_token).get("stacks/#{current_stack}")
+      provisioner = Central::Machine::DigitalOcean::NodeProvisioner.new(client(require_token), token)
+      provisioner.run!(
+        master_uri: api_url,
+        stack_token: stack['token'],
+        stack: current_stack,
+        ssh_key: ssh_key,
+        name: name,
+        size: size,
+        region: region,
+        version: version
+      )
+    end
+  end
+end

data/lib/central/cli/nodes/digital_ocean/restart_command.rb

@@ -0,0 +1,27 @@
+module Central::Cli::Nodes::DigitalOcean
+  class RestartCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+
+    parameter 'NAME', 'Node name'
+    option '--token', 'TOKEN', 'DigitalOcean API token', required: true
+
+    def execute
+      require_api_url
+      require_current_stack
+
+      require 'central/machine/digital_ocean'
+
+      client = DropletKit::Client.new(access_token: token)
+      droplet = client.droplets.all.find { |d| d.name == name }
+      if droplet
+        ShellSpinner "Restarting DigitalOcean droplet #{name.colorize(:cyan)} " do
+          client.droplet_actions.reboot(droplet_id: droplet.id)
+          sleep 5 until client.droplets.find(id: droplet.id).status == 'active'
+        end
+      else
+        abort "Cannot find droplet #{name.colorize(:cyan)} in DigitalOcean"
+      end
+    end
+  end
+end

data/lib/central/cli/nodes/digital_ocean/terminate_command.rb

@@ -0,0 +1,19 @@
+module Central::Cli::Nodes::DigitalOcean
+  class TerminateCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+
+    parameter 'NAME', 'Node name'
+    option '--token', 'TOKEN', 'DigitalOcean API token', required: true
+
+    def execute
+      require_api_url
+      require_current_stack
+
+      require 'central/machine/digital_ocean'
+      stack = client(require_token).get("stacks/#{current_stack}")
+      destroyer = Central::Machine::DigitalOcean::NodeDestroyer.new(client(require_token), token)
+      destroyer.run!(stack, name)
+    end
+  end
+end

data/lib/central/cli/nodes/digital_ocean_command.rb

@@ -0,0 +1,14 @@
+require_relative 'digital_ocean/create_command'
+require_relative 'digital_ocean/restart_command'
+require_relative 'digital_ocean/terminate_command'
+
+module Central::Cli::Nodes
+  class DigitalOceanCommand < Clamp::Command
+    subcommand 'create', 'Create a new DigitalOcean node', DigitalOcean::CreateCommand
+    subcommand 'restart', 'Restart DigitalOcean node', DigitalOcean::RestartCommand
+    subcommand 'terminate', 'Terminate DigitalOcean node', DigitalOcean::TerminateCommand
+
+    def execute
+    end
+  end
+end

data/lib/central/cli/nodes/label_command.rb

@@ -0,0 +1,12 @@
+module Central::Cli::Nodes
+  require_relative 'labels/add_command'
+  require_relative 'labels/remove_command'
+
+  class LabelCommand < Clamp::Command
+    subcommand 'add', 'Add label to node', Labels::AddCommand
+    subcommand %w(remove rm), 'Remove label from node', Labels::RemoveCommand
+
+    def execute
+    end
+  end
+end

data/lib/central/cli/nodes/labels/add_command.rb

@@ -0,0 +1,19 @@
+module Central::Cli::Nodes::Labels
+  class AddCommand < Clamp::Command
+    include Central::Cli::Common
+
+    parameter 'NODE_ID', 'Node id'
+    parameter 'LABEL', 'Label'
+
+    def execute
+      require_api_url
+      require_current_stack
+      token = require_token
+
+      node = client(token).get("stacks/#{current_stack}/nodes/#{node_id}")
+      data = {}
+      data[:labels] = node['labels'].to_a | [label]
+      client.put("nodes/#{node['id']}", data, {}, 'Central-Stack-Token' => node['stack']['token'])
+    end
+  end
+end