@@ -1,18 +1,18 @@ require 'socket' module Craken - def self.determine_raketab_file + def self.determine_raketab_files if File.directory?("#{DEPLOY_PATH}/config/craken/") # Use hostname specific raketab first. - File.exists?("#{DEPLOY_PATH}/config/craken/#{HOSTNAME}_raketab") ? - "#{DEPLOY_PATH}/config/craken/#{HOSTNAME}_raketab" : "#{DEPLOY_PATH}/config/craken/raketab" + raketabs = Dir["#{DEPLOY_PATH}/config/craken/*raketab*"].partition {|f| f =~ %r[/raketab.*$] } + raketabs.last.empty? ? raketabs.first : raketabs.last.grep(/#{HOSTNAME}_raketab/) else - "#{DEPLOY_PATH}/config/raketab" - end + Dir["#{DEPLOY_PATH}/config/raketab*"] + end end HOSTNAME = Socket.gethostname.split('.').first.downcase.strip DEPLOY_PATH = ENV['deploy_path'] || RAILS_ROOT - RAKETAB_FILE = ENV['raketab_file'] || determine_raketab_file + RAKETAB_FILES = ENV['raketab_files'].split(":") rescue determine_raketab_files CRONTAB_EXE = ENV['crontab_exe'] || "/usr/bin/crontab" RAKE_EXE = ENV['rake_exe'] || (rake = `which rake`.strip and rake.empty?) ? "/usr/bin/rake" : rake RAKETAB_RAILS_ENV = ENV['raketab_rails_env'] || RAILS_ENV @@ -61,4 +61,36 @@ module Craken `#{CRONTAB_EXE} #{filename}` FileUtils.rm filename end + + def raketab(files=RAKETAB_FILES) + files.map do |file| + next unless File.exist?(file) + builder = file =~ /.(\w+)$/ ? "build_raketab_from_#{$1}" : "build_raketab" + send(builder.to_sym, file) + end.join("\n") + end + + private + def build_raketab_from_yml(file) + yml = YAML::load(ERB.new(File.read(file)).result(binding)) + yml.map do |name,tab| + format = [] + format << (tab['min'] || tab['minute'] || '*') + format << (tab['hour'] || '*') + format << (tab['day'] || '*') + format << (tab['month'] =~ /\d+/ ? tab['month'] : Date._parse(tab['month'].to_s)[:mon] || '*') + format << ((day = tab['weekday'] || tab['wday'] and day =~ /^\d+/ ? day : Date._parse(day.to_s)[:wday]) || '*') + format << tab['command'] + format.join(' ') + end.join("\n") + end + alias_method :build_raketab_from_yaml, :build_raketab_from_yml + + def build_raketab(file) + ERB.new(File.read(file)).result(binding) + end + + def method_missing(method, *args) + method.to_s =~ /^build_raketab/ ? build_raketab(*args) : super + end end lib/craken.rb @@ -6,10 +6,9 @@ namespace :craken do task :install do require 'erb' include Craken - if File.exists? RAKETAB_FILE - puts "craken:install => Using raketab file #{RAKETAB_FILE}" - #raketab = File.read RAKETAB_FILE - raketab = ERB.new(File.read(RAKETAB_FILE)).result(binding) + unless RAKETAB_FILES.empty? + files = (plural = RAKETAB_FILES.size > 1) ? RAKETAB_FILES.join(", ") : RAKETAB_FILES.first + puts "craken:install => Using raketab file#{plural ? 's' : ''} #{files}" crontab = append_tasks(load_and_strip, raketab) install crontab end tasks/craken.rake ab0d13c58f17685a1e0731c8dd8d19e9a1d404c7 Reid MacDonald reid@laruby.com http://github.com/latimes/craken/commit/f860c244259f2e9120886e2d40e9adbbf091a993 f860c244259f2e9120886e2d40e9adbbf091a993 2008-11-07T17:23:16-08:00 2008-11-07T17:23:16-08:00 added way to have multiple raketab builders and added one for YAML f7ff99ec1b7ea7cfb8eceb245b45acf33d222c1e Reid MacDonald reid@laruby.com