ecs_cmd 0.1.0

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2018 Daniel Schaaff
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,45 @@
+# Ecs-Cmd
+
+[![Build Status](https://travis-ci.org/dschaaff/ecs-cmd.svg?branch=master)](https://travis-ci.org/dschaaff/ecs-cmd)
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/ecs_cmd`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'ecs_cmd'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install ecs_cmd
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/ecs_cmd. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Code of Conduct
+
+Everyone interacting in the EcsCmd project’s codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/[USERNAME]/ecs_cmd/blob/master/CODE_OF_CONDUCT.md).

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "ecs_cmd"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "pry"
+Pry.start(__FILE__)

data/bin/ecs-cmd

@@ -0,0 +1,115 @@
+#!/usr/bin/env ruby
+require 'gli'
+require 'ecs_cmd'
+include GLI::App
+include EcsCmd
+program_desc 'Command utility for interacting with AWS ECS'
+
+version EcsCmd::VERSION
+
+subcommand_option_handling :normal
+arguments :strict
+
+desc 'Describe some switch here'
+switch [:s,:switch]
+
+desc 'Set the aws region'
+default_value 'us-east-1'
+arg_name 'region'
+flag [:r,:region]
+
+desc 'Get Info '
+command :get do |c|
+  c.command :clusters do |clusters|
+    clusters.desc 'list ecs clusters'
+    clusters.action do |global_options,options,args|
+      clust = EcsCmd::Clusters.new(global_options[:region])
+      clust.list_clusters
+    end
+  end
+
+  c.arg_name '<cluster>'
+  c.command :services do |services|
+    services.desc 'list services in a given ecs cluster'
+    services.action do |global_options, options, args|
+      serv = EcsCmd::Services.new(global_options[:region])
+      serv.list_services(args[0])
+    end
+  end
+
+  c.arg_name '<service> <cluster'
+  c.command :service do |service|
+    service.default_desc 'get info about an ecs service'
+    service.action do |global_options, options, args|
+      EcsCmd::Service.new(args[1],args[0], global_options[:region]).list_service
+    end
+    service.arg_name '<service> <cluster>'
+    service.desc 'get event stream for ecs service'
+    service.command :events do |events|
+      events.desc 'list events for service'
+      events.action do |global_options, options, args|
+        EcsCmd::Service.new(args[1],args[0], global_options[:region]).events
+      end
+    end
+    service.arg_name '<service> <cluster> task_definition'
+    service.desc 'get json task definition for service'
+    service.command :task_definition do |task_definition|
+      task_definition.desc 'get the current task definition for the service'
+      task_definition.action do |global_options, options, args|
+        EcsCmd::TaskDefinition.new(EcsCmd::Service.new(args[1],args[0], global_options[:region]).task_definition).print_json
+      end
+    end
+  end
+end
+
+# desc 'Add description here'
+# arg_name 'Describe arguments here'
+# command :get do |c|
+#   c.action do |global_options, options, args|
+#     if args[0] == 'task_definition'
+#       task = EcsCmd::TaskDefinition.new(args[1])
+#       puts JSON.pretty_generate(JSON[task.json])
+#     elsif
+#       puts "not implemented"
+#     end
+#   end
+# end
+
+desc 'Describe add here'
+arg_name 'Describe arguments to add here'
+command :add do |c|
+  c.action do |global_options,options,args|
+    puts "add command ran"
+  end
+end
+
+# desc 'Describe complete here'
+# arg_name 'Describe arguments to complete here'
+# command :complete do |c|
+#   c.action do |global_options,options,args|
+#     puts "complete command ran"
+#   end
+# end
+
+pre do |global,command,options,args|
+  # Pre logic here
+  # Return true to proceed; false to abort and not call the
+  # chosen command
+  # Use skips_pre before a command to skip this block
+  # on that command only
+  true
+end
+
+post do |global,command,options,args|
+  # Post logic here
+  # Use skips_post before a command to skip this
+  # block on that command only
+end
+
+on_error do |exception|
+  # Error logic here
+  # return false to skip default error handling
+  true
+end
+
+exit run(ARGV)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/ecs_cmd.gemspec

@@ -0,0 +1,42 @@
+
+# frozen_string_literal: true
+
+lib = File.expand_path('lib', __dir__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ecs_cmd/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'ecs_cmd'
+  spec.version       = EcsCmd::VERSION
+  spec.authors       = ['Daniel Schaaff']
+  spec.email         = ['dbschaaff@gmail.com']
+
+  spec.summary       = 'AWS ECS CLI Utility'
+  spec.description   = 'AWS ECS CLI Utility'
+  spec.homepage      = 'https://danielschaaff.com'
+  spec.license       = 'MIT'
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+  # if spec.respond_to?(:metadata)
+  #   spec.metadata['allowed_push_host'] = "rubyge'"
+  # else
+  #   raise 'RubyGems 2.0 or newer is required to protect against ' \
+  #     'public gem pushes.'
+  # end
+
+  spec.files = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir = 'bin'
+  spec.executables << 'ecs-cmd'
+  spec.require_paths = ['lib']
+  spec.add_runtime_dependency 'aws-sdk'
+  spec.add_runtime_dependency 'gli', '2.17.1'
+  spec.add_runtime_dependency 'terminal-table'
+
+  spec.add_development_dependency 'bundler', '~> 1.16'
+  spec.add_development_dependency 'rake', '~> 10.0'
+  spec.add_development_dependency 'rspec', '~> 3.0'
+  spec.add_development_dependency 'pry'
+end

data/lib/ecs_cmd.rb

@@ -0,0 +1,9 @@
+require 'ecs_cmd/version'
+require 'ecs_cmd/clusters.rb'
+require 'ecs_cmd/services.rb'
+require 'ecs_cmd/service.rb'
+require 'ecs_cmd/task_definition.rb'
+
+module EcsCmd
+  # Your code goes here...
+end

data/lib/ecs_cmd/clusters.rb

@@ -0,0 +1,47 @@
+require 'aws-sdk'
+require 'terminal-table'
+
+module EcsCmd
+  class Clusters
+    def initialize(region)
+      @client = Aws::ECS::Client.new(region: region)
+    end
+
+    def get_clusters
+      @cluster_arns = @client.list_clusters
+    end
+
+    def get_container_instance_count(stats)
+      stats['registered_container_instances_count']
+    end
+
+    def get_service_count(stats)
+      stats['active_services_count']
+    end
+
+    def get_running_task_count(stats)
+      stats['running_tasks_count']
+    end
+
+    def get_pending_task_count(stats)
+      stats['pending_tasks_count']
+    end
+
+    def cluster_names
+      get_clusters[0]
+    end
+
+    def list_clusters
+      rows = []
+      cluster_names.map do |c|
+        cluster_name = c.split('/')[1]
+        stats = @client.describe_clusters(clusters: [cluster_name])[0][0]
+        rows << [cluster_name, get_container_instance_count(stats), get_service_count(stats),
+                 get_running_task_count(stats), get_pending_task_count(stats)]
+      end
+      table = Terminal::Table.new headings: ['CLUSTER NAME', 'CONTAINER_INSTANCE_COUNT',
+                                             'SERVICES', 'RUNNING_TASKS', 'PENDING_TASKS'], rows: rows
+      puts table
+    end
+  end
+end

data/lib/ecs_cmd/service.rb

@@ -0,0 +1,86 @@
+require 'aws-sdk'
+require 'terminal-table'
+
+module EcsCmd
+  class Service
+    def initialize(cluster, name, region)
+      @client = Aws::ECS::Client.new(region: region)
+      @service_stats = @client.describe_services(cluster: cluster, services: [name])[0]
+      raise 'service does not appear to exist' if @service_stats.empty?
+    end
+
+    def arn
+      @service_stats[0]['service_arn']
+    end
+
+    def status
+      @service_stats[0]['status']
+    end
+
+    def deployments
+      t = []
+      @service_stats[0]['deployments'].each do |e|
+        t << ['id', e['id']]
+        t << ['status', e['status']]
+        t << ['task definition', e['task_definition']]
+        t << ['desired count', e['desired_count']]
+        t << ['pending count', e['pending_count']]
+        t << ['running count', e['running_count']]
+        t << ['created at', e['created_at']]
+        t << ['updated at', e['updated_at']]
+        t << ["\n"]
+      end
+      table = Terminal::Table.new headings: ['DEPLOYMENTS'], rows: t
+      table
+    end
+
+    def deployment_configuration
+      @service_stats[0]['deployment_configuration']
+    end
+
+    def desired_count
+      @service_stats[0]['desired_count']
+    end
+
+    def running_count
+      @service_stats[0]['running_count']
+    end
+
+    def pending_count
+      @service_stats[0]['pending_count']
+    end
+
+    def task_definition
+      @service_stats[0]['task_definition']
+    end
+
+    def health_check_grace_period
+      @service_stats[0]['health_check_grace_period_seconds']
+    end
+
+    def events
+      @service_stats[0]['events'].each do |e|
+        puts "#{e['created_at']}: #{e['message']}"
+      end
+    end
+
+    def name
+      @service_stats[0]['service_name']
+    end
+
+    def list_service
+      row1 = []
+      row1 << [name, status, running_count, desired_count, pending_count,
+               deployment_configuration['maximum_percent'], deployment_configuration['minimum_healthy_percent']]
+      table1 = Terminal::Table.new headings: ['NAME', 'STATUS', 'RUNNING COUNT',
+                                              'DESIRED COUNT', 'PENDING COUNT',
+                                              'MAX HEALTHY', 'MIN HEALTHY'], rows: row1
+      row2 = []
+      row2 << [task_definition]
+      table2 = Terminal::Table.new headings: ['TASK DEFINITION'], rows: row2
+      puts table1
+      puts table2
+      puts deployments
+    end
+  end
+end

data/lib/ecs_cmd/services.rb

@@ -0,0 +1,33 @@
+require 'aws-sdk'
+require 'terminal-table'
+
+module EcsCmd
+  class Services
+    def initialize(region)
+      @client = Aws::ECS::Client.new(region: region)
+      @reg = region
+    end
+
+    def get_services(cluster)
+      @service_arns = []
+      @client.list_services(cluster: cluster).each do |r|
+        @service_arns << r[0]
+        @service_arns.flatten!
+      end
+      @service_arns
+    end
+
+    def list_services(cluster, reg = @reg)
+      @service_list = get_services(cluster)
+      rows = []
+      @service_list.map do |service|
+        service_name = service.split('/')[1]
+        serv = EcsCmd::Service.new(cluster, service_name, reg)
+        rows << [service_name, serv.desired_count, serv.running_count, serv.pending_count]
+      end
+      table = Terminal::Table.new headings: ['SERVICE NAME', 'DESIRED_COUNT',
+                                             'RUNNING_COUNT', 'PENDING_COUNT'], rows: rows
+      puts table
+    end
+  end
+end

data/lib/ecs_cmd/task_definition.rb

@@ -0,0 +1,31 @@
+require 'aws-sdk'
+require 'terminal-table'
+
+module EcsCmd
+  class TaskDefinition
+ 
+    def initialize(task_definition)
+      @client = Aws::ECS::Client.new(region: 'us-east-1')
+      @task_def = @client.describe_task_definition(task_definition: task_definition)[0]
+    end
+
+
+    def container_defintions
+      @task_def['container_definitions']
+    end
+
+    def hash
+      @task_def.to_hash
+    end
+
+    def json
+      hash.to_json
+    end
+
+    def print_json
+      puts JSON.pretty_generate(JSON[json])
+      
+    end
+
+  end
+end