Verify known image types on upload.
[lhc/web/wiklou.git] / includes / DatabasePostgreSQL.php
index f98814f..f38b70c 100644 (file)
@@ -1,21 +1,32 @@
 <?php
 # $Id$
-#
-# DO NOT USE !!!  Unless you want to help developping it.
-#
-# This file is an attempt to port the mysql database layer to postgreSQL. The
-# only thing done so far is s/mysql/pg/ and dieing if function haven't been
-# ported.
-# 
-# As said brion 07/06/2004 :
-# "table definitions need to be changed. fulltext index needs to work differently
-#  things that use the last insert id need to be changed. Probably other things
-#  need to be changed. various semantics may be different."
-#
-# Hashar
 
+/**
+ * DO NOT USE !!!  Unless you want to help developping it.
+ *
+ * This file is an attempt to port the mysql database layer to postgreSQL. The
+ * only thing done so far is s/mysql/pg/ and dieing if function haven't been
+ * ported.
+ *
+ * As said brion 07/06/2004 :
+ * "table definitions need to be changed. fulltext index needs to work differently
+ * things that use the last insert id need to be changed. Probably other things
+ * need to be changed. various semantics may be different."
+ *
+ * Hashar
+ *
+ * @package MediaWiki
+ */
+
+/**
+ * Depends on database
+ */
 require_once( 'Database.php' );
 
