Add extend method

README.md

@@ -26,6 +26,8 @@ into arbitrary code.
 - ConsumerInitializer = ClassName | Consumer
 - ServiceInitializer = ConsumerInitializer | Value
 
+Construction
+------------
 Container is initialized with raw values, no Initalizers allowed:
 
 ```php
@@ -40,6 +42,8 @@ You can set a name which will be used to refer to container itself:
 $container = new SD\DependencyInjection\Container([], 'container');
 ```
 
+Registering services
+--------------------
 Services are registered with ServiceInitializers:
 
 ```php
@@ -62,6 +66,19 @@ $container->register('helloWorld', function ($name) {
 $container->register('name', $container->value('Skywalker'));
 ```
 
+You can extend registered services:
+
+```php
+$container->register('currency', SD\Currency\Repository::class);
+$container->extend('currency', function ($container, $currency) {
+    $store = $container->produce(SD\CurrencyStore\Wpdb::class);
+    $currency->setStore($store);
+    return $currency;
+});
+```
+
+Consumer production
+-------------------
 Consumers are produced with ConsumerInitializers (Value is not supported):
 
 ```php

src/Container.php

@@ -21,6 +21,20 @@ public function register($name, $initializer) {
         }
     }
 
+    public function extend($name, $extender) {
+        if (isset($this->services[$name])) {
+            $this->services[$name] = $this->produce($extender);
+        } elseif (isset($this->initializers[$name])) {
+            $initializer = $this->initializers[$name];
+            $this->initializers[$name] = function () use ($extender, $initializer, $name) {
+                $this->services[$name] = $this->produce($initializer);
+                return $this->inject($extender);
+            };
+        } else {
+            throw new Exception("Cannot extend unknown service $name");
+        }
+    }
+
     public function value($value) {
         return new Value($value);
     }

tests/ContainerTest.php

@@ -14,7 +14,7 @@ public function testRegisterClassName() {
             ],
             'container'
         );
-        $serviceName = 'hello_world';
+        $serviceName = 'helloWorld';
         $className = HelloWorldService::class;
         $container->register($serviceName, $className);
         $service = $container->get($serviceName);
@@ -30,4 +30,23 @@ public function testDetectCyclicDependencies() {
         $this->expectException(Exception::class);
         $container->get('a');
     }
+
+    public function testExtend() {
+        $name1 = 'Jar Jar Binks';
+        $name2 = 'Palpatine';
+        $container = new Container(
+            [
+                'name' => $name1,
+            ],
+            'container'
+        );
+        $serviceName = 'helloWorld';
+        $container->register($serviceName, HelloWorldService::class);
+        $container->extend($serviceName, function ($helloWorld) use ($name2) {
+            $helloWorld->setName($name2);
+            return $helloWorld;
+        });
+        $service = $container->get($serviceName);
+        $this->assertEquals($name2, $service->getName(), 'Must return modified name');
+    }
 }

tests/HelloWorldService.php

@@ -9,15 +9,19 @@ class HelloWorldService implements DeclarerInterface {
 
     private $name;
 
-    public function __construct($name) {
+    public function __construct(string $name) {
         $this->name = $name;
     }
 
     public function declareDependencies() {
         return ['container'];
     }
 
-    public function getName() {
+    public function setName(string $name) {
+        $this->name = $name;
+    }
+
+    public function getName(): string {
         return $this->name;
     }