feat: support multiple inputs

algorandfoundation · Nov 30, 2023 · 0932916 · 0932916
1 parent 09573aa
commit 0932916
Showing 1 changed file with 55 additions and 50 deletions.
diff --git a/src/bin/tealscript.ts b/src/bin/tealscript.ts
@@ -8,11 +8,63 @@ import Compiler, { CompilerOptions } from '../lib/compiler';
 import 'dotenv/config';
 import { VERSION } from '../version';
 
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+async function processFile(filename: string, parsed: any) {
+  const content = fs.readFileSync(filename, 'utf-8');
+
+  // Output dir for artifacts
+  const dir = parsed.output;
+
+  const options = {
+    filename,
+    algodPort: process.env.ALGOD_PORT ? Number(process.env.ALGOD_PORT) : undefined,
+    algodServer: process.env.ALGOD_SERVER,
+    algodToken: process.env.ALGOD_TOKEN,
+    disableWarnings: parsed.disable_warnings as boolean,
+    disableOverflowChecks: parsed.unsafe_disable_overflow_checks as boolean,
+    disableTypeScript: parsed.unsafe_disable_typescript as boolean,
+  } as CompilerOptions;
+
+  const compilers = Compiler.compileAll(content, options);
+
+  compilers.forEach(async (compilerPromise) => {
+    const compiler = await compilerPromise;
+
+    const { name } = compiler;
+
+    const approvalPath = path.join(dir, `${name}.approval.teal`);
+    const clearPath = path.join(dir, `${name}.clear.teal`);
+    const abiPath = path.join(dir, `${name}.arc4.json`);
+    const appPath = path.join(dir, `${name}.arc32.json`);
+    const srcmapPath = path.join(dir, `${name}.src_map.json`);
+    const lsigPath = path.join(dir, `${name}.lsig.teal`);
+
+    if (fs.existsSync(approvalPath)) fs.rmSync(approvalPath);
+    if (fs.existsSync(clearPath)) fs.rmSync(clearPath);
+    if (fs.existsSync(abiPath)) fs.rmSync(abiPath);
+    if (fs.existsSync(appPath)) fs.rmSync(appPath);
+    if (fs.existsSync(srcmapPath)) fs.rmSync(srcmapPath);
+    if (fs.existsSync(lsigPath)) fs.rmSync(lsigPath);
+
+    if (compiler.teal.lsig.length > 0) {
+      fs.writeFileSync(lsigPath, compiler.teal.lsig.map((t) => t.teal).join('\n'));
+      fs.writeFileSync(srcmapPath, JSON.stringify(compiler.srcMap, null, 2));
+      return;
+    }
+
+    fs.writeFileSync(approvalPath, compiler.teal.approval.map((t) => t.teal).join('\n'));
+    fs.writeFileSync(clearPath, compiler.teal.clear.map((t) => t.teal).join('\n'));
+    fs.writeFileSync(abiPath, JSON.stringify(compiler.abi, null, 2));
+    fs.writeFileSync(srcmapPath, JSON.stringify(compiler.srcMap, null, 2));
+    fs.writeFileSync(appPath, JSON.stringify(compiler.appSpec(), null, 2));
+  });
+}
+
 const parser = new ArgumentParser({
   description: 'Argparse example',
 });
 
-parser.add_argument('input', { help: 'TEALScript source file to compile' });
+parser.add_argument('input', { help: 'TEALScript source file(s) to compile', nargs: '*' });
 parser.add_argument('output', { help: 'output directory for artifacts' });
 parser.add_argument('-v', '--version', { action: 'version', version: VERSION });
 parser.add_argument('--disable-warnings', {
@@ -32,52 +84,5 @@ parser.add_argument('--unsafe-disable-typescript', {
 });
 
 const parsed = parser.parse_args();
-const filename = parsed.input;
-const content = fs.readFileSync(filename, 'utf-8');
-
-// Output dir for artifacts
-const dir = parsed.output;
-
-const options = {
-  filename,
-  algodPort: process.env.ALGOD_PORT ? Number(process.env.ALGOD_PORT) : undefined,
-  algodServer: process.env.ALGOD_SERVER,
-  algodToken: process.env.ALGOD_TOKEN,
-  disableWarnings: parsed.disable_warnings as boolean,
-  disableOverflowChecks: parsed.unsafe_disable_overflow_checks as boolean,
-  disableTypeScript: parsed.unsafe_disable_typescript as boolean,
-} as CompilerOptions;
-
-const compilers = Compiler.compileAll(content, options);
-
-compilers.forEach(async (compilerPromise) => {
-  const compiler = await compilerPromise;
-
-  const { name } = compiler;
-
-  const approvalPath = path.join(dir, `${name}.approval.teal`);
-  const clearPath = path.join(dir, `${name}.clear.teal`);
-  const abiPath = path.join(dir, `${name}.arc4.json`);
-  const appPath = path.join(dir, `${name}.arc32.json`);
-  const srcmapPath = path.join(dir, `${name}.src_map.json`);
-  const lsigPath = path.join(dir, `${name}.lsig.teal`);
-
-  if (fs.existsSync(approvalPath)) fs.rmSync(approvalPath);
-  if (fs.existsSync(clearPath)) fs.rmSync(clearPath);
-  if (fs.existsSync(abiPath)) fs.rmSync(abiPath);
-  if (fs.existsSync(appPath)) fs.rmSync(appPath);
-  if (fs.existsSync(srcmapPath)) fs.rmSync(srcmapPath);
-  if (fs.existsSync(lsigPath)) fs.rmSync(lsigPath);
-
-  if (compiler.teal.lsig.length > 0) {
-    fs.writeFileSync(lsigPath, compiler.teal.lsig.map((t) => t.teal).join('\n'));
-    fs.writeFileSync(srcmapPath, JSON.stringify(compiler.srcMap, null, 2));
-    return;
-  }
-
-  fs.writeFileSync(approvalPath, compiler.teal.approval.map((t) => t.teal).join('\n'));
-  fs.writeFileSync(clearPath, compiler.teal.clear.map((t) => t.teal).join('\n'));
-  fs.writeFileSync(abiPath, JSON.stringify(compiler.abi, null, 2));
-  fs.writeFileSync(srcmapPath, JSON.stringify(compiler.srcMap, null, 2));
-  fs.writeFileSync(appPath, JSON.stringify(compiler.appSpec(), null, 2));
-});
+
+parsed.input.forEach((f: string) => processFile(f, parsed));