class Spring::Commands::RailsRunner

def call
  ARGV.replace extract_environment(ARGV).first
  super
end

def command_name
  "runner"
end

def env(args)
  extract_environment(args).last
end

def extract_environment(args)
  environment = nil
  args = args.select.with_index { |arg, i|
    case arg
    when "-e"
      false
    when /--environment=(\w+)/
      environment = $1
      false
    else
      if i > 0 && args[i - 1] == "-e"
        environment = arg
        false
      else
        true
      end
    end
  }
  [args, environment]
end