ors 0.2.10 → 0.3.0

data/lib/ors/commands/migrate.rb

@@ -1,13 +1,12 @@
-module ORS::Commands
+class ORS
+  module Commands
+    class Migrate < Base
 
-  class Migrate < Base
-
-    def execute
-      info "migrating #{name} #{environment}..."
-
-      run_migrations migration_server
-    end
+      def execute
+        info "migrating #{ORS.config[:name]} #{ORS.config[:environment]}..."
 
+        run_migrations ORS.config[:migration_server]
+      end
+    end # Migrate < Base
   end
-
 end

data/lib/ors/commands/restart.rb

@@ -1,13 +1,11 @@
-module ORS::Commands
-
-  class Restart < Base
-
-    def execute
-      info "restarting #{name} #{environment}..."
-
-      execute_in_parallel(app_servers) {|server| restart_server server }
-    end
-
+class ORS
+  module Commands
+    class Restart < Base
+      def execute
+        info "restarting #{ORS.config[:name]} #{ORS.config[:environment]}..."
+
+        execute_in_parallel(ORS.config[:app_servers]) {|server| restart_server server }
+      end
+    end # Restart < Base
   end
-
 end

data/lib/ors/commands/ruby.rb

@@ -1,8 +1,10 @@
-module ORS::Commands
-  class Ruby < Base
-    def execute
-      info "setting up ruby/rubygems for #{name} #{environment}..."
-      execute_in_parallel(ruby_servers) {|server| setup_ruby server }
-    end
+class ORS
+  module Commands
+    class Ruby < Base
+      def execute
+        info "setting up ruby/rubygems for #{ORS.config[:name]} #{ORS.config[:environment]}..."
+        execute_in_parallel(ORS.config[:ruby_servers]) {|server| setup_ruby server }
+      end
+    end # Ruby < Base
   end
 end

data/lib/ors/commands/runner.rb

@@ -1,33 +1,37 @@
-module ORS::Commands
-  class Runner < Base
-    def execute
-      begin
-        code = ORS::Config.options[ ORS::Config.options.index{|e| e =~ /^(-c|--code)$/} + 1 ]
-        raise if code.strip.empty?
-      rescue
-        fatal "ERROR: Missing option --code 'ruby code'."
+class ORS
+  module Commands
+    class Runner < Base
+      def setup
+        @code = ORS.config[:args].shift
       end
