module Rails::Generators::Actions

def add_source(source, options={})
  log :source, source
  in_root do
    prepend_file "Gemfile", "source #{source.inspect}\n", :verbose => false
  end
end

def capify!
  log :capify, ""
  in_root { run("#{extify(:capify)} .", :verbose => false) }
end

def environment(data=nil, options={}, &block)
  sentinel = /class [a-z_:]+ < Rails::Application/i
  data = block.call if !data && block_given?
  in_root do
    if options[:env].nil?
      inject_into_file 'config/application.rb', "\n  #{data}", :after => sentinel, :verbose => false
    else
      Array.wrap(options[:env]).each do|env|
        append_file "config/environments/#{env}.rb", "\n#{data}", :verbose => false
      end
    end
  end
end

def extify(name)
  if RbConfig::CONFIG['host_os'] =~ /mswin|mingw/
    "#{name}.bat"
  else
    name
  end
end

def freeze!(args={})
  ActiveSupport::Deprecation.warn "freeze! is deprecated since your rails app now comes bundled with Rails by default, please check your Gemfile"
end

def gem(*args)
  options = args.extract_options!
  name, version = args
  # Deal with deprecated options
  { :env => :group, :only => :group,
    :lib => :require, :require_as => :require }.each do |old, new|
    next unless options[old]
    options[new] = options.delete(old)
    ActiveSupport::Deprecation.warn "#{old.inspect} option in gem is deprecated, use #{new.inspect} instead"
  end
  # Deal with deprecated source
  if source = options.delete(:source)
    ActiveSupport::Deprecation.warn ":source option in gem is deprecated, use add_source method instead"
    add_source(source)
  end
  # Set the message to be shown in logs. Uses the git repo if one is given,
  # otherwise use name (version).
  parts, message = [ name.inspect ], name
  if version ||= options.delete(:version)
    parts   << version.inspect
    message << " (#{version})"
  end
  message = options[:git] if options[:git]
  log :gemfile, message
  options.each do |option, value|
    parts << ":#{option} => #{value.inspect}"
  end
  in_root do
    append_file "Gemfile", "gem #{parts.join(", ")}\n", :verbose => false
  end
end

def generate(what, *args)
  log :generate, what
  argument = args.map {|arg| arg.to_s }.flatten.join(" ")
  in_root { run_ruby_script("script/rails generate #{what} #{argument}", :verbose => false) }
end

def git(command={})
  if command.is_a?(Symbol)
    run "git #{command}"
  else
    command.each do |command, options|
      run "git #{command} #{options}"
    end
  end
end

def initializer(filename, data=nil, &block)
  log :initializer, filename
  create_file("config/initializers/#{filename}", data, :verbose => false, &block)
end

def lib(filename, data=nil, &block)
  log :lib, filename
  create_file("lib/#{filename}", data, :verbose => false, &block)
end

def log(*args)
  if args.size == 1
    say args.first.to_s
  else
    args << (self.behavior == :invoke ? :green : :red)
    say_status *args
  end
end

def plugin(name, options)
  log :plugin, name
  if options[:git] && options[:submodule]
    options[:git] = "-b #{options[:branch]} #{options[:git]}" if options[:branch]
    in_root do
      run "git submodule add #{options[:git]} vendor/plugins/#{name}", :verbose => false
    end
  elsif options[:git] || options[:svn]
    options[:git] = "-b #{options[:branch]} #{options[:git]}"   if options[:branch]
    options[:svn] = "-r #{options[:revision]} #{options[:svn]}" if options[:revision]
    in_root do
      run_ruby_script "script/rails plugin install #{options[:svn] || options[:git]}", :verbose => false
    end
  else
    log "! no git or svn provided for #{name}. Skipping..."
  end
end

def rake(command, options={})
  log :rake, command
  env  = options[:env] || 'development'
  sudo = options[:sudo] && RbConfig::CONFIG['host_os'] !~ /mswin|mingw/ ? 'sudo ' : ''
  in_root { run("#{sudo}#{extify(:rake)} #{command} RAILS_ENV=#{env}", :verbose => false) }
end

def rakefile(filename, data=nil, &block)
  log :rakefile, filename
  create_file("lib/tasks/#{filename}", data, :verbose => false, &block)
end

def readme(path)
  say File.read(find_in_source_paths(path))
end

def route(routing_code)
  log :route, routing_code
  sentinel = /\.routes\.draw do(?:\s*\|map\|)?\s*$/
  in_root do
    inject_into_file 'config/routes.rb', "\n  #{routing_code}\n", { :after => sentinel, :verbose => false }
  end
end

def vendor(filename, data=nil, &block)
  log :vendor, filename
  create_file("vendor/#{filename}", data, :verbose => false, &block)
end