Fix exif timestamp bug

[lhc/web/wiklou.git] / maintenance / protect.php

diff --git a/maintenance/protect.php b/maintenance/protect.php
index 6606746..2f6aa1a 100644 (file)
--- a/maintenance/protect.php
+++ b/maintenance/protect.php
@@ -1,5 +1,7 @@
 <?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
@@ -15,48 +17,66 @@
  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  * http://www.gnu.org/copyleft/gpl.html
  *
+ * @file
  * @ingroup Maintenance
  */
 
-require_once( "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 );
+               }
+
                $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" );
@@ -65,4 +85,4 @@ class Protect extends Maintenance {
 }
 
 $maintClass = "Protect";
-require_once( DO_MAINTENANCE );
+require_once( RUN_MAINTENANCE_IF_MAIN );