aell 0.1.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c21d73c186534e4f439d42d762ffe4cf97ac7590981fea017de02e384d5d9dc9
-  data.tar.gz: 6145658d3ef742ad3989e61b9d69ed1643f94a51eeea36016c96cd9c0e635810
+  metadata.gz: 6a366cd9a2b0a141f24e8270b81aa32e3ae1f6fe0c0a30c1c09b10bbd3056796
+  data.tar.gz: d890e5c9b5148cfb13653eef9fde2dcf993e0d0ea80eab4e4c54653240c886f3
 SHA512:
-  metadata.gz: 87fb47f0f0084fb13ede06a4acb00111d2531c5950f34f58d4bdaa27bb3f057fbbe1cb0fab60c17d67ca460e6ed5e3dc641b5eea1281bb3fb7c20fb41e86d8f0
-  data.tar.gz: 254553736aaa335d3d1dfa897e0eaa8fe51e0d011feb130009e4c719809e916e63f60483032a135ef2e517168be6d005e0d7cbc32f5ab54c557c21dd9d2cb03b
+  metadata.gz: 2858561b28f5779e03b3858803cef5e658796e88207ed5cd2074cafee0ab111ebfc9f79fc41e9b421f3ed2581d5cd5dc6e1faaa3762f7c516083fdd34a1300fd
+  data.tar.gz: 579d2b0931b72141f57d7e94a3ebe86c1784b9c4ea803764db3a7ffab8ae96cf00739fd4631045d1eab66276aca9e09050f3e2e5bf841ed06c635f4f3c1d6c10

data/Gemfile.lock

@@ -1,30 +1,30 @@
 PATH
   remote: .
   specs:
-    aell (0.1.0)
+    aell (0.3.0)
       aws-sdk-ec2 (~> 1.137)
       net-ssh-multi (~> 1.2.1)
 
 GEM
   remote: https://rubygems.org/
   specs:
-    aws-eventstream (1.0.3)
-    aws-partitions (1.267.0)
-    aws-sdk-core (3.89.1)
-      aws-eventstream (~> 1.0, >= 1.0.2)
+    aws-eventstream (1.1.0)
+    aws-partitions (1.354.0)
+    aws-sdk-core (3.104.3)
+      aws-eventstream (~> 1, >= 1.0.2)
       aws-partitions (~> 1, >= 1.239.0)
       aws-sigv4 (~> 1.1)
       jmespath (~> 1.0)
-    aws-sdk-ec2 (1.137.0)
-      aws-sdk-core (~> 3, >= 3.71.0)
+    aws-sdk-ec2 (1.186.0)
+      aws-sdk-core (~> 3, >= 3.99.0)
       aws-sigv4 (~> 1.1)
-    aws-sigv4 (1.1.0)
-      aws-eventstream (~> 1.0, >= 1.0.2)
+    aws-sigv4 (1.2.1)
+      aws-eventstream (~> 1, >= 1.0.2)
     coderay (1.1.2)
     diff-lcs (1.3)
     jmespath (1.4.0)
     method_source (0.9.2)
-    net-ssh (5.2.0)
+    net-ssh (6.1.0)
     net-ssh-gateway (2.0.0)
       net-ssh (>= 4.0.0)
     net-ssh-multi (1.2.1)
@@ -33,7 +33,7 @@ GEM
     pry (0.12.2)
       coderay (~> 1.1.0)
       method_source (~> 0.9.0)
-    rake (10.5.0)
+    rake (13.0.1)
     rspec (3.9.0)
       rspec-core (~> 3.9.0)
       rspec-expectations (~> 3.9.0)
@@ -55,8 +55,8 @@ DEPENDENCIES
   aell!
   bundler (~> 1.17)
   pry
-  rake (~> 10.0)
+  rake (>= 12.3.3)
   rspec (~> 3.0)
 
 BUNDLED WITH
-   1.17.2
+   1.17.3

