import { createConsoleLogger } from '@aztec/foundation/log';
import { codegen } from '@aztec/noir-noir_codegen';
import { type CompiledCircuit } from '@aztec/noir-types';

import { pascalCase } from 'change-case';
import { promises as fs } from 'fs';

const log = createConsoleLogger('mock-circuits');

const circuits = [
  'app_creator',
  'app_reader',
  'mock_hiding',
  'mock_private_kernel_init',
  'mock_private_kernel_inner',
  'mock_private_kernel_reset',
  'mock_private_kernel_tail',
  'mock_rollup_tx_base_public',
  'mock_rollup_tx_base_private',
  'mock_rollup_tx_merge',
  'mock_rollup_root',
];

const main = async () => {
  try {
    await fs.access('./src/types/');
  } catch (error) {
    await fs.mkdir('./src/types', { recursive: true });
  }
  const programs: [string, CompiledCircuit][] = [];
  // Collect all circuits
  for (const circuit of circuits) {
    const rawData = await fs.readFile(`./artifacts/${circuit}.json`, 'utf-8');
    const abiObj: CompiledCircuit = JSON.parse(rawData);
    programs.push([pascalCase(circuit), abiObj]);
  }
  const code = codegen(
    programs,
    false, // Don't embed artifacts
    true, // Use fixed length arrays
  );
  await fs.writeFile('./src/types/index.ts', code);
};

try {
  await main();
} catch (err: unknown) {
  log(`Error generating types ${err}`);
  process.exit(1);
}