Allow extensions to add post-update maintenance scripts

diff --git a/includes/installer/DatabaseUpdater.php b/includes/installer/DatabaseUpdater.php
index ba51beb..a394f19 100644 (file)
--- a/includes/installer/DatabaseUpdater.php
+++ b/includes/installer/DatabaseUpdater.php
@@ -190,6 +190,14 @@ abstract class DatabaseUpdater {
        public function addExtensionField( $tableName, $columnName, $sqlPath ) {
                $this->extensionUpdates[] = array( 'addField', $tableName, $columnName, $sqlPath, true );
        }
+       
+       /**
+        * Add a maintenance script to be run after the database updates are complete
+        * @param $class string Name of a Maintenance subclass
+        */
+       public function addPostDatabaseUpdateMaintenance( $class ) {
+               $this->postDatabaseUpdateMaintenance[] = $class;
+       }
 
        /**
         * Get the list of extension-defined updates