added DecoratorExtension

Author: dg

src/DI/Extensions/DecoratorExtension.php

@@ -0,0 +1,72 @@
+<?php
+
+/**
+ * This file is part of the Nette Framework (http://nette.org)
+ * Copyright (c) 2004 David Grudl (http://davidgrudl.com)
+ */
+
+namespace Nette\DI\Extensions;
+
+use Nette,
+	Nette\DI\Statement;
+
+
+/**
+ * Decorators for services.
+ *
+ * @author     David Grudl
+ */
+class DecoratorExtension extends Nette\DI\CompilerExtension
+{
+	public $defaults = array(
+		'setup' => array(),
+		'tags' => array(),
+		'inject' => NULL,
+	);
+
+
+	public function beforeCompile()
+	{
+		foreach ($this->getConfig() as $class => $info) {
+			$this->validate($info, $this->defaults, $this->prefix($class));
+			$info += $this->defaults;
+			if ($info['inject'] !== NULL) {
+				$info['tags'][InjectExtension::TAG_INJECT] = $info['inject'];
+			}
+			$this->addSetups($class, $info['setup']);
+			$this->addTags($class, $info['tags']);
+		}
+	}
+
+
+	public function addSetups($type, array $setups)
+	{
+		$builder = $this->getContainerBuilder();
+		foreach ($builder->findByType($type, FALSE) as $name) {
+			foreach ($setups as $setup) {
+				$builder->getDefinition($name)->addSetup($setup);
+			}
+		}
+	}
+
+
+	public function addTags($type, array $tags)
+	{
+		$tags = Nette\Utils\Arrays::normalize($tags, TRUE);
+		$builder = $this->getContainerBuilder();
+		foreach ($builder->findByType($type, FALSE) as $name) {
+			$def = $builder->getDefinition($name);
+			$def->setTags($def->getTags() + $tags);
+		}
+	}
+
+
+	private function validate(array $config, array $expected, $name)
+	{
+		if ($extra = array_diff_key($config, $expected)) {
+			$extra = implode(", $name.", array_keys($extra));
+			throw new Nette\InvalidStateException("Unknown option $name.$extra.");
+		}
+	}
+
+}

tests/DI/Compiler.extension.decorator.phpt

@@ -0,0 +1,68 @@
+<?php
+
+/**
+ * Test: Nette\DI\Compiler: service decorators.
+ */
+
+use Nette\DI,
+	Nette\DI\Statement,
+	Tester\Assert;
+
+
+require __DIR__ . '/../bootstrap.php';
+
+
+interface Iface
+{}
+
+
+class Service extends Nette\Object implements Iface
+{
+	public $setup;
+
+	function setup($a = NULL)
+	{
+		$this->setup[] = $a;
+	}
+}
+
+
+$compiler = new DI\Compiler;
+$compiler->addExtension('decorator', new Nette\DI\Extensions\DecoratorExtension);
+$container = createContainer($compiler, '
+decorator:
+	Nette\Object:
+		setup:
+			- setup(Object)
+		inject: yes
+
+	Iface:
+		setup:
+			- setup(Iface)
+			- setup
+		tags: [b, tag: 1]
+
+services:
+	one:
+		class: Service
+		tags: [a, tag: 2]
+		setup:
+			- setup(Service)
+');
+
+
+$builder = $compiler->getContainerBuilder();
+
+Assert::same(
+	array('a' => TRUE, 'tag' => 2, 'inject' => TRUE, 'b' => TRUE),
+	$builder->getDefinition('one')->tags
+);
+
+Assert::true( $builder->getDefinition('one')->getTag('inject') );
+
+Assert::equal( array(
+	new Statement(array('@self', 'setup'), array('Service')),
+	new Statement(array('@self', 'setup'), array('Object')),
+	new Statement(array('@self', 'setup'), array('Iface')),
+	new Statement(array('@self', 'setup')),
+), $builder->getDefinition('one')->getSetup() );