<?php
/**
+ * Protect or unprotect a page.
+ *
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
* http://www.gnu.org/copyleft/gpl.html
*
+ * @file
* @ingroup Maintenance
*/
-require_once( dirname(__FILE__) . '/Maintenance.php' );
+require_once( __DIR__ . '/Maintenance.php' );
+/**
+ * Maintenance script that protects or unprotects a page.
+ *
+ * @ingroup Maintenance
+ */
class Protect extends Maintenance {
public function __construct() {
parent::__construct();
- $this->mDescription = "Protect or unprotect an article from the command line.";
+ $this->mDescription = "Protect or unprotect a page from the command line.";
$this->addOption( 'unprotect', 'Removes protection' );
$this->addOption( 'semiprotect', 'Adds semi-protection' );
- $this->addOption( 'u', 'Username to protect with', false, true );
- $this->addOption( 'r', 'Reason for un/protection', false, true );
+ $this->addOption( 'cascade', 'Add cascading protection' );
+ $this->addOption( 'user', 'Username to protect with', false, true, 'u' );
+ $this->addOption( 'reason', 'Reason for un/protection', false, true, 'r' );
+ $this->addArg( 'title', 'Title to protect', true );
}
public function execute() {
- global $wgUser, $wgTitle, $wgArticle;
-
$userName = $this->getOption( 'u', 'Maintenance script' );
$reason = $this->getOption( 'r', '' );
+ $cascade = $this->hasOption( 'cascade' );
+
$protection = "sysop";
- if ( $this->hasOption('semiprotect') ) {
+ if ( $this->hasOption( 'semiprotect' ) ) {
$protection = "autoconfirmed";
- } elseif ( $this->hasOption('unprotect') ) {
+ } elseif ( $this->hasOption( 'unprotect' ) ) {
$protection = "";
}
- $wgUser = User::newFromName( $userName );
+ $user = User::newFromName( $userName );
+ if ( !$user ) {
+ $this->error( "Invalid username", true );
+ }
+
+ // @todo FIXME: This is reset 7 lines down.
$restrictions = array( 'edit' => $protection, 'move' => $protection );
- $wgTitle = Title::newFromText( $this->getArg() );
- if ( !$wgTitle ) {
+ $t = Title::newFromText( $this->getArg() );
+ if ( !$t ) {
$this->error( "Invalid title", true );
}
- $wgArticle = new Article( $wgTitle );
+ $restrictions = array();
+ foreach( $t->getRestrictionTypes() as $type ) {
+ $restrictions[$type] = $protection;
+ }
# un/protect the article
$this->output( "Updating protection status... " );
- $success = $wgArticle->updateRestrictions($restrictions, $reason);
- if ( $success ) {
+
+ $page = WikiPage::factory( $t );
+ $status = $page->doUpdateRestrictions( $restrictions, array(), $cascade, $reason, $user );
+
+ if ( $status->isOK() ) {
$this->output( "done\n" );
} else {
$this->output( "failed\n" );
}
$maintClass = "Protect";
-require_once( DO_MAINTENANCE );
+require_once( RUN_MAINTENANCE_IF_MAIN );