import { Field, MerkleTree, verify } from 'o1js';
import { Performance } from '../../../lib/testing/perf-regression.js';
import {
  MainProgramState,
  MerkleTreeWitness,
  SideloadedProgramProof,
  mainProgram,
  sideloadedProgram,
} from './dynamic-keys-merkletree.js';

const csSide = await sideloadedProgram.analyzeMethods();
const csMain = await mainProgram.analyzeMethods();

const perfSide = Performance.create(sideloadedProgram.name, csSide);
const perfMain = Performance.create(mainProgram.name, csMain);

perfSide.start('compile');
const sideVk = (await sideloadedProgram.compile()).verificationKey;
perfSide.end();

perfMain.start('compile');
const mainVk = (await mainProgram.compile()).verificationKey;
perfMain.end();

const tree = new MerkleTree(64);

console.log('\nProving deployment of side-loaded key');
const rootBefore = tree.getRoot();
tree.setLeaf(1n, sideVk.hash);
const witness = new MerkleTreeWitness(tree.getWitness(1n));

perfMain.start('prove', 'addSideloadedProgram');
const { proof: proof1 } = await mainProgram.addSideloadedProgram(
  new MainProgramState({
    treeRoot: rootBefore,
    state: Field(0),
  }),
  sideVk,
  witness
);
perfMain.end();

console.log('\nProving child program execution');
perfSide.start('prove', 'compute');
const { proof: childProof1 } = await sideloadedProgram.compute(Field(0), Field(10));
perfSide.end();

console.log('\nProving verification inside main program');
perfMain.start('prove', 'validateUsingTree');
const { proof: proof2 } = await mainProgram.validateUsingTree(
  proof1.publicOutput,
  proof1,
  sideVk,
  witness,
  SideloadedProgramProof.fromProof(childProof1)
);
perfMain.end();

perfMain.start('verify', 'validateUsingTree');
const validProof2 = await verify(proof2, mainVk);
perfMain.end();
console.log('isValid2? ', validProof2);
if (!validProof2) throw new Error('proof verification failed!');

console.log('\nProving different method of child program');
perfSide.start('prove', 'assertAndAdd');
const { proof: childProof2 } = await sideloadedProgram.assertAndAdd(Field(0), Field(10));
perfSide.end();

console.log('\nProving verification inside main program');
const { proof: proof3 } = await mainProgram.validateUsingTree(
  proof1.publicOutput,
  proof1,
  sideVk,
  witness,
  SideloadedProgramProof.fromProof(childProof2)
);

const validProof3 = await verify(proof3, mainVk);
console.log('isValid3? ', validProof3);
if (!validProof3) throw new Error('proof verification failed!');