protect.php

Go to the documentation of this file.

<?php
require_once( dirname( __FILE__ ) . '/Maintenance.php' );

class Protect extends Maintenance {
        public function __construct() {
                parent::__construct();
                $this->mDescription = "Protect or unprotect an article from the command line.";
                $this->addOption( 'unprotect', 'Removes protection' );
                $this->addOption( 'semiprotect', 'Adds semi-protection' );
                $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;

                $userName = $this->getOption( 'u', 'Maintenance script' );
                $reason = $this->getOption( 'r', '' );

                $cascade = $this->hasOption( 'cascade' );

                $protection = "sysop";
                if ( $this->hasOption( 'semiprotect' ) ) {
                        $protection = "autoconfirmed";
                } elseif ( $this->hasOption( 'unprotect' ) ) {
                        $protection = "";
                }

                $user = User::newFromName( $userName );
                if ( !$user ) {
                        $this->error( "Invalid username", true );
                }

                $restrictions = array( 'edit' => $protection, 'move' => $protection );

                $t = Title::newFromText( $this->getArg() );
                if ( !$t ) {
                        $this->error( "Invalid title", true );
                }

                $restrictions = array();
                foreach( $t->getRestrictionTypes() as $type ) {
                        $restrictions[$type] = $protection;
                }

                # un/protect the article
                $this->output( "Updating protection status... " );

                $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( RUN_MAINTENANCE_IF_MAIN );