feat: when tag with multiple values

harttle · May 6, 2021 · 8f9639f · 8f9639f
1 parent 3e130e0
commit 8f9639f
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 8 deletions.
diff --git a/docs/source/tags/case.md b/docs/source/tags/case.md
@@ -12,7 +12,7 @@ Input
 {% case handle %}
   {% when "cake" %}
      This is a cake
-  {% when "cookie" %}
+  {% when "cookie", "biscuit" %}
      This is a cookie
   {% else %}
      This is not a cake nor a cookie

diff --git a/docs/source/zh-cn/tags/case.md b/docs/source/zh-cn/tags/case.md
@@ -12,7 +12,7 @@ title: case
 {% case handle %}
   {% when "cake" %}
      This is a cake
-  {% when "cookie" %}
+  {% when "cookie", "biscuit" %}
      This is a cookie
   {% else %}
      This is not a cake nor a cookie

diff --git a/src/builtin/tags/case.ts b/src/builtin/tags/case.ts
@@ -1,4 +1,5 @@
-import { toValue, Value, Emitter, TagToken, TopLevelToken, Context, Template, TagImplOptions, ParseStream } from '../../types'
+import { toValue, evalToken, Value, Emitter, TagToken, TopLevelToken, Context, Template, TagImplOptions, ParseStream } from '../../types'
+import { Tokenizer } from '../../parser/tokenizer'
 
 export default {
   parse: function (tagToken: TagToken, remainTokens: TopLevelToken[]) {
@@ -9,10 +10,21 @@ export default {
     let p: Template[] = []
     const stream: ParseStream = this.liquid.parser.parseStream(remainTokens)
       .on('tag:when', (token: TagToken) => {
-        this.cases.push({
-          val: new Value(token.args, this.liquid),
-          templates: p = []
-        })
+        p = []
+
+        const tokenizer = new Tokenizer(token.args, this.liquid.options.operatorsTrie)
+
+        while (!tokenizer.end()) {
+          const value = tokenizer.readValue()
+          if (value) {
+            this.cases.push({
+              val: value,
+              templates: p
+            })
+          }
+
+          tokenizer.readTo(',')
+        }
       })
       .on('tag:else', () => (p = this.elseTemplates))
       .on('tag:endcase', () => stream.stop())
@@ -28,7 +40,7 @@ export default {
     const r = this.liquid.renderer
     const cond = toValue(yield this.cond.value(ctx, ctx.opts.lenientIf))
     for (const branch of this.cases) {
-      const val = toValue(yield branch.val.value(ctx, ctx.opts.lenientIf))
+      const val = evalToken(branch.val, ctx, ctx.opts.lenientIf)
       if (val === cond) {
         yield r.renderTemplates(branch.templates, ctx, emitter)
         return

diff --git a/test/integration/builtin/tags/case.ts b/test/integration/builtin/tags/case.ts
@@ -64,4 +64,11 @@ describe('tags/case', function () {
       return expect(html).to.equal('d')
     })
   })
+  it('should support case with multiple values', async function () {
+    const src = '{% case "b" %}' +
+            '{% when "a", "b" %}foo' +
+            '{%endcase%}'
+    const html = await liquid.parseAndRender(src)
+    return expect(html).to.equal('foo')
+  })
 })