Merge f016262 into 7da730a

NightlyCommit · May 11, 2020 · 129e23a · 129e23a
2 parents 7da730a + f016262
commit 129e23a
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 1 deletion.
diff --git a/src/lib/Transpiler.ts b/src/lib/Transpiler.ts
@@ -13,7 +13,7 @@ import {
     TwingNodeExpressionName,
     TwingNodeFor,
     TwingNodeIf,
-    TwingNodePrint,
+    TwingNodePrint, TwingNodeSet,
     TwingNodeText,
     TwingNodeType,
     TwingSource,
@@ -196,6 +196,19 @@ export class Transpiler {
         return `${node.getAttribute('name')}(${parameters.join(',')})`;
     }
 
+    private transpileSetNode(node: TwingNodeSet): string {
+        let results: Array<string> = [];
+
+        const names = node.getNode('names');
+        const values = node.getNode('values');
+
+        for (let [k, v] of names.getNodes()) {
+            results.push(`<?php $${v.getAttribute('name')} = ${this.transpileNode(values.getNode(k))} ?>`);
+        }
+
+        return results.join('\n');
+    }
+
     private transpileNode(node: TwingNode, raw: boolean = false): string {
         if (node.getType() === TwingNodeType.PRINT) {
             return this.transpilePrintNode(node as TwingNodePrint);
@@ -245,6 +258,10 @@ export class Transpiler {
             return this.transpileExpressionFunctionNode(node as TwingNodeExpressionFunction);
         }
 
+        if (node.getType() === TwingNodeType.SET) {
+            return this.transpileSetNode(node as TwingNodeSet);
+        }
+
         let results = [];
 
         for (let [name, child] of node.getNodes()) {

diff --git a/test/lib/Transpiler/test.ts b/test/lib/Transpiler/test.ts
@@ -85,6 +85,16 @@ bar*/ ?>`);
             test.end();
         });
 
+        test.test('supports set', (test) => {
+            test.same(transpiler.transpile('{% set foo = "bar" %}'), '<?php $foo = "bar" ?>');
+            test.same(transpiler.transpile('{% set foo, bar = "foo", "bar" %}'), `<?php $foo = "foo" ?>
+<?php $bar = "bar" ?>`);
+            test.same(transpiler.transpile('{% set foo = lorem.ipsum %}'), '<?php $foo = $lorem->ipsum ?>');
+            test.same(transpiler.transpile('{% set foo = lorem.ipsum() %}'), '<?php $foo = $lorem->ipsum() ?>');
+
+            test.end();
+        });
+
         test.end();
     });