module Wacktrace

# File lib/wacktrace.rb, line 23
def add_to_stack(lines, &real)
  if lines.length <= 0
    return real.call
  end

  # If stack traces are being printed newest-to-oldest, we need to reverse
  # the input lines in order to have them print out readably.
  if detect_order == :recent_first
    lines.reverse!
  end

  lines = fix_duplicate_methods(lines)

  # Create a namespace for all these methods so we don't polute the global
  # namespace.
  container_class = Class.new

  raise "add to stack missing block" unless block_given?

  lines = lines.map { |line| [clean_method_name(line[0]), line[1], line[2]] }

  # Define each method in series: a calls b, b calls c, and so on.  The last
  # line does not get a method defined here.
  lines.each_cons(2) do |line_1, line_2|
    # puts "defining '#{line_1[0]}'"
    success = true
    define_stack_level(container_class, line_1, line_2[0])
  end

  last = lines.last
  define_stack_level(container_class, last, 'ending')
  container_class.define_singleton_method("ending") { real.call }
  return container_class.send(lines.first[0])
end

# File lib/wacktrace.rb, line 61
def add_to_stack_from_lyrics(lyrics, filename, &block)
  lines = lyrics.split("\n").map.with_index do |line, i|
    [line, i, filename]
  end
  add_to_stack(lines, &block)
end

# File lib/wacktrace.rb, line 167
def add_lines_to_stack(lines, &real)
  add_to_stack(lyrics_to_stack_lines(lines), &real)
end

# File lib/wacktrace.rb, line 90
def clean_method_name(string)
  # Always start the method name with a non-breaking space so ruby doesn't
  # interpret it as a constant if you started the string with a capital.
  ' ' + string
    .gsub(' ', ' ')
    .gsub(',', '，')
    .gsub('.', '․')
    .gsub(':', '：')
    .gsub("'", '’')
    .gsub("@", '＠')
    .gsub("`", '`')
    .gsub("~", '～')
    .gsub("#", '＃')
    .gsub("$", '＄')
    .gsub("%", '％')
    .gsub("^", '＾')
    .gsub("&", '&')
    .gsub("*", '＊')
    .gsub("(", '（')
    .gsub(")", '）')
    .gsub("-", '–')
    .gsub("+", '＋')
    .gsub("[", '［')
    .gsub("]", '］')
    .gsub("|", '｜')
    .gsub("/", '／')
    .gsub('\\', '＼')
    .gsub(";", ';')
    .gsub("{", '｛')
    .gsub("}", '｝')
    .gsub("\"", '＂')
    .gsub("'", '＇')
    .gsub("<", '＜')
    .gsub(">", '＞')
    # These are technically allowed at the end of a method, but I'm too lazy to
    # write logic to allow that.
    .gsub('!', '︕')
    .gsub('?', '︖')
    .gsub('=', '＝')
end

# File lib/wacktrace.rb, line 140
def define_stack_level(container_class, line, body)
  success = true

  # For some reason line numbers come out as one higher than whatever we put in here.
  line_number = line[1].to_i - 1
  method_name = line[0]
  file_name = line[2]

  method_string = "def #{method_name}\n#{body}\nend"

  begin
    container_class.instance_eval(method_string, file_name, line_number)
  rescue SyntaxError => e
    success = false
  end

  if !success || !container_class.methods.include?(method_name.to_sym)
    warn("
      Unable to create a method named:
      #{method_name}
      with body
      #{method_string}
      One of them probably contains some special character that's not yet handled."
    )
  end
end

# File lib/wacktrace.rb, line 173
def detect_order
  begin
    detect_order_raise
  rescue Exception => e
    trace_lines = e.full_message.split("\n")
    if trace_lines.first.include?("detect_order_raise")
      return :recent_first
    elsif trace_lines.last.include?("detect_order_raise")
      return :recent_last
    else
      return :undetectable
    end
  end
end

# File lib/wacktrace.rb, line 187
def detect_order_raise
  raise 'error'
end

# File lib/wacktrace.rb, line 75
def fix_duplicate_methods(lines)
  counts = {}
  lines.map { |line|
    method_name = line[0]
    if counts[method_name]
      original_method_name = method_name
      method_name = method_name + (' ' * counts[method_name])
      counts[original_method_name] += 1
    else
      counts[method_name] = 1
    end
    [method_name, line[1], line[2]]
  }
end