ridoku 0.1.0

data/lib/ridoku/service.rb

@@ -0,0 +1,158 @@
+#
+# Command: service
+#
+
+require 'ridoku/base'
+
+
+module Ridoku
+  module Services
+    @service_list = []
+    def self.add_service(serv)
+      @service_list << serv.to_s
+    end
+
+    def self.list
+      @service_list
+    end
+  end
+end
+
+require_rel 'services/'
+
+module Ridoku
+  register :service
+
+  class Service < Base
+    attr_accessor :services
+
+    # "services":[{
+    #   "name": "rabbitmq",
+    #   "layers": [
+    #     "worker"
+    #   ]
+    # },...]
+
+    # Service name should be the same used to define configurations, ideally.
+
+    def run(command = nil)
+      command ||= Base.config[:command]
+      command.shift
+      sub_command = command.shift
+
+      environment = load_environment
+      case sub_command
+      when 'list'
+        list
+      when 'set', 'add'
+        add
+      when 'delete', 'remove', 'rm'
+        delete
+      when 'config', nil
+        config(ARGV)
+      else
+        print_service_help
+      end
+    end
+
+    protected
+
+    def load_environment
+      Base.fetch_stack
+      self.services = (Base.custom_json['services'] ||= {})
+    end
+
+    def print_service_help
+      $stderr.puts <<-EOF
+Command: service
+
+List/Modify the current app's associated workers.
+   service[:list]   lists the services and what layer they are assigned
+   service:add      service:layer, e.g., rabbitmq:workers
+   service:delete   delete service information from stack and layer
+   service:config   service:list for description of service configuration layer
+
+Example 'service:list' output:
+  Services for MyStack:
+    [layer] service 
+
+examples:
+  $ services
+  No services specified!
+  $ services:add rabbitmq:workers
+  $ services:add nagios:workers # not yet implemented
+  $ services:list
+  Services for MyStack:
+   [layer] service
+   [workers] rabbitmq
+   [workers] nagios
+  $ services:delete nagios
+  Services for MyStack:
+   [layer] service
+   [workers] rabbitmq
+  $ services:config rabbitmq:help
+    ...
+      EOF
+    end
+
+    def list
+      if services.length == 0
+        $stdout.puts 'No services specified!'
+      else
+        $stdout.puts "Services for #{$stdout.colorize(Base.config[:stack], [:bold, :green])}:"
+        services.each do |service|
+          if service['layers'].is_a?(Array)
+            $stdout.puts "  [#{$stdout.colorize(service['layers'].join(','), :bold)}] #{service['name']}"
+          else
+            $stdout.puts "  [#{$stdout.colorize('No Layers Specified', [:bold, :red])}] #{service['name']}"
+          end
+        end
+      end
+    end
+
+    def add
+    end
+
+    def delete
+    end
+
+    def config(argv)
+      return list unless argv.length > 0
+      klass, cmd = argv.shift.split(/:/)
+
+      if klass.nil?
+        $stdout.puts 'Available services:'
+        Ridoku::Services.list.each { |service| $stdout.puts " #{service}" }
+        return
+      end
+
+      begin
+        command = Ridoku::Services.const_get(
+          klass.capitalize
+        ).new
+      rescue => e
+        $stderr.puts "Invalid service:config command specified: #{klass}"
+        puts e.to_s if Ridoku::Base.config[:debug]
+        print_help
+        exit 1
+      end
+
+      begin
+        command.run cmd, argv
+      rescue Ridoku::InvalidConfig => e
+        $stderr.puts "#{e.error.to_s.capitalize} #{e.type.to_s} specified."
+        $stderr.puts 'Use the `list` command to see relavent info.'
+        print_help
+        exit 1
+      rescue Ridoku::NoSshAccess
+        $stderr.puts 'Your user does not have access to ssh on the specified stack.'
+        print_help
+        exit 1
+      rescue ArgumentError => e
+        raise e if Ridoku::Base.config[:debug]
+        $stderr.puts e.to_s
+      end
+
+    end
+  end
+end

data/lib/ridoku/services/postgres.rb

