3 * Makes the required database updates for Special:ProtectedPages
4 * to show all protected pages, even ones before the page restrictions
5 * schema change. All remaining page_restriction column values are moved
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
21 * http://www.gnu.org/copyleft/gpl.html
24 * @ingroup Maintenance
27 require_once __DIR__
. '/Maintenance.php';
30 * Maintenance script that updates page_restrictions table from
31 * old page_restriction column.
33 * @ingroup Maintenance
35 class UpdateRestrictions
extends Maintenance
{
36 public function __construct() {
37 parent
::__construct();
38 $this->addDescription( 'Updates page_restrictions table from old page_restriction column' );
39 $this->setBatchSize( 1000 );
42 public function execute() {
43 $dbw = $this->getDB( DB_PRIMARY
);
44 $batchSize = $this->getBatchSize();
46 if ( !$dbw->tableExists( 'page_restrictions', __METHOD__
) ) {
47 $this->fatalError( "page_restrictions table does not exist" );
50 if ( !$dbw->fieldExists( 'page', 'page_restrictions' ) ) {
51 $this->output( "Migration is not needed.\n" );
55 $encodedExpiry = $dbw->getInfinity();
57 $maxPageId = $dbw->newSelectQueryBuilder()
58 ->select( 'MAX(page_id)' )
60 ->caller( __METHOD__
)->fetchField();
61 $escapedEmptyBlobValue = $dbw->addQuotes( '' );
66 $batchMaxPageId = $batchMinPageId +
$batchSize;
68 $this->output( "...processing page IDs from $batchMinPageId to $batchMaxPageId.\n" );
70 $res = $dbw->newSelectQueryBuilder()
71 ->select( [ 'page_id', 'page_restrictions' ] )
74 "page_restrictions != $escapedEmptyBlobValue",
75 'page_id > ' . $dbw->addQuotes( $batchMinPageId ),
76 'page_id <= ' . $dbw->addQuotes( $batchMaxPageId ),
78 ->caller( __METHOD__
)->fetchResultSet();
80 // No pages have legacy protection settings in the current batch
81 if ( !$res->numRows() ) {
82 $batchMinPageId = $batchMaxPageId;
89 foreach ( $res as $row ) {
90 $pageIds[] = $row->page_id
;
92 $restrictionsByAction = $this->mapLegacyRestrictionBlob( $row->page_restrictions
);
94 # Update restrictions table
95 foreach ( $restrictionsByAction as $action => $restrictions ) {
97 'pr_page' => $row->page_id
,
99 'pr_level' => $restrictions,
101 'pr_expiry' => $encodedExpiry
106 $this->beginTransaction( $dbw, __METHOD__
);
108 // Insert new format protection settings for the pages in the current batch.
109 // Use INSERT IGNORE to ignore conflicts with new format settings that might exist for the page
117 // Clear out the legacy page.page_restrictions blob for this batch
118 $dbw->update( 'page', [ 'page_restrictions' => '' ], [ 'page_id' => $pageIds ], __METHOD__
);
120 $this->commitTransaction( $dbw, __METHOD__
);
122 $batchMinPageId = $batchMaxPageId;
123 } while ( $batchMaxPageId < $maxPageId );
125 $this->output( "...Done!\n" );
130 * Convert a legacy restriction specification from the page.page_restrictions blob to
131 * a map of action names to restriction levels.
133 * @param string $legacyBlob Legacy page.page_restrictions blob,
134 * e.g. "sysop" or "edit=sysop:move=autoconfirmed"
135 * @return string[] array of restriction levels keyed by action names
137 private function mapLegacyRestrictionBlob( $legacyBlob ) {
138 $oldRestrictions = [];
140 foreach ( explode( ':', trim( $legacyBlob ) ) as $restrict ) {
141 $temp = explode( '=', trim( $restrict ) );
143 // Treat old old format without action name as edit/move restriction
144 if ( count( $temp ) == 1 ) {
145 $level = trim( $temp[0] );
147 $oldRestrictions['edit'] = $level;
148 $oldRestrictions['move'] = $level;
150 $restriction = trim( $temp[1] );
151 // Some old entries are empty
152 if ( $restriction != '' ) {
153 $oldRestrictions[$temp[0]] = $restriction;
158 return $oldRestrictions;
162 $maintClass = UpdateRestrictions
::class;
163 require_once RUN_MAINTENANCE_IF_MAIN
;