Include pattern specific data

Author: aleksip

src/PatternLab/PatternEngine/Twig/IncludeNodeVisitor.php

@@ -0,0 +1,33 @@
+<?php
+
+namespace PatternLab\PatternEngine\Twig;
+
+use PatternLab\Data;
+
+class IncludeNodeVisitor extends \Twig_BaseNodeVisitor
+{
+    protected function doEnterNode(\Twig_Node $node, \Twig_Environment $env)
+    {
+        return $node;
+    }
+
+    protected function doLeaveNode(\Twig_Node $node, \Twig_Environment $env)
+    {
+        if ($node instanceof \Twig_Node_Include) {
+            if ($node->hasNode('expr') && $node->getNode('expr')->hasAttribute('value')) {
+                $patternStoreKey = $node->getNode('expr')->getAttribute('value');
+                $data = Data::getPatternSpecificData($patternStoreKey);
+                $dataNode = new PatternDataIncludeNode($node, $data);
+
+                return $dataNode;
+            }
+        }
+
+        return $node;
+    }
+
+    public function getPriority()
+    {
+        return 0;
+    }
+}

src/PatternLab/PatternEngine/Twig/Loaders/PatternLoader.php

@@ -14,6 +14,7 @@
 namespace PatternLab\PatternEngine\Twig\Loaders;
 
 use \PatternLab\Config;
+use \PatternLab\PatternEngine\Twig\IncludeNodeVisitor;
 use \PatternLab\PatternEngine\Twig\Loaders\Twig\PatternPartialLoader as Twig_Loader_PatternPartialLoader;
 use \PatternLab\PatternEngine\Twig\Loaders\Twig\PatternStringLoader as Twig_Loader_PatternStringLoader;
 use \PatternLab\PatternEngine\Loader;
@@ -66,7 +67,9 @@ public function __construct($options = array()) {
 		$this->instance       = TwigUtil::loadDateFormats($this->instance);
 		$this->instance       = TwigUtil::loadDebug($this->instance);
 		$this->instance       = TwigUtil::loadMacros($this->instance);
-		
+
+		// add node visitor
+		$this->instance->addNodeVisitor(new IncludeNodeVisitor());
 	}
 
 	/**

src/PatternLab/PatternEngine/Twig/PatternDataIncludeNode.php

@@ -0,0 +1,40 @@
+<?php
+
+namespace PatternLab\PatternEngine\Twig;
+
+class PatternDataIncludeNode extends \Twig_Node_Include
+{
+    protected $data;
+
+    public function __construct(\Twig_Node_Include $originalNode, $data)
+    {
+        \Twig_Node::__construct($originalNode->nodes, $originalNode->attributes, $originalNode->lineno, $originalNode->tag);
+        $this->data = $data;
+    }
+
+    protected function addTemplateArguments(\Twig_Compiler $compiler)
+    {
+        if (null === $this->getNode('variables')) {
+            if (false === $this->getAttribute('only')) {
+                $compiler
+                    ->raw('array_merge($context, ')
+                    ->repr($this->data)
+                    ->raw(')')
+                ;
+            }
+            else {
+                $compiler->raw('array()');
+            }
+        } elseif (false === $this->getAttribute('only')) {
+            $compiler
+                ->raw('array_merge($context, ')
+                ->repr($this->data)
+                ->raw(', ')
+                ->subcompile($this->getNode('variables'))
+                ->raw(')')
+            ;
+        } else {
+            $compiler->subcompile($this->getNode('variables'));
+        }
+    }
+}