naider 1.9.0 → 1.10.0

package/src/index.js

@@ -4,7 +4,23 @@ import { Generator } from './generator.js';
 import { preprocess } from './preprocess.js';
 import { TypeChecker } from './typechecker.js';
 
-export function compile(source, { mode = 'naide', runtimePath, sourceFile, typeCheck = false } = {}) {
+async function getGenerator(target, options) {
+  switch (target) {
+    case 'python':
+    case 'py': {
+      const { PythonGenerator } = await import('./generator-python.js');
+      return new PythonGenerator(options);
+    }
+    case 'bun': {
+      const { BunGenerator } = await import('./generator-bun.js');
+      return new BunGenerator(options);
+    }
+    default:
+      return new Generator(options);
+  }
+}
+
+export function compile(source, { mode = 'naide', runtimePath, sourceFile, typeCheck = false, target = 'node' } = {}) {
   let processedSource = source;
   if (mode === 'x') {
     processedSource = preprocess(source);
@@ -42,6 +58,44 @@ export function compile(source, { mode = 'naide', runtimePath, sourceFile, typeC
   }
 }
 
+export async function compileAsync(source, { mode = 'naide', runtimePath, sourceFile, typeCheck = false, target = 'node' } = {}) {
+  let processedSource = source;
+  if (mode === 'x') {
+    processedSource = preprocess(source);
+  }
+  try {
+    const lexer = new Lexer(processedSource);
+    const tokens = lexer.tokenize();
+    const parser = new Parser(tokens);
+    const ast = parser.parse();
+
+    let typeErrors = null;
+    if (typeCheck) {
+      const checker = new TypeChecker();
+      typeErrors = checker.check(ast);
+    }
+
+    const generator = await getGenerator(target, { runtimePath, sourceFile });
+    const code = generator.generate(ast);
+    return { js: code, code, ast, tokens, sourceMap: generator.sourceMap, naide: mode === 'x' ? processedSource : null, typeErrors, target };
+  } catch (e) {
+    const lineMatch = e.message.match(/line (\d+)/);
+    if (lineMatch) {
+      const lineNum = parseInt(lineMatch[1]);
+      const lines = processedSource.split('\n');
+      const start = Math.max(0, lineNum - 3);
+      const end = Math.min(lines.length, lineNum + 2);
+      const context = lines.slice(start, end).map((l, i) => {
+        const num = start + i + 1;
+        const marker = num === lineNum ? ' >> ' : '    ';
+        return `${marker}${num} | ${l}`;
+      }).join('\n');
+      e.message += `\n\n${context}\n`;
+    }
+    throw e;
+  }
+}
+
 export function transpile(source, opts) {
   return compile(source, opts).js;
 }