ridoku 0.1.0

data/lib/ridoku/domain.rb

@@ -0,0 +1,124 @@
+#
+# Command: domain
+#
+
+require 'ridoku/base'
+require 'json'
+
+module Ridoku
+  register :domain
+
+  class Domain < Base
+    attr_accessor :domains
+
+    def run
+      clist = Base.config[:command]
+      command = clist.shift
+      sub_command = clist.shift
+
+      environment = load_environment
+
+      case sub_command
+      when 'list', nil
+        list
+      when 'set', 'add'
+        add
+      when 'delete', 'remove', 'rm'
+        delete
+      when 'push'
+        push_update
+      else
+        print_domain_help
+      end
+    end
+
+    protected
+
+    def load_environment
+      Base.fetch_app
+      self.domains = (Base.app[:domains] ||= [])
+    end
+
+    def print_domain_help
+      $stderr.puts <<-EOF
+Command: domain
+
+List/Modify the current app's associated domains.
+   domain[:list]   lists the key value pairs
+   domain:add      domain, e.g., http://app.example.com
+   domain:delete   domain or index
+   domain:push     push updated domains to the server
+
+examples:
+  $ domain
+  No domains specified!
+  $ domain:add app.example.com
+  $ domain:list
+  Domains:
+   0: app.example.com
+  EOF
+    end
+
+    def list
+      if domains.length == 0
+        $stdout.puts 'No domains specified!'
+      else
+        $stdout.puts 'Domains:'
+        domains.each_index do |idx|
+          $stdout.puts "  #{$stdout.colorize(idx.to_s, :bold)}: #{domains[idx]}"
+        end
+      end
+    end
+
+    def add
+      ARGV.each do |domain|
+        if domains.index(domain) == nil
+          domains << domain 
+          $stdout.puts "Adding #{domain}."
+        end
+      end
+
+      Base.save_app(:domains)
+    end
+
+    def delete
+      ARGV.each do |domain|
+        if domain.match(/^[0-9]+$/)
+          value = domains.delete_at(domain.to_i)
+        else
+          value = domains.delete(domain)
+        end
+        $stdout.puts "Deleting domain: #{value}"
+      end
+      
+      Base.save_app(:domains)
+    end
+
+    def push_update
+      if domains.length == 0
+        $stdout.puts 'No domains specified!'
+        $stderr.puts 'Please specify at least 1 domain and try again.'
+        return
+      end
+
+      unless Base.config[:quiet]
+        $stdout.puts "Pushing domains:"
+
+        domains.each_index do |idx|
+          $stdout.puts "  #{$stdout.colorize(idx.to_s, :bold)}: #{domains[idx]}"
+        end
+      end
+
+      Base.standard_deploy('rails-app', 
+        {
+          opsworks_custom_cookbooks: {
+            recipes: [
+              "deploy::domains"
+            ]
+          }
+        }
+      )
+      
+    end
+  end
+end

data/lib/ridoku/dump.rb

@@ -0,0 +1,132 @@
+#
+# Command: dump
+#
+
+require 'ridoku/base'
+
+module Ridoku
+  register :dump
+
+  class Dump < Base
+
+    def run
+      command = Base.config[:command]
+      sub_command = (command.length > 0 && command[1]) || nil
+      sub_sub_command = (command.length > 1 && command[2]) || nil
+
+      case sub_command
+      when 'stack'
+        dump_stack(sub_sub_command == 'all')
+      when 'custom'
+        dump_custom(sub_sub_command == 'all')
+      when 'app'
+        dump_app(sub_sub_command == 'all')
+      when 'layer'
+        dump_layer(sub_sub_command == 'all')
+      when 'instance'
+        dump_instance
+      else
+        print_dump_help
+      end
+    end
+
+    protected
+
+    def print_dump_help
+      $stderr.puts <<-EOF
+  Command: dump
+
+  List/Modify the current app's database configuration.
+    dump                 displays this list
+    dump:stack[:all]     dump stack json or :all stack jsons
+    dump:custom[:all]    dump stack's custom json only
+    dump:app[:all]       dump app json or :all app jsons for specified stack
+    dump:layer[:all]     dump layer json or :all layer jsons for specified stack
+    dump:instance        dump instance json
+    EOF
+    end
+
+    def dump_custom(all = false)
+      Base.fetch_stack
+
+      if all
+        $stdout.print '['
+        Base.stack_list.each { |st| custom(st, true) }
+        $stdout.puts ']'
+      else
+        custom(Base.stack)
+      end
+    end
+
+    def dump_stack(all = false)
+      Base.fetch_stack
+
+      if all
+        $stdout.print '['
+        Base.stack_list.each { |st| stack(st, true) }
+        $stdout.puts ']'
+      else
+        stack(Base.stack)
+      end
+    end
+
+    def dump_app(all = false)
+      Base.fetch_app
+
+      if all
+        $stdout.print '['
+        Base.app_list.each { |ap| app(ap, true) }
+        $stdout.puts ']'
+      else
+        app(Base.app)
+      end
+    end
+
+    def dump_layer(all = false)
+      Base.fetch_layer(Base.config[:layer] || :all)
+
+      $stdout.print '['
+      if all
+        Base.layer_list.each { |ly| layer(ly) }
+      else
+        Base.layers.each { |ly| layer(ly)  }
+      end
+      $stdout.puts ']'
+    end
+
+    def dump_instance
+      Base.fetch_instance
+      $stdout.print '['
+      Base.instances.each { |ist| instance(ist) }
+      $stdout.puts ']'
+    end
+
+    def stack(st, multiple = false)
+      $stdout.print JSON.generate(st.to_hash)
+      $stdout.print ',' if multiple
+      $stdout.puts
+    end
+
+    def custom(st, multiple = false)
+      $stdout.print st[:custom_json]
+      $stdout.print ',' if multiple
+      $stdout.puts
+    end
+
+    def app(ap, multiple = false)
+      $stdout.print ap.to_json
+      $stdout.print ',' if multiple
+      $stdout.puts
+    end
+
+    def layer(ly)
+      $stdout.print ly.to_json
+      $stdout.puts ','
+    end
+
+    def instance(ist)
+      $stdout.print ist.to_json
+      $stdout.puts ','
+    end
+  end
+end

