wikimedia/mediawiki-extensions-Wikibase

<?php

namespace Wikibase\DataModel\Services\Diff;

use InvalidArgumentException;
use Wikibase\DataModel\Entity\EntityDocument;
use Wikibase\DataModel\Entity\Item;
use Wikibase\DataModel\Services\Diff\Internal\FingerprintPatcher;
use Wikibase\DataModel\Services\Diff\Internal\SiteLinkListPatcher;

/**
 * @since 1.0
 *
 * @license GPL-2.0-or-later
 * @author Jeroen De Dauw < jeroendedauw@gmail.com >
 */
class ItemPatcher implements EntityPatcherStrategy {

    /**
     * @var FingerprintPatcher
     */
    private $fingerprintPatcher;

    /**
     * @var StatementListPatcher
     */
    private $statementListPatcher;

    /**
     * @var SiteLinkListPatcher
     */
    private $siteLinkListPatcher;

    public function __construct() {
        $this->fingerprintPatcher = new FingerprintPatcher();
        $this->statementListPatcher = new StatementListPatcher();
        $this->siteLinkListPatcher = new SiteLinkListPatcher();
    }

    /**
     * @param string $entityType
     *
     * @return bool
     */
    public function canPatchEntityType( $entityType ) {
        return $entityType === 'item';
    }

    /**
     * @param EntityDocument $entity
     * @param EntityDiff $patch
     *
     * @throws InvalidArgumentException
     */
    public function patchEntity( EntityDocument $entity, EntityDiff $patch ) {
        $this->assertIsItem( $entity );

        $this->patchItem( $entity, $patch );
    }

    private function assertIsItem( EntityDocument $item ) {
        if ( !( $item instanceof Item ) ) {
            throw new InvalidArgumentException( '$item must be an instance of Item' );
        }
    }

    private function patchItem( Item $item, EntityDiff $patch ) {
        $this->fingerprintPatcher->patchFingerprint( $item->getFingerprint(), $patch );

        if ( $patch instanceof ItemDiff ) {
            $item->setSiteLinkList( $this->siteLinkListPatcher->getPatchedSiteLinkList(
                $item->getSiteLinkList(),
                $patch->getSiteLinkDiff()
            ) );
        }

        $this->statementListPatcher->patchStatementList(
            $item->getStatements(),
            $patch->getClaimsDiff()
        );
    }

}