#!/usr/bin/env ruby

JSPEC_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..'))
$:.unshift JSPEC_ROOT
 
require 'rubygems'
require 'commander'
require 'bind'
require 'fileutils'

RHINO = 'java org.mozilla.javascript.tools.shell.Main'

program :name, 'JSpec'
program :version, '2.8.2'
program :description, 'JavaScript BDD Testing Framework'
default_command :bind

command :init do |c|
  c.syntax = 'jspec init [dest]'
  c.summary = 'Initialize a JSpec project template'
  c.description = 'Initialize a JSpec project template. Defaults to the current directory
    when [dest] is not specified. The template includes several files for 
    running via Rhino, DOM, and the JSpec Rack server.'
  c.example 'Create a directory foo, initialized with a jspec template', 'jspec init foo'
  c.option '-R', '--rails', 'Initialize rails template'
  c.when_called do |args, options|
    dest = args.shift || '.'
    if options.rails
      initialize_rails_to dest
    else
      initialize_to dest
    end
    say "Template initialized at '#{dest}'"
  end
end

command :update do |c|
  c.syntax = 'jspec update [path ...]'
  c.summary = 'Update JSpec releases'
  c.description = 'Update JSpec release in [paths], this will allow you to utilize 
    the latest JSpec features. Execute from JSpec project root without [paths] to
    update the default template spec files.'
  c.when_called do |args, options|
    if args.empty?
      if rails?
        paths = 'jspec/spec.dom.html', 'jspec/spec.rhino.js'
      else
        paths = 'spec/spec.dom.html', 'spec/spec.rhino.js'
      end
    else
      paths = args
    end
    update_version_in *paths
  end
end

command :run do |c|
  c.syntax = 'jspec run [path] [options]'
  c.summary = 'Run specifications'
  c.description = 'Run specifications, defaulting [path] to spec/spec.dom.html.
    You will need to supply [path] if your specs do not reside 
    in this location. `run --bind` is the default sub-command of
    jspec so you may simply execute `jspec` in order to bind execution
    of your specs when a file is altered.
    
    JSpec supports Rhino execution when installed. The [path] is assumed
    to be spec/spec.rhino.js unless specified. See examples below for
    using the --rhino switch.
    
    JSpec\'s server is also available via --server, which defaults 
    the [path] to spec/server.html'
  c.example 'Run once in Safari', 'jspec run'
  c.example 'Run once in Safari and Firefox', 'jspec run --browsers Safari,Firefox'
  c.example 'Run custom spec file', 'jspec run foo.html'
  c.example 'Auto-run browsers when a file is altered', 'jspec run --bind --browsers Safari,Firefox'
  c.example 'Shortcut for the previous example', 'jspec --browsers Safari,Firefox'
  c.example 'Auto-run rhino when a file is altered', 'jspec --rhino'
  c.example 'Run Rhino specs once', 'jspec run specs/something.js --rhino'
  c.option '-b', '--browsers BROWSERS', Array, 'Specify browsers to test, defaults to Safari'
  c.option '-p', '--paths PATHS', Array, 'Specify paths when binding, defaults to javascript within ./lib and ./spec'
  c.option '-B', '--bind', 'Auto-run specs when source files or specs are altered'
  c.option '-R', '--rhino', 'Run specs using Rhino'
  c.option '-S', '--server', 'Run specs using the JSpec server'
  c.option '-s', '--server-only', 'Start JSpec server without running browsers'
  c.when_called do |args, options|
    
    # Rails
    if rails?
      options.default :browsers => %w( Safari ), :paths => ['public/javascripts/**/*.js', 'jspec/**/*.js']
    else
      options.default :browsers => %w( Safari ), :paths => ['lib/**/*.js', 'spec/**/*.js']
    end
    
    # Actions
    if options.rhino
      spec = args.shift || path_to('spec.rhino.js')
      action = lambda { rhino spec }
    elsif options.server_only
      start_server options, nil
    elsif options.server
      spec = args.shift || path_to('spec.server.html')
      action = lambda { start_server options, spec }
    else
      spec = args.shift || path_to('spec.dom.html')
      action = Bind::Actions::RefreshBrowsers.new spec, *options.browsers
    end
    
    # Binding
    if options.bind
      listener = Bind::Listener.new :paths => options.paths, :interval => 1, :actions => [action], :debug => $stdout
      listener.run!
    else
      action.call File.new(spec)
    end
  end
end
alias_command :bind, :run, '--bind'

def initialize_to dest
  unless Dir[dest + '/*'].empty?
    abort unless agree "'#{dest}' is not empty; continue? "
  end
  copy_template_to 'default', dest
  replace_root_in dest, 'spec/spec.dom.html', 'spec/spec.rhino.js'
end

def initialize_rails_to dest
  unless looks_like_rails_root?(dest)
    abort unless agree "'#{dest}' does not look like root of a rails project; continue? "
  end
  copy_template_to 'rails', "#{dest}/jspec"
  replace_root_in "#{dest}/jspec", 'spec.dom.html', 'spec.rhino.js'
end

def copy_template_to name, dest
  FileUtils.mkdir_p dest
  FileUtils.cp_r path_to_template(name), dest
end

def path_to_template name
  File.join JSPEC_ROOT, 'templates', name, '.'
end

def path_to file
  rails? ? "jspec/#{file}" : "spec/#{file}"
end

def rhino file
  abort "#{file} not found" unless File.exists? file
  system "#{RHINO} #{file}"
end

def start_server options, spec
  require 'server/server'
  JSpec::Server.start options, spec
end

def rails?
  File.directory? 'jspec'  
end

def replace_root_in dest, *paths
  paths.each do |path|
    path = File.join dest, path
    contents = File.read(path).gsub 'JSPEC_ROOT', JSPEC_ROOT
    File.open(path, 'w') { |file| file.write contents }
  end
end

def update_version_in *paths
  paths.each do |path|
    next unless File.exists? path
    contents = File.read(path).gsub /visionmedia-jspec-(\d+\.\d+\.\d+)/, "visionmedia-jspec-#{program(:version)}"
    File.open(path, 'r+'){ |file| file.write contents } 
    say "Updated #{path}; #{$1} -> #{program(:version)}"
  end
  say "Finished updating JSpec"
end

def looks_like_rails_root? path = '.'
  File.directory? "#{path}/vendor"
end