From: Tom Needham <needham.thomas@gmail.com>
Date: Tue, 13 Mar 2012 17:18:42 +0000 (+0000)
Subject: added cleanUp() method to OC_Migrate
X-Git-Tag: v4.0.0beta~421
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=0f3eebbbd95b75e3dfe8f9322d8fd486925ac54a;p=nextcloud-server.git

added cleanUp() method to OC_Migrate
---

diff --git a/apps/user_migrate/settings.php b/apps/user_migrate/settings.php
index 00f46660cf5..c017ba226cf 100644
--- a/apps/user_migrate/settings.php
+++ b/apps/user_migrate/settings.php
@@ -70,8 +70,8 @@ if (isset($_POST['user_export'])) {
     readfile($filename);
     // Cleanup
     unlink($filename);
-    unlink($infofile);
     rmdir($exportdir);
+    OC_Migrate::cleanUp();
     
 } if( isset( $_POST['user_import'] ) ){
 	// TODO
diff --git a/lib/migrate.php b/lib/migrate.php
index 88c0e7cfc28..1be229d8fdd 100644
--- a/lib/migrate.php
+++ b/lib/migrate.php
@@ -438,4 +438,19 @@ class OC_Migrate{
 		
 	}
 	
+	/**
+	* @breif removes migration.db and exportinfo.json from the users data dir
+	* @return void
+	*/
+	static public function cleanUp(){
+		if( !self::$uid ){
+			OC_Log::write('migration', 'Failed to cleanup after migration', OC_Log::ERROR);
+			return false;	
+		}
+		// Remove migration.db
+		unlink(  OC::$SERVERROOT . '/data/' . self::$uid . '/migration.db' );
+		// Remove exportinfo.json
+		unlink(  OC::$SERVERROOT . '/data/' . self::$uid . '/exportinfo.json' );
+		return true;	
+	}
 }