fix: handle mapping that omit key part

zackad · Aug 19, 2024 · b0dccf8 · b0dccf8
1 parent de25ec4
commit b0dccf8
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 7 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,9 @@
 
 ## unreleased
 
+### Bugfixes
+- Fix handling mapping that omit key part
+
 ---
 ## 0.8.0 (2024-08-09)
 

diff --git a/src/melody/melody-parser/Parser.js b/src/melody/melody-parser/Parser.js
@@ -52,6 +52,10 @@ const TAG = Symbol("TAG");
 const TEST = Symbol("TEST");
 
 export default class Parser {
+    /**
+     * @param {TokenStream} tokenStream
+     * @param {Object} options
+     */
     constructor(tokenStream, options) {
         this.tokens = tokenStream;
         this[UNARY] = {};
@@ -765,6 +769,9 @@ export default class Parser {
                 if (!n.is(key, "StringLiteral")) {
                     computed = true;
                 }
+            } else if ((token = tokens.nextIf(Types.EXPRESSION_START))) {
+                key = this.matchExpression();
+                computed = true;
             } else if ((token = tokens.nextIf(Types.SYMBOL))) {
                 key = createNode(n.Identifier, token, token.text);
             } else if ((token = tokens.nextIf(Types.NUMBER))) {
@@ -781,12 +788,20 @@ export default class Parser {
                         tokens.next()
                 });
             }
-            tokens.expect(Types.COLON);
-            const value = this.matchExpression();
-            const prop = new n.ObjectProperty(key, value, computed);
-            copyStart(prop, key);
-            copyEnd(prop, value);
-            obj.properties.push(prop);
+            if (tokens.test(Types.COLON)) {
+                tokens.expect(Types.COLON);
+                const value = this.matchExpression();
+                const prop = new n.ObjectProperty(key, value, computed);
+                copyStart(prop, key);
+                copyEnd(prop, value);
+                obj.properties.push(prop);
+            } else {
+                const value = key;
+                const prop = new n.ObjectProperty(key, value, computed, true);
+                copyStart(prop, key);
+                copyEnd(prop, value);
+                obj.properties.push(prop);
+            }
             if (!tokens.test(Types.RBRACKET)) {
                 tokens.expect(Types.COMMA);
                 // support trailing comma

diff --git a/src/melody/melody-types/index.js b/src/melody/melody-types/index.js
@@ -356,12 +356,14 @@ export class ObjectProperty extends Node {
      * @param {Node} key
      * @param {Node} value
      * @param {boolean} computed
+     * @param {boolean} omitKey
      */
-    constructor(key, value, computed) {
+    constructor(key, value, computed, omitKey = false) {
         super();
         this.key = key;
         this.value = value;
         this.computed = computed;
+        this.omitKey = omitKey;
     }
 }
 type(ObjectProperty, "ObjectProperty");

diff --git a/src/print/ObjectProperty.js b/src/print/ObjectProperty.js
@@ -16,6 +16,10 @@ const p = (node, path, print, options) => {
     if (needsParentheses) {
         parts.push(")");
     }
+    // handle property that omit key
+    if (node.omitKey) {
+        return parts;
+    }
     parts.push(": ");
     node[STRING_NEEDS_QUOTES] = true;
     parts.push(path.call(print, "value"));