kontena-cli 0.0.1

data/lib/kontena/cli/server/user.rb

@@ -0,0 +1,111 @@
+require 'kontena/client'
+require_relative '../common'
+
+module Kontena::Cli::Server
+  class User
+    include Kontena::Cli::Common
+
+    def login
+      require_api_url
+      username = ask("Email: ")
+      password = password("Password: ")
+      params = {
+          username: username,
+          password: password,
+          grant_type: 'password',
+          scope: 'user'
+      }
+
+      response = client.post('auth', params)
+
+      if response
+        inifile['server']['token'] = response['access_token']
+        inifile.save(filename: ini_filename)
+        print color('Login Successful', :green)
+        true
+      else
+        print color('Login Failed', :red)
+        false
+      end
+    end
+
+    def logout
+      inifile['server'].delete('token')
+      inifile.save(filename: ini_filename)
+    end
+
+    def whoami
+      require_api_url
+      puts "Server: #{inifile['server']['url']}"
+      token = require_token
+      response = client(token).get('user')
+      puts "User: #{response['email']}"
+    end
+
+    def invite(email)
+      require_api_url
+      token = require_token
+      data = { email: email }
+      response = client(token).post('users', data)
+      puts 'User invited' if response
+    end
+
+    def register
+      require_api_url
+      email = ask("Email: ")
+      password = password("Password: ")
+      password2 = password("Password again: ")
+      if password != password2
+        raise ArgumentError.new("Passwords don't match")
+      end
+      params = {email: email, password: password}
+      client.post('users/register', params)
+    end
+
+    def verify_account(token)
+      require_api_url
+
+      params = {token: token}
+      client.post('user/email_confirm', params)
+      print color('Account verified', :green)
+    end
+
+    def request_password_reset(email)
+      require_api_url
+
+      params = {email: email}
+      client.post('user/password_reset', params)
+      puts 'Email with password reset instructions is sent to your email address. Please follow the instructions to change your password.'
+    end
+
+    def reset_password(token)
+      require_api_url
+      password = password("Password: ")
+      password2 = password("Password again: ")
+      if password != password2
+        raise ArgumentError.new("Passwords don't match")
+      end
+      params = {token: token, password: password}
+      client.put('user/password_reset', params)
+      puts 'Password is now changed. To login with the new password, please run: kontena login'
+    end
+
+    def add_registry
+      default_url = 'https://index.docker.io/v1/'
+      require_api_url
+      username = ask("Username: ")
+      password = password("Password: ")
+      email = ask("Email: ")
+      url = ask("URL [#{default_url}]: ")
+      url = default_url if url.strip == ''
+      data = { username: username, password: password, email: email, url: url }
+      client(token).post("user/registries", data)
+    end
+
+    private
+
+    def token
+      @token ||= require_token
+    end
+  end
+end

data/lib/kontena/cli/services/commands.rb

