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
  env_file_sentinel = /::Application\.configure do/
  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(options[:env]).each do |env|
        inject_into_file "config/environments/#{env}.rb", "\n  #{data}", after: env_file_sentinel, verbose: false
      end
    end
  end
end

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

def gem(*args)
  options = args.extract_options!
  name, version = args
  # 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
    str = "gem #{parts.join(", ")}"
    str = "  " + str if @in_group
    str = "\n" + str
    append_file "Gemfile", str, verbose: false
  end
end

def gem_group(*names, &block)
  name = names.map(&:inspect).join(", ")
  log :gemfile, "group #{name}"
  in_root do
    append_file "Gemfile", "\ngroup #{name} do", force: true
    @in_group = true
    instance_eval(&block)
    @in_group = false
    append_file "Gemfile", "\nend\n", force: true
  end
end

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

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

def initialize(*) # :nodoc:
  super
  @in_group = nil
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 unless options.quiet?
  else
    args << (self.behavior == :invoke ? :green : :red)
    say_status(*args)
  end
end

def rake(command, options={})
  log :rake, command
  env  = options[:env] || ENV["RAILS_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)
  log File.read(find_in_source_paths(path))
end

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

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