Fix issue #2

src/index.js

@@ -25,12 +25,15 @@ class Depper extends Transform {
             });
 
             let processToken = function (token, template) {
-                if (token.type == 'logic') {
+                if (token.type === 'logic') {
                     token = token.token;
 
                     switch (token.type) {
                         case 'Twig.logic.type.include':
-                        case 'Twig.logic.type.import': {
+                        case 'Twig.logic.type.import':
+                        case 'Twig.logic.type.extends':
+                        case 'Twig.logic.type.embed':
+                        case 'Twig.logic.type.use': {
                             token.stack.forEach(function (stackEntry) {
                                 switch (stackEntry.type) {
                                     case 'Twig.expression.type.string':

test/depper.js

@@ -4,7 +4,7 @@ const fs = require('fs');
 const path = require('path');
 
 tap.test('depper', function (test) {
-    test.plan(8);
+    test.plan(11);
 
     test.test('should handle Twig.logic.type.include', function (test) {
         let d = new Depper();
@@ -193,4 +193,70 @@ tap.test('depper', function (test) {
 
         d.end(entry);
     });
+
+    test.test('should handle Twig.logic.type.extends', function (test) {
+        let d = new Depper();
+        let entry = path.join(__dirname, '/fixtures/extends/entry.twig');
+
+        let rows = [];
+
+        d.on('data', function (row) {
+            rows.push(row);
+        });
+
+        d.on('finish', function () {
+            test.same(rows.sort(), [
+                path.join(__dirname, '/fixtures/extends/entry.twig'),
+                path.join(__dirname, '/fixtures/extends/import.twig')
+            ].sort());
+
+            test.end();
+        });
+
+        d.end(entry);
+    });
+
+    test.test('should handle Twig.logic.type.embed', function (test) {
+        let d = new Depper();
+        let entry = path.join(__dirname, '/fixtures/embed/entry.twig');
+
+        let rows = [];
+
+        d.on('data', function (row) {
+            rows.push(row);
+        });
+
+        d.on('finish', function () {
+            test.same(rows.sort(), [
+                path.join(__dirname, '/fixtures/embed/entry.twig'),
+                path.join(__dirname, '/fixtures/embed/import.twig')
+            ].sort());
+
+            test.end();
+        });
+
+        d.end(entry);
+    });
+
+    test.test('should handle Twig.logic.type.use', function (test) {
+        let d = new Depper();
+        let entry = path.join(__dirname, '/fixtures/use/entry.twig');
+
+        let rows = [];
+
+        d.on('data', function (row) {
+            rows.push(row);
+        });
+
+        d.on('finish', function () {
+            test.same(rows.sort(), [
+                path.join(__dirname, '/fixtures/use/entry.twig'),
+                path.join(__dirname, '/fixtures/use/import.twig')
+            ].sort());
+
+            test.end();
+        });
+
+        d.end(entry);
+    });
 });

test/fixtures/embed/entry.twig

@@ -0,0 +1,2 @@
+{% embed "import.twig" %}
+{% endembed %}

test/fixtures/embed/import.twig

@@ -0,0 +1 @@
+foo

test/fixtures/extends/entry.twig

@@ -0,0 +1 @@
+{% extends "import.twig" %}

test/fixtures/extends/import.twig

@@ -0,0 +1 @@
+foo

test/fixtures/use/entry.twig

@@ -0,0 +1 @@
+{% use "import.twig" %}

test/fixtures/use/import.twig

@@ -0,0 +1 @@
+foo