@@ -0,0 +1,77 @@
+# encoding: utf-8
+
+module Ridoku
+  module Services
+    add_service :postgres
+
+    class Postgres
+      attr_accessor :postgresql
+
+      def run(cmd, args)
+        sub_command = args.shift
+
+        case cmd
+        when 'config'
+          config(sub_command)
+        when 'describe', 'list', 'show', nil
+          show(sub_command)
+        else
+          print_help
+        end
+      end
+
+      def print_help
+        $stderr.puts <<-EOF
+Command: service postgres
+
+List/Modify the current app's associated workers.
+  postgres[:config]   lists configuration
+
+        EOF
+      end
+
+      def setup
+        Ridoku::Base.fetch_stack
+        self.postgresql = (Base.custom_json['postgresql'] ||= {})
+      end
+
+      def config(sub)
+        $stderr.puts ''
+      end
+
+      def show_url()
+        Ridoku::Base.fetch_app
+        Ridoku::Base.fetch_instance('postgresql', force: true)
+
+        app = Ridoku::Base.app[:shortname]
+        dbase = postgresql['databases'].select do |db|
+          db['app'] == app
+        end.first
+
+        unless dbase
+          $stderr.puts "Application #{$stderr.colorize('app', :red)} has no "\
+            "databases configured."
+          return
+        end
+
+        dbase['adapter'] = 'postgres'
+        dbase['password'] = dbase['user_password']
+        dbase['port'] = postgresql['config']['port']
+        dbase['host'] = Ridoku::Base.instances.first[:public_ip]
+
+        $stdout.puts $stdout.colorize(Ridoku::Db.gen_dbase_url(dbase), :bold)
+      end
+
+      def show(sub)
+        setup
+
+        if sub == 'url'
+          return show_url
+        end
+
+        $stdout.puts 'Postgresql configuration:'
+        puts JSON.pretty_generate(postgresql)
+      end
+    end
+  end
+end

data/lib/ridoku/services/rabbitmq.rb

@@ -0,0 +1,48 @@
+# encoding: utf-8
+
+module Ridoku
+  module Services
+    add_service :rabbitmq
+
+    class Rabbitmq
+      attr_accessor :mqconfig
+
+      def run(cmd, args)
+        sub_command = args.shift
+
+        case cmd
+        when 'config', nil
+          config(sub_command)
+        when 'describe', 'list', 'show'
+          show(sub_command)
+        else
+          print_help
+        end
+      end
+
+      def print_help
+        $stderr.puts <<-EOF
+Command: service:change rabbitmq
+
+List/Modify the current app's associated workers.
+  rabbitmq[:config]   lists configuration
+
+        EOF
+      end
+
+      def setup
+        Ridoku::Base.fetch_stack
+        self.mqconfig = (Base.custom_json['rabbitmq'] ||= {})
+      end
+
+      def config(sub)
+        $stderr.puts 'RabbitMQ Config'
+      end
+
+      def show(sub)
+        setup
+        puts JSON.pretty_generate(mqconfig)
+      end
+    end
+  end
+end

data/lib/ridoku/version.rb

@@ -0,0 +1,5 @@
+# encoding: utf-8
+
+module Ridoku
+  VERSION = "0.1.0"
+end

data/lib/ridoku/workers.rb

