createAndPromote.php

Go to the documentation of this file.

<?php
require_once __DIR__ . '/Maintenance.php';

class CreateAndPromote extends Maintenance {
    private static $permitRoles = array( 'sysop', 'bureaucrat', 'bot' );

    public function __construct() {
        parent::__construct();
        $this->mDescription = "Create a new user account and/or grant it additional rights";
        $this->addOption(
            'force',
            'If acccount exists already, just grant it rights or change password.'
        );
        foreach ( self::$permitRoles as $role ) {
            $this->addOption( $role, "Add the account to the {$role} group" );
        }
        $this->addArg( "username", "Username of new user" );
        $this->addArg( "password", "Password to set (not required if --force is used)", false );
    }

    public function execute() {
        $username = $this->getArg( 0 );
        $password = $this->getArg( 1 );
        $force = $this->hasOption( 'force' );
        $inGroups = array();

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

        $exists = ( 0 !== $user->idForName() );

        if ( $exists && !$force ) {
            $this->error( "Account exists. Perhaps you want the --force option?", true );
        } elseif ( !$exists && !$password ) {
            $this->error( "Argument <password> required!", false );
            $this->maybeHelp( true );
        } elseif ( $exists ) {
            $inGroups = $user->getGroups();
        }

        $promotions = array_diff(
            array_filter( self::$permitRoles, array( $this, 'hasOption' ) ),
            $inGroups
        );

        if ( $exists && !$password && count( $promotions ) === 0 ) {
            $this->output( "Account exists and nothing to do.\n" );

            return;
        } elseif ( count( $promotions ) !== 0 ) {
            $promoText = "User:{$username} into " . implode( ', ', $promotions ) . "...\n";
            if ( $exists ) {
                $this->output( wfWikiID() . ": Promoting $promoText" );
            } else {
                $this->output( wfWikiID() . ": Creating and promoting $promoText" );
            }
        }

        if ( $password ) {
            # Try to set the password
            try {
                $user->setPassword( $password );
                if ( $exists ) {
                    $this->output( "Password set.\n" );
                    $user->saveSettings();
                }
            } catch ( PasswordError $pwe ) {
                $this->error( $pwe->getText(), true );
            }
        }

        if ( !$exists ) {
            # Insert the account into the database
            $user->addToDatabase();
            $user->saveSettings();
        }

        # Promote user
        array_map( array( $user, 'addGroup' ), $promotions );

        if ( !$exists ) {
            # Increment site_stats.ss_users
            $ssu = new SiteStatsUpdate( 0, 0, 0, 0, 1 );
            $ssu->doUpdate();
        }

        $this->output( "done.\n" );
    }
}

$maintClass = "CreateAndPromote";
require_once RUN_MAINTENANCE_IF_MAIN;