オブジェクトリテラル内で省略記法に対応

Author: uzmoi

src/parser/syntaxes/expressions.ts

@@ -577,7 +577,7 @@ function parseReference(s: ITokenStream): Ast.Identifier {
 
 /**
  * ```abnf
- * Object = "{" [ObjectKey ":" Expr *(SEP IDENT ":" Expr) [SEP]] "}"
+ * Object = "{" [ObjectKey [":" Expr] *(SEP ObjectKey [":" Expr]) [SEP]] "}"
  * ```
 */
 function parseObject(s: ITokenStream, isStatic: boolean): Ast.Obj {
@@ -592,16 +592,23 @@ function parseObject(s: ITokenStream, isStatic: boolean): Ast.Obj {
 
 	const map = new Map<string, Ast.Expression>();
 	while (!s.is(TokenKind.CloseBrace)) {
+		const startPos = s.getPos();
+
 		const k = parseObjectKey(s);
 		if (map.has(k)) {
 			throw new AiScriptSyntaxError(`Key ${k} is duplicated.`, s.getPos());
 		}
 		s.next();
 
-		s.expect(TokenKind.Colon);
-		s.next();
+		let v: Ast.Expression;
 
-		const v = parseExpr(s, isStatic);
+		if (s.is(TokenKind.Colon)){
+			s.next();
+
+			v = parseExpr(s, isStatic);
+		} else {
+			v = NODE("identifier", { name: k }, startPos, s.getPos());
+		}
 
 		map.set(k, v);
 

test/literals.ts

@@ -243,6 +243,14 @@ describe('literal', () => {
 		});
 	});
 
+	test.concurrent('obj (shorthand)', async () => {
+		const res = await exe(`
+		let a = 1
+		<: { a }
+		`);
+		eq(res, OBJ(new Map([['a', NUM(1)]])));
+	});
+
 	test.concurrent('obj (escaped reserved word as key)', async () => {
 		await expect(async () => await exe(`
 		<: {

test/syntax.ts

@@ -635,6 +635,13 @@ describe('Variable declaration', () => {
 		`);
 		eq(res, ARR([NUM(1), NUM(2)]));
 	});
+	test.concurrent('destructuring declaration with shorthand', async () => {
+		const res = await exe(`
+			let { value } = { value: 1 }
+			<: value
+		`);
+		eq(res, NUM(1));
+	});
 	test.concurrent('empty function', async () => {
 		const res = await exe(`
 			@hoge() { }

unreleased/obj-shorthand-sugar-syntax.md

@@ -0,0 +1,2 @@
+- オブジェクトリテラルとオブジェクトの分割代入構文内で、キーと同名の変数を参照する省略記法を使用できるようになりました。
+	- 例えば`{ hoge }`は`{ hoge: hoge }`と等価です。