module Ag

# File lib/the_ruby_searcher.rb, line 4
def search(pattern, path, e = "ag", lno=false)
  engine = find_engine(e)
  warn(e,engine) if e != engine
  if lno
    case engine
      when "ag"
        flag = "--numbers"
      when "grep"
        flag = "-n"
      else
        flag = ""
    end
  else
    case engine
      when "ag"
        flag = "--nonumbers"
      else
        flag = ""
    end
  end
  if flag == ""
        cmd = [engine, pattern, path]
  else
    cmd = [engine, pattern, path, flag]
  end
  out = IO.popen(cmd).read
  if out != ""
    if lno
      out.split("\n").map{ |l| l.split(":") }.to_h
    else
      out.split("\n")
    end
  else
    raise SystemCallError, "Something went wrong, check the logs."
  end
end

# File lib/the_ruby_searcher.rb, line 47
def check(engine)
  cmd = ["which", engine].join(" ")
  stdout = IO.popen(cmd).read
  if stdout != "" and engines.include? engine
    return false
  else
    return true
  end
end

# File lib/the_ruby_searcher.rb, line 57
def engines
  ["ag", "ack", "grep"]
end

# File lib/the_ruby_searcher.rb, line 61
def find_engine(e)
  while check(e)
    index = engines.index(e)
    index = index + 1 rescue 0
    if index < engines.size
      e = engines[index]
    else
      raise ArgumentError, "Could not find any compatible search engines installed on system. Try installing Ag: https://github.com/ggreer/the_silver_searcher"
    end
  end
  e
end

# File lib/the_ruby_searcher.rb, line 43
def warn(e,engine)
  puts "WARN: Could not find search engine: #{e}. Using #{engine}."
end