@@ -0,0 +1,131 @@
+module Kontena::Cli::Services; end;
+
+require_relative 'containers'
+require_relative 'logs'
+require_relative 'services'
+require_relative 'stats'
+
+command 'service list' do |c|
+  c.syntax = 'kontena service list'
+  c.description = 'List all services'
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.list
+  end
+end
+
+command 'service show' do |c|
+  c.syntax = 'kontena service show <service_id>'
+  c.description = 'Show service details'
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.show(args[0])
+  end
+end
+
+command 'service deploy' do |c|
+  c.syntax = 'kontena service deploy <service_id>'
+  c.description = 'Deploy service to nodes'
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.deploy(args[0])
+  end
+end
+
+command 'service restart' do |c|
+  c.syntax = 'kontena service restart <service_id>'
+  c.description = 'Restart service containers'
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.restart(args[0])
+  end
+end
+
+command 'service stop' do |c|
+  c.syntax = 'kontena service stop <service_id>'
+  c.description = 'Stop service containers'
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.stop(args[0])
+  end
+end
+
+command 'service start' do |c|
+  c.syntax = 'kontena service start <service_id>'
+  c.description = 'Start service containers'
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.start(args[0])
+  end
+end
+
+command 'service create' do |c|
+  c.syntax = 'kontena service create <name> <image>'
+  c.description = 'Create new service'
+  c.option '-p', '--ports Array', Array, 'Publish a service\'s port to the host'
+  c.option '-e', '--env Array', Array, 'Set environment variables'
+  c.option '-l', '--link Array', Array, 'Add link to another service in the form of name:alias'
+  c.option '-v', '--volume Array', Array, 'Mount a volume'
+  c.option '--volumes_from Array', Array, 'Mount volumes from another container'
+  c.option '-a', '--affinity Array', Array, 'Set service affinity'
+  c.option '-c', '--cpu-shares INTEGER', Integer, 'CPU shares (relative weight)'
+  c.option '-m', '--memory INTEGER', String, 'Memory limit (format: <number><optional unit>, where unit = b, k, m or g)'
+  c.option '--memory-swap INTEGER', String, 'Total memory usage (memory + swap), set \'-1\' to disable swap (format: <number><optional unit>, where unit = b, k, m or g)'
+  c.option '--cmd STRING', String, 'Command to execute'
+  c.option '--instances INTEGER', Integer, 'How many instances should be deployed'
+  c.option '-u', '--user String', String, 'Username who executes first process inside container'
+  c.option '--stateful', 'Set service as stateful'
+
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.create(args[0], args[1], options)
+  end
+end
+
+command 'service update' do |c|
+  c.syntax = 'kontena service update <service_id>'
+  c.description = 'Update service'
+  c.option '-p', '--ports Array', Array, 'Exposed ports'
+  c.option '-e', '--env Array', Array, 'Environment variables'
+  c.option '--image STRING', String, 'Service image'
+  c.option '--instances INTEGER', Integer, 'How many instances should be deployed'
+  c.option '--cmd STRING', String, 'Command to execute'
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.update(args[0], options)
+  end
+end
+
+command 'service scale' do |c|
+  c.syntax = 'kontena service scale <service_id> <instances>'
+  c.description = 'Scale service horizontally'
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.scale(args[0], args[1])
+  end
+end
+
+command 'service delete' do |c|
+  c.syntax = 'kontena service delete <service_id>'
+  c.description = 'Delete service'
+  c.action do |args, options|
+    Kontena::Cli::Services::Services.new.destroy(args[0])
+  end
+end
+
+command 'service containers' do |c|
+  c.syntax = 'kontena service containers <service_id>'
+  c.description = 'Show service containers'
+  c.action do |args, options|
+    Kontena::Cli::Services::Containers.new.list(args[0])
+  end
+end
+
+command 'service logs' do |c|
+  c.syntax = 'kontena service logs <service_id>'
+  c.description = 'Show service logs'
+  c.option '-f', '--follow', 'Follow logs in real time'
+  c.action do |args, options|
+    Kontena::Cli::Services::Logs.new.show(args[0], options)
+  end
+end
+
+command 'service stats' do |c|
+  c.syntax = 'kontena service stats <name>'
+  c.description = 'Show service stats'
+  c.option '-f', '--follow', 'Follow stats in real time'
+  c.action do |args, options|
+    Kontena::Cli::Services::Stats.new.show(args[0], options)
+  end
+end

data/lib/kontena/cli/services/containers.rb

@@ -0,0 +1,24 @@
+require 'kontena/client'
+require_relative '../common'
+
+module Kontena::Cli::Services
+  class Containers
+    include Kontena::Cli::Common
+
+    ##
+    # @param [String] service_id
+    def list(service_id)
+      require_api_url
+      token = require_token
+
+      result = client(token).get("services/#{service_id}/containers")
+      result['containers'].each do |container|
+        puts "#{container['id']}:"
+        puts "  node: #{container['node']['name']}"
+        puts "  ip (internal): #{container['network_settings']['ip_address']}"
+        puts "  status: #{container['status']}"
+        puts ""
+      end
+    end
+  end
+end

data/lib/kontena/cli/services/logs.rb

@@ -0,0 +1,26 @@
+require 'kontena/client'
+require_relative '../common'
+
+module Kontena::Cli::Services
+  class Logs
+    include Kontena::Cli::Common
+
+    ##
+    # @param [String] service_id
+    def show(service_id, options)
+      require_api_url
+      token = require_token
+      last_id = nil
+      loop do
+        query_params = last_id.nil? ? '' : "from=#{last_id}"
+        result = client(token).get("services/#{service_id}/container_logs?#{query_params}")
+        result['logs'].each do |log|
+          puts log['data']
+          last_id = log['id']
+        end
+        break unless options.follow
+        sleep(2)
+      end
+    end
+  end
+end

data/lib/kontena/cli/services/services.rb

