Collection-recursive AssetReference for Assetic

AssetReference.php

<?php

// Implements \IteratorAggregate and AssetCollectionInterface instead of only AssetInterface
class AssetReference implements \IteratorAggregate, AssetCollectionInterface
{
    /* [snip, snip] */

    public function add(AssetInterface $asset)
    {
        $asset = $this->am->get($this->name);

        if($asset instanceof AssetCollectionInterface) {
            return $asset->add($asset);
        }
    }

    public function all()
    {
        $this->flushFilters();

        $asset = $this->am->get($this->name);

        if($asset instanceof AssetCollectionInterface) {
            return $asset->all();
        } else {
            // Simply dereference
            return [ $asset ];
        }
    }

    public function removeLeaf(AssetInterface $leaf, $graceful = false)
    {
        $asset = $this->am->get($this->name);

        if($asset instanceof AssetCollectionInterface) {
            return $asset->removeLeaf($leaf, $graceful);
        }
    }

    public function replaceLeaf(AssetInterface $needle, AssetInterface $replacement, $graceful = false)
    {
        $asset = $this->am->get($this->name);

        if($asset instanceof AssetCollectionInterface) {
            return $asset->replaceLeaf($needle, $replacement, $graceful);
        }
    }

    public function getIterator()
    {
        $asset = $this->am->get($this->name);

        if($asset instanceof \IteratorAggregate) {
            return $asset->getIterator();
        } else {
            return new \ArrayIterator([$asset]);
        }
    }

    /* [snip, snip] */
}