Add option to specify functionName for compiled regex circuit

Shigoto-dev19 · Jul 20, 2024 · a41bdb2 · a41bdb2
1 parent c428904
commit a41bdb2
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/src/cli.ts b/src/cli.ts
@@ -15,10 +15,17 @@ program
     'Partial state transitions to reveal'
   )
   .option('-s, --revealSubpatterns <values...>', 'Regex subpatterns to reveal')
+  .option(
+    '-n, --functionName <name>',
+    'Function name to give to the regex circuit'
+  )
   .action((rawRegex, options) => {
     // Extract and set the options
     const countEnabled = options.count || false;
     let revealEnabled = false;
+    const functionName = options.functionName
+      ? 'function ' + options.functionName
+      : '';
 
     let revealInput: string[] | [number, number][][] | undefined = undefined;
 
@@ -48,7 +55,12 @@ program
     }
 
     // Print the regex circuit based on the options
-    compiler.printRegexCircuit(countEnabled, revealEnabled, revealInput);
+    compiler.printRegexCircuit(
+      countEnabled,
+      revealEnabled,
+      revealInput,
+      functionName
+    );
   });
 
 // Parse the command-line arguments

diff --git a/src/compiler.ts b/src/compiler.ts
@@ -494,7 +494,8 @@ export class RegexCompiler {
   printRegexCircuit(
     countEnabled: boolean,
     revealEnabled: boolean,
-    transitionInput?: string[] | [number, number][][]
+    transitionInput?: string[] | [number, number][][],
+    functionName?: string
   ) {
     let circuitLines: string[] = [];
     circuitLines = this.writeDeclarationLines()
@@ -503,7 +504,7 @@ export class RegexCompiler {
       .concat(this.writeAcceptLines(countEnabled));
 
     const stringRegexCircuit =
-      '\n(input: UInt8[]) {\n' +
+      `\n${functionName}(input: UInt8[]) {\n` +
       circuitLines.join('\n\t') +
       this.writeRevealLines(revealEnabled, transitionInput) +
       '\n}';