<?php
/**
+ * Protect or unprotect an article.
+ *
* 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
* @ingroup Maintenance
*/
-require_once( dirname(__FILE__) . '/Maintenance.php' );
+require_once( dirname( __FILE__ ) . '/Maintenance.php' );
class Protect extends Maintenance {
public function __construct() {
$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->addArg( 'title', 'Title to protect', true );
}
public function execute() {
- global $wgUser, $wgTitle, $wgArticle;
+ global $wgUser;
$userName = $this->getOption( 'u', 'Maintenance script' );
$reason = $this->getOption( 'r', '' );
$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 );
+ if ( !$wgUser ) {
+ $this->error( "Invalid username", true );
+ }
+
$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 );
+ $article = new Article( $t );
# un/protect the article
$this->output( "Updating protection status... " );
- $success = $wgArticle->updateRestrictions($restrictions, $reason);
+ $success = $article->updateRestrictions( $restrictions, $reason );
if ( $success ) {
$this->output( "done\n" );
} else {
}
$maintClass = "Protect";
-require_once( DO_MAINTENANCE );
+require_once( RUN_MAINTENANCE_IF_MAIN );