From f96a9f6f2b03070ce2307584e339554a969c0e95 Mon Sep 17 00:00:00 2001
From: Kotlyar Maksim <kotlyar.maksim@gmail.com>
Date: Thu, 23 Apr 2015 15:43:31 +0000
Subject: [PATCH] Auto register storages as extension in a gateways.

---
 src/Payum/Silex/DynamicRegistry.php | 31 +++++++++++++++++++++++++++++
 src/Payum/Silex/PayumProvider.php   |  3 ---
 2 files changed, 31 insertions(+), 3 deletions(-)
 create mode 100644 src/Payum/Silex/DynamicRegistry.php

diff --git a/src/Payum/Silex/DynamicRegistry.php b/src/Payum/Silex/DynamicRegistry.php
new file mode 100644
index 0000000..d2a176c
--- /dev/null
+++ b/src/Payum/Silex/DynamicRegistry.php
@@ -0,0 +1,31 @@
+<?php
+namespace Payum\Silex;
+
+use Payum\Core\Extension\StorageExtension;
+use Payum\Core\Gateway;
+use Payum\Core\Registry\DynamicRegistry as CoreDynamicRegistry;
+
+class DynamicRegistry extends CoreDynamicRegistry
+{
+    /**
+     * {@inheritDoc}
+     */
+    public function getGateway($name)
+    {
+        $gateway = parent::getGateway($name);
+
+        $this->addStorageToGateway($gateway);
+
+        return $gateway;
+    }
+
+    /**
+     * @param Gateway $gateway
+     */
+    protected function addStorageToGateway(Gateway $gateway)
+    {
+        foreach ($this->getStorages() as $storage) {
+            $gateway->addExtension(new StorageExtension($storage));
+        }
+    }
+}
diff --git a/src/Payum/Silex/PayumProvider.php b/src/Payum/Silex/PayumProvider.php
index 402282f..660ec3e 100644
--- a/src/Payum/Silex/PayumProvider.php
+++ b/src/Payum/Silex/PayumProvider.php
@@ -12,9 +12,6 @@
 use Payum\Core\Bridge\Symfony\Security\TokenFactory;
 use Payum\Core\Bridge\Twig\TwigFactory;
 use Payum\Core\GatewayFactory;
-use Payum\Core\GatewayFactoryInterface;
-use Payum\Core\Registry\DynamicRegistry;
-use Payum\Core\Registry\SimpleRegistry;
 use Payum\Core\Reply\ReplyInterface;
 use Payum\Core\Security\GenericTokenFactory;
 use Payum\Silex\Controller\AuthorizeController;