Add PatternDataNodeVisitor

Author: aleksip

composer.json

@@ -23,6 +23,7 @@
         }
     },
     "require": {
-        "php": ">=5.5.9"
+        "php": ">=5.5.9",
+        "pattern-lab/patternengine-twig": "0.*"
     }
 }

src/aleksip/DataTransformPlugin/PatternLabListener.php

@@ -2,14 +2,17 @@
 
 namespace aleksip\DataTransformPlugin;
 
+use aleksip\DataTransformPlugin\Twig\PatternDataNodeVisitor;
 use PatternLab\Listener;
 use PatternLab\PatternData\Event;
+use PatternLab\PatternEngine\Twig\TwigUtil;
 
 class PatternLabListener extends Listener
 {
     public function __construct()
     {
         $this->addListener('patternData.codeHelperStart', 'runHelper');
+        $this->addListener('twigPatternLoader.customize', 'addNodeVisitor');
     }
 
     public function runHelper(Event $event)
@@ -18,4 +21,11 @@ public function runHelper(Event $event)
         $helper = new Helper($options);
         $helper->run();
     }
+
+    public function addNodeVisitor()
+    {
+        $instance = TwigUtil::getInstance();
+        $instance->addNodeVisitor(new PatternDataNodeVisitor());
+        TwigUtil::setInstance($instance);
+    }
 }

src/aleksip/DataTransformPlugin/Twig/PatternDataEmbedNode.php

@@ -0,0 +1,23 @@
+<?php
+
+namespace aleksip\DataTransformPlugin\Twig;
+
+class PatternDataEmbedNode extends \Twig_Node_Embed
+{
+    use PatternDataNodeTrait;
+
+    public function __construct(\Twig_Node_Embed $originalNode, $data)
+    {
+        parent::__construct(
+          $originalNode->getAttribute('filename'),
+          $originalNode->getAttribute('index'),
+          $originalNode->getNode('variables'),
+          $originalNode->getAttribute('only'),
+          $originalNode->getAttribute('ignore_missing'),
+          $originalNode->getLine(),
+          $originalNode->getNodeTag()
+        );
+
+        $this->data = $data;
+    }
+}

src/aleksip/DataTransformPlugin/Twig/PatternDataIncludeNode.php

@@ -0,0 +1,22 @@
+<?php
+
+namespace aleksip\DataTransformPlugin\Twig;
+
+class PatternDataIncludeNode extends \Twig_Node_Include
+{
+    use PatternDataNodeTrait;
+
+    public function __construct(\Twig_Node_Include $originalNode, $data)
+    {
+        parent::__construct(
+          $originalNode->getNode('expr'),
+          $originalNode->getNode('variables'),
+          $originalNode->getAttribute('only'),
+          $originalNode->getAttribute('ignore_missing'),
+          $originalNode->getLine(),
+          $originalNode->getNodeTag()
+        );
+
+        $this->data = $data;
+    }
+}

src/aleksip/DataTransformPlugin/Twig/PatternDataNodeTrait.php

@@ -0,0 +1,34 @@
+<?php
+
+namespace aleksip\DataTransformPlugin\Twig;
+
+trait PatternDataNodeTrait
+{
+    protected $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'));
+        }
+    }
+}

src/aleksip/DataTransformPlugin/Twig/PatternDataNodeVisitor.php

@@ -0,0 +1,38 @@
+<?php
+
+namespace aleksip\DataTransformPlugin\Twig;
+
+use PatternLab\Data;
+
+class PatternDataNodeVisitor 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);
+                if ($node instanceof \Twig_Node_Embed) {
+                    $dataNode = new PatternDataEmbedNode($node, $data);
+                }
+                else {
+                    $dataNode = new PatternDataIncludeNode($node, $data);
+                }
+
+                return $dataNode;
+            }
+        }
+
+        return $node;
+    }
+
+    public function getPriority()
+    {
+        return 0;
+    }
+}