+/**
+ *
+ * @package MediaWiki
+ */
 class DatabasePgsql extends Database {
        var $mInsertId = NULL;
        var $mLastResult = NULL;
@@ -32,15 +43,18 @@ class DatabasePgsql extends Database {
                return new DatabasePgsql( $server, $user, $password, $dbName, $failFunction, $flags, $tablePrefix );
        }
 
-       # Usually aborts on failure
-       # If the failFunction is set to a non-zero integer, returns success
-       function open( $server, $user, $password, $dbName )
-       {
+       /**
+        * Usually aborts on failure
+        * If the failFunction is set to a non-zero integer, returns success
+        */
+       function open( $server, $user, $password, $dbName ) {
                # Test for PostgreSQL support, to avoid suppressed fatal error
                if ( !function_exists( 'pg_connect' ) ) {
                        die( "PostgreSQL functions missing, have you compiled PHP with the --with-pgsql option?\n" );
                }
 
+               global $wgDBschema;
+
                $this->close();
                $this->mServer = $server;
                $this->mUser = $user;
@@ -51,7 +65,11 @@ class DatabasePgsql extends Database {
                
                if ( '' != $dbName ) {
                        # start a database connection
-                       @$this->mConn = pg_connect("host=$server dbname=$dbName user=$user password=$password");
+                       $hstring="";
+                       if ($server!=false && $server!="") {
+                               $hstring="host=$server ";
+                       }
+                       @$this->mConn = pg_connect("$hstring dbname=$dbName user=$user password=$password");
                        if ( $this->mConn == false ) {
                                wfDebug( "DB connection error\n" );
                                wfDebug( "Server: $server, Database: $dbName, User: $user, Password: " . substr( $password, 0, 3 ) . "...\n" );
@@ -59,14 +77,16 @@ class DatabasePgsql extends Database {
                        } else { 
                                $this->mOpened = true;
                        }
+                       $this->query("SET search_path = $wgDBschema,public");
                }
                return $this->mConn;
        }
        
-       # Closes a database connection, if it is open
-       # Returns success, true if already closed
-       function close()
-       {
+       /**
+        * Closes a database connection, if it is open
+        * Returns success, true if already closed
+        */
+       function close() {
                $this->mOpened = false;
                if ( $this->mConn ) {
                        return pg_close( $this->mConn );
@@ -88,6 +108,7 @@ class DatabasePgsql extends Database {
                        wfDebugDieBacktrace( "Unable to free PostgreSQL result\n" );
                }
        }
+       
        function fetchObject( $res ) {
                @$row = pg_fetch_object( $res );
                # FIXME: HACK HACK HACK HACK debug
@@ -119,7 +140,9 @@ class DatabasePgsql extends Database {
        function numFields( $res ) { return pg_num_fields( $res ); }
        function fieldName( $res, $n ) { return pg_field_name( $res, $n ); }
        
-       # This must be called after nextSequenceVal
+       /**
+        * This must be called after nextSequenceVal
+        */
        function insertId() { 
                return $this->mInsertId;
        }
@@ -132,10 +155,11 @@ class DatabasePgsql extends Database {
                return pg_affected_rows( $this->mLastResult ); 
        }
        
-       # Returns information about an index
-       # If errors are explicitly ignored, returns NULL on failure
-       function indexInfo( $table, $index, $fname = 'Database::indexExists' ) 
-       {
+       /**
+        * Returns information about an index
+        * If errors are explicitly ignored, returns NULL on failure
+        */
+       function indexInfo( $table, $index, $fname = 'Database::indexExists' ) {
                $sql = "SELECT indexname FROM pg_indexes WHERE tablename='$table'";
                $res = $this->query( $sql, $fname );
                if ( !$res ) {
@@ -150,8 +174,19 @@ class DatabasePgsql extends Database {
                return false;
        }
 
-       function fieldInfo( $table, $field )
-       {
+       function indexUnique ($table, $index, $fname = 'Database::indexUnique' ) {
+               $sql = "SELECT indexname FROM pg_indexes WHERE tablename='{$table}'".
+                       " AND indexdef LIKE 'CREATE UNIQUE%({$index})'";
+               $res = $this->query( $sql, $fname );
+               if ( !$res )
+                       return NULL;
+               while ($row = $this->fetchObject( $res )) 
+                       return true;
+               return false;
+               
+       }
+
+       function fieldInfo( $table, $field ) {
                wfDebugDieBacktrace( 'Database::fieldInfo() error : mysql_fetch_field() not implemented for postgre' );
                /*
                $res = $this->query( "SELECT * FROM '$table' LIMIT 1" );
@@ -172,21 +207,27 @@ class DatabasePgsql extends Database {
                # We have a go at faking one of them
                # TODO: DELAYED, LOW_PRIORITY 
 
-               # IGNORE is performed using single-row inserts, ignoring errors in each
-               if ( in_array( 'IGNORE', $options ) ) {
-                       # FIXME: need some way to distiguish between key collision and other types of error
+               if ( !is_array($options))
+                       $options = array($options);
+
+               if ( in_array( 'IGNORE', $options ) ) 
                        $oldIgnore = $this->ignoreErrors( true );
-                       if ( !is_array( reset( $a ) ) ) {
-                               $a = array( $a );
-                       }
-                       foreach ( $a as $row ) {
-                               parent::insertArray( $table, $row, $fname, array() );
-                       }
-                       $this->ignoreErrors( $oldIgnore );
-                       $retVal = true;
-               } else {
-                       $retVal = parent::insertArray( $table, $a, $fname, array() );
+
+               # IGNORE is performed using single-row inserts, ignoring errors in each
+               # FIXME: need some way to distiguish between key collision and other types of error
+               $oldIgnore = $this->ignoreErrors( true );
+               if ( !is_array( reset( $a ) ) ) {
+                       $a = array( $a );
+               }
+               foreach ( $a as $row ) {
+                       parent::insertArray( $table, $row, $fname, array() );
                }
+               $this->ignoreErrors( $oldIgnore );
+               $retVal = true;
+
+               if ( in_array( 'IGNORE', $options ) ) 
+                       $this->ignoreErrors( $oldIgnore );
+
                return $retVal;
        }
        
@@ -217,15 +258,19 @@ class DatabasePgsql extends Database {
                return pg_escape_string( $s );
        }
 
-       # Return the next in a sequence, save the value for retrieval via insertId()
+       /**
+        * Return the next in a sequence, save the value for retrieval via insertId()
+        */
        function nextSequenceValue( $seqName ) {
                $value = $this->getField(''," nextval('" . $seqName . "')");
                $this->mInsertId = $value;
                return $value;
        }
 
-       # USE INDEX clause
-       # PostgreSQL doesn't have them and returns ""
+       /**
+        * USE INDEX clause
+        * PostgreSQL doesn't have them and returns ""
+        */
        function useIndexClause( $index ) {
                return '';
        }
@@ -241,6 +286,10 @@ class DatabasePgsql extends Database {
        # occurred in MySQL
        function replace( $table, $uniqueIndexes, $rows, $fname = 'Database::replace' ) {
                $table = $this->tableName( $table );
+       
+               if (count($rows)==0) {
+                       return;
+               }
 
                # Single row case
                if ( !is_array( reset( $rows ) ) ) {
@@ -250,17 +299,17 @@ class DatabasePgsql extends Database {
                foreach( $rows as $row ) {
                        # Delete rows which collide
                        if ( $uniqueIndexes ) {
-                               $sql = "DELETE FROM $table WHERE (";
+                               $sql = "DELETE FROM $table WHERE ";
                                $first = true;
                                foreach ( $uniqueIndexes as $index ) {
                                        if ( $first ) {
                                                $first = false;
+                                               $sql .= "(";
                                        } else {
                                                $sql .= ') OR (';
                                        }
                                        if ( is_array( $index ) ) {
                                                $first2 = true;
-                                               $sql .= "(";
                                                foreach ( $index as $col ) {
                                                        if ( $first2 ) { 
                                                                $first2 = false;
@@ -269,16 +318,16 @@ class DatabasePgsql extends Database {
                                                        }
                                                        $sql .= $col.'=' . $this->addQuotes( $row[$col] );
                                                }
-                               } else {
+                                       } else {
                                                $sql .= $index.'=' . $this->addQuotes( $row[$index] );
-                               }
+                                       }
                                }
                                $sql .= ')';
                                $this->query( $sql, $fname );
                        }
 
                        # Now insert the row
-                       $sql = "INSERT INTO $table (" . $this->makeList( array_flip( $row ) ) .') VALUES (' .
+                       $sql = "INSERT INTO $table (" . $this->makeList( array_keys( $row ), LIST_NAMES ) .') VALUES (' .
                                $this->makeList( $row, LIST_COMMA ) . ')';
                        $this->query( $sql, $fname );
                }
@@ -304,8 +353,17 @@ class DatabasePgsql extends Database {
        # Returns the size of a text field, or -1 for "unlimited"
        function textFieldSize( $table, $field ) {
                $table = $this->tableName( $table );
-               $res = $this->query( "SELECT $field FROM $table LIMIT 1", "Database::textFieldLength" );
-               $size = pg_field_size( $res, 0 );
+               $sql = "SELECT t.typname as ftype,a.atttypmod as size
+                       FROM pg_class c, pg_attribute a, pg_type t 
+                       WHERE relname='$table' AND a.attrelid=c.oid AND 
+                               a.atttypid=t.oid and a.attname='$field'";
+               $res =$this->query($sql);
+               $row=$this->fetchObject($res);
+               if ($row->ftype=="varchar") {
+                       $size=$row->size-4;     
+               } else {
+                       $size=$row->size;
+               }
                $this->freeResult( $res );
                return $size;
        }
@@ -317,27 +375,69 @@ class DatabasePgsql extends Database {
        function limitResult($limit,$offset) {
                return " LIMIT $limit ".(is_numeric($offset)?" OFFSET {$offset} ":"");
        }
+       
+       /**
+        * Returns an SQL expression for a simple conditional.
+        * Uses CASE on PostgreSQL.
+        *
+        * @param string $cond SQL expression which will result in a boolean value
+        * @param string $trueVal SQL expression to return if true
+        * @param string $falseVal SQL expression to return if false
+        * @return string SQL fragment
+        */
+       function conditional( $cond, $trueVal, $falseVal ) {
+               return " (CASE WHEN $cond THEN $trueVal ELSE $falseVal END) ";
+       }
 
        # FIXME: actually detecting deadlocks might be nice
        function wasDeadlock() {
                return false;
        }
 
-        # Return DB-style timestamp used for MySQL schema
-        function timestamp( $ts=0 ) {
-                return wfTimestamp(TS_DB,$ts);
+       # Return DB-style timestamp used for MySQL schema
+       function timestamp( $ts=0 ) {
+               return wfTimestamp(TS_DB,$ts);
+       }
+
+        /**
+         * Return aggregated value function call
+         */
+        function aggregateValue ($valuedata,$valuename='value') {
+                return $valuedata;
         }
 
-        function reportQueryError( $error, $errno, $sql, $fname, $tempIgnore = false ) {
-               $message = "A database error has occurred\n" .
-                                  "Query: $sql\n" .
-                                  "Function: $fname\n" .
-                                  "Error: $errno $error\n";
+
+       function reportQueryError( $error, $errno, $sql, $fname, $tempIgnore = false ) {
+               $message = "A database error has occurred\n" .
+                       "Query: $sql\n" .
+                       "Function: $fname\n" .
+                       "Error: $errno $error\n";
                wfDebugDieBacktrace($message);
        }
+
+       /**
+        * @return string wikitext of a link to the server software's web site
+        */
+       function getSoftwareLink() {
+               return "[http://www.postgresql.org/ PostgreSQL]";
+       }
+       
+       /**
+        * @return string Version information from the database
+        */
+       function getServerVersion() {
+               $res = $this->query( "SELECT version()" );
+               $row = $this->fetchRow( $res );
+               $version = $row[0];
+               $this->freeResult( $res );
+               return $version;
+       }
 }
 
-# Just an alias.
+/**
+ * Just an alias.
+ * @package MediaWiki
+ */
 class DatabasePostgreSQL extends DatabasePgsql {
 }