Add functionality to add functions with config

Author: kenjis

libraries/Twig.php

@@ -26,9 +26,9 @@ class Twig
 	];
 
 	/**
-	 * @var bool Whether added CodeIgniter functions or not
+	 * @var bool Whether functions are added or not
 	 */
-	private $add_ci_functions = FALSE;
+	private $functions_added = FALSE;
 
 	/**
 	 * @var Twig_Environment
@@ -49,6 +49,21 @@ public function __construct($params = [])
 		];
 
 		$this->config = array_merge($this->config, $params);
+
+		if (isset($params['functions_asis']))
+		{
+			$this->functions_asis = 
+				array_unique(
+					array_merge($this->functions_asis, $params['functions_asis'])
+				);
+		}
+		if (isset($params['functions_safe']))
+		{
+			$this->functions_safe = 
+				array_unique(
+					array_merge($this->functions_safe, $params['functions_safe'])
+				);
+		}
 	}
 
 	protected function resetTwig()
@@ -132,18 +147,18 @@ public function display($view, $params = [])
 	public function render($view, $params = [])
 	{
 		$this->createTwig();
-		// We call addCIFunctions() here, because we must call addCIFunctions()
+		// We call addFunctions() here, because we must call addFunctions()
 		// after loading CodeIgniter functions in a controller.
-		$this->addCIFunctions();
+		$this->addFunctions();
 
 		$view = $view . '.twig';
 		return $this->twig->render($view, $params);
 	}
 
-	protected function addCIFunctions()
+	protected function addFunctions()
 	{
 		// Runs only once
-		if ($this->add_ci_functions)
+		if ($this->functions_added)
 		{
 			return;
 		}
@@ -189,7 +204,7 @@ protected function addCIFunctions()
 			);
 		}
 
-		$this->add_ci_functions = TRUE;
+		$this->functions_added = TRUE;
 	}
 
 	/**

tests/libraries/TwigTest.php

@@ -44,7 +44,7 @@ public function testAddGlobal()
         $this->assertEquals('<title>Twig Test Site</title>'."\n", $output);
     }
 
-    public function testAddCIFunctionsRunsOnlyOnce()
+    public function testAddFunctionsRunsOnlyOnce()
     {
         $obj = new Twig(['paths' => __DIR__ . '/../templates/']);
 
@@ -53,26 +53,55 @@ public function testAddCIFunctionsRunsOnlyOnce()
         ];
 
         $ref_obj = new ReflectionObject($obj);
-        $ref_property = $ref_obj->getProperty('add_ci_functions');
+        $ref_property = $ref_obj->getProperty('functions_added');
         $ref_property->setAccessible(true);
-        $add_ci_functions = $ref_property->getValue($obj);
-        $this->assertEquals(false, $add_ci_functions);
+        $functions_added = $ref_property->getValue($obj);
+        $this->assertEquals(false, $functions_added);
 
         $output = $obj->render('welcome', $data);
 
         $ref_obj = new ReflectionObject($obj);
-        $ref_property = $ref_obj->getProperty('add_ci_functions');
+        $ref_property = $ref_obj->getProperty('functions_added');
         $ref_property->setAccessible(true);
-        $add_ci_functions = $ref_property->getValue($obj);
-        $this->assertEquals(true, $add_ci_functions);
+        $functions_added = $ref_property->getValue($obj);
+        $this->assertEquals(true, $functions_added);
 
         // Calls render() twice
         $output = $obj->render('welcome', $data);
 
         $ref_obj = new ReflectionObject($obj);
-        $ref_property = $ref_obj->getProperty('add_ci_functions');
+        $ref_property = $ref_obj->getProperty('functions_added');
         $ref_property->setAccessible(true);
-        $add_ci_functions = $ref_property->getValue($obj);
-        $this->assertEquals(true, $add_ci_functions);
+        $functions_added = $ref_property->getValue($obj);
+        $this->assertEquals(true, $functions_added);
     }
+
+    public function testFunctionAsIs()
+    {
+        $obj = new Twig([
+            'paths' => __DIR__ . '/../templates/',
+            'functions_asis' => ['md5'],
+            'cache' => false,
+        ]);
+        
+        $output = $obj->render('functions_asis');
+        $this->assertEquals('900150983cd24fb0d6963f7d28e17f72'."\n", $output);
+    }
+
+    public function testFunctionSafe()
+    {
+        $obj = new Twig([
+            'paths' => __DIR__ . '/../templates/',
+            'functions_safe' => ['test_safe'],
+            'cache' => false,
+        ]);
+        
+        $output = $obj->render('functions_safe');
+        $this->assertEquals('<s>test</s>'."\n", $output);
+    }
+}
+
+function test_safe()
+{
+    return '<s>test</s>';
 }

tests/templates/functions_asis.twig

@@ -0,0 +1 @@
+{{ md5('abc') }}

tests/templates/functions_safe.twig

@@ -0,0 +1 @@
+{{ test_safe() }}