1 files changed, 75 insertions, 0 deletions
diff --git a/maintenance/cleanupUploadStash.php b/maintenance/cleanupUploadStash.php
new file mode 100644
index 00000000..4b47f397
--- /dev/null
+++ b/maintenance/cleanupUploadStash.php
@@ -0,0 +1,75 @@
+<?php
+/**
+ * Remove old or broken uploads from temporary uploaded file storage,
+ * clean up associated database records
+ *
+ * Copyright © 2011, Wikimedia Foundation
+ *
+ * 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
+ *
+ * @file
+ * @author Ian Baker <ibaker@wikimedia.org>
+ * @ingroup Maintenance
+ */
+
+require_once( dirname( __FILE__ ) . '/Maintenance.php' );
+
+class UploadStashCleanup extends Maintenance {
+
+	public function __construct() {
+		parent::__construct();
+		$this->mDescription = "Clean up abandoned files in temporary uploaded file stash";
+  	}
+
+  	public function execute() {
+		$repo = RepoGroup::singleton()->getLocalRepo();
+	
+		$dbr = $repo->getSlaveDb();
+		
+		$this->output( "Getting list of files to clean up...\n" );
+		$res = $dbr->select(
+			'uploadstash',
+			'us_key',
+			'us_timestamp < ' . $dbr->timestamp( time() - UploadStash::REPO_AGE * 3600 ),
+			__METHOD__
+		);
+		
+		if( !is_object( $res ) || $res->numRows() == 0 ) {
+			// nothing to do.
+			return false;
+		}
+
+		// finish the read before starting writes.
+		$keys = array();
+		foreach($res as $row) {
+			array_push( $keys, $row->us_key );
+		}
+		
+		$this->output( 'Removing ' . count($keys) . " file(s)...\n" );
+		// this could be done some other, more direct/efficient way, but using
+		// UploadStash's own methods means it's less likely to fall accidentally
+		// out-of-date someday
+		$stash = new UploadStash( $repo );
+		
+		foreach( $keys as $key ) {
+			$stash->getFile( $key, true );
+			$stash->removeFileNoAuth( $key );
+		}
+  	}
+}
+
+$maintClass = "UploadStashCleanup";
+require_once( RUN_MAINTENANCE_IF_MAIN );
+\ No newline at end of file