Feat: Support ObjectLiteral; keep IIFE for blocks

This commit is contained in:
2025-09-26 10:09:48 -07:00
parent 1d1e168756
commit 673fb04494

View File

@@ -11,7 +11,8 @@ export function generate(ast) {
ExpressionStatement: (node) => `${generate(node.expression)};`, ExpressionStatement: (node) => `${generate(node.expression)};`,
VariableDeclaration: (node) => { VariableDeclaration: (node) => {
const keyword = node.value.type.includes('Function') ? 'const' : 'let'; const t = node.value.type;
const keyword = (t === 'FunctionExpression' || t === 'ArrowFunctionExpression') ? 'const' : 'let';
return `${keyword} ${node.identifier} = ${generate(node.value)};`; return `${keyword} ${node.identifier} = ${generate(node.value)};`;
}, },
AssignmentExpression: (node) => `${generate(node.left)} = ${generate(node.right)}`, AssignmentExpression: (node) => `${generate(node.left)} = ${generate(node.right)}`,
@@ -57,13 +58,17 @@ export function generate(ast) {
ArrowFunctionExpression: (node) => { ArrowFunctionExpression: (node) => {
const params = node.params.map(generate).join(', '); const params = node.params.map(generate).join(', ');
const body = generate(node.body); const body = generate(node.body);
// If body is not a block, it's an implicit return
const bodyStr = node.body.type === 'Block' || node.body.type === 'BlockStatement' ? body : `(${body})`; const bodyStr = node.body.type === 'Block' || node.body.type === 'BlockStatement' ? body : `(${body})`;
return `(${params}) => ${bodyStr}`; return `(${params}) => ${bodyStr}`;
}, },
ArrayLiteral: (node) => `[${node.elements.map(generate).join(', ')}]`, ArrayLiteral: (node) => `[${node.elements.map(generate).join(', ')}]`,
ObjectLiteral: (node) => {
const props = node.properties.map(p => `${p.key}: ${generate(p.value)}`).join(', ');
return `{ ${props} }`;
},
Identifier: (node) => node.name, Identifier: (node) => node.name,
ThisExpression: () => 'this', ThisExpression: () => 'this',
NumericLiteral: (node) => node.value, NumericLiteral: (node) => node.value,
@@ -77,3 +82,4 @@ export function generate(ast) {
} }
return generators[ast.type](ast); return generators[ast.type](ast);
} }