Adds setAsync Nunjucks shortcode. set isn’t async friendly. Resto…

…res `this.ctx` in Nunjucks shortcodes. Fixes #2108. Fixes #1584. Fixes #1596.

src/Engines/Nunjucks.js

@@ -155,6 +155,12 @@ class Nunjucks extends TemplateEngine {
     this.addFilters(this.config.nunjucksAsyncFilters, true);
 
     // TODO these all go to the same place (addTag), add warnings for overwrites
+    // TODO(zachleat): variableName should work with quotes or without quotes (same as {% set %})
+    this.addPairedShortcode("setAsync", function (content, variableName) {
+      this.ctx[variableName] = content;
+      return "";
+    });
+
     this.addCustomTags(this.config.nunjucksTags);
     this.addAllShortcodes(this.config.nunjucksShortcodes);
     this.addAllPairedShortcodes(this.config.nunjucksPairedShortcodes);
@@ -211,6 +217,7 @@ class Nunjucks extends TemplateEngine {
   static _normalizeShortcodeContext(context) {
     let obj = {};
     if (context.ctx && context.ctx.page) {
+      obj.ctx = context.ctx;
       obj.page = context.ctx.page;
     }
     return obj;

src/Plugins/RenderPlugin.js

@@ -193,6 +193,7 @@ function EleventyPlugin(eleventyConfig, options = {}) {
 
         let normalizedContext = {};
         if (context.ctx && context.ctx.page) {
+          normalizedContext.ctx = context.ctx;
           normalizedContext.page = context.ctx.page;
         }
 

test/TemplateRenderPluginTest.js

@@ -200,3 +200,12 @@ test("Capture nunjucks render output to a liquid variable", async (t) => {
   );
   t.is(html, `4`);
 });
+
+// Idea from https://twitter.com/raymondcamden/status/1460961620247650312
+// Possibly blocked by async in {% set %} https://github.com/mozilla/nunjucks/issues/815
+test("Capture liquid render output to a njk variable", async (t) => {
+  let html = await getTestOutputForFile(
+    "./test/stubs-render-plugin/capture-liquid.njk"
+  );
+  t.is(html, `4`);
+});

test/stubs-render-plugin/capture-liquid.njk

@@ -0,0 +1,11 @@
+---
+argData:
+  num: 2
+---
+{% setAsync "liquidOutput" %}
+{% renderTemplate "liquid", argData %}
+{% assign liquidVar = num | times: 2 %}
+{{ liquidVar }}
+{% endrenderTemplate %}
+{% endsetAsync %}
+{{ liquidOutput }}