hi-language/transpiler
1
0
Fork 0
mirror of https://github.com/hi-language/transpiler.git synced 2026-01-14 00:28:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Support ObjectLiteral; keep IIFE for blocks

Browse Source
This commit is contained in:
multipleof4
2025-09-26 10:09:48 -07:00
parent 1d1e168756
commit 673fb04494
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/generator.js
View File

@@ -5,13 +5,14 @@
*/ */
export function generate(ast) { export function generate(ast) {
if (!ast) return ''; if (!ast) return '';
const generators = { const generators = {
Program: (node) => node.body.map(generate).join('\n'), Program: (node) => node.body.map(generate).join('\n'),
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)}`,
@@ -33,13 +34,13 @@ export function generate(ast) {
if (callee === '_') return `console.log(${args})`; if (callee === '_') return `console.log(${args})`;
return `${callee}(${args})`; return `${callee}(${args})`;
}, },
MemberExpression: (node) => node.computed MemberExpression: (node) => node.computed
? `${generate(node.object)}[${generate(node.property)}]` ? `${generate(node.object)}[${generate(node.property)}]`
: `${generate(node.object)}.${generate(node.property)}`, : `${generate(node.object)}.${generate(node.property)}`,
BinaryExpression: (node) => `(${generate(node.left)} ${node.operator} ${generate(node.right)})`, BinaryExpression: (node) => `(${generate(node.left)} ${node.operator} ${generate(node.right)})`,
Block: (node) => { Block: (node) => {
let bodyCode = node.body.map(generate).join('\n'); let bodyCode = node.body.map(generate).join('\n');
// Check for implicit return // Check for implicit return
@@ -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);
} }