ecs-rails 0.0.4 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c08e4e91434f407ea8df658c3493bffca339b819540d0c20a9cc6a46e62f8c2c
-  data.tar.gz: 982257395f3f3f4aff5eef242a1a8d3608ffa59c71bf9d87c4e6238ecd6843e1
+  metadata.gz: 28b6dcfb2734b269fbb5da27e70b0e7b43e63cbfb025f904724f804108b21573
+  data.tar.gz: 47e6dc119ec99803eb887f8f1170e6766c28f7c086cbc1b22e7fbdc6d01c6323
 SHA512:
-  metadata.gz: 6fc00dadac2b0d17c2ab0c96312b2157f79b38f6c278e811080636d9d53e90986f8d731eccf2b1b78a994c7e8f5548b9735258aa1949b9352c6e2910d7de34e7
-  data.tar.gz: f18f580b67798d22634fd3e89e4d1fcd7cc96aac200653efb704aeb940ab4c4f8b504f221cd9574e4e8d612b05b5d6bf294d3bd713137a00e761e078d410a5bd
+  metadata.gz: 726c40bafbb598ce2673eeaf48b075b12ddd8f79a0fcdaba2c00c3a2c0acf953508d3783495dc6087fa520f73e4895c8733fb6f85caed5807306a7f97cdcacc7
+  data.tar.gz: e474ae8db2e7eb9b660c5a81e9dd9a76524cc769a50d56b45843b3b30fc65ef55773165552c5c726be704d8b722b3c57dd52bac68b79cad774b1fc17626431ef

data/README.md

@@ -9,6 +9,8 @@ This gem helps to get the correct cluster arn and task id so that you don't have
 ```bash
 ecs console
 ecs bash
+ecs logs
+ecs scale
 ```
 
 # Installation
@@ -25,7 +27,7 @@ gem install 'ecs-rails'
 
 # Configuration
 
-## Plain Ruby
+## environment variables
 
 Via environment variables:
 
@@ -36,18 +38,16 @@ export ENV['AWS_SECRET_ACCESS_KEY'] = 'your-secret-access-key'
 export ENV['CONTAINER_NAME'] = 'your-container-name'
 ```
 
-## Rails
+## AWS SSO
 
-```ruby
-# config/initializers/ecs-rails.rb
-EcsRails.aws_region = 'us-east-1'
-EcsRails.aws_access_key_id = ENV['AWS_ACCESS_KEY_ID']
-EcsRails.aws_secret_access_key = ENV['AWS_SECRET_ACCESS_KEY']
-EcsRails.container_name = 'webapp'
+```
+aws sso login
 ```
 
 # Usage
 
+## Console
+
 Connect to Rails console on a running Task.
 
 ```ruby
@@ -75,3 +75,67 @@ You can specify service name via -s option by giving a string included in servic
 ecs console -c prod -s app
 irb(main)>
 ```
+
+## Bash
+
+Connect to bash session on a running Task.
+
+```bash
+ecs bash -c prod -s app
+bash-4.2#
+```
+
+## Logs
+
+Tail CloudWatch logs for a service.
+
+```bash
+ecs logs -c prod -s app
+
+Fetching log group from task definition...
+Log group: webapp-app-log-group-prod-827a419
+Executing command: aws logs tail webapp-app-log-group-prod-827a419 --region us-east-1 --follow --format short
+```
+
+The logs command will:
+- Automatically discover the CloudWatch log group from your ECS task definition
+- Tail logs in real-time with `--follow` flag
+- Display logs in a short, readable format
+- Press Ctrl-C to exit
+
+## Scale
+
+View the current scaling status of a service (refreshes every 5 seconds):
+
+```bash
+ecs scale -c prod -s app
+
+Press Ctrl-C to exit
+
+Fetching service status... (refreshes every 5 seconds)
+
+Service: webapp-app-prod-7c7cad7
+Cluster: webapp-cluster-prod
+Status: ACTIVE
+
+Task counts:
+  Desired: 2
+  Running: 2
+  Pending: 0
+```
+
+Scale a service up or down by changing the desired task count:
+
+```bash
+ecs scale -c prod -s app --count 5
+
+Fetching service status...
+Current desired count: 2
+Scaling service 'webapp-app-prod-7c7cad7' to 5 tasks...
+✓ Service scaled successfully!
+New desired count: 5
+```
+
+The scale command will:
+- Without `--count`: Show the current status including desired, running, and pending task counts, refreshing every 5 seconds. Press Ctrl-C to exit.
+- With `--count`: Update the service's desired count to the specified number and confirm the operation

