# Watchr: Autotest like functionality
#
# gem install watchr
#
# Run me with:
#
#   $ watchr spec/watchr.rb

require 'term/ansicolor'

$c = Term::ANSIColor

def getch
  state = `stty -g`
  begin
    `stty raw -echo cbreak`
    $stdin.getc
  ensure
    `stty #{state}`
  end
end

# --------------------------------------------------
# Convenience Methods
# --------------------------------------------------
def all_feature_files
  Dir['features/*.feature']
end

def all_spec_files
  files = Dir['spec/**/*_spec\.rb']
end

def run(cmd)

  pid = fork do
    puts "\n"
    print $c.cyan, cmd, $c.clear, "\n"
    exec(cmd)
  end
  Signal.trap('INT') do
    puts "sending KILL to pid: #{pid}"
    Process.kill("KILL", pid)
  end
  Process.waitpid(pid)

  prompt
end

def run_all
  run_all_specs
  run_default_cucumber
end

# allow cucumber rerun.txt smarts
def run_default_cucumber
  cmd = "cucumber"
  run(cmd)
end

def run_all_features
  cmd = "cucumber #{all_feature_files.join(' ')}"
  run(cmd)
end

def run_feature(feature)
  cmd = "cucumber #{feature}"
  $last_feature = feature
  run(cmd)
end

def run_last_feature
  run_feature($last_feature) if $last_feature
end

def run_default_spec
  cmd = "spec --color --format s ./spec"
  run(cmd)
end

def run_all_specs
  cmd = "spec --color --format s #{all_spec_files.join(' ')}"
  p cmd
  run(cmd)
end

def run_spec(spec)
  cmd = "spec --color --format s #{spec}"
  $last_spec = spec
  run(cmd)
end

def run_last_spec
  run_spec($last_spec) if $last_spec
end

def prompt
  puts "Ctrl-\\ for menu, Ctrl-C to quit"
end

# init
$last_feature = nil
prompt

# --------------------------------------------------
# Watchr Rules
# --------------------------------------------------
watch( '^features/(.*)\.feature'   )   { run_default_cucumber }

watch( '^bin/(.*)'   )   { run_default_cucumber }
watch( '^lib/(.*)'   )   { run_default_cucumber }

watch( '^features/step_definitions/(.*)\.rb' )   { run_default_cucumber }
watch( '^features/support/(.*)\.rb' )   { run_default_cucumber }

watch( '^spec/(.*)_spec\.rb'   )   { |m| run_spec(m[0]) }
# specify just the lib files that have specs
# TODO: This can be determined automatically from the spec file naming convention
watch( '^lib/oct.rb'   )   { run_default_spec }

# --------------------------------------------------
# Signal Handling
# --------------------------------------------------

# Ctrl-\
Signal.trap('QUIT') do

  puts "\n\nMENU: a = all , f = features  s = specs, l = last feature (#{$last_feature ? $last_feature : 'none'}), q = quit\n\n"
  c = getch
  puts c.chr
  if c.chr == "a"
    run_all
  elsif c.chr == "f"
    run_default_cucumber
  elsif c.chr == "s"
    run_all_specs
  elsif c.chr == "q"
    abort("exiting\n")
  elsif c.chr == "l"
    run_last_feature
  end

end