contexto 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 195f6f1cf5e2290f2e30504fc730ceba7dcec5db
-  data.tar.gz: 8bc00df2ae7f8bc35b8a8a0e55d99bd7e1fc1ba1
+  metadata.gz: 70a1d561afdac4ad1e724a0f4e5174b2e302b5c6
+  data.tar.gz: 899491e29583693bcde3f42d5f02b17dfb943765
 SHA512:
-  metadata.gz: 12dfdc35a021d798a55ae622d9c86e863e14e45724508ee4add2904f899dea55ce0015b5ea1d3acec4a4443e43ae2cb838382da3ea2ae9adfb5abe0ba5e64a7c
-  data.tar.gz: 508654ea0e00261c36fc85bb5619ac450dd1c1784f95fbcd7577f6a6bf35af17c1c3e4ebfd1d51dc42e0565d21803b7a71c74ee037c4299758753d24ca3822b2
+  metadata.gz: ed11c289dd07940251a6a75eb67f6d8d1f158a0b36b98380871103f503203ae84a728cebd400c117eb6c2819de7259eea377c3055211caa0cf5a1b3f074e7b79
+  data.tar.gz: e0c9492d095b2af5b3284295c9a47e1e4839a84579d49b0006053e00e0532439c289f71eb7091e39fe5f57b2b0366bd071cc6ebb2710823f2ec5cc945cf1a8fa

data/README.md

@@ -68,25 +68,40 @@ Containers
 
 ### SSH
 
-You can also get Contexto to spit out an SSH command you can use to connect to
-a specific container.
+You can also get Contexto to connect via SSH to a Rails console or run a `rake` task.
 
-To do so specify the cluster, servicec and container and the `--ssh`
-flag.
+#### Rails console
+
+To connect to a Rails console, specify the cluster, service and container and the `--console` flag.
 
 ```shell
-$ contexto -c sandbox -s backend-app -t app --ssh
+$ contexto -c sandbox -s backend-app -t app --console
 
                          ** Contexto Contextualizes **
 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
 
-ssh -t -A bastion.empatico.xyz "ssh -t ec2-user@10.2.1.218 'docker exec -it \$(docker ps | grep ecs-sandbox-backend-app-.*-app | cut -d\  -f1) bin/rails c'"
+I, [2017-03-28T22:36:35.555604 #181]  INFO -- sentry: ** [Raven] Raven 2.4.0 ready to catch errors
+Loading sandbox environment (Rails 5.0.1)
+irb(main):001:0>
 ```
 
-## TODO
+#### Rake tasks
+
+You can run `rake` tasks. To run the `test` task you would do:
+
+```shell
+$ contexto -c sandbox -s backend-app -t sidekiq --rake test
+
 
-1. Automate the SSH login process.
-2. ???
+                         ** Contexto Contextualizes **
+━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+
+Run options: --seed 52808
+
+# Running:
+
+................................................E
+```
 
 ## Development
 

data/bin/contexto

@@ -11,20 +11,20 @@ require 'English'
 $stderr.sync = true
 
 options = {}
-options[:cluster] = 'sandbox'
-options[:ssh]     = false
+options[:console]     = false
 
 optparse = OptionParser.new do |opts|
   opts.banner = 'Usage: contexto [options]'
   opts.on('-c', '--cluster CLUSTER', 'Cluster') { |v| options[:cluster] = v }
   opts.on('-s', '--service SERVICE', 'Service') { |v| options[:service] = v }
   opts.on('-t', '--container CONTAINER', 'Container') { |v| options[:container] = v }
-  opts.on('-b', '--ssh') { options[:ssh] = true }
+  opts.on('-r', '--rake COMMAND', 'Rake command') { |v| options[:rake] = v }
+  opts.on('-b', '--console') { options[:console] = true }
 end
 
 begin
   optparse.parse!
-  mandatory = [:cluster]
+  mandatory = []
   missing = mandatory.select { |param| options[param].nil? }
   unless missing.empty?
     raise OptionParser::MissingArgument.new(missing.join(', '))

data/lib/contexto/contextualize.rb

@@ -7,54 +7,80 @@ module Contexto
     attr_reader :display, :cluster, :service
 
     def initialize(params = {})
-      @cluster = params.fetch(:cluster)
+      @cluster = params.fetch(:cluster) if params[:cluster]
       @service = params.fetch(:service) if params[:service]
       @container = params.fetch(:container) if params[:container]
-      @ssh = params.fetch(:ssh)
+      @rake = params.fetch(:rake) if params[:rake]
+      @console = params.fetch(:console)
       @display = Contexto::Display.new
     end
 
     def run
-      if @ssh
+      if (@console || @rake)
         task = describe_tasks
         ec2_instance_id = describe_container_instance(task[:container_instance_arn])
