wikimedia/mediawiki-extensions-CirrusSearch

<?php

namespace CirrusSearch\Maintenance\Validators;

use CirrusSearch\Maintenance\Printer;
use Elastica\Client;
use MediaWiki\Status\Status;

class IndexAllAliasValidator extends IndexAliasValidator {
    /**
     * @var string prefix of names of indices that should be removed
     */
    protected $shouldRemovePrefix;

    /**
     * @param Client $client
     * @param string $aliasName
     * @param string $specificIndexName
     * @param bool $startOver
     * @param string $type
     * @param Printer|null $out
     */
    public function __construct( Client $client, $aliasName, $specificIndexName, $startOver, $type, Printer $out = null ) {
        parent::__construct( $client, $aliasName, $specificIndexName, $startOver, $out );
        $this->shouldRemovePrefix = $type;
    }

    /**
     * @param string[] $add
     * @param string[] $remove
     * @return Status
     */
    protected function updateIndices( array $add, array $remove ) {
        $data = [];

        $this->output( "alias not already assigned to this index..." );

        // We'll remove the all alias from the indices that we're about to delete while
        // we add it to this index.  Elastica doesn't support this well so we have to
        // build the request to Elasticsearch ourselves.

        foreach ( $add as $indexName ) {
            $data['actions'][] = [ 'add' => [ 'index' => $indexName, 'alias' => $this->aliasName ] ];
        }

        foreach ( $remove as $indexName ) {
            $data['actions'][] = [ 'remove' => [ 'index' => $indexName, 'alias' => $this->aliasName ] ];
        }

        $this->client->request( '_aliases', \Elastica\Request::POST, $data );
        $this->output( "corrected\n" );

        return parent::updateIndices( $add, $remove );
    }

    /**
     * @param string $name
     * @return bool
     */
    protected function shouldRemoveFromAlias( $name ) {
        // Only if the name starts with the type being processed otherwise we'd
        // remove the content index from the all alias.
        return str_starts_with( $name, $this->shouldRemovePrefix );
    }
}