data/lib/ridoku/env.rb

@@ -0,0 +1,118 @@
+#
+# Command: env
+# Description: List/Modify the current apps configuration.
+#  env - lists the key value pairs
+#  env:set KEY:VALUE [...]
+#  env:delete KEY
+#
+
+require 'ridoku/base'
+
+module Ridoku
+  register :env
+
+  class Env < Base
+    attr_accessor :environment
+
+    def run
+      command = Base.config[:command]
+      sub_command = (command.length > 0 && command[1]) || nil
+
+      environment = load_environment
+
+      case sub_command
+      when 'list', nil
+        list
+      when 'set', 'add'
+        set
+      when 'push'
+        push
+      when 'delete', 'remove', 'rm'
+        delete
+      else
+        print_env_help
+      end
+    end
+
+    protected
+
+    def load_environment
+      Base.fetch_stack
+      Base.fetch_app
+      self.environment =
+        Base.custom_json['deploy'][Base.app[:shortname]]['app_env']
+    end
+
+    def print_env_help
+      $stderr.puts <<-EOF
+    Command: env
+
+    List/Modify the current app's environment.
+       env        lists the key value pairs
+       env:push   push recipe changes
+       env:set    KEY:VALUE [...]
+       env:delete KEY [...]
+
+    examples:
+      $ env
+      Environment Empty!
+      $ env:set AWS_ACCESS_KEY:'jas8dyfawenfi9f'
+      $ env:set AWS_SECRET_KEY:'SJHDF3HSDOFJS4DFJ3E'
+      $ env:delete AWS_SECRET_KEY
+      $ env
+      Environment:
+        AWS_ACCESS_KEY: 'jas8dyfawenfi9f'
+      EOF
+    end
+
+    def push
+      $stdout.puts "Pushing current environment..."
+
+      Base.standard_deploy(['rails-app','workers'],
+        {
+          opsworks_custom_cookbooks: {
+            recipes: [
+              "workers::default",
+              "deploy::environment"
+            ]
+          }
+        }
+      )
+    end
+
+    def list
+      if environment.keys.length == 0
+        $stdout.puts 'Environment Empty!'
+      else
+        $stdout.puts 'Environment:'
+        environment.each do |key, value|
+          $stdout.puts "  #{$stdout.colorize(key, :bold)}: '#{value}'"
+        end
+      end
+    end
+
+    def set
+      ARGV.each do |kvpair|
+        kvpair.match(%r((^[^:]+):(.*))) do |m|
+          key = m[1]
+          value = m[2]
+
+          update = environment.key?(key)
+          environment[key] = value
+          $stdout.puts "#{update && 'Updating' || 'Adding'}: #{key} as '#{value}'"
+        end
+      end
+
+      Base.save_stack
+    end
+
+    def delete
+      ARGV.each do |key|
+        value = environment.delete(key)
+        $stdout.puts "Deleting key: #{key}, '#{value}'"
+      end
+      
+      Base.save_stack
+    end
+  end
+end

data/lib/ridoku/list.rb

