protect.php

Go to the documentation of this file.

<?php
require_once __DIR__ . '/Maintenance.php';
 
class Protect extends Maintenance {
    public function __construct() {
        parent::__construct();
        $this->addDescription( 'Protect or unprotect a page 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() {
        $userName = $this->getOption( 'user', false );
        $reason = $this->getOption( 'reason', '' );
 
        $cascade = $this->hasOption( 'cascade' );
 
        $protection = "sysop";
        if ( $this->hasOption( 'semiprotect' ) ) {
            $protection = "autoconfirmed";
        } elseif ( $this->hasOption( 'unprotect' ) ) {
            $protection = "";
        }
 
        if ( $userName === false ) {
            $user = User::newSystemUser( 'Maintenance script', [ 'steal' => true ] );
        } else {
            $user = User::newFromName( $userName );
        }
        if ( !$user ) {
            $this->error( "Invalid username", true );
        }
 
        // @todo FIXME: This is reset 7 lines down.
        $restrictions = [ 'edit' => $protection, 'move' => $protection ];
 
        $t = Title::newFromText( $this->getArg() );
        if ( !$t ) {
            $this->error( "Invalid title", true );
        }
 
        $restrictions = [];
        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, [], $cascade, $reason, $user );
 
        if ( $status->isOK() ) {
            $this->output( "done\n" );
        } else {
            $this->output( "failed\n" );
        }
    }
}
 
$maintClass = "Protect";
require_once RUN_MAINTENANCE_IF_MAIN;