Fix CSRF Token

khanhicetea · Jun 27, 2016 · 4d3b1fd · 4d3b1fd
1 parent 19ae022
commit 4d3b1fd
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/Sifoni/Controller/Base.php b/src/Sifoni/Controller/Base.php
@@ -78,6 +78,6 @@ public function isFormValid($action = '', $token_name = '_token')
     {
         $token = new CsrfToken($action, $this->request->get($token_name));
 
-        return $this->app['form.csrf_provider']->isTokenValid($token);
+        return $this->app['csrf.token_manager']->isTokenValid($token);
     }
 }
diff --git a/src/Sifoni/Engine.php b/src/Sifoni/Engine.php
@@ -16,6 +16,8 @@
 use Sifoni\Provider\SessionServiceProvider;
 use Sifoni\Provider\HttpCacheServiceProvider;
 use Monolog\Logger;
+use Twig_SimpleFunction;
+use Twig_Environment;
 
 class Engine
 {
@@ -188,6 +190,17 @@ private function registerServices()
 
         if ($app['enabled_csrf']) {
             $app->register(new CsrfServiceProvider());
+
+            if ($app['enabled_twig']) {
+                $app['twig'] = $app->extend('twig', function (Twig_Environment $twig, $app) {
+                    $csrf_token = new Twig_SimpleFunction('csrf_token', function ($token_id) use ($app) {
+                        return $app['csrf.token_manager']->getToken($token_id);
+                    });
+                    $twig->addFunction($csrf_token);
+
+                    return $twig;
+                });
+            }
         }
 
         if ($app['enabled_capsule']) {