#!/usr/bin/env ruby

require 'rubygems'
require 'fileutils'
require 'optparse'
require 'irt/utils'

copy = IRT::Utils.copyright

options = {}

optparse = OptionParser.new do |opts|

  opts.banner = <<EOB
Interactive Ruby Tools:
    Improved irb and rails console with lots of easy and powerful tools.
Usage:
    irt [PATH] [options]
Options:
EOB

  options[:interactive_eof] = false
  opts.on( '-i', '--interactive-eof', 'Opens an interactive session at EOF') do
    options[:interactive_eof] = true
  end

  options[:irb_options] = nil
  opts.on( '-b', '--irb-options [OPTIONS]', 'Sets the irb or rails console options' ) do |opt|
    options[:irb_options] = opt
  end

  options[:rails_env] = 'development'
  opts.on( '-e', '--rails-env [ENVIRONMENT]', 'Sets the Rails Environment' ) do |env|
     options[:rails_env] = env
   end

  options[:no_rails] = false
  opts.on( '-n', '--no-rails', 'Does not autoload the Rails Environment' ) do
     options[:no_rails] = true
   end

  opts.on( '-v', '--version', 'Shows the version and exits' ) do
    puts IRT::Utils.version
    exit
  end

  opts.on( '-h', '--help', 'Display this screen' ) do
    puts copy
    puts opts
    exit
  end

end

optparse.parse!

puts copy

paths = if ARGV.empty?
          options[:interactive_eof] = true
          [ IRT::Utils.create_tmp_file ]
        else
          ARGV.map {|p| File.expand_path(p) }
        end

files = paths.map do |path|
          unless File.exists?(path)
            next if IRT::Prompter.no? %(Do you want to create the file "#{path}"?), :hint => '[<enter=y|n]', :default => 'y'
            options[:interactive_eof] = true
            dirname = File.dirname(path)
            FileUtils.mkdir_p(dirname) unless File.directory?(dirname)
            File.open(path, 'w') {|f| f.puts "\n" } # one empty line makes irb of 1.9.2 happy
          end
          File.directory?(path) ? Dir.glob(File.join(path, '**/*.irt')) : path
        end.flatten.compact

if files.empty?
  IRT::Prompter.say_notice 'No *.irt files to run'
  exit
end

cmd_format = if File.exists?('./config/environment.rb') && !options[:no_rails]
               ENV['RAILS_ENV'] = options[:rails_env]
               if File.exists?('./script/rails')
                 gemfile = File.read('Gemfile')
                 unless gemfile.match(/\bgem\b.+\birt\b/)
                   IRT::Prompter.say_warning %(The Gemfile doesn't look to include any 'gem "irt"' statement.\nIRT will probably not work until you add it!)
                   IRT::Prompter.yes?("Do you want to add irt to your Gemfile?", :hint => '[enter=y|n]', :default => 'y') do
                     File.open('Gemfile', 'a') do |f|
                       f.puts %(gem "irt")
                     end
                   end
                 end
                 'rails c %s %s %s'
               elsif File.exists?('./script/console')
                 'ruby script/console --irb="irt_rails2 %2$s" %1$s %3$s'
               end
             else
               'irt_irb %s %s'
             end

ENV['IRT_INTERACTIVE_EOF'] = options[:interactive_eof].inspect if options[:interactive_eof]

files.each do |file|
  ENV['IRT_COMMAND'] = sprintf cmd_format, options[:irb_options], file, options[:rails_env]
  unless system(ENV['IRT_COMMAND'])
    puts Dye.sgr(:clear) if Dye.color?
    exit(1)
  end
end