ridoku 0.1.0

data/lib/ridoku/db.rb

@@ -0,0 +1,235 @@
+#
+# Command: db
+# Description: List/Modify the current apps database configuration
+#  db - lists the database parameters
+#  config:set KEY:VALUE [...]
+#  config:delete KEY
+#
+
+require 'ridoku/base'
+
+module Ridoku
+  register :db
+
+  class Db < Base
+    attr_accessor :dbase
+
+    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 'list', nil, 'info'
+        list(false)
+      when 'credentials'
+        list(true)
+      when 'set', 'add'
+        set
+      when 'push', 'update'
+        push_update
+      when 'delete', 'remove', 'rm'
+        delete
+      when 'url', 'path'
+        url(sub_sub_command)
+      else
+        print_db_help
+      end
+    end
+
+    protected
+
+    def load_database
+      Base.fetch_stack
+      Base.fetch_app
+      self.dbase =
+        Base.custom_json['deploy'][Base.app[:shortname]]['database']
+    end
+
+    def print_db_help
+      $stderr.puts <<-EOF
+Command: database
+
+List/Modify the current app's database configuration.
+   db           lists the key value pairs
+   db:push      push changes to the servers
+   db:set:url   attribute:value [...]
+   db:delete    attribute [...]
+   db:url       get the URL form of the database info
+   db:url:set   set attributes using a URL
+
+examples:
+  $ db:set database:'Survly'
+  Database:
+    adapter: postgresql
+    host: ec2-50-47-234-2.amazonaws.com
+    port: 5234
+    database: Survly
+    username: survly
+    reconnect: true
+      EOF
+    end
+
+    def push_update
+      if Base.config[:wait]
+        $stdout.puts 'Disabling command wait (2 commands to issue)...' 
+        Base.config[:wait] = false
+      end
+
+      Base.fetch_app
+
+      cjson = {
+        opsworks: {
+          rails_stack: {
+            restart_command: '../../shared/scripts/unicorn force-restart'
+          }  
+        },
+        deploy: {
+          Base.app[:shortname] => {
+            application_type: 'rails'
+          }
+        }
+      }
+
+      begin
+        $stdout.puts 'This operation will '\
+          "#{$stdout.colorize('RESTART', :red)} (not soft reload) "\
+          'your application servers and worker servers.'
+        sleep 2
+        $stdout.puts 'Hold your seats: db push 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] = ['rails-app']
+      Ridoku::Cook.cook_recipe('deploy::database', cjson)
+
+      Base.config[:layers] = ['workers']
+      Ridoku::Cook.cook_recipe(['rails::configure', 'workers::configure'], cjson)
+    end
+
+    def list(cred)
+      load_database
+      if dbase.keys.length == 0
+        $stdout.puts 'Database Not Configured!'
+      else
+        $stdout.puts 'Database:'
+        dbase.each do |key, value|
+          $stdout.puts "  #{$stdout.colorize(key, :bold)}: #{value}" if
+            (cred || (key != 'password' && key != 'username'))
+        end
+      end
+    end
+
+    def set
+      load_database
+      ARGV.each do |kvpair|
+        kvpair.match(%r((^[^:=]+)(:|=)(.*))) do |m|
+          key = m[1]
+          value = m[3]
+
+          update = dbase.key?(key)
+          dbase[key] = value
+          $stdout.puts "#{update && 'Updating' || 'Adding'}: #{key} as '#{value}'"
+        end
+      end
+
+      Base.save_stack
+    end
+
+    def delete
+      load_database
+      ARGV.each do |key|
+        value = dbase.delete(key)
+        $stdout.puts "Deleting key: #{key}, '#{value}'"
+      end
+
+      Base.save_stack
+    end
+
+    def set_url_database(subc)
+      if subc != 'set'
+        print_db_help 
+        exit 1
+      end
+
+      regex = %r(^([^:]+)://([^:]+):([^@]+)@([^:]+):([^/]+)/(.*)$)
+      ARGV[0].match(regex) do |m|
+        dbase['adapter'] = Db.adapter_from_scheme(m[1])
+        dbase['username'] = m[2]
+        dbase['password'] = m[3]
+        dbase['host'] = m[4]
+        dbase['port'] = m[5]
+        dbase['database'] = m[6]
+      end
+    end
+
+    class << self
+      def scheme_hash
+        {
+          'postgresql' => 'postgres',
+          'mysql' => 'mysql'
+        }
+      end
+
+      def scheme_from_adapter(adapter)
+        val = scheme_hash
+        return val[adapter] if val.key?(adapter)
+        adapter
+      end
+
+      def adapter_from_scheme(scheme)
+        val = scheme_hash.invert
+        return val[scheme] if val.key?(scheme)
+        scheme
+      end
+
+      def gen_dbase_url(dbase)
+        scheme = scheme_from_adapter(dbase['adapter'])
+        username = dbase['username']
+        password = dbase['password']
+        host = dbase['host']
+        port = dbase['port']
+        database = dbase['database']
+
+        url = "#{scheme}://"
+        url += username if username
+        url += ":#{password}"
+        url += '@' if username || password
+        url += host
+        url += ":#{port}" if port
+        url += "/#{database}" if database
+        url
+      end
+    end
+
+    def get_url_database
+      scheme = Db.scheme_from_adapter(dbase['adapter'])
+
+      unless  dbase['database'] && scheme && dbase['host']
+        $stdout.puts $stdout.colorize(
+          "One or more required fields are not specified!",
+          :bold
+        )
+        $stdout.puts $stdout.colorize("adapter, host, and database", :red)
+        list_database
+      end
+
+      url = Db.gen_dbase_url(dbase)
+      $stdout.puts $stdout.colorize(url, :bold)
+    end
+
+    def url(subc)
+      load_database
+      if subc
+        set_url_database(subc)
+        Base.save_stack
+      else
+        get_url_database
+      end
+    end
+  end
+end

data/lib/ridoku/defaults.rb

@@ -0,0 +1,68 @@
+#
+# Base class for Component Generation
+#
+
+require 'active_support/core_ext/hash/deep_merge'
+
+module Ridoku
+  class PropertyDefaultsUndefined < StandardError
+    attr_accessor :method
+    def initialize(_method)
+      self.method = _method
+      super "Inheritors of 'ClassProperties' must define: #{method}"
+    end
+  end
+
+  class ClassProperties
+    attr_accessor :default, :required, :input, :warnings
+
+    def initialize
+      fail StandardError.new('IAM/OpsWorks permission are not yet configured.') unless
+        Base.roles_configured?
+    end
+
+    def defaults_with_wizard(type, _input)
+      defaults(type, _input, true)
+    end
+
+    def defaults(type, _input, wizard = false)
+      self.input = _input
+      
+      fail ArgumentError.new("No default parameters for type: #{type}") unless
+        default.key?(type)
+
+      merge_input(type, wizard)
+    end
+
+    protected
+
+    def merge_input(type, with_wizard = false)
+      fail ArgumentError.new(":default and :required lack type: #{type}") unless
+        default.key?(type) && required.key?(type)
+
+      fail ArgumentError.new('Inputs :default, :required, and :user must be hashes!') unless
+        default[type].is_a?(Hash) && input.is_a?(Hash) && required[type].is_a?(Hash)
+
+      type_default = default[type].deep_merge(input)
+
+      errors = []
+      collect = {}
+
+      required[type].each do |k|
+        unless input.key?(k)
+          errors << k
+          collect[k] = required[k]
+        end
+      end
+
+      if with_wizard
+        ConfigWizard.fetch_input(type_default, required[type], warnings)
+      else
+        fail ArgumentError.new("User input required: #{errors}") if
+          errors.length
+      end
+
+      type_default
+    end
+  end
+end

data/lib/ridoku/deploy.rb

@@ -0,0 +1,157 @@
+#
+# Command: deploy
+#
+
+require 'ridoku/base'
+
+module Ridoku
+  register :deploy
+
+  class Deploy < Base
+    attr_accessor :app
+
+    def run
+      clist = Base.config[:command]
+      command = clist.shift
+      sub_command = clist.shift
+
+      case sub_command
+      when 'to', nil
+        deploy
+      when 'rollback'
+        rollback
+      when 'restart'
+        restart
+      when 'info'
+        info
+      else
+        print_deploy_help
+      end
+    end
+
+    protected
+
+    def rollback
+      Base.fetch_instance
+      Base.fetch_app
+
+      Base.instances.select! { |inst| inst[:status] == 'online' }
+      instance_ids = Base.instances.map { |inst| inst[:instance_id] }
+
+      # :app_source => {
+      #   :type => "git",
+      #   :url => "git@github.com:Protobit/Survly.git",
+      #   :ssh_key => "*****FILTERED*****",
+      #   :revision => "opsworks-staging"
+      # }
+
+      $stdout.puts "Application:"
+      $stdout.puts "  #{$stdout.colorize(Base.app[:name], :bold)}"
+      $stdout.puts "Rolling back on #{Base.instances.length} instance(s):"
+      Base.pretty_instances($stdout).each do |inst|
+        $stdout.puts "  #{inst}"
+      end
+      $stdout.puts "Repository:"
+      $stdout.puts "  #{$stdout.colorize(Base.app[:app_source][:url], :bold)} " +
+        "@ #{$stdout.colorize(Base.app[:app_source][:revision], :bold)}"
+
+      command = Base.rollback(Base.app[:app_id], instance_ids,
+        Base.config[:comment], {
+          opsworks_custom_cookbooks: {
+            recipes: [
+              "workers::rollback"
+            ]
+          }
+        }
+      )
+
+      Base.run_command(command)
+    end
+
+    def deploy
+      custom_json = {}.tap do |json|
+        app = Base.config[:app]
+
+        json[:deploy] = {}
+        json[:deploy][app] = {}
+        json[:deploy][app][:action] = 'force_deploy' if Base.config[:force]
+        json[:deploy][app][:migrate] = true if Base.config[:migrate]
+
+        json[:opsworks_custom_cookbooks] = {
+          recipes: [
+            "workers::deploy"
+          ]
+        }
+      end
+
+      if Base.config[:force]
+        begin
+          $stdout.puts 'Hold your seats: force deploying 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", :green)
+          exit 1
+        end
+      end
+
+      $stdout.puts 'Database will be migrated.' if Base.config[:migrate]
+      Base.standard_deploy(:all, custom_json)
+    end
+
+    def restart
+      Base.fetch_app
+
+      Base.config[:layers] = 'rails-app'
+      custom_json = {}.tap do |json|
+        json[Base.app[:name]] = {
+          deploy: {
+            application_type: 'rails'
+          }
+        }
+      end
+
+      $stdout.puts "Restarting application: #{Base.app[:name]}."
+      Ridoku::Cook.cook_recipe('unicorn::force-restart', custom_json)
+    end
+
+    def info
+      Base.fetch_instance('rails-app') unless Base.instances
+
+      Base.instances = Base.instances.select do |inst|
+         inst[:status] == 'online'
+      end
+
+      $stdout.puts 'Instances which will be deployed:'
+      $stdout.puts Base.pretty_instances($stdout)
+    end
+
+    def print_deploy_help      
+      $stderr.puts <<-EOF
+  Command: deploy
+
+  Deploy the specified application:
+    deploy            deploy the given application to stack instances
+      --layers/-l <layers,...>:
+        used to specify which layers in the stack to deploy to if not
+        specified, all online stack instances are deployed
+      --practice/-p:
+        print what would be done, but don't actually do it
+    deploy:rollback   rollback the most recently deployed application.
+        NOTE: This will not rollback environment, database, or domain changes.
+              It will only rollback source code changes.  Configurations will
+              remain the same.
+    deploy:restart   Force Restart the unicorn servers (service will go down).
+
+  examples:
+    $ deploy
+    Application:
+      test-app
+    Deploying to 2 instances:
+      mukujara, hinoenma
+    Using git repository:
+      git@github.com:ridoku/example-app.git @ master
+    EOF
+    end
+  end
+end