From 96b2d33423329477d811b0cfdb461941c3f53232 Mon Sep 17 00:00:00 2001
From: Dragory <dragory.up@gmail.com>
Date: Sat, 16 Mar 2019 17:54:05 +0200
Subject: [PATCH] templateFormatter: add cases() template function; Tags: add
 set/get tag functions to store/retrieve variables within tags

---
 src/plugins/Tags.ts      | 8 ++++++++
 src/templateFormatter.ts | 6 ++++++
 2 files changed, 14 insertions(+)

diff --git a/src/plugins/Tags.ts b/src/plugins/Tags.ts
index 061819cc..fab77e4b 100644
--- a/src/plugins/Tags.ts
+++ b/src/plugins/Tags.ts
@@ -196,9 +196,17 @@ export class TagsPlugin extends ZeppelinPlugin<ITagsPluginConfig, ITagsPluginPer
 
     // Format the string
     try {
+      const dynamicVars = {};
       body = await renderTemplate(body, {
         args: tagArgs,
         ...this.tagFunctions,
+        set(name, val) {
+          if (typeof name !== "string") return;
+          dynamicVars[name] = val;
+        },
+        get(name) {
+          return dynamicVars[name] == null ? "" : dynamicVars[name];
+        },
       });
     } catch (e) {
       if (e instanceof TemplateParseError) {
diff --git a/src/templateFormatter.ts b/src/templateFormatter.ts
index e05dd389..b5d5af49 100644
--- a/src/templateFormatter.ts
+++ b/src/templateFormatter.ts
@@ -342,6 +342,12 @@ const baseValues = {
       return result / parseFloat(arg);
     }, args[0]);
   },
+  cases(mod, ...cases) {
+    if (cases.length === 0) return "";
+    if (isNaN(mod)) return "";
+    mod = parseInt(mod, 10) - 1;
+    return cases[Math.max(0, mod % cases.length)];
+  },
 };
 
 export async function renderTemplate(template: string, values = {}, includeBaseValues = true) {