require 'darkext/io'
require 'darkext/symbol'

module Beagle
  class BeagleError < RuntimeError
    def initialize(msg)
      super(msg)
    end
  end

  def self.home
    ENV['BEAGLE_HOME']
  end

  def self.home=(home)
    ENV['BEAGLE_HOME'] = home
  end

  def self.start
    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
    system('beagled')
  end

  def self.stop
    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
    system('beagle-shutdown')
  end

  def self.query(query, verbose = false)
    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
    args = Array.new
    args << 'beagle-query'
    args << '--verbose' if verbose
    args << query
    output = DarkIO.capture_output(:stderr => false) { system(*args) }
    parser = verbose ? verbose_parser : regular_parser
    parser.call(output)
  end

  def self.running?
    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
    return false if status.include?("Could not connect")
    return true
  end

  def self.status
    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
    DarkIO.capture_output(:stderr => false) { system('beagle-info','--status') }.strip
  end

private

  def self.verbose_parser
    lambda do |output|
      lines = output.split("\n")
      2.times do
        lines.shift
        lines.pop
      end
      sections = lines.join("\n").strip.split("\n\n")
      results = sections.map do |section|
        section_lines = section.split("\n")
        section_hash = Hash.new
        section_lines.each do |line|
          if line.include?('=')
            k,v = line.split('=')
            k = k.split(':').last
            v.gsub!("'",'')
          else
            k,v = line.split(':')
          end
          section_hash[k.strip] = v.strip
        end
        section_hash
      end
      results.sort { |l,r| l['Score'] <=> r['Score'] }
    end
  end

  def self.regular_parser
    lambda do |output|
      output.split("\n").map(&:strip)
    end
  end
end