Two new parser tests related to bug 6200
[lhc/web/wiklou.git] / maintenance / deleteArchivedFiles.inc
index 1880ed9..e0ac225 100644 (file)
@@ -1,61 +1,62 @@
-<?php\r
-\r
-/**\r
- * Support functions for the deleteArchivedFiles script\r
- *\r
- * @addtogroup Maintenance\r
- * @author Aaron Schulz\r
- */\r
-\r
-require_once( "$IP/includes/FileStore.php" );\r
-\r
-function DeleteArchivedFiles( $delete = false ) {\r
-\r
-       # Data should come off the master, wrapped in a transaction\r
-       $dbw = wfGetDB( DB_MASTER );\r
-       $dbw->begin();\r
-       \r
-       $transaction = new FSTransaction();\r
-       if( !FileStore::lock() ) {\r
-               wfDebug( __METHOD__.": failed to acquire file store lock, aborting\n" );\r
-               return false;\r
-       }\r
-       \r
-       $tbl_arch = $dbw->tableName( 'filearchive' );\r
-       \r
-       # Get "active" revisions from the filearchive table\r
-       echo( "Searching for and deleting archived files...\n" );\r
-       $res = $dbw->query( "SELECT fa_storage_group,fa_storage_key FROM $tbl_arch" );\r
-       while( $row = $dbw->fetchObject( $res ) ) {\r
-               $key = $row->fa_storage_key;\r
-               $group = $row->fa_storage_group;\r
-               \r
-               $store = FileStore::get( $group );\r
-               if ( $store ) {\r
-                       $path = $store->filePath( $key );\r
-                       if ( $path && file_exists($path) ) {\r
-                               $transaction->addCommit( FSTransaction::DELETE_FILE, $path );\r
-                       } else {\r
-                               echo( "Notice - file '$key' not found in group '$group'\n" );\r
-                       }\r
-               } else {\r
-                       echo( "Notice - invalid file storage group '$group'\n" );\r
-               }\r
-       }\r
-       echo( "done.\n" );\r
-       \r
-       $transaction->commit();\r
-       \r
-       \r
-       # Delete as appropriate\r
-       echo( "Deleting filearchive rows..." );\r
-       $dbw->query( "TRUNCATE TABLE $tbl_arch" );\r
-       echo( "done.\n" );\r
-       \r
-       # This bit's done\r
-       # Purge redundant text records\r
-       $dbw->commit();\r
-\r
-}\r
-\r
-?>
\ No newline at end of file
+<?php
+/**
+ * Core functions for deleteArchivedFiles.php
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
+ * @ingroup Maintenance
+ */
+
+class DeleteArchivedFilesImplementation {
+       static public function doDelete( $output, $force ) {
+               # Data should come off the master, wrapped in a transaction
+               $dbw = wfGetDB( DB_MASTER );
+               $dbw->begin();
+               $tbl_arch = $dbw->tableName( 'filearchive' );
+               $repo = RepoGroup::singleton()->getLocalRepo();
+               # Get "active" revisions from the filearchive table
+               $output->handleOutput( "Searching for and deleting archived files...\n" );
+               $res = $dbw->query( "SELECT fa_id,fa_storage_group,fa_storage_key FROM $tbl_arch" );
+               $count = 0;
+               foreach ( $res as $row ) {
+                       $key = $row->fa_storage_key;
+                       $group = $row->fa_storage_group;
+                       $id = $row->fa_id;
+                       $path = $repo->getZonePath( 'deleted' ) . '/' . $repo->getDeletedHashPath( $key ) . $key;
+                       $sha1 = substr( $key, 0, strcspn( $key, '.' ) );
+                       // Check if the file is used anywhere...
+                       $inuse = $dbw->selectField( 'oldimage', '1',
+                               array( 'oi_sha1' => $sha1,
+                               'oi_deleted & ' . File::DELETED_FILE => File::DELETED_FILE ),
+                               __METHOD__,
+                               array( 'FOR UPDATE' )
+                       );
+                       if ( $path && file_exists( $path ) && !$inuse ) {
+                               unlink( $path ); // delete
+                               $count++;
+                               $dbw->query( "DELETE FROM $tbl_arch WHERE fa_id = $id" );
+                       } else {
+                               $output->handleOutput( "Notice - file '$key' not found in group '$group'\n" );
+                               if ( $force ) {
+                                       $output->handleOutput( "Got --force, deleting DB entry\n" );
+                                       $dbw->query( "DELETE FROM $tbl_arch WHERE fa_id = $id" );
+                               }
+                       }
+               }
+               $dbw->commit();
+               $output->handleOutput( "Done! [$count file(s)]\n" );
+       }
+}
\ No newline at end of file