adamwathan/faktory

<?php namespace AdamWathan\Faktory;

use AdamWathan\Faktory\Factory\Factory;
use AdamWathan\Faktory\Factory\FactoryProxy;
use Closure;

class Faktory
{
    protected $factories = [];

    public function define($model, $name, $definitionCallback = null)
    {
        if ($name instanceof Closure) {
            $definitionCallback = $name;
            $name = $model;
        }

        $factory = Factory::make($model, $this);
        $this->addFactory($name, $factory);
        $definitionCallback($factory);
    }

    protected function addFactory($name, $factory)
    {
        $this->factories[$name] = $factory;
    }

    public function build($name, $attributes = [])
    {
        return $this->getFactory($name)->build($attributes);
    }

    public function create($name, $attributes = [])
    {
        return $this->getFactory($name)->create($attributes);
    }

    public function buildMany($name, $count, $attributes = [])
    {
        return $this->getFactory($name)->buildMany($count, $attributes);
    }

    public function createMany($name, $count, $attributes = [])
    {
        return $this->getFactory($name)->createMany($count, $attributes);
    }

    public function getFactory($name)
    {
        return $this->getProxyFactory($name);
    }

    protected function getProxyFactory($name)
    {
        return new FactoryProxy(function () use ($name) {
            return $this->fetchFactory($name);
        });
    }

    protected function fetchFactory($name)
    {
        if (! isset($this->factories[$name])) {
            throw new FactoryNotRegisteredException("'{$name}' is not a registered factory.");
        }
        return $this->factories[$name];
    }
}