omc 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: dc3cffe813d70187f7415b6ee368a1355301cbb4
-  data.tar.gz: 188d8b778d92830e879a016968529ebd133cdd70
+  metadata.gz: 40766a4dd46b353e2b9bfdd9cc5628f83845d90b
+  data.tar.gz: 591dc4424274502ba89c3a1c487745ac19293955
 SHA512:
-  metadata.gz: 2f36c9dff7a20e16e2bb17718c55d6b548ba336fcc67ae1337f7743fa2ec27d34205dbe2b04a6fd1e21ab6a9d798a1b687aebda9b46f7701a34874ba1498fd88
-  data.tar.gz: afaf005acb4df2b401b1015d57109009e98fabe957509772f70aadfbd38f094399c2186997762cdd6947e94d952adf3fab7a3797b7a7d294b65027829c4edfcb
+  metadata.gz: 0d280a904f65b44571e6a4f9c45f6b5ae34f6c26b921e95eff6df3395adc3007443893b9a5bf7afd3d75fdcef9cb44666fbeaefcf64389bd602c14efd225d1f3
+  data.tar.gz: db7927aca1c1e0a498bdfd73b4454549b29d984ef39aef69aebe06bb31c5f39cccd0f3653dd0b697d870be4167ea27f7e4b19e1fb6fe14e705fcfb4fd4ec8b3e

data/lib/omc.rb

@@ -2,5 +2,4 @@ require "omc/version"
 require "omc/cli"
 
 module Omc
-  # Your code goes here...
 end

data/lib/omc/cli.rb

@@ -1,30 +1,30 @@
 require 'thor'
 require 'aws_cred_vault'
 require 'aws-sdk'
+require "omc/stack_command"
 
 module Omc
   class Cli < Thor
     desc 'ssh', 'Connect to an instance on a stack on an account'
-    def ssh(account, stack)
-      iam_account = vault.account account
-      user = iam_account.users.first
-      ops = ::AWS::OpsWorks::Client.new user.credentials
+    def ssh(account_name, stack)
+      command = StackCommand.new(user(account), stack)
+      command.ssh
+    end
 
-      ops_stack = get_by_name ops.describe_stacks[:stacks], stack
-      instances = ops.describe_instances(stack_id: ops_stack[:stack_id])[:instances]
-      instance = instances.first
-      system "ssh #{user.name}@#{instance[:public_ip]}"
+    desc 'console', 'Run a rails console on the given stack'
+    def console(account, stack)
+      command = StackCommand.new(user(account), stack)
+      command.console
     end
 
     private
-    def vault
-      AwsCredVault::Toml.new File.join(ENV['HOME'], '.aws_cred_vault')
+    def user(account)
+      iam_account = vault.account(account) || abort("Account #{account.inspect} not configured")
+      iam_account.users.first || abort("No users configured under #{account.inspect}")
     end
 
-    def get_by_name collection, name
-      collection.detect do |x|
-        x[:name] == name
-      end || abort("Can't find #{name.inspect} among #{collection.map{|x| x[:name] }.inspect}")
+    def vault
+      AwsCredVault::Toml.new File.join(ENV['HOME'], '.aws_cred_vault')
     end
   end
 end

data/lib/omc/stack_command.rb

@@ -0,0 +1,48 @@
+require 'aws-sdk'
+
+module Omc
+  class StackCommand
+    def initialize user, stack_name
+      @user = user
+      @stack_name = stack_name
+    end
+
+    def ssh
+      exec "ssh", ssh_host
+    end
+
+    def console
+      app = applications.first
+      exec 'ssh', '-t', ssh_host, "sudo su deploy -c 'cd /srv/www/#{app[:name]}/current && RAILS_ENV=#{app[:attributes]['RailsEnv']} bundle exec rails c'"
+    end
+
+    private
+    def applications
+      ops.describe_apps(stack_id: stack[:stack_id])[:apps]
+    end
+
+    def instance
+      instances = ops.describe_instances(stack_id: stack[:stack_id])[:instances]
+      instances.reject!{|i| i[:status] != "online" }
+      instance = instances.first || abort("No running instances")
+    end
+
+    def ssh_host
+      "#{@user.name}@#{instance[:public_ip]}"
+    end
+
+    def ops
+      @ops ||= ::AWS::OpsWorks::Client.new @user.credentials
+    end
+
+    def stack
+      @stack ||= get_by_name ops.describe_stacks[:stacks], @stack_name
+    end
+
+    def get_by_name collection, name
+      collection.detect do |x|
+        x[:name] == name
+      end || abort("Can't find #{name.inspect} among #{collection.map{|x| x[:name] }.inspect}")
+    end
+  end
+end

data/lib/omc/version.rb

@@ -1,3 +1,3 @@
 module Omc
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omc
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Clarke Brunsdon
@@ -96,6 +96,7 @@ files:
 - bin/omc
 - lib/omc.rb
 - lib/omc/cli.rb
+- lib/omc/stack_command.rb
 - lib/omc/version.rb
 - omc.gemspec
 homepage: http://github.com/freerunningtech/omc