X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;f=includes%2Finstaller%2FMssqlUpdater.php;h=4d86d116d7538b65fde8410aba1c242c051cbc5d;hb=ef6d5493500b8e82a38d7f949d9b5131120abb5d;hp=f9c4287b418c7936ecf9d45358e4912d333c7550;hpb=57a0864cece51b476d55116f9c65f0606a2f0df5;p=lhc%2Fweb%2Fwiklou.git

diff --git a/includes/installer/MssqlUpdater.php b/includes/installer/MssqlUpdater.php
index f9c4287b41..4d86d116d7 100644
--- a/includes/installer/MssqlUpdater.php
+++ b/includes/installer/MssqlUpdater.php
@@ -41,6 +41,91 @@ class MssqlUpdater extends DatabaseUpdater {
 			array( 'addField', 'mwuser', 'user_password_expires', 'patch-user_password_expires.sql' ),
 
 			// 1.24
+			array( 'addField', 'page', 'page_lang', 'patch-page-page_lang.sql'),
+			// Constraint updates
+			array( 'updateConstraints', 'category_types', 'categorylinks', 'cl_type' ),
+			array( 'updateConstraints', 'major_mime', 'filearchive', 'fa_major_mime' ),
+			array( 'updateConstraints', 'media_type', 'filearchive', 'fa_media_type' ),
+			array( 'updateConstraints', 'major_mime', 'oldimage', 'oi_major_mime' ),
+			array( 'updateConstraints', 'media_type', 'oldimage', 'oi_media_type' ),
+			array( 'updateConstraints', 'major_mime', 'image', 'img_major_mime' ),
+			array( 'updateConstraints', 'media_type', 'image', 'img_media_type' ),
+			array( 'updateConstraints', 'media_type', 'uploadstash', 'us_media_type' ),
+			// END: Constraint updates
 		);
 	}
+
+	/**
+	 * Drops unnamed and creates named constraints following the pattern
+	 * <column>_ckc
+	 *
+	 * @param string $constraintType
+	 * @param string $table Name of the table to which the field belongs
+	 * @param string $field Name of the field to modify
+	 * @return bool False if patch is skipped.
+	 */
+	protected function updateConstraints( $constraintType, $table, $field ) {
+		global $wgDBname, $wgDBmwschema;
+
+		if ( !$this->doTable( $table ) ) {
+			return true;
+		}
+
+		$this->output( "...updating constraints on [$table].[$field] ..." );
+		$updateKey = "$field-$constraintType-ck";
+		if ( !$this->db->tableExists( $table, __METHOD__ ) ) {
+			$this->output( "...$table table does not exist, skipping modify field patch.\n" );
+			return true;
+		} elseif ( !$this->db->fieldExists( $table, $field, __METHOD__ ) ) {
+			$this->output( "...$field field does not exist in $table table, " .
+				"skipping modify field patch.\n" );
+			return true;
+		} elseif ( $this->updateRowExists( $updateKey ) ) {
+			$this->output( "...$field in table $table already patched.\n" );
+			return true;
+		}
+
+		# After all checks passed, start the update
+		$this->insertUpdateRow( $updateKey );
+		$path = 'named_constraints.sql';
+		$constraintMap = array(
+			'category_types' =>
+				"($field in('page', 'subcat', 'file'))",
+			'major_mime'     =>
+				"($field in('unknown', 'application', 'audio', 'image', 'text', 'video'," .
+				" 'message', 'model', 'multipart'))",
+			'media_type'     =>
+				"($field in('UNKNOWN', 'BITMAP', 'DRAWING', 'AUDIO', 'VIDEO', 'MULTIMEDIA'," .
+				"'OFFICE', 'TEXT', 'EXECUTABLE', 'ARCHIVE'))"
+		);
+		$constraint = $constraintMap[$constraintType];
+
+		# and hack-in those variables that should be replaced
+		# in our template file right now
+		$this->db->setSchemaVars( array(
+			'tableName'       => $table,
+			'fieldName'       => $field,
+			'checkConstraint' => $constraint,
+			'wgDBname'        => $wgDBname,
+			'wgDBmwschema'    => $wgDBmwschema,
+		) );
+
+		# Full path from file name
+		$path = $this->db->patchPath( $path );
+
+		# No need for a cursor allowing result-iteration; just apply a patch
+		# store old value for re-setting later
+		$wasScrollable = $this->db->scrollableCursor( false );
+
+		# Apply patch
+		$this->db->sourceFile( $path );
+
+		# Reset DB instance to have original state
+		$this->db->setSchemaVars( false );
+		$this->db->scrollableCursor( $wasScrollable );
+
+		$this->output( "done.\n" );
+
+		return true;
+	}
 }