class RbFind

# File lib/rbfind.rb, line 283
def initialize path, count, params = nil, &block
  @levels = []
  @block = block

  if params then
    params = params.dup
    dl = :do_level_depth if params.delete :depth
    md = params.delete :max_depth ; @max_depth = md.to_i if md
    st = params.delete :sort ; @sort = sort_parser st
    @follow = params.delete :follow
    @error = params.delete :error
    params.empty?  or
      raise RuntimeError, "Unknown parameter(s): #{params.keys.join ','}."
  end
  @do_level = method dl||:do_level

  @start, @count = Time.now, count
  @wd = Dir.getwd
  if path then
    File.lstat path
    @wd = nil unless absolute_path? path
    @levels.push path
    walk
  else
    build_path
    scan_dir
  end
end

# File lib/rbfind.rb, line 372
def age ; @start - stat.mtime ; end

# File lib/rbfind.rb, line 386
def age_days  ; age / DAY    ; end

# File lib/rbfind.rb, line 384
def age_hours ; age / HOUR   ; end

# File lib/rbfind.rb, line 382
def age_mins  ; age / MINUTE ; end

# File lib/rbfind.rb, line 362
def arrow
  ARROW + (File.readlink @path) if stat.symlink?
end

# File lib/rbfind.rb, line 664
def binary? n = 1
  open { |file|
    loop do
      if n then
        break if n <= 0
        n -= 1
      end
      b = file.read BLOCK_SIZE
      b or break
      return true if b[ "\0"]
    end
  }
  false
end

# File lib/rbfind.rb, line 355
def broken_link?
  return unless stat.symlink?
  !File.stat @path rescue true
end

# File lib/rbfind.rb, line 366
def carrow
  r = creadlink
  ARROW + r if r
end

# File lib/rbfind.rb, line 463
def cfullpath  ; color fullpath  ; end

# File lib/rbfind.rb, line 465
def cfullpath! ; color fullpath! ; end

# File lib/rbfind.rb, line 461
def cname      ; color name      ; end

# File lib/rbfind.rb, line 467
def color arg
  col_stat arg, stat
end

# File lib/rbfind.rb, line 494
def colored arg, num
  @cols or colors col_str
  "\e[#{@cols[num]}m#{arg}\e[m"
end

# File lib/rbfind.rb, line 476
def colors str
  @cols = []
  str.scan /(.)(.)/i do
    fg, bg = $~.captures.map { |x| x.downcase.ord - ?a.ord }
    a = []
    case fg
      when 0..7 then a.push 30 + fg
    end
    a.push 1 if $1 == $1.upcase
    case bg
      when 0..7 then a.push 40 + bg
    end
    e = a.join ";"
    @cols.push e
  end
end

# File lib/rbfind.rb, line 718
def colsep *args
  csv COLON, *args
end

# File lib/rbfind.rb, line 575
def contains? name
  c = File.join @path, name
  File.exists? c
end

# File lib/rbfind.rb, line 462
def cpath      ; color path      ; end

# File lib/rbfind.rb, line 464
def cpath!     ; color path!     ; end

# File lib/rbfind.rb, line 344
def creadlink
  if stat.symlink? then
    s = File.stat @path rescue nil
    l = File.readlink @path
    (col_stat l, s)
  end
end

# File lib/rbfind.rb, line 735
def csv sep, *args
  e = args.join sep
  puts e
end

# File lib/rbfind.rb, line 332
def depth ; @levels.size ; end

# File lib/rbfind.rb, line 399
def dir? ; stat.directory? ; end

# File lib/rbfind.rb, line 324
def dirname
  d = File.dirname @fullpath
  File.basename d
end

# File lib/rbfind.rb, line 564
def empty?
  read_dir.each! { |f| return false }
  true
rescue Errno::ENOTDIR
end

# File lib/rbfind.rb, line 586
def entries
  read_dir.entries!
rescue Errno::ENOTDIR
end

# File lib/rbfind.rb, line 329
def ext         ; File.extname name ; end

# File lib/rbfind.rb, line 452
def filesize
  if block_given? then
    yield stat.size if file?
  else
    stat.size if file?
  end
end

# File lib/rbfind.rb, line 320
def fullpath  ; @fullpath    ; end

# File lib/rbfind.rb, line 322
def fullpath! ; append_slash @fullpath ; end

# File lib/rbfind.rb, line 642
def grep re, color = nil
  case color
    when /\A\d+(?:;\d+)*\z/, nil, false then
    when true then color = "31;1"  # red
    else           raise "Illegal color spec: #{color}"
  end
  lines { |l,i|
    l =~ re or next
    if color then
      l = "#$`\e[#{color}m#$&\e[m#$'"
    end
    colsep @path, i, l
  }
