1 files changed, 21 insertions, 0 deletions
diff --git a/lib/roo/tempdir.rb b/lib/roo/tempdir.rb
new file mode 100644
index 0000000..4be755b
--- /dev/null
+++ b/lib/roo/tempdir.rb
@@ -0,0 +1,21 @@
+module Roo
+  module Tempdir
+    def finalize_tempdirs(object_id)
+      if @tempdirs && (dirs_to_remove = @tempdirs[object_id])
+        @tempdirs.delete(object_id)
+        dirs_to_remove.each do |dir|
+          ::FileUtils.remove_entry(dir)
+        end
+      end
+    end
+
+    def make_tempdir(object, prefix, root)
+      root ||= ENV["ROO_TMP"]
+      # NOTE: This folder is cleaned up by finalize_tempdirs.
+      ::Dir.mktmpdir("#{Roo::TEMP_PREFIX}#{prefix}", root).tap do |tmpdir|
+        @tempdirs ||= Hash.new { |h, k| h[k] = [] }
+        @tempdirs[object.object_id] << tmpdir
+      end
+    end
+  end
+end