envirobly 1.3.1 → 1.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2991e127dfd64a68b47720bf70561343e3b7107304d9f5da9c1423cd4a3f0105
-  data.tar.gz: 8d80a93f0af7427bd44668aaf47b68598ae9a62abd27168a35b52e9102cf9bf2
+  metadata.gz: 7a02e37a74f5ed588bf8438579f5ff30035ac8eee90d58a97eb95aa90e78c6c9
+  data.tar.gz: a4e2bb4dedcc8801a69796ae41ceacc63ce09430ce74d96a28c540a7188ac025
 SHA512:
-  metadata.gz: ab2ba4141f8bbc95fa907bdbfe78d30a9b1600862c9a97a34412de4bc295a4db6a8d46d52d0c97bc744792b15d57b051cf467b7d638287fb2c23e26593832e8f
-  data.tar.gz: d9a69cb24f0d30fa89bdef0dfd06bfb07c78bc99f9fd32e9b2a2fbcfb136f81e1d72adfc39548b834b1c9ebe499fc6bb429e0dd5f9eeb1272448cf21f0018f26
+  metadata.gz: 953ef3a5819627d2910b41970e34c530b5a06603a46a715f96741d683171e8bd7b5db3ee862174e1e101f342a7d5af5b17fd0cbe1730fc9eacc7c07d8285ad66
+  data.tar.gz: 656fa66b9495e7936c15dc0d33c8661c59c3b1dde8e78982d7b577b3f6e325655a4e3567e3f6b415b7dd13c01002201a98ec73109273dae3f4e814a87350d485

data/lib/envirobly/access_token.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "fileutils"
 require "pathname"
 

data/lib/envirobly/api.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "json"
 require "net/http"
 require "socket"

data/lib/envirobly/aws/credentials.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 # @deprecated
 class Envirobly::Aws::Credentials
   def initialize(params)

data/lib/envirobly/aws/s3.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "zlib"
 require "json"
 require "open3"

data/lib/envirobly/aws.rb

@@ -1,2 +1,4 @@
+# frozen_string_literal: true
+
 module Envirobly::Aws
 end

data/lib/envirobly/base.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "thor"
 
 class Envirobly::Base < Thor

data/lib/envirobly/cli/main.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class Envirobly::Cli::Main < Envirobly::Base
   include Envirobly::Colorize
 
@@ -126,6 +128,27 @@ class Envirobly::Cli::Main < Envirobly::Base
   method_option :shell, type: :string
   method_option :user, type: :string
   def exec(service_name, *command)
-    Envirobly::ContainerShell.new(service_name, command, options).connect
+    Envirobly::ContainerShell.new(service_name, options).exec(command)
+  end
+
+  desc "rsync [SERVICE_NAME:]SOURCE_PATH [SERVICE_NAME:]DESTINATION_PATH", <<~TXT
+    Synchronize files between you and your service's data volume.
+  TXT
+  method_option :account_id, type: :numeric
+  method_option :project_id, type: :numeric
+  method_option :project_name, type: :string
+  method_option :environ_name, type: :string
+  method_option :args, type: :string, default: "-avzP"
+  def rsync(source, destination)
+    service_name = nil
+
+    [ source, destination ].each do |path|
+      if path =~ /\A([a-z0-9\-_]+):/i
+        service_name = $1
+        break
+      end
+    end
+
+    Envirobly::ContainerShell.new(service_name, options).rsync(source, destination)
   end
 end

data/lib/envirobly/cli.rb

@@ -1,2 +1,4 @@
+# frozen_string_literal: true
+
 module Envirobly::Cli
 end

data/lib/envirobly/colorize.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Envirobly::Colorize
   GREEN  = "\e[32m"
   RED    = "\e[31m"

data/lib/envirobly/config.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "dotenv"
 
 class Envirobly::Config

data/lib/envirobly/container_shell.rb

@@ -1,17 +1,27 @@
+# frozen_string_literal: true
+
 class Envirobly::ContainerShell
-  CMD_TEMLATE =
-    "AWS_ACCESS_KEY_ID='%s' " +
-    "AWS_SECRET_ACCESS_KEY='%s' " +
-    "AWS_SESSION_TOKEN='%s' " +
-    "ssh -i %s " +
-    "envirobly-service@%s " +
-    "-o StrictHostKeyChecking=accept-new " +
+  AWS_ENV = [
+    "AWS_ACCESS_KEY_ID='%s'",
+    "AWS_SECRET_ACCESS_KEY='%s'",
+    "AWS_SESSION_TOKEN='%s'"
+  ]
+  SSH = [
+    "ssh -i %s",
+    "-o StrictHostKeyChecking=accept-new",
+    "-o SendEnv=ENVIROBLY_SERVICE_INTERACTIVE_SHELL",
+    "-o SendEnv=ENVIROBLY_SERVICE_SHELL_USER",
     "-o ProxyCommand='aws ec2-instance-connect open-tunnel --instance-id %s --region %s'"
+  ]
+  USER_AND_HOST = "envirobly-service@%s"
+
+  attr_reader :options, :service_name
 
-  def initialize(service_name, command, options)
-    @command = command
+  def initialize(service_name, options)
+    @service_name = service_name
     @options = options
     commit = Envirobly::Git::Commit.new "HEAD"
+
     @params = {
       project: {
         account_id: options.account_id || Envirobly::Defaults::Account.new.id,
@@ -20,7 +30,7 @@ class Envirobly::ContainerShell
       },
       environ: { name: options.environ_name || commit.current_branch },
       service: { name: service_name },
-      instance: { slot: options.instance_slot }
+      instance: { slot: options.instance_slot || 0 }
     }
 
     if options.project_name.blank? && options.account_id.blank? && options.project_id.blank?
