40 lines
595 B
Ruby
40 lines
595 B
Ruby
|
# Usage:
|
||
|
|
# require "find.rb"
|
||
|
|
#
|
||
|
|
# Find.find('/foo','/bar') {|f| ...}
|
||
|
|
# or
|
||
|
|
# include Find
|
||
|
|
# find('/foo','/bar') {|f| ...}
|
||
|
|
#
|
||
|
|
|
||
|
|
module Find
|
||
|
|
def find(*path)
|
||
|
|
while file = path.shift
|
||
|
|
catch(:prune) {
|
||
|
|
yield file
|
||
|
|
if File.directory? file and not File.symlink? file then
|
||
|
|
d = Dir.open(file)
|
||
|
|
begin
|
||
|
|
for f in d
|
||
|
|
next if f =~ /^\.\.?$/
|
||
|
|
if file == "/" then
|
||
|
|
f = "/" + f
|
||
|
|
else
|
||
|
|
f = file + "/" + f
|
||
|
|
end
|
||
|
|
path.unshift f
|
||
|
|
end
|
||
|
|
ensure
|
||
|
|
d.close
|
||
|
|
end
|
||
|
|
end
|
||
|
|
}
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def prune
|
||
|
|
throw :prune
|
||
|
|
end
|
||
|
|
module_function :find, :prune
|
||
|
|
end
|