<?php

namespace BookStack\References;

use BookStack\Entities\Models\Entity;

class ReferenceChangeContext
{
    /**
     * Entity pairs where the first is the old entity and the second is the new entity.
     * @var array<array{0: Entity, 1: Entity}>
     */
    protected array $changes = [];

    public function add(Entity $oldEntity, Entity $newEntity): void
    {
        $this->changes[] = [$oldEntity, $newEntity];
    }

    /**
     * Get all the new entities from the changes.
     */
    public function getNewEntities(): array
    {
        return array_column($this->changes, 1);
    }

    /**
     * Get all the old entities from the changes.
     */
    public function getOldEntities(): array
    {
        return array_column($this->changes, 0);
    }

    public function getNewForOld(Entity $oldEntity): ?Entity
    {
        foreach ($this->changes as [$old, $new]) {
            if ($old->id === $oldEntity->id && $old->type === $oldEntity->type) {
                return $new;
            }
        }
        return null;
    }
}