data/README.md

@@ -22,20 +22,30 @@ To start using aell you need a `AWS_ACCESS_KEY_ID`, `AWS_SECRET_ACCESS_KEY` and
 export AWS_ACCESS_KEY_ID=my_access_key
 export AWS_SECRET_ACCESS_KEY=secret_access_key
 export AWS_REGION=aws_region(eg: us-east-1)
+
+# OPTIONAL (when using MFA)
+export AWS_SESSION_TOKEN=secret_session_token
 ```
 
 Or set with the aell options in each command:
 
 ```bash
-aell - Application:some-app -u deploy-user --access_key_id access_key_id --secret_access_key secret_access_key --region us-east-1 -c 'echo "Hello Aell"'
+aell -t Application:some-app -u deploy-user --access_key_id access_key_id --secret_access_key secret_access_key --region us-east-1 -c 'echo "Hello Aell"'
 ```
+
+Aell can also use aws configuration in ~/.aws/, or use a specific profile with `--profile` option, eg:
+
+
+```bash
+aell --profile my_other_profile -t Application:some-app -u deploy-user -c 'echo "Hello Aell"'
+```
+
 Aell use AWS tags to query the servers with the `-t`/ `--tag` flags, to know more just type `aell -h`
 
 ## TODO
 
- 1. Configure a Docker image
- 2. Separate concerns
- 3. Add specs
+ 1. Separate concerns
+ 2. Add specs
 
 ## Contributing
 

data/aell.gemspec

@@ -28,6 +28,6 @@ Gem::Specification.new do |spec|
 
   spec.add_development_dependency "pry"
   spec.add_development_dependency "bundler", "~> 1.17"
-  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rake", ">= 12.3.3"
   spec.add_development_dependency "rspec", "~> 3.0"
 end

data/exe/aell

@@ -6,18 +6,18 @@ require 'aws-sdk-ec2'
 AWS_ACCESS_KEY_ID = ENV['AWS_ACCESS_KEY_ID']
 AWS_SECRET_ACCESS_KEY = ENV['AWS_SECRET_ACCESS_KEY']
 AWS_REGION = ENV['AWS_REGION']
+AWS_SESSION_TOKEN = ENV['AWS_SESSION_TOKEN']
+
+Signal.trap("INT") { exit 2 }
 
 def verify_options!(options)
-u error('Error: Missing user "-u my_user"') unless options[:user]
-  error('Error: Missing tag "-t some_tag:example" or hosts "-h 10.0.0.1,my_domain.intra"') unless options[:tag] || options[:hosts]
-  error('Error: Missing command "-c some_command"') unless options[:command]
-  error('Error: Missing AWS_ACCESS_KEY_ID, set with AWS_ACCESS_KEY_ID environment variable or with  "--access_key_id my_key"') unless options[:access_key_id] || options[:hosts]
-  error('Error: Missing AWS_SECRET_ACCESS_KEY, set with AWS_SECRET_ACCESS_KEY environment variable or with  "--secret_access_key my_key"') unless options[:secret_access_key] || options[:hosts]
-  error('Error: Missing AWS_REGION, set with AWS_REGION environment variable or with  "--region region"') unless options[:region] || options[:hosts]
+  error('Missing user "-u my_user"') unless options[:user]
+  error('Missing tag "-t some_tag:example" or hosts "-h 10.0.0.1,my_domain.intra"') unless options[:tag] || options[:hosts]
+  error('Missing command "-c some_command"') unless options[:command]
 end
 
 def error(message)
-  $stderr.puts message
+  $stderr.puts "[ERROR] #{message}"
   exit 1
 end
 
@@ -28,27 +28,39 @@ def parse_tags(tags)
   [tag, parsed_values]
 end
 
-def get_servers_ips(access_key_id, secret_access_key, region, tags)
-  tag, values = parse_tags(tags)
-
-  ec2_client = Aws::EC2::Client.new(access_key_id: access_key_id, secret_access_key: secret_access_key, region: region)
+def get_servers_ips(tag:, **aws_options)
+  tag_name, values = parse_tags(tag)
+  ec2_client = Aws::EC2::Client.new(**aws_options.compact)
   ec2 = Aws::EC2::Resource.new(client: ec2_client)
   ec2.instances({
     filters:
       [{
-        name: "tag:#{tag}",
+        name: "tag:#{tag_name}",
         values: values
       }]
     })
-  .collect { |i| i.private_ip_address }
+  .collect { |i| i.private_ip_address }.compact
+
+rescue Aws::EC2::Errors::ServiceError => ex
+  error(ex.message)
+rescue Interrupt
+  exit 2
 end
 
 options = {
   access_key_id: AWS_ACCESS_KEY_ID,
   secret_access_key: AWS_SECRET_ACCESS_KEY,
-  region: AWS_REGION
+  session_token: AWS_SESSION_TOKEN,
+  region: AWS_REGION,
+  verify_host_key: :never,
 }
 
+class Hash
+  def except(*keys)
+    self.reject { |k,v| keys.include?(k) }
+  end
+end
+
 OptionParser.new do |opts|
   opts.banner = <<-BANNER
 Aell - Run commands in multiple servers querying by your AWS Infrastructure
@@ -75,6 +87,14 @@ BANNER
     options[:user] = user
   end
 
+  opts.on("--single", "Only runs in one host") do
+    options[:single] = true
+  end
+
+  opts.on("--verify-host-key HOST_VERIFICATION", "Check host key when connecting (never, accept_new_or_local_tunnel, accept_new or always), default: never") do |host_verification|
+    options[:verify_host_key] = host_verification.to_sym
+  end
+
   opts.on("-v", "--version", "Shows aell version") do
     puts Aell::VERSION
     exit
@@ -92,6 +112,15 @@ BANNER
     options[:region] = region
   end
 
+  opts.on("--profile PROFILE", "Sets the profile (using ~/.aws/credentials), overrides --access_key_id, --secret_access_key, --region, --session_token options and environment variables") do |profile|
+    options[:profile] = profile
+    options.reject! { |k, v| [:access_key_id, :secret_access_key, :region, :session_token].include?(k) }
+  end
+
+  opts.on("--session-token SESSION_TOKEN ", "Sets the session token when using AWS MFA") do |session_token|
+    options[:session_token] = session_token
+  end
+
   opts.on_tail("-h", "--help", "Show this message") do
     puts opts
     exit
@@ -100,13 +129,17 @@ end.parse!
 
 verify_options!(options)
 
-servers = options[:hosts] || get_servers_ips(options[:access_key_id], options[:secret_access_key], options[:region], options[:tag])
+servers = options[:hosts] || get_servers_ips(**options.except(:user, :hosts, :command, :single, :verify_host_key))
+
+servers = [servers.first] if options[:single]
 
 begin
   Thread.report_on_exception = false
   Net::SSH::Multi.start do |session|
     servers.each do |server|
-      session.use(options[:user] ? "#{options[:user]}@#{server}" : server)
+      connection = options[:user] ? "#{options[:user]}@#{server}" : server
+
+      session.use(connection, verify_host_key: options[:verify_host_key])
     end
 
     session.exec options[:command]

data/lib/aell/version.rb

@@ -1,3 +1,3 @@
 module Aell
-  VERSION = "0.1.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: aell
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Carlos Atkinson
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-01-22 00:00:00.000000000 Z
+date: 2021-01-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-ec2
@@ -70,16 +70,16 @@ dependencies:
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: 12.3.3
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: 12.3.3
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
@@ -135,7 +135,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubyforge_project: 
+rubygems_version: 2.7.6
 signing_key: 
 specification_version: 4
 summary: Run commands on multiple servers in your AWS Infrastructure