Seeder.php

<?php

/*
 * This file is part of the WouterJEloquentBundle package.
 *
 * (c) 2014 Wouter de Jong
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace WouterJ\EloquentBundle;

use Symfony\Component\DependencyInjection\ContainerInterface;
use Illuminate\Database\Seeder as BaseSeeder;
use Illuminate\Database\ConnectionInterface;

/**
 * @author Wouter J <wouter@wouterj.nl>
 */
class Seeder extends BaseSeeder
{
    /** @var null|ContainerInterface */
    protected $container;
    protected $seededClasses = array();
    /** @var ConnectionInterface */
    protected $connection;

    public function call($class)
	{
        $seeder = $this->resolve($class);
        $seeder->setConnection($this->connection);

        $seeder->run();
    }

    /**
     * {@inheritDoc}
     *
     * @return Seeder
     */
    public function resolve($class)
    {
        if ($this->getContainer()->has($class)) {
            $seeder = $this->getContainer()->get($class);
        } else {
            $seeder = new $class;
        }

        if (!$seeder instanceof self) {
            throw new \LogicException(sprintf('The seeder "%s" does not extend WouterJ\EloquentBundle\Seeder', get_class($seeder)));
        }
        $seeder->setSfContainer($this->getContainer());

        $this->addSeededClass($seeder);

        return $seeder;
    }

    public function setSfContainer(ContainerInterface $container)
    {
        $this->container = $container;
    }

    protected function getContainer()
    {
        return $this->container;
    }

    public function getSeedClasses()
    {
        return $this->seededClasses;
    }

    protected function addSeededClass($object)
    {
        $this->seededClasses[] = is_string($object) ? $object : get_class($object);
    }

    public function setConnection(ConnectionInterface $connection)
    {
        $this->connection = $connection;
    }
}