PageMoveCollection.php

<?php
declare( strict_types = 1 );
 
namespace MediaWiki\Extension\Translate\PageTranslation;
 
use MediaWiki\Title\Title;
use Status;
 
class PageMoveCollection {
    private PageMoveOperation $translatablePage;
    private array $translationPagePairs;
    private array $unitPagesPairs;
    private array $subpagesPairs;
    private array $talkpagesPairs;
    private array $translatableSubpages;
    private array $nonMovableSubpages;
 
    public function __construct(
        PageMoveOperation $translatablePage,
        array $translationPagePairs,
        array $unitPagesPairs,
        array $subpagesPairs,
        array $translatableSubpages
    ) {
        $this->translatablePage = $translatablePage;
        $this->translationPagePairs = $translationPagePairs;
        $this->unitPagesPairs = $unitPagesPairs;
        $this->subpagesPairs = $subpagesPairs;
        $this->translatableSubpages = $translatableSubpages;
 
        // Populate the talk pages from the various inputs.
        $this->talkpagesPairs = $this->getTalkpages(
            $this->translatablePage, ...$translationPagePairs, ...$unitPagesPairs, ...$subpagesPairs
        );
 
        $this->nonMovableSubpages = [];
    }
 
    public function getTranslatablePage(): PageMoveOperation {
        return $this->translatablePage;
    }
 
    public function getTranslationPagesPair(): array {
        return $this->translationPagePairs;
    }
 
    public function getUnitPagesPair(): array {
        return $this->unitPagesPairs;
    }
 
    public function getSubpagesPair(): array {
        return $this->subpagesPairs;
    }
 
    public function getTranslatableSubpages(): array {
        return $this->translatableSubpages;
    }
 
    public function getTranslationPages(): array {
        return $this->getOldPagesFromList( $this->translationPagePairs );
    }
 
    public function getUnitPages(): array {
        return $this->getOldPagesFromList( $this->unitPagesPairs );
    }
 
    public function getSubpages(): array {
        return $this->getOldPagesFromList( $this->subpagesPairs );
    }
 
    public function getListOfPages(): array {
        $pageList = [
            $this->translatablePage->getOldTitle()->getPrefixedText() =>
                $this->translatablePage->getNewTitle() ?
                    $this->translatablePage->getNewTitle()->getPrefixedText() : null
        ];
        $pageList = array_merge( $pageList, $this->getPagePairFromList( $this->translationPagePairs ) );
        $pageList = array_merge( $pageList, $this->getPagePairFromList( $this->unitPagesPairs ) );
        $pageList = array_merge( $pageList, $this->getPagePairFromList( $this->subpagesPairs ) );
        $pageList = array_merge( $pageList, $this->getPagePairFromList( $this->talkpagesPairs ) );
 
        return $pageList;
    }
 
    public function getListOfPagesToRedirect(): array {
        $redirects = [
            $this->translatablePage->getOldTitle()->getPrefixedText() =>
                    $this->translatablePage->shouldLeaveRedirect()
        ];
 
        $redirects = array_merge( $redirects, $this->getLeaveRedirectPairFromList( $this->translationPagePairs ) );
        $redirects = array_merge( $redirects, $this->getLeaveRedirectPairFromList( $this->unitPagesPairs ) );
        $redirects = array_merge( $redirects, $this->getLeaveRedirectPairFromList( $this->subpagesPairs ) );
        $redirects = array_merge( $redirects, $this->getLeaveRedirectPairFromList( $this->talkpagesPairs ) );
 
        return $redirects;
    }
 
    public function addNonMovableSubpage( Title $page, Status $invalidityReason ): bool {
        $this->nonMovableSubpages[ $page->getPrefixedText() ] = $invalidityReason;
 
        return $this->removeFromSubpageList( $page );
    }
 
    public function getNonMovableSubpages(): array {
        return $this->nonMovableSubpages;
    }
 
    private function removeFromSubpageList( Title $page ): bool {
        foreach ( $this->subpagesPairs as $key => $pair ) {
            if ( $pair->getOldTitle() === $page ) {
                unset( $this->subpagesPairs[ $key ] );
                return true;
            }
        }
        return false;
    }
 
    private function getOldPagesFromList( array $pagePairs ): array {
        $oldTitles = [];
        foreach ( $pagePairs as $pair ) {
            $oldTitles[] = $pair->getOldTitle();
        }
 
        return $oldTitles;
    }
 
    private function getPagePairFromList( array $pagePairs ): array {
        $pairs = [];
        foreach ( $pagePairs as $pair ) {
            $pairs[ $pair->getOldTitle()->getPrefixedText() ] =
                $pair->getNewTitle() ? $pair->getNewTitle()->getPrefixedText() : null;
        }
 
        return $pairs;
    }
 
    private function getLeaveRedirectPairFromList( array $pagePairs ): array {
        $pairs = [];
        foreach ( $pagePairs as $pair ) {
            if ( $pair->shouldLeaveRedirect() ) {
                $pairs[ $pair->getOldTitle()->getPrefixedText() ] = true;
                $talkpage = $pair->getOldTalkPage();
                if ( $talkpage ) {
                    $pairs[ $talkpage->getPrefixedText() ] = true;
                }
            }
        }
 
        return $pairs;
    }
 
    private function getTalkpages( PageMoveOperation ...$allMoveOperations ): array {
        $talkpagesPairs = [];
        foreach ( $allMoveOperations as $moveOperation ) {
            if ( $moveOperation->hasTalkpage() ) {
                $talkpagesPairs[] = new PageMoveOperation(
                    $moveOperation->getOldTalkpage(), $moveOperation->getNewTalkpage()
                );
            }
        }
        return $talkpagesPairs;
    }
}