data/bin/ecs

@@ -21,6 +21,10 @@ OptionParser.new do |opts|
     options[:config_file] = c
   end
 
+  opts.on("--count COUNT", "Number of tasks to scale to. Ex: 'ecs scale -c prod -s app --count 3'") do |c|
+    options[:count] = c
+  end
+
 end.parse!
 
 command_keyword = ARGV[0]

data/lib/ecs-rails/command_factory.rb

@@ -1,10 +1,10 @@
 module EcsRails
   class CommandFactory
-
     def initialize(command_keyword, options = {})
       @command_keyword = command_keyword
       @cluster_name = options[:cluster]
       @service_name = options[:service]
+      @count = options[:count]
     end
 
     def command
@@ -13,6 +13,10 @@ module EcsRails
         EcsRails::Console.new(cluster_name, service_name)
       when 'bash'
         EcsRails::Bash.new(cluster_name, service_name)
+      when 'logs'
+        EcsRails::Logs.new(cluster_name, service_name)
+      when 'scale'
+        EcsRails::Scale.new(cluster_name, service_name, count)
       else
         EcsRails::NullCommand.new
       end
@@ -20,7 +24,6 @@ module EcsRails
 
     private
 
-      attr_reader :command_keyword, :cluster_name, :service_name
-
+    attr_reader :command_keyword, :cluster_name, :service_name, :count
   end
 end

data/lib/ecs-rails/ecs_rails_configuration.rb

@@ -11,7 +11,7 @@ module EcsRails
     attr_accessor :container_name
 
     def self.delegated
-      public_instance_methods - Singleton.instance_methods
+      public_instance_methods - superclass.public_instance_methods - Singleton.instance_methods
     end
 
     def initialize

data/lib/ecs-rails/logs.rb

@@ -0,0 +1,43 @@
+module EcsRails
+  class Logs < Command
+    def call
+      return full_command if test_mode?
+
+      puts "Executing command: #{full_command}"
+      system(full_command)
+    end
+
+    private
+
+    def full_command
+      @full_command ||= "aws logs tail #{log_group_name} --region #{region} --follow --format short"
+    end
+
+    def log_group_name
+      puts('Fetching log group from task definition...')
+
+      service_descriptions = client.describe_services(
+        cluster: selected_cluster,
+        services: [selected_service]
+      ).services
+
+      task_definition_arn = service_descriptions.first.task_definition
+
+      task_definition = client.describe_task_definition(
+        task_definition: task_definition_arn
+      ).task_definition
+
+      log_config = task_definition.container_definitions.find { |c| c.name == container_name }&.log_configuration
+
+      if log_config && log_config.log_driver == 'awslogs'
+        group_name = log_config.options['awslogs-group']
+        puts("Log group: #{group_name}")
+        group_name
+      else
+        puts('Warning: Could not find awslogs configuration, trying default /ecs/[service-name] format')
+        service_name_from_arn = selected_service.split('/').last
+        "/ecs/#{service_name_from_arn}"
+      end
+    end
+  end
+end

data/lib/ecs-rails/scale.rb