@@ -0,0 +1,138 @@
+#
+# Command: worker
+#
+
+require 'ridoku/base'
+
+module Ridoku
+  register :worker
+
+  class Worker < Base
+    attr_accessor :workers
+
+    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'
+        add
+      when 'delete', 'remove', 'rm'
+        delete
+      when 'push', 'reconfigure'
+        reconfigure
+      else
+        print_worker_help
+      end
+    end
+
+    protected
+
+    def load_environment
+      Base.fetch_stack
+      self.workers = (Base.custom_json['deploy'][Base.config[:app].downcase]['workers'] ||= {})
+    end
+
+    def print_worker_help
+      $stderr.puts <<-EOF
+Command: worker
+
+List/Modify the current app's associated workers.
+   worker[:list]   lists the worker type and separate process queues
+   worker:add      worker, e.g., delayed_job:queue_name
+   worker:delete   delete all keys associated with a worker
+   worker:push     reconfigure the worker server.
+                   Updates environment and DB changes.
+
+Example 'worker:list' output:
+   delayed_job: ["mail", "sms"]
+
+The above list output indicates a DelayedJob worker and two separate
+delayed_job processes.  One processing 'mail'; the other processing 'sms'.
+
+examples:
+  $ worker
+  No workers specified!
+  $ worker:add delayed_job=mail
+  $ worker:add delayed_job=sms
+  $ worker:add sneakers=Workers::MailWorker
+  $ worker:list
+  Workers for MyCurrentApp:
+   delayed_job: ["mail", "sms"]
+  $ worker:delete delayed_job
+  No workers specified!
+EOF
+    end
+
+    def list
+      if workers.length == 0
+        $stdout.puts 'No workers specified!'
+      else
+        $stdout.puts "Workers for #{Base.config[:app]}:"
+        workers.each do |type, queues|
+          $stdout.puts "  #{$stdout.colorize(type.to_s, :bold)}: #{queues.to_s}"
+        end
+      end
+    end
+
+    def reconfigure
+      if Base.config[:wait]
+        $stdout.puts 'Disabling command wait (2 commands to issue)...' 
+        Base.config[:wait] = false
+      end
+
+      Base.fetch_app
+
+      cjson = {
+        deploy: {
+          Base.app[:shortname] => {
+            application_type: 'rails'
+          }
+        }
+      }
+
+      begin
+        $stdout.puts 'This operation will '\
+          "#{$stdout.colorize('RESTART', :red)} (not soft reload) "\
+          'your application workers.'
+        sleep 2
+        $stdout.puts 'Hold your seats: worker reconfigure and force restart in... (Press CTRL-C to Stop)'
+        5.times { |t| $stdout.print "#{$stdout.colorize(5-t, :red)} "; sleep 1 }
+        $stdout.puts "\nSilence is acceptance..."
+      rescue Interrupt
+        $stdout.puts $stdout.colorize("\nCommand canceled successfully!", :green)
+        exit 1
+      end
+
+      Base.config[:layers] = ['workers']
+      Ridoku::Cook.cook_recipe(['rails::configure', 'workers::configure'], cjson)
+    end
+
+    def add
+      ARGV.each do |worker|
+        new_workers = worker.split('=')
+        workers[new_workers[0]] ||= []
+        workers[new_workers[0]] << new_workers[1]
+        workers[new_workers[0]].flatten!
+        $stdout.puts "Adding #{worker}."
+      end
+
+      list
+      Base.save_stack
+    end
+
+    def delete
+      ARGV.each do |worker|
+        workers.delete(worker)
+        $stdout.puts "Deleting worker: #{worker}"
+      end
+
+      list
+      Base.save_stack
+    end
+  end
+end

data/ridoku.gemspec

@@ -0,0 +1,32 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'ridoku/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "ridoku"
+  spec.version       = Ridoku::VERSION
+  spec.authors       = ["Terry Meacham","Joel Clay"]
+  spec.email         = ["zv1n.fire@gmail.com","ra3don92@gmail.com"]
+  spec.description   = %q{Ridoku: OpsWork management CLI.}
+  spec.summary       = %q{Ridoku: OpsWork management CLI.}
+  spec.homepage      = "http://ridoku-cli.blogspot.com"
+  spec.license       = "BSD 3-clause"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = ['ridoku', 'rid']
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ['lib']
+
+  spec.add_runtime_dependency 'aws-sdk', '~> 1.50'
+  spec.add_runtime_dependency 'rugged', '~> 0.21'
+  spec.add_runtime_dependency 'json', '~> 1.8'
+  spec.add_runtime_dependency 'rest-client', '~> 1.7'
+  spec.add_runtime_dependency 'require_all', '~> 1.3'
+  spec.add_runtime_dependency 'activesupport', '~> 4.1'
+  spec.add_runtime_dependency 'activesupport-inflector', '~> 0'
+
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake", '~> 0'
+end