ROOT_PATH = File.expand_path(Dir.pwd)
ENV_PATH  = File.expand_path('config/environment',  ROOT_PATH)
BOOT_PATH = File.expand_path('config/boot',  ROOT_PATH)
APP_PATH  = File.expand_path('config/application',  ROOT_PATH)

require 'zeus'

module Zeus ; module Rails ; end ; end
Zeus.plan ||= Zeus::Rails

module Zeus
  module Rails
    class << self

      def after_fork
        reconnect_activerecord
        restart_girl_friday
      end

      def boot
        require BOOT_PATH
        require 'rails/all'
      end

      def default_bundle
        Bundler.require(:default)
      end

      def development_environment
        Bundler.require(:development)
        ::Rails.env = ENV['RAILS_ENV'] = "development"
        require APP_PATH
        ::Rails.application.require_environment!
      end

      def prerake
        require 'rake'
        load 'Rakefile'
      end

      def rake
        Rake.application.run
      end

      def generate
        begin
          require 'rails/generators'
          ::Rails.application.load_generators
        rescue LoadError # Rails 3.0 doesn't require this block to be run, but 3.2+ does
        end
        require 'rails/commands/generate'
      end

      def runner
        require 'rails/commands/runner'
      end

      def console
        require 'rails/commands/console'
        ::Rails::Console.start(::Rails.application)
      end

      def server
        require 'rails/commands/server'
        server = ::Rails::Server.new
        Dir.chdir(::Rails.application.root)
        server.start
      end

      def test_environment
        Bundler.require(:test)

        ::Rails.env = ENV['RAILS_ENV'] = 'test'
        require APP_PATH

        $rails_rake_task = 'yup' # lie to skip eager loading
        ::Rails.application.require_environment!
        $rails_rake_task = nil
        $LOAD_PATH.unshift(ROOT_PATH) unless $LOAD_PATH.include?(ROOT_PATH)

        if Dir.exist?(ROOT_PATH + "/test")
          test = File.join(ROOT_PATH, 'test')
          $LOAD_PATH.unshift(test) unless $LOAD_PATH.include?(test)
        end

        if Dir.exist?(ROOT_PATH + "/spec")
          spec = File.join(ROOT_PATH, 'spec')
          $LOAD_PATH.unshift(spec) unless $LOAD_PATH.include?(spec)
        end
      end

      def test_helper
        require 'test_helper'
      end

      def testrb
        argv = ARGV

        # try to find pattern by line using testrbl
        if defined?(Testrbl) && argv.size == 1 and argv.first =~ /^\S+:\d+$/
          file, line = argv.first.split(':')
          argv = [file, '-n', "/#{Testrbl.send(:pattern_from_file, File.readlines(file), line)}/"]
          puts "using -n '#{argv[2]}'" # let users copy/paste or adjust the pattern
        end

        runner = Test::Unit::AutoRunner.new(true)
        if runner.process_args(argv)
          exit runner.run
        else
          abort runner.options.banner + " tests..."
        end
      end

      def spec_helper
        require 'spec_helper'
      end

      def rspec
        exit RSpec::Core::Runner.run(ARGV)
      end


      private

      def restart_girl_friday
        return unless defined?(GirlFriday::WorkQueue)
        # The Actor is run in a thread, and threads don't persist post-fork.
        # We just need to restart each one in the newly-forked process.
        ObjectSpace.each_object(GirlFriday::WorkQueue) do |obj|
          obj.send(:start)
        end
      end

      def reconnect_activerecord
        ActiveRecord::Base.clear_all_connections! rescue nil
        ActiveRecord::Base.establish_connection   rescue nil
      end

    end
  end
end