capistrano-hivequeen 7.6.0 → 7.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 501195df16e5bf37fcbfd70296ba9241a784f7d56670baf7722f6b627ecce049
-  data.tar.gz: 1c684d9f8f7bd13a28a11d21e987b73f972cadd50f8374599bbcf68a81c6f5e4
+  metadata.gz: 72b660b02784362a5c9a2f8a96a3355a3fd18fb399b57d138942488f89b16b10
+  data.tar.gz: da9503dc5137e661476abba23820112d59bd4033d0775dbd015e449a0ee889f2
 SHA512:
-  metadata.gz: 8e45901e1a5a9968f4b0703a88cabfc7733a9959177321a860c00f35e3492d702b51454a5353c593573e6a35396d6e5a15611bbe7486fe8c597f0acf4523d3ea
-  data.tar.gz: '09c60243c11e62f0f8bbc5d2d7ae4d95e8795c5fbe0039f4c1d338ba2b0a41caa8f3f5d04d22cbda30d81f69db7c328d8c9bfddfb237ff986f29e075609348b4'
+  metadata.gz: 96e6bb88e860b84984b32501ac7be9135364b4ad16a4a644540e8b3cb9f30e655f20f829b5bdcf085e4fc8ba4ddc54d18cc1f36a9a59bef356de48af5f37080a
+  data.tar.gz: ca952a870f3db6dafbc36701dcb6708a2902d611fd1abf57783be5607bd3fc0b62523f81ebfd2e7c8f19b45741a549ab7704d79b548c56220185392069bcb6ed

data/capistrano-hivequeen.gemspec

@@ -38,6 +38,8 @@ Gem::Specification.new do |s|
 
   ## List your runtime dependencies here. Runtime dependencies are those
   ## that are needed for an end user to actually USE your code.
+  s.add_dependency('aws-sdk-ec2', '~> 1.0')
+  s.add_dependency('aws-sdk-ec2instanceconnect', '~> 1.0')
   s.add_dependency('capistrano', '>= 2.11.0')
   s.add_dependency('activesupport', '>= 3.0.0')
   s.add_dependency('json')

data/lib/capistrano/hivequeen/capistrano_configuration.rb

@@ -97,14 +97,18 @@ Capistrano::Configuration.instance(:must_exist).load do
         command_format = "ssh -t -A -l %s %s"
         env['roles'].keys.each do |role_name|
           task role_name do
-            cmd = command_format % [user, roles[role_name.to_sym].servers.first]
+            server = roles[role_name.to_sym].servers.sample
+            HiveQueen.ec2_instance_connect(server.host)
+            cmd = command_format % [user, server]
             puts "Executing #{cmd}"
             exec cmd
           end
         end
 
         task :default do
-          cmd = command_format % [user, roles.values.sample.servers.sample]
+          server = roles.values.sample.servers.sample
+          HiveQueen.ec2_instance_connect(server.host)
+          cmd = command_format % [user, server]
           puts "Executing #{cmd}"
           exec cmd
         end
@@ -114,14 +118,18 @@ Capistrano::Configuration.instance(:must_exist).load do
         command_format = "ssh -t -A -l %s %s 'source /etc/profile; cd /apps/#{HiveQueen.project}/current && bundle exec rails console'"
         env['roles'].keys.each do |role_name|
           task role_name do
+            server = roles[role_name.to_sym].servers.sample
+            HiveQueen.ec2_instance_connect(server.host)
             puts "Opening console"
-            exec command_format % [user, roles[role_name.to_sym].servers.first]
+            exec command_format % [user, server]
           end
         end
 
         task :default do
+          server = roles.values.sample.servers.sample
+          HiveQueen.ec2_instance_connect(server.host)
           puts "Opening console"
-          exec command_format % [user, roles.values.sample.servers.sample]
+          exec command_format % [user, server]
         end
       end
     end

data/lib/capistrano/hivequeen/ec2_instance_connect.rb

@@ -0,0 +1,36 @@
+class HiveQueen
+  def self.ec2_client
+    @ec2_client ||= Aws::EC2::Client.new
+  end
+
+  def self.ec2_instance_connect_client
+    @ec2_instance_connect_client ||= Aws::EC2InstanceConnect::Client.new
+  end
+
+  def self.ec2_instance_connect(*private_dns)
+    # EC2 Instance Connect
+    ssh_public_key = File.read(File.expand_path('~/.ssh/ksr_ed25519.pub'))
+    ec2_params = {
+      filters: [{
+        name:   'network-interface.private-dns-name',
+        values: private_dns
+      }]
+    }
+    logger.trace("ec2:DescribeInstances #{ec2_params.to_json}")
+    instances = ec2_client.describe_instances(**ec2_params).reservations.map(&:instances).flatten
+    threads = instances.map do |instance|
+      Thread.new do
+        ec2ic_params = {
+          availability_zone: instance.placement.availability_zone,
+          instance_id:       instance.instance_id,
+          instance_os_user:  'ksr',
+          ssh_public_key:    ssh_public_key,
+        }
+        logger.trace("ec2-instance-connect:SendSSHPublicKey #{ec2ic_params.to_json}")
+        ec2_instance_connect_client.send_ssh_public_key(**ec2ic_params)
+      end
+    end
+
+    threads.each(&:join)
+  end
+end

data/lib/capistrano/hivequeen/version.rb

@@ -1,6 +1,6 @@
 class HiveQueen
   class Version
-    @@version = '7.6.0'
+    @@version = '7.7.0'
 
     def self.to_s
       @@version

data/lib/capistrano/hivequeen.rb

@@ -1,12 +1,17 @@
 # HTTP Client for Hive Queen environment configuration
+require 'base64'
+require 'fileutils'
 require 'json'
+
 require 'active_support'
 require 'active_support/core_ext'
-require 'fileutils'
+require 'aws-sdk-ec2'
+require 'aws-sdk-ec2instanceconnect'
 require 'excon'
-require 'base64'
+
 require 'capistrano/hivequeen/version'
 require 'capistrano/hivequeen/multiio'
+require 'capistrano/hivequeen/ec2_instance_connect'
 
 # Special cases:
 # - environment not found
@@ -135,4 +140,3 @@ class HiveQueen
 end
 
 require "capistrano/hivequeen/capistrano_configuration"
-

metadata

@@ -1,15 +1,43 @@
 --- !ruby/object:Gem::Specification
 name: capistrano-hivequeen
 version: !ruby/object:Gem::Version
-  version: 7.6.0
+  version: 7.7.0
 platform: ruby
 authors:
 - Aaron Suggs
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-09-15 00:00:00.000000000 Z
+date: 2022-07-28 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ec2
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ec2instanceconnect
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: capistrano
   requirement: !ruby/object:Gem::Requirement
@@ -93,6 +121,7 @@ files:
 - lib/capistrano/hivequeen.rb
 - lib/capistrano/hivequeen/capistrano_configuration.rb
 - lib/capistrano/hivequeen/deploy.rb
+- lib/capistrano/hivequeen/ec2_instance_connect.rb
 - lib/capistrano/hivequeen/multiio.rb
 - lib/capistrano/hivequeen/setup.rb
 - lib/capistrano/hivequeen/version.rb
@@ -116,7 +145,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubygems_version: 3.3.7
 signing_key:
 specification_version: 2
 summary: Capistrano extensions for interacting with HiveQueen