end

# File lib/rbfind.rb, line 552
def group
  g = stat.gid
  (Etc.getgrgid g).name rescue g.to_s
end

# File lib/rbfind.rb, line 334
def hidden?  ; name =~ /^\./ ; end

# File lib/rbfind.rb, line 628
def lines
  block_given? or return lines do end
  open { |file|
    n = 0
    file.each_line { |l|
      l.chomp!
      n += 1
      set_predefs l, n
      yield l, n
    }
    n
  }
end

# File lib/rbfind.rb, line 338
def mode ; stat.mode        ; end

# File lib/rbfind.rb, line 423
def modes
  m = stat.mode
  r = ""
  3.times {
    h = m & 07
    m >>= 3
    r.insert 0, ((h & 01).nonzero? ? "x" : "-")
    r.insert 0, ((h & 02).nonzero? ? "w" : "-")
    r.insert 0, ((h & 04).nonzero? ? "r" : "-")
  }
  if (m & 04).nonzero? then
    r[ 2] = r[ 2, 1] == "x" ? "s" : "S"
  end
  if (m & 02).nonzero? then
    r[ 5] = r[ 5, 1] == "x" ? "s" : "S"
  end
  if (m & 01).nonzero? then
    r[ 8] = r[ 8, 1] == "x" ? "t" : "T"
  end
  r
end

# File lib/rbfind.rb, line 318
def name      ; @levels.last ; end

# File lib/rbfind.rb, line 698
def novcs
  prune if %w(CVS .svn .git).include? name
end

# File lib/rbfind.rb, line 371
def now ; @start ; end

# File lib/rbfind.rb, line 260
def open *args, &block
  params = case args.last
    when Hash then args.pop
  end
  args.flatten!
  if args.any? then
    count = 0
    args.each do |path|
      f = new path, count, params, &block
      count = f.count
    end
  else
    f = new nil, 0, params, &block
    f.count
  end
end

# File lib/rbfind.rb, line 319
def path      ; @path        ; end

# File lib/rbfind.rb, line 321
def path!     ; append_slash @path     ; end

# File lib/rbfind.rb, line 690
def prune ; raise Prune ; end

# File lib/rbfind.rb, line 611
def read n = nil
  open { |o|
    if block_given? then
      while (r = o.read n) do
        yield r
      end
    else
      o.read n
    end
  }
end

# File lib/rbfind.rb, line 340
def readlink
  File.readlink @path if stat.symlink?
end

# File lib/rbfind.rb, line 740
def rename newname
  p = @path
  nb = File.basename newname
  newname == nb or raise RuntimeError,
        "#{self.class}: rename to `#{newname}' may not be a path."
  @levels.pop
  @levels.push newname
  build_path
  File.rename p, @path
  nil
end

# File lib/rbfind.rb, line 276
def run *args, &block
  open *args do |f| f.instance_eval &block end
end

# File lib/rbfind.rb, line 728
def spcsep *args
  csv " ", *args
end

# File lib/rbfind.rb, line 337
def stat ; File.lstat @path ; end

# File lib/rbfind.rb, line 404
def stype
  m = stat.mode >> 12 rescue nil
  case m
    when 001 then "p"
    when 002 then "c"
    when 004 then "d"
    when 006 then "b"
    when 010 then "-"
    when 012 then "l"
    when 014 then "s"
    when 016 then "w"
    when nil then "#"
    else          "?"
  end
end

# File lib/rbfind.rb, line 519
def suffix
  m = stat.mode >> 12 rescue nil
  case m
    when 001 then "|"
    when 002 then " "
    when 004 then "/"
    when 006 then " "
    when 010 then stat.executable? ? "*" : " "
    when 012 then "@"
    when 014 then "="
    when 016 then "%"
    else          "?"
  end
end

# File lib/rbfind.rb, line 723
def tabsep *args
  csv "\t", *args
end

# File lib/rbfind.rb, line 541
def user
  u = stat.uid
  (Etc.getpwuid u).name rescue u.to_s
end

# File lib/rbfind.rb, line 708
def vimswap?
  if name =~ /\A(\..+)?\.sw[a-z]\z/i then
    mark = read 5
    mark == "b0VIM"
  end
end

# File lib/rbfind.rb, line 335
def visible? ; not hidden?   ; end

