require 'term/ansicolor'
require 'spruz/xt'

class ::String
  include Term::ANSIColor
end

class ::File
  include Utils::FileXt
end

class Utils::Finder
  include Utils::Find
  include Utils::Patterns

  def initialize(opts = {})
    @args    = opts[:args] || {}
    @roots   = opts[:roots] || []
    pattern_opts = opts.subhash(:pattern) | {
      :cset  => @args['a'],
      :icase => @args['i'],
    }
    @pattern = @args['r'] ?
      RegexpPattern.new(pattern_opts) :
      FuzzyPattern.new(pattern_opts)
    @directory = @args['d']
    @only_directory = @args['D']
    @pathes  = []
  end

  attr_reader :pathes

  attr_reader :output

  def attempt_match?(path)
    stat = File.stat(path)
    stat.symlink? and stat = File.lstat(path)
    if @only_directory
      stat.directory?
    elsif @directory
      stat.directory? || stat.file? && (stat.size == 0 || File.ascii?(path))
    else
      stat.file? && (stat.size == 0 || File.ascii?(path))
    end
  rescue SystemCallError => e
    warn "Caught #{e.class}: #{e}"
    nil
  end

  def search
    pathes = []
    find(*@roots) { |file| pathes << file }
    pathes.uniq!
    pathes.map! { |p| a = File.split(p) ; a.unshift(p) ; a }
    @suffix = @args['I']
    pathes = pathes.map! do |path, dir, file|
      @suffix && @suffix != File.extname(file)[1..-1] and next
      if do_match = attempt_match?(path) and $DEBUG
        warn "Attempt match of #{path.inspect}"
      end
      if do_match and match = @pattern.match(file)
        if FuzzyPattern === @pattern
          current = 0
          marked_file = ''
          score, e = 0, 0
          for i in 1...(match.size)
            match[i] or next
            b = match.begin(i)
            marked_file << file[current...b]
            marked_file << file[b, 1].red
            score += (b - e)
            e = match.end(i)
            current = b + 1
          end
          marked_file << match.post_match
          [ score, file.size, path, File.join(dir, marked_file) ]
        else
          marked_file = file[0...match.begin(0)] <<
            file[match.begin(0)...match.end(0)].red <<
            file[match.end(0)..-1]
          [ 0, file.size, path, File.join(dir, marked_file) ]
        end
      end
    end
    pathes.compact!
    @pathes, @output = pathes.sort.transpose.values_at(-2, -1)
    if !@args['e'] && @output && !@output.empty?
      yield @output if block_given?
    end
    self
  end
end