class ActionDispatch::Journey::GTG::TransitionTable

def []=(from, to, sym)
  to_mappings = states_hash_for(sym)[from] ||= {}
  to_mappings[sym] = to
end

def accepting?(state)
  @accepting[state]
end

def accepting_states
  @accepting.keys
end

def add_accepting(state)
  @accepting[state] = true
end

def add_memo(idx, memo)
  @memos[idx] << memo
end

def as_json(options = nil)
  simple_regexp = Hash.new { |h,k| h[k] = {} }
  @regexp_states.each do |from, hash|
    hash.each do |re, to|
      simple_regexp[from][re.source] = to
    end
  end
  {
    regexp_states: simple_regexp,
    string_states: @string_states,
    accepting:     @accepting
  }
end

def eclosure(t)
  Array(t)
end

def initialize
  @regexp_states = {}
  @string_states = {}
  @accepting     = {}
  @memos         = Hash.new { |h,k| h[k] = [] }
end

def memo(idx)
  @memos[idx]
end

def move(t, a)
  return [] if t.empty?
  regexps = []
  t.map { |s|
    if states = @regexp_states[s]
      regexps.concat states.map { |re, v| re === a ? v : nil }
    end
    if states = @string_states[s]
      states[a]
    end
  }.compact.concat regexps
end

def states
  ss = @string_states.keys + @string_states.values.flat_map(&:values)
  rs = @regexp_states.keys + @regexp_states.values.flat_map(&:values)
  (ss + rs).uniq
end

def states_hash_for(sym)
  case sym
  when String
    @string_states
  when Regexp
    @regexp_states
  else
    raise ArgumentError, 'unknown symbol: %s' % sym.class
  end
end

def to_svg
  svg = IO.popen('dot -Tsvg', 'w+') { |f|
    f.write(to_dot)
    f.close_write
    f.readlines
  }
  3.times { svg.shift }
  svg.join.sub(/width="[^"]*"/, '').sub(/height="[^"]*"/, '')
end

def transitions
  @string_states.flat_map { |from, hash|
    hash.map { |s, to| [from, s, to] }
  } + @regexp_states.flat_map { |from, hash|
    hash.map { |s, to| [from, s, to] }
  }
end

def visualizer(paths, title = 'FSM')
  viz_dir   = File.join File.dirname(__FILE__), '..', 'visualizer'
  fsm_js    = File.read File.join(viz_dir, 'fsm.js')
  fsm_css   = File.read File.join(viz_dir, 'fsm.css')
  erb       = File.read File.join(viz_dir, 'index.html.erb')
  states    = "function tt() { return #{to_json}; }"
  fun_routes = paths.sample(3).map do |ast|
    ast.map { |n|
      case n
      when Nodes::Symbol
        case n.left
        when ':id' then rand(100).to_s
        when ':format' then %w{ xml json }.sample
        else
          'omg'
        end
      when Nodes::Terminal then n.symbol
      else
        nil
      end
    }.compact.join
  end
  stylesheets = [fsm_css]
  svg         = to_svg
  javascripts = [states, fsm_js]
  # Annoying hack for 1.9 warnings
  fun_routes  = fun_routes
  stylesheets = stylesheets
  svg         = svg
  javascripts = javascripts
  require 'erb'
  template = ERB.new erb
  template.result(binding)
end