require 'ripper'
require 'coypond/parser'

module Coypond
  class CoypondRunner
    
    def initialize(files)
      @files = files
    end
    
    def search(options)
      final_result = {}
      each_file do |file_name, source, parser|
        final_result[file_name] = []
        classes = modules = methods = {}
        classes = search_in_collection(parser.classes, options) if options[:class] || options[:all]
        modules = search_in_collection(parser.modules, options) if options[:module] || options[:all]
        methods = search_in_collection(parser.methods, options) if options[:method] || options[:all]
        result = classes.merge(modules.merge(methods))
        result.each do |dec, location|
          context = source.lines.to_a[location.first - 1]
          final_result[file_name] << [dec, location, context]
        end
        
        yield(file_name, final_result[file_name]) if block_given?
        
      end
      return final_result
    end
    
    private
    def search_in_collection(collection, options)
      search_term = options[:search_term]
      search_term = search_term.downcase if options[:ignore_case]
      collection.select do |k,v|
        k = k.downcase if options[:ignore_case]
        if options[:regexp]
          Regexp.new("(#{search_term})$").match(k)
        else
          k.end_with?(search_term)
        end
      end
    end
    
    def each_file
      @files.each do |file|
        source = File.read(file)
        parse_tree = Ripper::SexpBuilder.new(source).parse
        parser = Coypond::Parser.new.parse(parse_tree)
        yield(file, source, parser)
      end
    end
    
  end
end