@@ -0,0 +1,168 @@
+#
+# Command: list
+# Description: Used to list all stacks on an AWS OpsStack account.
+#   The current selection is colorized green.
+#
+
+require 'ridoku/base'
+
+module Ridoku
+  register :list
+
+  class List < Base    
+    def run
+      command = Base.config[:command]
+      sub_command = (command.length > 0 && command[1]) || nil
+    
+      case sub_command
+      when nil
+        return apps if Base.config[:stack]
+        stacks
+
+      when 'stacks'
+        stacks
+
+      when 'apps'
+        apps
+
+      when 'layers'
+        layers
+
+      when 'instances'
+        instances
+
+      when 'config'
+        config
+
+      when 'services'
+        services
+
+      else
+        print_list_help
+      end
+    end
+
+    protected
+
+    def print_list_help
+      $stderr.puts <<-EOF
+    Command: list
+
+    List/Modify the current app's database configuration.
+       list           lists stacks or apps if stack is specified
+       list:config    lists current configuration information (app, stack, etc)
+       list:stacks    lists stacks by name
+       list:apps      lists apps if stack is specified
+       list:layers    lists layers if stack is specified
+       list:instances lists instances by layer
+      EOF
+    end
+
+    def config
+      config = [
+        'Current:',
+        "  #{$stdout.colorize('Stack', :bold)}: #{Base.config[:stack]}",
+        "  #{$stdout.colorize('App', :bold)}: #{Base.config[:app]}",
+        "  #{$stdout.colorize('Shell User', :bold)}: #{Base.config[:shell_user]}",
+        "  #{$stdout.colorize('Service ARN', :bold)}: #{Base.config[:service_arn]}",
+        "  #{$stdout.colorize('Instance ARN', :bold)}: #{Base.config[:instance_arn]}",
+        "  #{$stdout.colorize('Default SSH Key', :bold)}: #{Base.config[:ssh_key]}",
+      ]
+      $stdout.puts config
+    end
+
+    def stacks
+      Base.fetch_stack(force: true)
+
+      stack_arr = Base.stack_list.map do |stack|
+        name = stack[:name]
+        (name == Base.config[:stack] && $stdout.colorize(name, :green)) || name
+      end
+
+      list = stack_arr.join(', ')
+      $stdout.puts 'Application stacks on your account:'
+      $stdout.puts " #{$stdout.colorize(list, :bold)}"
+    end
+
+    def apps
+      Base.fetch_app
+
+      app_arr = Base.app_list.map do |app|
+        name = app[:name]
+        (name == Base.config[:app] && $stdout.colorize(name, :green)) || name
+      end
+
+      list = app_arr.join(', ')
+      $stdout.puts "Application apps on stack " +
+        "#{$stdout.colorize(Base.stack[:name], [:green, :bold])}:"
+      $stdout.puts " #{$stdout.colorize(list, :bold)}"
+    end
+
+    def layers
+      Base.fetch_layer
+
+      max = 0
+      Base.layer_list.each do |layer|
+        shortname = $stdout.colorize(layer[:shortname], :bold)
+        max = shortname.length if max < shortname.length
+      end
+
+      layer_arr = Base.layer_list.map do |layer|
+        fmt = "%#{max}s"
+        shortname = sprintf(fmt, $stdout.colorize(layer[:shortname], :bold))
+        name = "[#{shortname}] #{layer[:name]}"
+        if layer[:shortname] == Base.config[:layer]
+          $stdout.colorize(name, :green) 
+        else
+          name
+        end
+      end
+
+      $stdout.puts 'Layers on stack ' +
+        "#{$stdout.colorize(Base.stack[:name], [:bold, :green])}:"
+      $stdout.puts layer_arr
+    end
+
+    def instances
+      Base.fetch_layer
+      Base.fetch_instance
+
+      $stdout.puts 'Application instances on stack ' +
+        "#{$stdout.colorize(Base.stack[:name], [:bold, :green])}:"
+
+      Base.layer_list.each do |layer|
+        selected = Base.config[:instances]
+
+        linstances = Base.instances.select do |inst|
+          inst[:layer_ids].index(layer[:layer_id]) != nil
+        end
+
+        instance_arr = linstances.map do |instance|
+          name = "  #{instance[:hostname]}: #{$stdout.colorize(
+            instance[:status], instance[:status] == 'online' ? :green : :red)}"
+          if selected && selected.index(instance[:hostname]) != nil
+            $stdout.colorize(name, :green) 
+          else
+            name
+          end
+        end
+
+        name = "Layer: #{layer[:name]} [#{layer[:shortname]}]"
+
+        $stdout.puts (layer[:shortname] == Base.config[:layer] &&
+          $stdout.colorize(name, :green)) || name
+
+        if instance_arr.length
+          $stdout.puts instance_arr
+        else
+          $stdout.puts '  No instances in this layer.'
+        end
+        $stdout.puts
+      end
+    end
+
+    def services
+      Ridoku::Service.new.run(['list'])
+    end
+  end
+end