-        @connect = Contexto::SSH.new(describe_instances(ec2_instance_id), @cluster, @service, @container)
+        @ssh = Contexto::SSH.new(describe_instances(ec2_instance_id), @cluster, @service, @container)
+        if @console
+          @ssh.console
+        elsif @rake
+          @ssh.rake(@rake)
+        end
+        return
+      end
+      if !@cluster
+        clusters
         return
       end
       cluster
       puts "\n"
-      return unless @service
-      service
-      puts "\n"
-      tasks
+      if !@service
+        services
+      end
+      if @service
+        puts "\n"
+        service
+        puts "\n"
+        tasks
+      end
     end
 
     private
 
+    def clusters
+      title = "Clusters"
+      headings = %w(Name)
+      clusters = list_clusters
+      rows = []
+      clusters.each do |cluster|
+        rows << [cluster.split('/')[-1]]
+      end
+      @display.create_display(title, headings, rows)
+    end
+
     def cluster
-      resp = ecs_client.describe_clusters(
-        clusters: [
-          @cluster
-        ]
-      )
       title = "Cluster #{@cluster}"
       headings = %w(Name Status Instances Services RunningTasks PendingTasks)
-      cluster = resp.clusters[0].to_h
+      cluster = describe_clusters
       rows = []
       rows << [cluster[:cluster_name], cluster[:status], cluster[:registered_container_instances_count], cluster[:active_services_count], cluster[:running_tasks_count], cluster[:pending_tasks_count]]
       @display.create_display(title, headings, rows)
     end
 
+    def services
+      title = "Services"
+      headings = %w(Name)
+      services = list_services
+      rows = []
+      services.each do |service|
+        rows << [service.split('/')[-1]]
+      end
+      @display.create_display(title, headings, rows)
+    end
+
     def service
-      resp = ecs_client.describe_services(
-        cluster: @cluster,
-        services: [
-          @service
-        ]
-      )
       title = "Service #{@service}"
       headings = %w(Name Status Desired Running Pending)
-      service = resp.services[0].to_h
+      service = describe_services
       rows = []
       rows << [service[:service_name], service[:status], service[:desired_count], service[:running_count], service[:pending_count]]
       @display.create_display(title, headings, rows)
@@ -75,6 +101,37 @@ module Contexto
       @display.create_display(title, headings, rows)
     end
 
+    def list_clusters
+      resp = ecs_client.list_clusters
+      resp.cluster_arns
+    end
+
+    def describe_clusters
+      resp = ecs_client.describe_clusters(
+        clusters: [
+          @cluster
+        ]
+      )
+      resp.clusters[0].to_h
+    end
+
+    def list_services
+      resp = ecs_client.list_services(
+        cluster: @cluster
+      )
+      resp.service_arns
+    end
+
+    def describe_services
+      resp = ecs_client.describe_services(
+        cluster: @cluster,
+        services: [
+          @service
+        ]
+      )
+      resp.services[0].to_h
+    end
+
     def describe_tasks
       resp = ecs_client.describe_tasks(
         cluster: @cluster,

data/lib/contexto/display.rb

@@ -64,7 +64,7 @@ module Contexto
     end
 
     def col_width
-      @col_width ||= @rows.map(&:size).max + 6
+      @col_width ||= @rows.map(&:size).max + 10
     end
   end
 end

data/lib/contexto/ssh.rb

@@ -11,12 +11,18 @@ module Contexto
       @bastion_user = ''
       @user_name    = 'ec2-user'
       @options      = {}
-      connect
     end
 
-    def connect
-      cmd = "ssh -t -A #{@bastion_host} \"ssh -t ec2-user@#{@ipaddress} 'docker exec -it \\$(docker ps | grep ecs-#{@cluster}-#{@service}-.*-#{@container} | cut -d\\  -f1) bin/rails c'\""
-      puts cmd
+    def console
+      exec "#{cmd} c'\""
+    end
+
+    def rake(rake_cmd)
+      exec "#{cmd} #{rake_cmd}'\""
+    end
+
+    def cmd
+      "ssh -t -A #{@bastion_host} \"ssh -t ec2-user@#{@ipaddress} 'docker exec -it \\$(docker ps | grep ecs-#{@cluster}-#{@service}-.*-#{@container} | cut -d\\  -f1) bin/rails "
     end
   end
 end

data/lib/contexto/version.rb

@@ -1,4 +1,4 @@
 # Contexto version
 module Contexto
-  VERSION = '0.1.1'.freeze
+  VERSION = '0.2.0'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: contexto
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - James Turnbull
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-03-27 00:00:00.000000000 Z
+date: 2017-03-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk