Control.php

<?php
namespace Ytnuk\Shop;

use Nette;
use Ytnuk;

final class Control
	extends Ytnuk\Application\Control
{

	/**
	 * @var Category\Control\Factory
	 */
	private $categoryControl;

	/**
	 * @var Product\Control\Factory
	 */
	private $productControl;

	/**
	 * @var Category\Entity
	 */
	private $category;

	/**
	 * @var Product\Entity
	 */
	private $product;

	public function __construct(
		Category\Control\Factory $categoryControl,
		Product\Control\Factory $productControl
	) {
		parent::__construct();
		$this->categoryControl = $categoryControl;
		$this->productControl = $productControl;
	}

	public function setCategory(Ytnuk\Shop\Category\Entity $category)
	{
		$this->category = $category;
		unset($this['category']);
	}

	public function setProduct(Ytnuk\Shop\Product\Entity $product)
	{
		$this->product = $product;
		unset($this['product']);
	}

	protected function createComponentCategory() : Category\Control
	{
		return $this->categoryControl->create($this->category ? : new Category\Entity);
	}

	protected function createComponentProduct() : Product\Control
	{
		return $this->productControl->create($this->product ? : new Product\Entity);
	}
}