@@ -0,0 +1,94 @@
+module EcsRails
+  class Scale < Command
+    def initialize(cluster_name = nil, service_name = nil, count = nil)
+      super(cluster_name, service_name)
+      @count = count
+    end
+
+    def call
+      if count.nil?
+        show_status_loop
+      else
+        scale_service_once
+      end
+    end
+
+    private
+
+    attr_reader :count
+
+    def show_status_loop
+      puts "Press Ctrl-C to exit\n\n"
+
+      trap('INT') do
+        puts "\nExiting..."
+        exit 0
+      end
+
+      loop do
+        print "\033[2J\033[H"
+
+        puts "Fetching service status... (refreshes every 5 seconds)\n"
+
+        service_descriptions = client.describe_services(
+          cluster: selected_cluster,
+          services: [selected_service]
+        ).services
+
+        service = service_descriptions.first
+        service_name = service.service_name
+
+        display_status(service, service_name)
+
+        sleep 5
+      end
+    end
+
+    def scale_service_once
+      puts 'Fetching service status...'
+
+      service_descriptions = client.describe_services(
+        cluster: selected_cluster,
+        services: [selected_service]
+      ).services
+
+      service = service_descriptions.first
+      service_name = service.service_name
+      current_count = service.desired_count
+
+      puts "Current desired count: #{current_count}"
+      puts "Scaling service '#{service_name}' to #{count} tasks..."
+
+      response = client.update_service(
+        cluster: selected_cluster,
+        service: selected_service,
+        desired_count: count.to_i
+      )
+
+      new_count = response.service.desired_count
+      puts '✓ Service scaled successfully!'
+      puts "New desired count: #{new_count}"
+    end
+
+    def display_status(service, service_name)
+      puts "\nService: #{service_name}"
+      puts "Cluster: #{service.cluster_arn.split('/').last}"
+      puts "Status: #{service.status}"
+      puts "\nTask counts:"
+      puts "  Desired: #{service.desired_count}"
+      puts "  Running: #{service.running_count}"
+      puts "  Pending: #{service.pending_count}"
+
+      return unless service.deployments.length > 1
+
+      puts "\nDeployments:"
+      service.deployments.each_with_index do |deployment, index|
+        status_label = deployment.rollout_state == 'COMPLETED' ? '(primary)' : '(in progress)'
+        puts "  #{index + 1}. #{status_label}"
+        puts "     Desired: #{deployment.desired_count}"
+        puts "     Running: #{deployment.running_count}"
+        puts "     Pending: #{deployment.pending_count}"
+      end
+    end
+  end
+end

data/lib/ecs-rails/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module EcsRails
-  VERSION = '0.0.4'
+  VERSION = '0.0.6'
 end

data/lib/ecs-rails.rb

@@ -22,21 +22,21 @@ require_relative 'ecs-rails/command_factory'
 require_relative 'ecs-rails/command'
 require_relative 'ecs-rails/console'
 require_relative 'ecs-rails/bash'
+require_relative 'ecs-rails/logs'
+require_relative 'ecs-rails/scale'
 require_relative 'ecs-rails/null_command'
 
 # delegate
 require 'active_support/core_ext/module/delegation'
 
 # prompt
-require "tty-prompt"
+require 'tty-prompt'
 
 module EcsRails
-
   class << self
     def config
       EcsRails::EcsRailsConfiguration.instance
     end
     delegate(*EcsRails::EcsRailsConfiguration.delegated, to: :config)
   end
-
 end

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: ecs-rails
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.6
 platform: ruby
 authors:
 - Franck D'agostini
-autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-05-16 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: byebug
@@ -140,7 +139,9 @@ files:
 - lib/ecs-rails/command_factory.rb
 - lib/ecs-rails/console.rb
 - lib/ecs-rails/ecs_rails_configuration.rb
+- lib/ecs-rails/logs.rb
 - lib/ecs-rails/null_command.rb
+- lib/ecs-rails/scale.rb
 - lib/ecs-rails/service_selector.rb
 - lib/ecs-rails/task_selector.rb
 - lib/ecs-rails/version.rb
@@ -151,7 +152,6 @@ homepage: https://rubygems.org/gems/ecs-rails
 licenses:
 - MIT
 metadata: {}
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -166,8 +166,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.12
-signing_key:
+rubygems_version: 3.6.9
 specification_version: 4
 summary: Connect to your AWS ECS tasks
 test_files: []