# File lib/rbfind.rb, line 330
def without_ext ; name[ /^(.+?)(?:\.[^.]+)?$/, 1 ].to_s ; end

# File lib/rbfind.rb, line 770
def absolute_path? p
  loop do
    q = File.dirname p
    break if q == p
    p = q
  end
  p != Dir::CUR_DIR
end

# File lib/rbfind.rb, line 314
def append_slash s ; (File.directory? s) ? (File.join s, "") : s ; end

# File lib/rbfind.rb, line 779
def build_path
  @path = File.join @levels
  @fullpath = if @wd then
    File.join @wd, @path
  else
    @path
  end
  if @path.empty? then @path = Dir::CUR_DIR end
end

# File lib/rbfind.rb, line 819
def call_block
  set_predefs name, count
  @block.call self
end

# File lib/rbfind.rb, line 878
def col_stat arg, s
  m = s.mode if s
  code = case m && m >> 12
    when 001 then                        5
    when 002 then                        8
    when 004 then
      if (m & 0002).nonzero? then
        if (m & 01000).nonzero? then    11
        else                            12
        end
      else                               2
      end
    when 006 then                        7
    when 010 then
      col_type or          if (m & 0111).nonzero? then
        if (m & 04000).nonzero? then     9
        elsif (m & 02000).nonzero? then 10
        else                             6
        end
      else                               0
      end
    when 012 then                        3
    when 014 then                        4
    when 016 then                       13
    when nil then                        1
    else                                14
  end
  self.class.colored arg, code
end

# File lib/rbfind.rb, line 502
def col_str
  ENV[ "RBFIND_COLORS"] || ENV[ "RBFIND_COLOURS"] || (
    env = DEFAULT_COLORS.dup
    els = ENV[ "LSCOLORS"]
    if els then
      env[ 2*2, els.length] = els
    end
    env
  )
end

# File lib/rbfind.rb, line 909
def col_type
  # Overwrite this to define custom colors
  # Example:
  #   case ext
  #     when ".png", /\.jpe?g$/, /\.tiff?$/ then 15
  #     when /\.tar\.(gz|bz2)$/             then 16
  #   end
end

# File lib/rbfind.rb, line 796
def do_level
  begin
    call_block
  rescue Prune
    return
  end
  scan_dir
end

# File lib/rbfind.rb, line 805
def do_level_depth
  begin
    path, fullpath = @path, @fullpath
    scan_dir
  ensure
    @path, @fullpath = path, fullpath
  end
  begin
    call_block
  rescue Prune
    raise RuntimeError, "#{self.class}: prune doesn't work with :depth."
  end
end

# File lib/rbfind.rb, line 919
def find *args
  raise NotImplementedError, "This is not the standard Find."
end

# File lib/rbfind.rb, line 865
def handle_error err = nil
  yield
rescue err||StandardError
  if @error.respond_to? :call then
    @error.call
  elsif @error then
    instance_eval @error
  else
    raise
  end
  nil
end

# File lib/rbfind.rb, line 391
def method_missing sym, *args, &block
  stat.send sym, *args, &block
rescue NoMethodError
  super
end

# File lib/rbfind.rb, line 830
def read_dir
  handle_error Errno::EACCES do
    Dir.new @path
  end
end

# File lib/rbfind.rb, line 836
def scan_dir
  return unless File.directory? @path
  if File.symlink? @path then
    return unless @follow and handle_error do
      d = @path
      while d != Dir::CUR_DIR do
        d, = File.split d
        raise "circular recursion in #@path" if File.identical? d, @path
      end
      true
    end
  end
  dir = (read_dir or return).entries!
  if @sort.respond_to? :call then
    dir = dir.sort_by &@sort
  elsif @sort and @sort.nonzero? then
    dir.sort!
    dir.reverse! if @sort < 0
  end
  dir.each { |f|
    begin
      @levels.push f
      walk
    ensure
      @levels.pop
    end
  }
end

# File lib/rbfind.rb, line 824
def set_predefs l, n
  b = @block.binding
  b.local_variable_set "_", [ l, n]
  b.eval "$_, $. = *_"
end

# File lib/rbfind.rb, line 754
def sort_parser st
  case st
    when Proc    then st
    when Numeric then st
    when true    then +1
    when false   then  0
    when nil     then +1
    else
      case st.to_s
        when "^", "reverse", /^desc/, /^-/ then -1
        when "unsorted", "*"               then  0
        else                                    +1
      end
  end
end

# File lib/rbfind.rb, line 789
def walk
  return if @max_depth and depth > @max_depth
  build_path
  @count += 1
  @do_level.call
end