-      results = execute_command console_server, prepare_environment,
-                                                %(if [ -f script/rails ]; then bundle exec rails runner -e #{environment} \\"#{code}\\"; else ./script/runner -e #{environment} \\"#{code}\\"; fi),
-                                                :capture => true, :quiet_ssh => true
-      results.sub!(/\AUsing BufferedLogger due to exception: .*?\n/, '') # The central_logger gem spits this out without any way of shutting it up
-      puts results
-    end
 
-    def help
-      puts <<-END
-Usage: ./ors runner [environment=production] --code "ruby code here" [options]
+      def execute
+        # need code to run
+        fatal "ERROR: Missing 'ruby code'." if @code.nil?
 
-=== Description
+        # get results
+        results = execute_command(ORS.config[:console_server],
+                                  prepare_environment,
+                                  %(if [ -f script/rails ]; then bundle exec rails runner -e #{ORS.config[:environment]} \\"#{@code}\\"; else ./script/runner -e #{ORS.config[:environment]} \\"#{@code}\\"; fi),
+                                  :capture => true, :quiet_ssh => true)
+
+        # The central_logger gem spits this out
+        results.sub!(/\AUsing BufferedLogger due to exception: .*?\n/, '')
+
+        puts results
+      end
+
+      def usage
+        "./ors runner 'ruby code here' [options]"
+      end
+
+      def description
+        <<-END
 Runs rails runner returning the result on STDOUT.
 The ruby code will be transmitted within several nested quotes:  '"\"ruby code here\""'
 Keep that in mind if you need to use quotes.
-
-=== Options
---code       (or -c)   The code to run.
---pretend    (or -p)   Don't execute anything, just show me what you're going to do
---no-gateway (or -ng)  Don't use a gateway (if you're inside the firewall)
-      END
-    end
+        END
+      end
+    end # Runner < Base
   end
 end

data/lib/ors/commands/setup.rb

@@ -1,23 +1,23 @@
-module ORS::Commands
+class ORS
+  module Commands
+    class Setup < Base
 
-  class Setup < Base
+      def execute
+        info "setting up #{ORS.config[:name]} #{ORS.config[:environment]}..."
 
-    def execute
-      info "setting up #{name} #{environment}..."
+        info "Are you sure? ('yes' + ctrl+D^2)"
+        if ORS.config[:pretending] || STDIN.read == "yes"
+          execute_in_parallel(ORS.config[:all_servers]) {|server| setup_repo server }
 
-      info "Are you sure? ('yes' + ctrl+D^2)"
-      if pretending || STDIN.read == "yes"
-        execute_in_parallel(all_servers) {|server| setup_repo server }
+          Ruby.run_without_setup
 
-        run Ruby
-
-        execute_command migration_server, prepare_environment,
-                                          %(RAILS_ENV=#{environment} bundle exec rake db:create)
-      else
-        info "Setup aborted."
+          execute_command(ORS.config[:migration_server],
+                          prepare_environment,
+                          %(RAILS_ENV=#{environment} bundle exec rake db:create))
+        else
+          info "Setup aborted."
+        end
       end
-    end
-
+    end # Setup < Base
   end
-
 end

data/lib/ors/commands/start.rb

@@ -1,13 +1,11 @@
-module ORS::Commands
-
-  class Start < Base
-
-    def execute
-      info "starting #{name} #{environment}..."
-
-      execute_in_parallel(app_servers) {|server| start_server server }
-    end
-
+class ORS
+  module Commands
+    class Start < Base
+      def execute
+        info "starting #{ORS.config[:name]} #{ORS.config[:environment]}..."
+
+        execute_in_parallel(ORS.config[:app_servers]) {|server| start_server server }
+      end
+    end # Start < Base
   end
-
 end

data/lib/ors/commands/stop.rb

@@ -1,13 +1,11 @@
-module ORS::Commands
-
-  class Stop < Base
-
-    def execute
-      info "stopping #{name} #{environment}..."
-
-      execute_in_parallel(app_servers) {|server| stop_server server }
-    end
-
+class ORS
+  module Commands
+    class Stop < Base
+      def execute
+        info "stopping #{ORS.config[:name]} #{ORS.config[:environment]}..."
+
+        execute_in_parallel(ORS.config[:app_servers]) {|server| stop_server server }
+      end
+    end # Stop < Base
   end
-
 end

data/lib/ors/commands/symlink.rb

@@ -0,0 +1,31 @@
+class ORS
+  module Commands
+    class Symlink < Base
+
+      def setup
+        parse_remote_and_or_branch
+      end
+
+      def execute
+        info "symlinking #{ORS.config[:name]} #{ORS.config[:environment]}..."
+
+        execute_in_parallel(ORS.config[:ruby_servers]) do |server|
+          info "[#{server}] symlinking..."
+          execute_command(server,
+                          prepare_environment,
+                          #%(if [ -d config/deploy ]; then cd config/deploy; for i in ./**/*; do if [ -f \\\$i ]; then echo \\\$i; fi; done; cd ../../; fi))
+                          %(if [ -d config/deploy ]; then cd config/deploy; for i in ./**/*; do if [ -f \\\$i ]; then ln -nfs #{ORS.config[:deploy_directory]}/config/deploy/\\\$i ../../\\\$i; fi; done; cd ../../; fi))
+
+        end
+      end
+
+      def usage
+        "./ors symlink [remote|remote/branch] [options]"
+      end
+
+      def description
+        "Symlinks files in config/deploy/ in given branch (origin/environment by default) to given environment"
+      end
+    end # Deploy < Base
+  end
+end

data/lib/ors/commands/timestamps.rb

@@ -0,0 +1,27 @@
+class ORS
+  module Commands
+
+    # ors timestamps
+    # ors timestamps from staging
+    class Timestamps < Base
+      def execute
+        timestamps = ORS.config[:app_servers].map do |server|
+          [
+           "[#{server}] ",
+           execute_command(server, prepare_environment, %(cat restart.timestamp), :capture => true)
+          ].join
+        end.join("\n")
+
+        puts timestamps unless ORS.config[:pretending]
+      end
+
+      def usage
+        "./ors check [options]"
+      end
+
+      def description
+        "Prints out contents of restart.timestamp on the app servers"
+      end
+    end # Timestamps < Base
+  end
+end

data/lib/ors/commands/update.rb

@@ -1,17 +1,21 @@
-module ORS::Commands
+class ORS
+  module Commands
+    class Update < Base
 
-  class Update < Base
+      def setup
+        parse_remote_and_or_branch
+      end
 
-    def execute
-      info "updating #{name} #{environment}..."
+      def execute
+        info "updating #{ORS.config[:name]} #{ORS.config[:environment]}..."
 
-      execute_in_parallel(all_servers) {|server| update_code server }
-      execute_in_parallel(ruby_servers) {|server| bundle_install server }
-
-      execute_command cron_server, prepare_environment,
-                                   %(if [ -f config/schedule.rb ]; then bundle exec whenever --update-crontab --set environment=#{environment} -i #{name}_#{environment}; fi)
-    end
+        execute_in_parallel(ORS.config[:all_servers]) {|server| update_code(server) }
+        execute_in_parallel(ORS.config[:ruby_servers]) {|server| bundle_install(server) }
 
+        execute_command(ORS.config[:cron_server],
+                        prepare_environment,
+                        %(if [ -f config/schedule.rb ]; then bundle exec whenever --update-crontab --set environment=#{ORS.config[:environment]} -i #{ORS.config[:name]}_#{ORS.config[:environment]}; fi))
+      end
+    end # Update < Base
   end
-
 end

data/lib/ors/config.rb

@@ -1,90 +1,120 @@
-module ORS
-  module Config
+class ORS
+  class Config
 
-    CONFIG_FILENAME="config/deploy.yml"
+    CONFIG_FILENAME = "config/deploy.yml"
 
-    mattr_accessor :name, :environment, :use_gateway, :pretending, :log_lines, :rails2, :deploy_hook, :remote_deploy_hook
-    mattr_accessor :gateway, :deploy_user, :repo, :base_path, :web_servers, :app_servers, :migration_server, :console_server, :cron_server
-    mattr_accessor :options
+    @@args = []
+    @@options = {}
 
-    self.environment = "production"
-    self.pretending = false
-    self.use_gateway = true
-    self.log_lines = 100
+    def initialize(options)
+      parse_options(options)
+      parse_config_file
+    end
 
-    module ModuleMethods
+    def _options
+      @@options
+    end
 
-      def parse_options options
-        self.name = name_from_git
-        self.environment = options.shift unless options.empty? or options.first.match(/^-/)
-        self.options = options.dup
+    def [](key)
+      @@options[key]
+    end
 
-        options.each do |option|
-          case option
-            when "-p", "--pretend" then self.pretending = true
-            when "-ng", "--no-gateway" then self.use_gateway = false
-          end
-        end
-      end
+    def []=(key, value)
+      @@options[key] = value
+    end
 
-      def parse_config_file
-        if File.exists?(CONFIG_FILENAME)
-          YAML.load(File.read(CONFIG_FILENAME)).each {|(name, value)| send "#{name}=", value }
-        else
-          self.gateway          = "deploy-gateway"
-          self.deploy_user      = "deployer"
-          self.repo             = "ors_git"
-          self.base_path        = "/var/www"
-          self.web_servers      = %w(koala)
-          self.app_servers      = %w(eel jellyfish squid)
-          self.migration_server = "tuna"
-          self.console_server   = "tuna"
-          self.cron_server      = "tuna"
-        end
-      end
+    def parse_options(options)
+      @@options[:pretending] = true if options.delete("-p") || options.delete("--pretend")
 
-      def valid_options?
-        name.to_s.size > 0 and valid_environments.include?(environment)
+      if options.delete("-ng") || options.delete("--no-gateway")
+        @@options[:use_gateway] = false
+      else
+        @@options[:use_gateway] = true
       end
 
-      def valid_environments
-        %w(production demo staging)
+      # grab environment
+      index = options.index("to") || options.index("from")
+      unless index.nil?
+        @@options[:environment] = options.delete_at(index + 1)
+        options.delete_at(index)
       end
 
-      def git
-        @git ||= Git.open(Dir.pwd)
-      end
+      @@options[:args] = options.dup
 
-      private
+      set_default_options
+    end
+
+    def set_default_options
+      @@options[:environment]    ||= "production"
+      @@options[:remote]         ||= "origin"
+      @@options[:branch]         ||= @@options[:environment]
+      @@options[:pretending]     ||= false
+      @@options[:log_lines]        = 100
+
+      @@options[:gateway]          = "deploy-gateway"
+      @@options[:user]             = "deployer"
+      @@options[:base_path]        = "/var/www"
+      @@options[:web_servers]      = %w(koala)
+      @@options[:app_servers]      = %w(eel jellyfish squid)
+      @@options[:migration_server] = "tuna"
+      @@options[:console_server]   = "tuna"
+      @@options[:cron_server]      = "tuna"
+    end
 
-      def name_from_git
-        git.config["remote.origin.url"].gsub(/^[\w]*(@|:\/\/)[^\/:]*(\/|:)([a-zA-Z0-9\/]*)(.git)?$/i, '\3')
+    def parse_config_file
+      if File.exists?(CONFIG_FILENAME)
+        YAML.load(File.read(CONFIG_FILENAME)).each do |(name, value)|
+          @@options[name.to_sym] = value
+        end
       end
+    end
+
+    def finalize!
+      @@options[:name] = name
+      @@options[:remote_url] = remote_url
+      @@options[:deploy_directory] = deploy_directory
+
+      @@options[:revision] = git.log(1).first.sha
+
+      @@options[:ruby_servers] = [@@options[:app_servers], @@options[:console_server], @@options[:cron_server], @@options[:migration_server]].flatten.compact.uniq
+      @@options[:all_servers] = [@@options[:web_servers], @@options[:ruby_servers]].flatten.compact.uniq
+    end
 
+    def git
+      @git ||= Git.open(Dir.pwd)
     end
-    extend ModuleMethods
 
-    def ruby_servers
-      (app_servers + [console_server, cron_server, migration_server]).uniq
+    def valid?
+      name.to_s.size > 0
     end
 
-    def all_servers
-      (web_servers + ruby_servers).uniq
+
+    private
+
+    #
+    # methods to help finalize the config
+    #
+
+    def remote_url
+      if git.config.has_key?("remote.#{@@options[:remote]}.url")
+        git.config["remote.#{@@options[:remote]}.url"]
+      else
+        raise StandardError, "There is no #{@@options[:remote]} remote in your git config, please check your config/deploy.yml"
+      end
     end
 
-    def revision
-      Config.git.log(1).first.sha
+    def name
+      remote_url.gsub(/^[\w]*(@|:\/\/)[^\/:]*(\/|:)([a-zA-Z0-9\/_\-]*)(.git)?$/i, '\3')
     end
 
     def deploy_directory
-      directory = File.join base_path, name
+      directory = File.join(@@options[:base_path], @@options[:name])
 
-      if environment == "production"
+      if @@options[:environment] == "production"
         directory
       else
-        "#{directory}_#{environment}"
+        "#{directory}_#{@@options[:environment]}"
       end
     end
-
   end
 end