@@ -28,39 +38,77 @@ class Envirobly::ContainerShell
     end
   end
 
-  def connect
-    api = Envirobly::Api.new
-    response = api.create_service_shell_connection @params
-    ssh_params = response.object
-
-    Tempfile.create do |tempkey|
-      tempkey.write ssh_params.fetch("instance").fetch("private_key")
-      tempkey.flush
-
-      cmd = sprintf(
-        CMD_TEMLATE,
-        ssh_params.fetch("open_tunnel_credentials").fetch("access_key_id"),
-        ssh_params.fetch("open_tunnel_credentials").fetch("secret_access_key"),
-        ssh_params.fetch("open_tunnel_credentials").fetch("session_token"),
-        tempkey.path,
-        ssh_params.fetch("instance").fetch("private_ipv4"),
-        ssh_params.fetch("instance").fetch("aws_id"),
-        ssh_params.fetch("region")
+  def exec(command = nil)
+    with_private_key do
+      system join(env_vars, ssh, user_and_host, command)
+    end
+  end
+
+  def rsync(source, destination)
+    with_private_key do
+      system join(
+        env_vars,
+        %(rsync #{options.args} -e "#{ssh}"),
+        source.sub("#{service_name}:", "#{user_and_host}:"),
+        destination.sub("#{service_name}:", "#{user_and_host}:")
       )
+    end
+  end
+
+  private
+    def join(*parts)
+      parts.flatten.compact.join(" ")
+    end
 
-      if @options.shell.present?
-        cmd = "ENVIROBLY_SERVICE_INTERACTIVE_SHELL='#{@options.shell}' #{cmd} -o SendEnv=ENVIROBLY_SERVICE_INTERACTIVE_SHELL"
+    def connect_data
+      @connect_data ||= begin
+        api = Envirobly::Api.new
+        api.create_service_shell_connection(@params).object
       end
+    end
+
+    def with_private_key
+      Tempfile.create do |file|
+        file.write connect_data.fetch("instance").fetch("private_key")
+        file.flush
+
+        @private_key_path = file.path
 
-      if @options.user.present?
-        cmd = "ENVIROBLY_SERVICE_SHELL_USER='#{@options.user}' #{cmd} -o SendEnv=ENVIROBLY_SERVICE_SHELL_USER"
+        yield
+      end
+    end
+
+    def env_vars
+      credentials = connect_data.fetch("open_tunnel_credentials")
+
+      result = sprintf(
+        join(AWS_ENV),
+        credentials.fetch("access_key_id"),
+        credentials.fetch("secret_access_key"),
+        credentials.fetch("session_token")
+      )
+
+      if options.shell.present?
+        result = join "ENVIROBLY_SERVICE_INTERACTIVE_SHELL='#{options.shell}'", result
       end
 
-      if @command.present?
-        cmd = "#{cmd} #{@command.join(" ")}"
+      if options.user.present?
+        result = join "ENVIROBLY_SERVICE_SHELL_USER='#{options.user}'", result
       end
 
-      system cmd
+      result
+    end
+
+    def ssh
+      sprintf(
+        join(SSH),
+        @private_key_path,
+        connect_data.fetch("instance").fetch("aws_id"),
+        connect_data.fetch("region")
+      )
+    end
+
+    def user_and_host
+      sprintf USER_AND_HOST, connect_data.fetch("instance").fetch("private_ipv4")
     end
-  end
 end

data/lib/envirobly/default.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class Envirobly::Default
   attr_accessor :shell
 

data/lib/envirobly/defaults/account.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class Envirobly::Defaults::Account < Envirobly::Default
   include Envirobly::Colorize
 

data/lib/envirobly/defaults/project.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class Envirobly::Defaults::Project < Envirobly::Default
   def self.file = "project.yml"
   def self.regexp = /projects\/(\d+)/

data/lib/envirobly/defaults/region.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class Envirobly::Defaults::Region < Envirobly::Default
   include Envirobly::Colorize
 

data/lib/envirobly/defaults.rb

@@ -1,2 +1,4 @@
+# frozen_string_literal: true
+
 module Envirobly::Defaults
 end

data/lib/envirobly/deployment.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "yaml"
 
 class Envirobly::Deployment

data/lib/envirobly/duration.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "benchmark"
 
 class Envirobly::Duration

data/lib/envirobly/git/commit.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "time"
 require "open3"
 

data/lib/envirobly/git/unstaged.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class Envirobly::Git::Unstaged < Envirobly::Git::Commit
   def initialize(working_dir: Dir.getwd)
     @working_dir = working_dir

data/lib/envirobly/git.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class Envirobly::Git
   def initialize(working_dir = Dir.getwd)
     @working_dir = working_dir

data/lib/envirobly/numeric.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 class Envirobly::Numeric < Numeric
   def initialize(value, short: false)
     @value = value

data/lib/envirobly/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Envirobly
-  VERSION = "1.3.1"
+  VERSION = "1.4.1"
 end

data/lib/envirobly.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Envirobly
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: envirobly
 version: !ruby/object:Gem::Version
-  version: 1.3.1
+  version: 1.4.1
 platform: ruby
 authors:
 - Robert Starsi
@@ -149,6 +149,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '8.0'
+- !ruby/object:Gem::Dependency
+  name: rubocop-rails-omakase
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
 email: klevo@klevo.sk
 executables:
 - envirobly