lib/find.rb

   1 #
   2 # find.rb: the Find module for processing all files under a given directory.
   3 #
   4
   5 #
   6 # The +Find+ module supports the top-down traversal of a set of file paths.
   7 #
   8 # For example, to total the size of all files under your home directory,
   9 # ignoring anything in a "dot" directory (e.g. $HOME/.ssh):
  10 #
  11 #   require 'find'
  12 #
  13 #   total_size = 0
  14 #
  15 #   Find.find(ENV["HOME"]) do |path|
  16 #     if FileTest.directory?(path)
  17 #       if File.basename(path)[0] == ?.
  18 #         Find.prune       # Don't look any further into this directory.
  19 #       else
  20 #         next
  21 #       end
  22 #     else
  23 #       total_size += FileTest.size(path)
  24 #     end
  25 #   end
  26 #
  27 module Find
  28
  29   #
  30   # Calls the associated block with the name of every file and directory listed
  31   # as arguments, then recursively on their subdirectories, and so on.
  32   #
  33   # See the +Find+ module documentation for an example.
  34   #
  35   def find(*paths) # :yield: path
  36     paths.collect!{|d| d.dup}
  37     while file = paths.shift
  38       catch(:prune) do
  39         yield file.dup.taint
  40         next unless File.exist? file
  41         begin
  42           if File.lstat(file).directory? then
  43             d = Dir.open(file)
  44             begin
  45               for f in d
  46                 next if f == "." or f == ".."
  47                 if File::ALT_SEPARATOR and file =~ /^(?:[\/\\]|[A-Za-z]:[\/\\]?)$/ then
  48                   f = file + f
  49                 elsif file == "/" then
  50                   f = "/" + f
  51                 else
  52                   f = File.join(file, f)
  53                 end
  54                 paths.unshift f.untaint
  55               end
  56             ensure
  57               d.close
  58             end
  59           end
  60         rescue Errno::ENOENT, Errno::EACCES
  61         end
  62       end
  63     end
  64   end
  65
  66   #
  67   # Skips the current file or directory, restarting the loop with the next
  68   # entry. If the current file is a directory, that directory will not be
  69   # recursively entered. Meaningful only within the block associated with
  70   # Find::find.
  71   #
  72   # See the +Find+ module documentation for an example.
  73   #
  74   def prune
  75     throw :prune
  76   end
  77
  78   module_function :find, :prune
  79 end