Remove-on-destroy option for simgear::Dir, to help with cleaning up temporary directo...

diff --git a/simgear/misc/sg_dir.cxx b/simgear/misc/sg_dir.cxx
index 95f9949d9893e55800f69c706c27a5d9cb5caef2..8922b030e0d038ca506a46c85b094532f6173284 100644 (file)
--- a/simgear/misc/sg_dir.cxx
+++ b/simgear/misc/sg_dir.cxx
@@ -47,22 +47,37 @@ using std::string;
 namespace simgear
 {
 
-Dir::Dir()
+Dir::Dir() :
+    _removeOnDestroy(false)
 {
 }
 
 Dir::Dir(const SGPath& path) :
-  _path(path)
+  _path(path),
+  _removeOnDestroy(false)
 {
     _path.set_cached(false); // disable caching, so create/remove work
 }
 
 Dir::Dir(const Dir& rel, const SGPath& relPath) :
-  _path(rel.file(relPath.str()))
+  _path(rel.file(relPath.str())),
+  _removeOnDestroy(false)
 {
     _path.set_cached(false); // disable caching, so create/remove work
 }
 
+Dir::~Dir()
+{
+    if (_removeOnDestroy) {
+        remove(true);
+    }
+}
+
+void Dir::setRemoveOnDestroy()
+{
+    _removeOnDestroy = true;
+}
+
 Dir Dir::current()
 {
 #ifdef _WIN32
@@ -79,7 +94,7 @@ Dir Dir::tempDir(const std::string& templ)
 {
 #ifdef HAVE_MKDTEMP
     char buf[1024];
-    char* tempPath = ::getenv("TMPDIR");
+    const char* tempPath = ::getenv("TMPDIR");
     if (!tempPath) {
         tempPath = "/tmp/";
     }

diff --git a/simgear/misc/sg_dir.hxx b/simgear/misc/sg_dir.hxx
index 7befbc8cd99158b04e2aea2cad5457feafb67a05..109a0d4b0d846b1a51732683d2aec1f6882cfbf5 100644 (file)
--- a/simgear/misc/sg_dir.hxx
+++ b/simgear/misc/sg_dir.hxx
@@ -43,7 +43,15 @@ namespace simgear
   {
   public:
     Dir();
-      
+    ~Dir();
+    
+    /**
+     * when this directory object is destroyed, remove the corresponding
+     * diretory (and its contents) from the disk. Often used with temporary
+     * directories to ensure they are cleaned up.
+     */
+    void setRemoveOnDestroy();
+    
     static Dir current();
     
     /**
@@ -93,6 +101,7 @@ namespace simgear
     Dir parent() const;
   private:
     mutable SGPath _path;
+    bool _removeOnDestroy;
   };
 } // of namespace simgear