@@ -0,0 +1,196 @@
+require 'kontena/client'
+require_relative '../common'
+
+module Kontena::Cli::Services
+  class Services
+    include Kontena::Cli::Common
+
+    def list
+      require_api_url
+      token = require_token
+
+      grids = client(token).get("grids/#{current_grid}/services")
+      puts "%-30.30s %-40.40s %-15s %-8s" % ['NAME', 'IMAGE', 'INSTANCES', 'STATE?']
+      grids['services'].each do |service|
+        state = service['stateful'] ? 'yes' : 'no'
+        puts "%-30.30s %-40.40s %-15.15s %-8s" % [service['id'], service['image'], service['container_count'], state]
+      end
+    end
+
+    def show(service_id)
+      require_api_url
+      token = require_token
+
+      service = client(token).get("services/#{service_id}")
+      puts "#{service['id']}:"
+      puts "  status: #{service['state'] }"
+      puts "  stateful: #{service['stateful'] == true ? 'yes' : 'no' }"
+      puts "  scaling: #{service['container_count'] }"
+      puts "  image: #{service['image']}"
+      if service['cmd']
+        puts "  cmd: #{service['cmd'].join(' ')}"
+      else
+        puts "  cmd: -"
+      end
+
+      puts "  env: "
+      if service['env']
+        service['env'].each{|e| puts "    - #{e}"}
+      end
+      puts "  ports:"
+      service['ports'].each do |p|
+        puts "    - #{p['node_port']}:#{p['container_port']}/#{p['protocol']}"
+      end
+      puts "  links: "
+      if service['links']
+        service['links'].each do |p|
+          puts "    - #{p['alias']}"
+        end
+      end
+      puts "  containers:"
+      result = client(token).get("services/#{service_id}/containers")
+      result['containers'].each do |container|
+        puts "    #{container['id']}:"
+        puts "      rev: #{container['deploy_rev']}"
+        puts "      node: #{container['node']['name']}"
+        puts "      dns: #{container['id']}.kontena.local"
+        puts "      ip: #{container['network_settings']['ip_address']}"
+        if container['status'] == 'unknown'
+          puts "      status: #{container['status'].colorize(:yellow)}"
+        else
+          puts "      status: #{container['status']}"
+        end
+      end
+    end
+
+    def scale(service_id, count)
+      client(require_token).put("services/#{service_id}", {container_count: count})
+      self.deploy(service_id)
+    end
+
+    def deploy(service_id)
+      require_api_url
+      token = require_token
+
+      result = client(token).post("services/#{service_id}/deploy", {})
+
+      print 'deploying '
+      until client(token).get("services/#{service_id}")['state'] != 'deploying' do
+        print '.'
+        sleep 1
+      end
+      puts ' done'
+      puts ''
+      self.show(service_id)
+    end
+
+    def restart(service_id)
+      require_api_url
+      token = require_token
+
+      result = client(token).post("services/#{service_id}/restart", {})
+    end
+
+    def stop(service_id)
+      require_api_url
+      token = require_token
+
+      result = client(token).post("services/#{service_id}/stop", {})
+    end
+
+    def start(service_id)
+      require_api_url
+      token = require_token
+
+      result = client(token).post("services/#{service_id}/start", {})
+    end
+
+    def create(name, image, options)
+      require_api_url
+      token = require_token
+      if options.ports
+        ports = parse_ports(options.ports)
+      end
+      data = {
+        name: name,
+        image: image,
+        stateful: !!options.stateful
+      }
+      if options.link
+        links = parse_links(options.link)
+      end
+      data[:ports] = ports if options.ports
+      data[:links] = links if options.link
+      data[:volumes] = options.volume if options.volume
+      data[:volumes_from] = options.volumes_from if options.volumes_from
+      data[:memory] = parse_memory(options.memory) if options.memory
+      data[:memory_swap] = parse_memory(options.memory_swap) if options.memory_swap
+      data[:cpu_shares] = options.cpu_shares if options.cpu_shares
+      data[:affinity] = options.affinity if options.affinity
+      data[:env] = options.env if options.env
+      data[:container_count] = options.instances if options.instances
+      data[:cmd] = options.cmd.split(" ") if options.cmd
+      data[:user] = options.user if options.user
+      data[:cpu] = options.cpu if options.cpu
+      if options.memory
+        memory = human_size_to_number(options.memory)
+        raise ArgumentError.new('Invalid --memory')
+        data[:memory] = memory
+      end
+      data[:memory] = options.memory if options.memory
+      client(token).post("grids/#{current_grid}/services", data)
+    end
+
+    def update(service_id, options)
+      require_api_url
+      token = require_token
+
+      data = {}
+      data[:env] = options.env if options.env
+      data[:container_count] = options.instances if options.instances
+      data[:cmd] = options.cmd.split(" ") if options.cmd
+      data[:ports] = parse_ports(options.ports) if options.ports
+      data[:image] = options.image if options.image
+
+      client(require_token).put("services/#{service_id}", data)
+    end
+
+    def destroy(service_id)
+      require_api_url
+      token = require_token
+
+      result = client(token).delete("services/#{service_id}")
+    end
+
+    private
+    def current_grid
+      inifile['server']['grid']
+    end
+
+    def parse_ports(port_options)
+      port_options.map{|p|
+        node_port, container_port = p.split(':')
+        if node_port.nil? || container_port.nil?
+          raise ArgumentError.new("Invalid port value #{p}")
+        end
+        {
+          container_port: container_port,
+          node_port: node_port
+        }
+      }
+    end
+
+    def parse_links(link_options)
+      link_options.map{|l|
+        service_name, alias_name = l.split(':')
+        if service_name.nil? || alias_name.nil?
+          raise ArgumentError.new("Invalid link value #{l}")
+        end
+        {
+            name: service_name,
+            alias: alias_name
+        }
+      }
+    end
+  end
+end