wikimedia/mediawiki-extensions-Wikibase

<?php

namespace Wikibase\Repo\Maintenance;

use InvalidArgumentException;
use MediaWiki\Maintenance\Maintenance;
use MediaWiki\User\User;
use Wikibase\DataModel\Term\AliasesProvider;
use Wikibase\DataModel\Term\DescriptionsProvider;
use Wikibase\DataModel\Term\LabelsProvider;
use Wikibase\Lib\Store\LookupConstants;
use Wikibase\Lib\Store\StorageException;
use Wikibase\Lib\WikibaseSettings;
use Wikibase\Repo\WikibaseRepo;

$basePath = getenv( 'MW_INSTALL_PATH' ) !== false ? getenv( 'MW_INSTALL_PATH' ) : __DIR__ . '/../../../..';

require_once $basePath . '/maintenance/Maintenance.php';

/**
 * @license GPL-2.0-or-later
 */
class RemoveTermsInLanguage extends Maintenance {

    public function __construct() {
        parent::__construct();
        $this->addDescription( "Removes terms in the given language in the given entities." );

        $this->addOption( 'entity-id', 'Id of the entity', true, true, false, true );
        $this->addOption( 'language', 'Language to remove', true, true );
    }

    public function execute() {
        if ( !WikibaseSettings::isRepoEnabled() ) {
            $this->fatalError( "You need to have Wikibase enabled in order to use this maintenance script!\n" );
        }

        $idSerializations = $this->getOption( 'entity-id' );
        $language = $this->getOption( 'language' );

        $entityIdParser = WikibaseRepo::getEntityIdParser();
        $entityRevisionLookup = WikibaseRepo::getEntityRevisionLookup();
        $entityStore = WikibaseRepo::getEntityStore();
        $user = User::newSystemUser( User::MAINTENANCE_SCRIPT_USER, [ 'steal' => true ] );

        foreach ( $idSerializations as $idSerialization ) {

            try {
                $entityId = $entityIdParser->parse( $idSerialization );
            } catch ( InvalidArgumentException $e ) {
                $this->fatalError( "Invalid entity id: " . $idSerialization );
            }

            $entityRevision = $entityRevisionLookup->getEntityRevision(
                $entityId,
                0,
                 LookupConstants::LATEST_FROM_MASTER
            );

            if ( $entityRevision === null ) {
                throw new StorageException( "Could not load entity: " . $entityId->getSerialization() );
            }

            $entity = $entityRevision->getEntity();
            if ( $entity instanceof LabelsProvider ) {
                $entity->getLabels()->removeByLanguage( $language );
            }

            if ( $entity instanceof DescriptionsProvider ) {
                $entity->getDescriptions()->removeByLanguage( $language );
            }

            if ( $entity instanceof AliasesProvider ) {
                $entity->getAliasGroups()->removeByLanguage( $language );
            }

            $entityStore->saveEntity(
                $entity,
                'Removed terms in language ' . $language,
                $user,
                EDIT_UPDATE,
                $entityRevision->getRevisionId()
            );
        }

        $this->output( "Successfully removed terms in language $language from " . implode( ", ", $idSerializations ) . "\n" );
    }

}

$maintClass = RemoveTermsInLanguage::class;
require_once RUN_MAINTENANCE_IF_MAIN;