# If the config/spring.rb contains requires for commands from other gems,
# then we need to be under bundler.
require "bundler/setup"

module Spring
  @commands = {}

  class << self
    attr_reader :commands
  end

  def self.register_command(name, klass)
    commands[name] = klass
  end

  def self.command?(name)
    commands.include? name
  end

  def self.command(name)
    commands.fetch name
  end

  module Commands
    class Command
      @preloads = []

      def self.preloads
        @preloads ||= superclass.preloads.dup
      end

      def self.preloads=(val)
        @preloads = val
      end

      def preloads
        self.class.preloads
      end

      def setup
        preload_files
      end

      private

      def preload_files
        preloads.each do |file|
          begin
            require file
          rescue LoadError => e
            $stderr.puts <<-MESSAGE
The #{self.class} command tried to preload #{file} but could not find it.
You can configure what to preload in your `config/spring.rb` with:
  #{self.class}.preloads = %w(files to preload)
MESSAGE
          end
        end
      end
    end

    class TestUnit < Command
      preloads << "test_helper"

      def env(*)
        "test"
      end

      def setup
        $LOAD_PATH.unshift "test"
        super
      end

      def call(args)
        if args.size > 0
          ARGV.replace args
          path = File.expand_path(args.first)

          if File.directory?(path)
            Dir[File.join path, "**", "*_test.rb"].each { |f| require f }
          else
            require path
          end
        else
          $stderr.puts "you need to specify what test to run: spring test TEST_NAME"
        end
      end

      def description
        "Execute a Test::Unit test."
      end
    end
    Spring.register_command "testunit", TestUnit.new

    class RSpec < Command
      preloads << "spec_helper"

      def env(*)
        "test"
      end

      def setup
        $LOAD_PATH.unshift "spec"
        require 'rspec/core'
        ::RSpec::Core::Runner.disable_autorun!
        super
      end

      def call(args)
        $0 = "rspec"
        ::RSpec::Core::Runner.run(args)
      end

      def description
        "Execute an RSpec spec."
      end
    end
    Spring.register_command "rspec", RSpec.new

    class Cucumber < Command
      def env(*)
        "test"
      end

      def setup
        super
        require 'cucumber'
      end

      def call(args)
        ::Cucumber::Cli::Main.execute(args)
      end

      def description
        "Execute a Cucumber feature."
      end
    end
    Spring.register_command "cucumber", Cucumber.new

    class Rake < Command
      def setup
        super
        require "rake"
      end

      def call(args)
        ARGV.replace args
        ::Rake.application.run
      end

      def description
        "Run a rake task."
      end
    end
    Spring.register_command "rake", Rake.new

    class RailsConsole < Command
      def env(tail)
        tail.first if tail.first && !tail.first.index("-")
      end

      def setup
        require "rails/commands/console"
      end

      def call(args)
        ARGV.replace args
        ::Rails::Console.start(::Rails.application)
      end

      def description
        nil
      end
    end
    Spring.register_command "rails_console", RailsConsole.new

    class RailsGenerate < Command
      def setup
        super
        Rails.application.load_generators
      end

      def call(args)
        ARGV.replace args
        require "rails/commands/generate"
      end

      def description
        nil
      end
    end
    Spring.register_command "rails_generate", RailsGenerate.new

    class RailsRunner < Command
      def env(tail)
        previous_option = nil
        tail.reverse.each do |option|
          case option
          when /--environment=(\w+)/ then return $1
          when '-e' then return previous_option
          end
          previous_option = option
        end
        nil
      end

      def call(args)
        Object.const_set(:APP_PATH, Rails.root.join('config/application'))
        ARGV.replace args
        require "rails/commands/runner"
      end

      def description
        nil
      end
    end
    Spring.register_command "rails_runner", RailsRunner.new
  end

  # Load custom commands, if any.
  # needs to be at the end to allow modification of existing commands.
  config = File.expand_path("./config/spring.rb")
  require config if File.exist?(config)
end