Since we're making other changes, let's bring in that rev_parent_id field from branch...
[lhc/web/wiklou.git] / maintenance / updaters.inc
index dbddc47..ca2da34 100644 (file)
@@ -1,14 +1,18 @@
 <?php
 /**
- * @package MediaWiki
- * @subpackage Maintenance
+ * @addtogroup Maintenance
  */
 
  /** */
 
+if ( !defined( 'MEDIAWIKI' ) ) {
+       echo "This file is not a valid entry point\n";
+       exit( 1 );
+}
+
 require_once 'convertLinks.inc';
-require_once 'InitialiseMessages.inc';
 require_once 'userDupes.inc';
+require_once 'deleteDefaultMessages.php';
 
 $wgRenamedTables = array(
 #           from             to                  patch file
@@ -30,7 +34,6 @@ $wgNewTables = array(
        array( 'langlinks',     'patch-langlinks.sql' ),
        array( 'querycache_info', 'patch-querycacheinfo.sql' ),
        array( 'filearchive',   'patch-filearchive.sql' ),
-       array( 'redirect',      'patch-redirect.sql' ),
        array( 'querycachetwo', 'patch-querycachetwo.sql' ),
 );
 
@@ -38,11 +41,11 @@ $wgNewFields = array(
 #           table            field             patch file (in maintenance/archives)
        array( 'ipblocks',      'ipb_id',           'patch-ipblocks.sql' ),
        array( 'ipblocks',      'ipb_expiry',       'patch-ipb_expiry.sql' ),
-       array( 'ipblocks',      'ipb_enable_autoblock', 'patch-ipb_optional_autoblock.sql' ),
        array( 'recentchanges', 'rc_type',          'patch-rc_type.sql' ),
        array( 'recentchanges', 'rc_ip',            'patch-rc_ip.sql' ),
        array( 'recentchanges', 'rc_id',            'patch-rc_id.sql' ),
        array( 'recentchanges', 'rc_patrolled',     'patch-rc-patrol.sql' ),
+       array( 'recentchanges', 'rc_old_len',       'patch-rc_len.sql' ),
        array( 'user',          'user_real_name',   'patch-user-realname.sql' ),
        array( 'user',          'user_token',       'patch-user_token.sql' ),
        array( 'user',          'user_email_token', 'patch-user_email_token.sql' ),
@@ -60,7 +63,13 @@ $wgNewFields = array(
        array( 'ipblocks',      'ipb_range_start',  'patch-ipb_range_start.sql' ),
        array( 'site_stats',    'ss_images',        'patch-ss_images.sql' ),
        array( 'ipblocks',      'ipb_anon_only',    'patch-ipb_anon_only.sql' ),
+       array( 'ipblocks',      'ipb_enable_autoblock', 'patch-ipb_optional_autoblock.sql' ),
        array( 'user',          'user_newpass_time','patch-user_newpass_time.sql' ),
+       array( 'user',          'user_editcount',   'patch-user_editcount.sql' ),
+       array( 'logging',       'log_id',               'patch-log_id.sql' ),
+       array( 'page_restrictions', 'pr_id',            'patch-page_restrictions_sortkey.sql' ),
+       array( 'revision',      'rev_len',              'patch-rev_len.sql' ),
+       array( 'revision',      'rev_parent_id',        'patch-rev_parent_id.sql' ),
 );
 
 function rename_table( $from, $to, $patch ) {
@@ -315,16 +324,16 @@ function do_schema_restructuring() {
                echo "...page table already exists.\n";
        } else {
                echo "...converting from cur/old to page/revision/text DB structure.\n"; flush();
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                echo "......checking for duplicate entries.\n"; flush();
 
-               extract( $wgDatabase->tableNames( 'cur', 'old', 'page', 'revision', 'text' ) );
+               list ($cur, $old, $page, $revision, $text) = $wgDatabase->tableNamesN( 'cur', 'old', 'page', 'revision', 'text' );
 
                $rows = $wgDatabase->query( "SELECT cur_title, cur_namespace, COUNT(cur_namespace) AS c
                                FROM $cur GROUP BY cur_title, cur_namespace HAVING c>1", $fname );
 
                if ( $wgDatabase->numRows( $rows ) > 0 ) {
-                       echo wfTimestamp();
+                       echo wfTimestamp( TS_DB );
                        echo "......<b>Found duplicate entries</b>\n";
                        echo ( sprintf( "<b>      %-60s %3s %5s</b>\n", 'Title', 'NS', 'Count' ) );
                        while ( $row = $wgDatabase->fetchObject( $rows ) ) {
@@ -372,18 +381,18 @@ function do_schema_restructuring() {
                        }
                        $sql = "DELETE FROM $cur WHERE cur_id IN ( " . join( ',', $deleteId ) . ')';
                        $rows = $wgDatabase->query( $sql, $fname );
-                       echo wfTimestamp();
+                       echo wfTimestamp( TS_DB );
                        echo "......<b>Deleted</b> ".$wgDatabase->affectedRows()." records.\n";
                }
 
 
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                echo "......Creating tables.\n";
                $wgDatabase->query("CREATE TABLE $page (
                        page_id int(8) unsigned NOT NULL auto_increment,
                        page_namespace int NOT NULL,
                        page_title varchar(255) binary NOT NULL,
-                       page_restrictions tinyblob NOT NULL default '',
+                       page_restrictions tinyblob NOT NULL,
                        page_counter bigint(20) unsigned NOT NULL default '0',
                        page_is_redirect tinyint(1) unsigned NOT NULL default '0',
                        page_is_new tinyint(1) unsigned NOT NULL default '0',
@@ -400,13 +409,13 @@ function do_schema_restructuring() {
                $wgDatabase->query("CREATE TABLE $revision (
                        rev_id int(8) unsigned NOT NULL auto_increment,
                        rev_page int(8) unsigned NOT NULL,
-                       rev_comment tinyblob NOT NULL default '',
+                       rev_comment tinyblob NOT NULL,
                        rev_user int(5) unsigned NOT NULL default '0',
                        rev_user_text varchar(255) binary NOT NULL default '',
                        rev_timestamp char(14) binary NOT NULL default '',
                        rev_minor_edit tinyint(1) unsigned NOT NULL default '0',
                        rev_deleted tinyint(1) unsigned NOT NULL default '0',
-
+                       rev_parent_id int(8) unsigned default NULL,
                        PRIMARY KEY rev_page_id (rev_page, rev_id),
                        UNIQUE INDEX rev_id (rev_id),
                        INDEX rev_timestamp (rev_timestamp),
@@ -415,15 +424,15 @@ function do_schema_restructuring() {
                        INDEX usertext_timestamp (rev_user_text,rev_timestamp)
                        ) TYPE=InnoDB", $fname );
 
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                echo "......Locking tables.\n";
                $wgDatabase->query( "LOCK TABLES $page WRITE, $revision WRITE, $old WRITE, $cur WRITE", $fname );
 
                $maxold = intval( $wgDatabase->selectField( 'old', 'max(old_id)', '', $fname ) );
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                echo "......maxold is {$maxold}\n";
 
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                global $wgLegacySchemaConversion;
                if( $wgLegacySchemaConversion ) {
                        // Create HistoryBlobCurStub entries.
@@ -443,7 +452,7 @@ function do_schema_restructuring() {
                        SELECT cur_namespace, cur_title, $cur_text, cur_comment, cur_user, cur_user_text, cur_timestamp, cur_minor_edit, $cur_flags
                        FROM $cur", $fname );
 
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                echo "......Setting up revision table.\n";
                $wgDatabase->query( "INSERT INTO $revision (rev_id, rev_page, rev_comment, rev_user, rev_user_text, rev_timestamp,
                                rev_minor_edit)
@@ -451,7 +460,7 @@ function do_schema_restructuring() {
                                old_timestamp, old_minor_edit
                        FROM $old,$cur WHERE old_namespace=cur_namespace AND old_title=cur_title", $fname );
 
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                echo "......Setting up page table.\n";
                $wgDatabase->query( "INSERT INTO $page (page_id, page_namespace, page_title, page_restrictions, page_counter,
                                page_is_redirect, page_is_new, page_random, page_touched, page_latest, page_len)
@@ -460,22 +469,21 @@ function do_schema_restructuring() {
                        FROM $cur,$revision
                        WHERE cur_id=rev_page AND rev_timestamp=cur_timestamp AND rev_id > {$maxold}", $fname );
 
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                echo "......Unlocking tables.\n";
                $wgDatabase->query( "UNLOCK TABLES", $fname );
 
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                echo "......Renaming old.\n";
                $wgDatabase->query( "ALTER TABLE $old RENAME TO $text", $fname );
 
-               echo wfTimestamp();
+               echo wfTimestamp( TS_DB );
                echo "...done.\n";
        }
 }
 
 function do_inverse_timestamp() {
        global $wgDatabase;
-       $fname="do_schema_restructuring";
        if( $wgDatabase->fieldExists( 'revision', 'inverse_timestamp' ) ) {
                echo "Removing revision.inverse_timestamp and fixing indexes... ";
                dbsource( archive( 'patch-inverse_timestamp.sql' ), $wgDatabase );
@@ -779,14 +787,77 @@ function do_rc_indices_update() {
                dbsource( archive( 'patch-recentchanges-utindex.sql' ) );
        } else {
                # Index seems to exist
-               echo( "...seems to be ok\n" );
+               echo( "...index on ( rc_namespace, rc_user_text ) seems to be ok\n" );
+       }
+
+       #Add (rc_user_text, rc_timestamp) index [A. Garrett], November 2006
+       # See if we can find the index we want
+       $info = $wgDatabase->indexInfo( 'recentchanges', 'rc_user_text', __METHOD__ );
+       if( !$info ) {
+               # None, so create
+               echo( "...index on ( rc_user_text, rc_timestamp ) not found; creating\n" );
+               dbsource( archive( 'patch-rc_user_text-index.sql' ) );
+       } else {
+               # Index seems to exist
+               echo( "...index on ( rc_user_text, rc_timestamp ) seems to be ok\n" );
+       }
+}
+
+function index_has_field($table, $index, $field) {
+       global $wgDatabase;
+       echo( "Checking if $table index $index includes field $field...\n" );
+       $info = $wgDatabase->indexInfo( $table, $index, __METHOD__ );
+       if( $info ) {
+               foreach($info as $row) {
+                       if($row->Column_name == $field) {
+                               echo( "...index $index on table $table seems to be ok\n" );
+                               return true;
+                       }
+               }
+       }
+       echo( "...index $index on table $table has no field $field; adding\n" );
+       return false;
+}
+
+function do_backlinking_indices_update() {
+       echo( "Checking for backlinking indices...\n" );
+       if (!index_has_field('pagelinks', 'pl_namespace', 'pl_from') ||
+               !index_has_field('templatelinks', 'tl_namespace', 'tl_from') ||
+               !index_has_field('imagelinks', 'il_to', 'il_from'))
+       {       
+               dbsource( archive( 'patch-backlinkindexes.sql' ) );
+       }
+}
+
+function do_stats_init() {
+       // Sometimes site_stats table is not properly populated.
+       global $wgDatabase;
+       echo "Checking site_stats row...";
+       $row = $wgDatabase->selectRow( 'site_stats', '*', array( 'ss_row_id' => 1 ), __METHOD__ );
+       if( $row === false ) {
+               echo "data is missing! rebuilding...\n";
+               
+               global $IP;
+               require_once "$IP/maintenance/initStats.inc";
+               wfInitStats();
+       } else {
+               echo "ok.\n";
        }
 }
 
-function do_all_updates( $doShared = false ) {
-       global $wgNewTables, $wgNewFields, $wgRenamedTables, $wgSharedDB, $wgDatabase, $wgDBtype;
+function purge_cache() {
+       global $wgDatabase;
+       # We can't guarantee that the user will be able to use TRUNCATE,
+       # but we know that DELETE is available to us
+       echo( "Purging caches..." );
+       $wgDatabase->delete( 'objectcache', '*', __METHOD__ );
+       echo( "done.\n" );
+}
 
-       $doUser = !$wgSharedDB || $doShared;
+function do_all_updates( $shared = false, $purge = true ) {
+       global $wgNewTables, $wgNewFields, $wgRenamedTables, $wgSharedDB, $wgDatabase, $wgDBtype, $IP;
+
+       $doUser = !$wgSharedDB || $shared;
 
        if ($wgDBtype === 'postgres') {
                do_postgres_updates();
@@ -849,175 +920,496 @@ function do_all_updates( $doShared = false ) {
        
        do_rc_indices_update(); flush();
 
-       initialiseMessages(); flush();
+       add_table( 'redirect', 'patch-redirect.sql' );
+
+       do_backlinking_indices_update(); flush();
+
+       do_restrictions_update(); flush ();
+
+       echo "Deleting old default messages (this may take a long time!)..."; flush();
+       deleteDefaultMessages();
+       echo "Done\n"; flush();
+       
+       do_stats_init(); flush();
+       
+       if( $purge ) {
+               purge_cache();
+               flush();
+       }
 }
 
 function archive($name) {
        global $wgDBtype, $IP;
        switch ($wgDBtype) {
-       case "oracle":
-               return "$IP/maintenance/oracle/archives/$name";
+       case "postgres":
+               return "$IP/maintenance/postgres/archives/$name";
        default:
                return "$IP/maintenance/archives/$name";
        }
 }
 
-function do_postgres_updates() {
-       global $wgDatabase, $wgVersion, $wgDBmwschema;
+function do_restrictions_update() {
+       # Adding page_restrictions table, obsoleting page.page_restrictions.
+       #  Migrating old restrictions to new table
+       # -- Andrew Garrett, January 2007.
 
-       $version = "1.7.1";
+       global $wgDatabase;
 
-       # Just in case their LocalSetings.php does not have this:
-       if ( !isset( $wgDBmwschema ))
-               $wgDBmwschema = 'mediawiki';
+       $name = 'page_restrictions';
+       $patch = 'patch-page_restrictions.sql';
+
+       if ( $wgDatabase->tableExists( $name ) ) {
+               echo "...$name table already exists.\n";
+       } else {
+               echo "Creating $name table...";
+               dbsource( archive($patch), $wgDatabase );
+               echo "ok\n";
+
+               echo "Migrating old restrictions to new table...";
+
+               $res = $wgDatabase->select( 'page', array( 'page_id', 'page_restrictions' ), array("page_restrictions!=''", "page_restrictions!='edit=:move='"), __METHOD__ );
+
+               $count = 0;
+
+               while ($row = $wgDatabase->fetchObject($res) ) {
+                       $count = ($count + 1) % 100;
+
+                       if ($count == 0) {
+                               if ( function_exists( 'wfWaitForSlaves' ) ) {
+                                       wfWaitForSlaves( 10 );
+                               } else {
+                                       sleep( 1 );
+                               }
+                       }
+
+                       # Figure out what the restrictions are..
+                       $id = $row->page_id;
+                       $flatrestrictions = explode( ':', $row->page_restrictions );
+
+                       $restrictions = array ();
+                       foreach( $flatrestrictions as $restriction ) {
+                               $thisrestriction = explode( '=', $restriction, 2 );
+                               if( count( $thisrestriction ) == 1 ) {
+                                       // Compatibility with old protections from before
+                                       // separate move protection was added.
+                                       list( $level ) = $thisrestriction;
+                                       if( $level ) {
+                                               $restrictions['edit'] = $level;
+                                               $restrictions['move'] = $level;
+                                       }
+                               } else {
+                                       list( $type, $level ) = $thisrestriction;
+                                       if( $level ) {
+                                               $restrictions[$type] = $level;
+                                       }
+                               }
+
+                       $wgDatabase->update( 'page', array ( 'page_restrictions' => ''), array( 'page_id' => $id ), __METHOD__ );
+
+                       }
+                       
+                       foreach( $restrictions as $type => $level ) {
+                               $wgDatabase->insert( 'page_restrictions', array ( 'pr_page' => $id,
+                                                                                       'pr_type' => $type,
+                                                                                       'pr_level' => $level,
+                                                                                       'pr_cascade' => 0 ),
+                                                                                       __METHOD__ );
+                       }
+               }
+               print "ok\n";
+       }
+       
+}
 
-       if ($wgDatabase->tableExists("mediawiki_version")) {
-               $version = "1.8";
+function
+pg_describe_table($table)
+{
+global $wgDatabase, $wgDBmwschema;
+       $q = <<<END
+SELECT attname, attnum FROM pg_namespace, pg_class, pg_attribute
+       WHERE pg_class.relnamespace = pg_namespace.oid 
+         AND attrelid=pg_class.oid AND attnum > 0
+         AND relname=%s AND nspname=%s
+END;
+       $res = $wgDatabase->query(sprintf($q,
+                       $wgDatabase->addQuotes($table),
+                       $wgDatabase->addQuotes($wgDBmwschema)));
+       if (!$res)
+               return null;
+
+       $cols = array();
+       while ($r = $wgDatabase->fetchRow($res)) {
+               $cols[] = array(        
+                               "name" => $r[0],
+                               "ord" => $r[1],
+                       );
        }
+       return $cols;
+}
 
-       if ($version == '1.7.1') {
-               $upgrade = <<<PGEND
+function
+pg_describe_index($idx)
+{
+global $wgDatabase, $wgDBmwschema;
+
+       // first fetch the key (which is a list of columns ords) and
+       // the table the index applies to (an oid)
+       $q = <<<END
+SELECT indkey, indrelid FROM pg_namespace, pg_class, pg_index
+       WHERE nspname=%s
+         AND pg_class.relnamespace = pg_namespace.oid
+         AND relname=%s
+         AND indexrelid=pg_class.oid
+END;
+       $res = $wgDatabase->query(sprintf($q,
+                       $wgDatabase->addQuotes($wgDBmwschema),
+                       $wgDatabase->addQuotes($idx)));
+       if (!$res)
+               return null;
+       if (!($r = $wgDatabase->fetchRow($res))) {
+               $wgDatabase->freeResult($res);
+               return null;
+       }
 
-BEGIN;
+       $indkey = $r[0];
+       $relid = intval($r[1]);
+       $indkeys = explode(" ", $indkey);
+       $wgDatabase->freeResult($res);
+
+       $colnames = array();
+       foreach ($indkeys as $rid) {
+               $query = <<<END
+SELECT attname FROM pg_class, pg_attribute
+       WHERE attrelid=$relid
+         AND attnum=%d
+         AND attrelid=pg_class.oid
+END;
+               $r2 = $wgDatabase->query(sprintf($query, $rid));
+               if (!$r2)
+                       return null;
+               if (!($row2 = $wgDatabase->fetchRow($r2))) {
+                       $wgDatabase->freeResult($r2);
+                       return null;
+               }
+               $colnames[] = $row2[0];
+               $wgDatabase->freeResult($r2);
+       }
 
--- Type tweaking:
-ALTER TABLE oldimage ALTER oi_size TYPE INTEGER;
-ALTER TABLE oldimage ALTER oi_width TYPE INTEGER;
-ALTER TABLE oldimage ALTER oi_height TYPE INTEGER;
+       return $colnames;
+}
 
-ALTER TABLE image ALTER img_size TYPE INTEGER;
-ALTER TABLE image ALTER img_width TYPE INTEGER;
-ALTER TABLE image ALTER img_height TYPE INTEGER;
+function
+pg_column_has_type($table, $column, $wanttype)
+{
+global $wgDatabase, $wgDBname, $wgDBmwschema;
 
--- Constraint tweaking:
-ALTER TABLE recentchanges ALTER rc_cur_id DROP NOT NULL;
+       $q = <<<END
+SELECT typname FROM pg_class, pg_namespace, pg_attribute, pg_type
+       WHERE relnamespace=pg_namespace.oid AND relkind='r'
+             AND attrelid=pg_class.oid AND atttypid=pg_type.oid
+             AND nspname=%s AND relname=%s AND attname=%s;
+END;
+       $res = $wgDatabase->query(sprintf($q,
+                       $wgDatabase->addQuotes($wgDBmwschema),
+                       $wgDatabase->addQuotes($table),
+                       $wgDatabase->addQuotes($column)));
+       $row = $wgDatabase->fetchRow($res);
+       $istype = false;
+       if ($row)
+               $istype = $row[0] === $wanttype;
+       $wgDatabase->freeResult($res);
+       return $istype;
+}
 
--- New columns:
-ALTER TABLE ipblocks ADD ipb_anon_only CHAR NOT NULL DEFAULT '0';
-ALTER TABLE ipblocks ADD ipb_create_account CHAR NOT NULL DEFAULT '1';
-ALTER TABLE ipblocks ADD ipb_enable_autoblock CHAR NOT NULL default '1';
+function
+pg_column_exists($table, $column)
+{
+global $wgDatabase, $wgDBname, $wgDBmwschema;
 
--- Index order rearrangements:
-DROP INDEX pagelink_unique;
-CREATE UNIQUE INDEX pagelink_unique ON pagelinks (pl_from,pl_namespace,pl_title);
+       $q = <<<END
+SELECT 1 FROM pg_class, pg_namespace, pg_attribute
+       WHERE relnamespace=pg_namespace.oid AND relkind='r'
+             AND attrelid=pg_class.oid
+             AND nspname=%s AND relname=%s AND attname=%s;
+END;
+       $res = $wgDatabase->query(sprintf($q,
+                       $wgDatabase->addQuotes($wgDBmwschema),
+                       $wgDatabase->addQuotes($table),
+                       $wgDatabase->addQuotes($column)));
+       $row = $wgDatabase->fetchRow($res);
+       $exists = !!$row;
+       $wgDatabase->freeResult($res);
+       return $exists;
+}
 
--- Rename tables
-ALTER TABLE "user" RENAME TO mwuser;
-ALTER TABLE "text" RENAME to pagecontent;
+function
+pg_column_is_nullable($table, $column)
+{
+global $wgDatabase, $wgDBname, $wgDBmwschema;
 
--- New tables:
-CREATE TABLE profiling (
-  pf_count   INTEGER         NOT NULL DEFAULT 0,
-  pf_time    NUMERIC(18,10)  NOT NULL DEFAULT 0,
-  pf_name    TEXT            NOT NULL,
-  pf_server  TEXT            NULL
-);
-CREATE UNIQUE INDEX pf_name_server ON profiling (pf_name, pf_server);
-
-CREATE TABLE mediawiki_version (
-  type         TEXT         NOT NULL,
-  mw_version   TEXT         NOT NULL,
-  notes        TEXT             NULL,
-
-  pg_version   TEXT             NULL,
-  pg_dbname    TEXT             NULL,
-  pg_user      TEXT             NULL,
-  pg_port      TEXT             NULL,
-  mw_schema    TEXT             NULL,
-  ts2_schema   TEXT             NULL,
-  ctype        TEXT             NULL,
-
-  sql_version  TEXT             NULL,
-  sql_date     TEXT             NULL,
-  cdate        TIMESTAMPTZ  NOT NULL DEFAULT now()
-);
+       $q = <<<END
+SELECT attnotnull FROM pg_class, pg_namespace, pg_attribute
+       WHERE relnamespace=pg_namespace.oid AND relkind='r'
+             AND attrelid=pg_class.oid
+             AND nspname=%s AND relname=%s AND attname=%s;
+END;
+       $res = $wgDatabase->query(sprintf($q,
+                       $wgDatabase->addQuotes($wgDBmwschema),
+                       $wgDatabase->addQuotes($table),
+                       $wgDatabase->addQuotes($column)));
+       $row = $wgDatabase->fetchRow($res);
+       $nullable = ($row[0] === 'f');
+       $wgDatabase->freeResult($res);
+       return $nullable;
+}
 
-INSERT INTO mediawiki_version (type,mw_version,notes)
-VALUES ('Upgrade','MWVERSION','Upgrade from older version 1.7.1');
-
--- Special modifications
-ALTER TABLE archive RENAME to archive2;
-CREATE VIEW archive AS 
-SELECT 
-  ar_namespace, ar_title, ar_text, ar_comment, ar_user, ar_user_text, 
-  ar_minor_edit, ar_flags, ar_rev_id, ar_text_id,
-       TO_CHAR(ar_timestamp, 'YYYYMMDDHH24MISS') AS ar_timestamp
-FROM archive2;
-
-CREATE RULE archive_insert AS ON INSERT TO archive
-DO INSTEAD INSERT INTO archive2 VALUES (
-  NEW.ar_namespace, NEW.ar_title, NEW.ar_text, NEW.ar_comment, NEW.ar_user, NEW.ar_user_text, 
-  TO_DATE(NEW.ar_timestamp, 'YYYYMMDDHH24MISS'),
-  NEW.ar_minor_edit, NEW.ar_flags, NEW.ar_rev_id, NEW.ar_text_id
-);
+function
+pg_table_exists($table)
+{
+global $wgDatabase, $wgDBname, $wgDBmwschema;
 
-CREATE FUNCTION page_deleted() RETURNS TRIGGER LANGUAGE plpgsql AS
-\$mw\$
-BEGIN
-DELETE FROM recentchanges WHERE rc_namespace = OLD.page_namespace AND rc_title = OLD.page_title;
-RETURN NULL;
+       $q = <<<END
+SELECT 1 FROM pg_class, pg_namespace
+       WHERE relnamespace=pg_namespace.oid AND relkind='r'
+             AND nspname=%s AND relname=%s
 END;
-\$mw\$;
+       $res = $wgDatabase->query(sprintf($q,
+                       $wgDatabase->addQuotes($wgDBmwschema),
+                       $wgDatabase->addQuotes($table)));
+       $row = $wgDatabase->fetchRow($res);
+       $exists = !!$row;
+       $wgDatabase->freeResult($res);
+       return $exists;
+}
 
-CREATE TRIGGER page_deleted AFTER DELETE ON page
-  FOR EACH ROW EXECUTE PROCEDURE page_deleted();
+function
+pg_trigger_exists($table, $trigger)
+{
+global $wgDatabase, $wgDBname, $wgDBmwschema;
 
-COMMIT;
+       $q = <<<END
+SELECT 1 FROM pg_class, pg_namespace, pg_trigger
+       WHERE relnamespace=pg_namespace.oid AND relkind='r'
+             AND tgrelid=pg_class.oid
+             AND nspname=%s AND relname=%s AND tgname=%s
+END;
+       $res = $wgDatabase->query(sprintf($q,
+                       $wgDatabase->addQuotes($wgDBmwschema),
+                       $wgDatabase->addQuotes($table),
+                       $wgDatabase->addQuotes($trigger)));
+       $row = $wgDatabase->fetchRow($res);
+       $exists = !!$row;
+       $wgDatabase->freeResult($res);
+       return $exists;
+}
 
-PGEND;
 
-       } ## end version 1.7.1 upgrade
-       else if ($version == '1.8') {
-               $upgrade = <<<PGEND
+function
+pg_index_exists($table, $index)
+{
+global $wgDatabase, $wgDBmwschema;
+       $exists = $wgDatabase->selectField("pg_indexes", "indexname",
+                       array(  "indexname" => $index,
+                               "tablename" => $table,
+                               "schemaname" => $wgDBmwschema));
+       return $exists === $index;
+}
 
-BEGIN;
+function
+pg_rule_exists($table, $rule)
+{
+global $wgDatabase, $wgDBmwschema;
+       $exists = $wgDatabase->selectField("pg_rules", "rulename",
+                       array(  "rulename" => $rule,
+                               "tablename" => $table,
+                               "schemaname" => $wgDBmwschema));
+       return $exists === $rule;
+}
 
--- Tighten up restrictions on the revisio table so we don't lose data:
-ALTER TABLE revision DROP CONSTRAINT revision_rev_user_fkey;
-ALTER TABLE revision ADD CONSTRAINT revision_rev_user_fkey
-  FOREIGN KEY (rev_user) REFERENCES mwuser(user_id) ON DELETE RESTRICT;
+function
+pg_fkey_deltype($fkey)
+{
+global $wgDatabase, $wgDBmwschema;
+       $q = <<<END
+SELECT confdeltype FROM pg_constraint, pg_namespace
+       WHERE connamespace=pg_namespace.oid
+         AND nspname=%s
+         AND conname=%s;
+END;
+       $r = $wgDatabase->query(sprintf($q,
+               $wgDatabase->addQuotes($wgDBmwschema),
+               $wgDatabase->addQuotes($fkey)));
+       if (!($row = $wgDatabase->fetchRow($r)))
+               return null;
+       return $row[0];
+}
 
--- New column for better password tracking:
-ALTER TABLE mwuser ADD user_newpass_time TIMESTAMPTZ;
+function
+pg_rule_def($table, $rule)
+{
+global $wgDatabase, $wgDBmwschema;
+       $q = <<<END
+SELECT definition FROM pg_rules
+       WHERE schemaname = %s
+         AND tablename = %s
+         AND rulename = %s
+END;
+       $r = $wgDatabase->query(sprintf($q,
+                       $wgDatabase->addQuotes($wgDBmwschema),
+                       $wgDatabase->addQuotes($table),
+                       $wgDatabase->addQuotes($rule)));
+       $row = $wgDatabase->fetchRow($r);
+       if (!$row)
+               return null;
+       $d = $row[0];
+       $wgDatabase->freeResult($r);
+       return $d;
+}
 
--- New tables:
-CREATE TABLE redirect (
-  rd_from       INTEGER  NOT NULL  REFERENCES page(page_id) ON DELETE CASCADE,
-  rd_namespace  SMALLINT NOT NULL,
-  rd_title      TEXT     NOT NULL
-);
-CREATE INDEX redirect_ns_title ON redirect (rd_namespace,rd_title,rd_from);
-
-CREATE TABLE querycachetwo (
-  qcc_type          TEXT     NOT NULL,
-  qcc_value         SMALLINT NOT NULL  DEFAULT 0,
-  qcc_namespace     INTEGER  NOT NULL  DEFAULT 0,
-  qcc_title         TEXT     NOT NULL  DEFAULT '',
-  qcc_namespacetwo  INTEGER  NOT NULL  DEFAULT 0,
-  qcc_titletwo      TEXT     NOT NULL  DEFAULT ''
-);
-CREATE INDEX querycachetwo_type_value ON querycachetwo (qcc_type, qcc_value);
-CREATE INDEX querycachetwo_title      ON querycachetwo (qcc_type,qcc_namespace,qcc_title);
-CREATE INDEX querycachetwo_titletwo   ON querycachetwo (qcc_type,qcc_namespacetwo,qcc_titletwo);
+function do_postgres_updates() {
+       global $wgDatabase, $wgVersion, $wgDBmwschema;
 
--- Note this upgrade
-INSERT INTO mediawiki_version (type,mw_version,notes)
-VALUES ('Upgrade','MWVERSION','Upgrade from older version 1.8');
+       # Just in case their LocalSetings.php does not have this:
+       if ( !isset( $wgDBmwschema ))
+               $wgDBmwschema = 'mediawiki';
+
+       $typechanges = array(
+               array("oldimage",     "oi_size",    "int4"),
+               array("oldimage",     "oi_width",   "int4"),
+               array("oldimage",     "oi_height",  "int4"),
+               array("image",        "img_size",   "int4"),
+               array("image",        "img_width",  "int4"),
+               array("image",        "img_height", "int4"),
+       );
+
+       $newcols = array(
+               array("mwuser",        "user_newpass_time",    "TIMESTAMPTZ"),
+               array("mwuser",        "user_editcount",       "INTEGER"),
+               array("ipblocks",      "ipb_anon_only",        "CHAR NOT NULL DEFAULT '0'"),
+               array("ipblocks",      "ipb_create_account",   "CHAR NOT NULL DEFAULT '1'"),
+               array("ipblocks",      "ipb_enable_autoblock", "CHAR NOT NULL DEFAULT '1'"),
+               array("recentchanges", "rc_old_len",           "INT"),
+               array("recentchanges", "rc_new_len",           "INT"),
+               array("revision",      "rev_len",              "INT")
+       );
+
+       $newtables = array(
+               array("mwuser",            "patch-mwuser.sql"),
+               array("pagecontent",       "patch-pagecontent.sql"),
+               array("querycachetwo",     "patch-querycachetwo.sql"),
+               array("redirect",          "patch-redirect.sql"),
+               array("page_restrictions", "patch-page_restrictions.sql"),
+               array("profiling",         "patch-profiling.sql"),
+               array("mediawiki_version", "patch-mediawiki_version.sql"),
+               array("archive2",          "patch-archive2.sql")
+       );
 
-COMMIT;
+       $newindexes = array(
+               array("revision", "rev_text_id_idx",          "patch-rev_text_id_idx.sql")
+       );
 
-PGEND;
+       $newrules = array(
+               array("archive", "archive_delete", "patch-archive_delete.sql")
+       );
 
+       foreach ($newtables as $nt) {
+               if (pg_table_exists($nt[0])) {
+                       echo "... table $nt[0] already exists\n";
+                       continue;
+               }
+
+               echo "... create table $nt[0]\n";
+               dbsource(archive($nt[1]));
        }
 
-       else {
-               print "No updates needed for version $version\n";
-               return;
+       foreach ($newcols as $nc) {
+               if (pg_column_exists($nc[0], $nc[1])) {
+                       echo "... column $nc[0].$nc[1] already exists\n";
+                       continue;
+               }
+
+               echo "... add column $nc[0].$nc[1]\n";
+               $wgDatabase->query("ALTER TABLE $nc[0] ADD $nc[1] $nc[2]");
+       }
+
+       foreach ($typechanges as $tc) {
+               if (!pg_column_exists($tc[0], $tc[1])) {
+                       echo "... error: expected column $tc[0].$tc[1] to exist\n";
+                       exit(1);
+               }
+
+               if (pg_column_has_type($tc[0], $tc[1], $tc[2]))
+                       echo "... $tc[0].$tc[1] is already $tc[2]\n";
+               else {
+                       echo "... change $tc[0].$tc[1] to $tc[2]\n";
+                       $wgDatabase->query("ALTER TABLE $tc[0] ALTER $tc[1] TYPE $tc[2];\nCOMMIT;\n");
+               }
+       }
+
+       if (!pg_column_has_type("user_newtalk", "user_ip", "text")) {
+               echo "... convert user_newtalk.user_ip to text\n";
+               $wgDatabase->query("ALTER TABLE user_newtalk ALTER user_ip TYPE TEXT USING host(user_ip)");
+       }
+
+       foreach ($newindexes as $ni) {
+               if (pg_index_exists($ni[0], $ni[1])) {
+                       echo "... index $ni[1] on $ni[0] already exists\n";
+                       continue;
+               }
+               dbsource(archive($ni[2]));
+       }
+
+       foreach ($newrules as $nr) {
+               if (pg_rule_exists($nr[0], $nr[1])) {
+                       echo "... rule $nr[1] on $nr[0] already exists\n";
+                       continue;
+               }
+               dbsource(archive($nr[2]));
+       }
+
+       if (!pg_column_has_type("ipblocks", "ipb_address", "text")) {
+               echo "... change ipblocks.ipb_address to TEXT\n";
+               $wgDatabase->query("ALTER TABLE ipblocks ALTER ipb_address TYPE TEXT USING ipb_address::TEXT");
+       } else
+               echo "... ipblocks.ipb_address is already TEXT\n";
+
+       if (!pg_trigger_exists("page", "page_deleted")) {
+               echo "... create page_deleted trigger\n";
+               dbsource(archive('patch-page_deleted.sql'));
        }
 
-       $upgrade = str_replace( 'MWVERSION', $wgVersion, $upgrade );
-       $res = $wgDatabase->query($upgrade);
+       if (!pg_column_is_nullable("recentchanges", "rc_cur_id")) {
+               echo "... remove NOT NULL constraint on recentchanges.rc_cur_id\n";
+               dbsource(archive('patch-rc_cur_id-not-null.sql'));
+       }
+
+       $pu = pg_describe_index("pagelink_unique");
+       if (!is_null($pu) && ($pu[0] != "pl_from" || $pu[1] != "pl_namespace" || $pu[2] != "pl_title")) {
+               echo "... dropping obsolete pagelink_unique index\n";
+               $wgDatabase->query("DROP INDEX pagelink_unique");
+               $pu = null;
+       } else
+               echo "... obsolete pagelink_unique index not present\n";
+
+       if (is_null($pu)) {
+               echo "... adding new pagelink_unique index\n";
+               $wgDatabase->query("CREATE UNIQUE INDEX pagelink_unique ON pagelinks (pl_from,pl_namespace,pl_title)");
+       } else
+               echo "... already have current pagelink_unique index\n";
+
+       if (pg_fkey_deltype("revision_rev_user_fkey") == 'r') {
+               echo "... revision_rev_user_fkey is already ON DELETE RESTRICT\n";
+       } else {
+               echo "... change revision_rev_user_fkey to ON DELETE RESTRICT\n";
+               dbsource(archive('patch-revision_rev_user_fkey.sql'));
+       }
 
+       $ai_def = pg_rule_def("archive", "archive_insert");
+       if (strstr($ai_def, "to_date") !== false) {
+               echo "... fix archive_insert rule\n";
+               dbsource(archive('patch-archive_insert.sql'));
+       } else
+               echo "... already have correct archive_insert rule\n";
 
        return;
 }