numbata · June 13, 2019 07:41 · Jun 13, 2019
diff --git a/directory_rwalk.rb b/directory_rwalk.rb
@@ -0,0 +1,33 @@
+require 'pathname'
+
+def rwalk(path)
+  current_directory = Pathname.new(path)
+  breadcrumbs = [[current_directory, Dir.foreach(current_directory)]]
+
+  Enumerator.new do |y|
+    loop do
+      current_directory, current_iterator = breadcrumbs.last
+
+      begin
+        fname = current_iterator.next
+      rescue StopIteration
+        breadcrumbs.pop
+        break if breadcrumbs.empty?
+
+        next
+      end
+
+      next if fname == '.' || fname == '..'
+
+      fullpath = current_directory.join(fname)
+
+      if File.file?(fullpath)
+        y << fullpath
+      else
+        breadcrumbs << [fullpath, Dir.foreach(fullpath)]
+      end
+
+      next
+    end
+  end
+end