From fb984e9ae5f7f0a059809c323760d0eab4c3111d Mon Sep 17 00:00:00 2001
From: Luiz <lvj253@gmail.com>
Date: Fri, 25 Oct 2024 01:06:55 -0300
Subject: [PATCH] updating parseUtils.ts and adding test to risc0 proof

---
 .../groth16ContractGenerator/parsingUtils.ts  | 33 +++++++++++++++++--
 .../starknet/groth16VkProofParsing.test.ts    |  3 +-
 2 files changed, 32 insertions(+), 4 deletions(-)

diff --git a/tools/npm/garaga_ts/src/node/starknet/groth16ContractGenerator/parsingUtils.ts b/tools/npm/garaga_ts/src/node/starknet/groth16ContractGenerator/parsingUtils.ts
index daebf7ae..1ff78f98 100644
--- a/tools/npm/garaga_ts/src/node/starknet/groth16ContractGenerator/parsingUtils.ts
+++ b/tools/npm/garaga_ts/src/node/starknet/groth16ContractGenerator/parsingUtils.ts
@@ -51,6 +51,12 @@ interface ReceiptClaim {
     tagDigest?: Uint8Array;
 }
 
+export class KeyPatternNotFoundError extends Error {
+    constructor(message: string) {
+        super(message);
+        this.name = "KeyPatternNotFoundError";
+    }
+}
 
 
 
@@ -188,10 +194,16 @@ export const parseGroth16ProofFromObject = (data: any, publicInputsData?: bigint
 
         try{
 
+            console.log("AQUI BEFORE")
+
             const sealHex = toHexStr(findItemFromKeyPatterns(proof, ['seal']));
             const imageIdHex = toHexStr(findItemFromKeyPatterns(proof, [ 'image_id']));
             const journalHex = toHexStr(findItemFromKeyPatterns(proof, ['journal']));
 
+            console.log("journalHex: ", journalHex);
+
+            console.log("LATER: ", journalHex);
+
             const sealBytes = hexStringToBytes(sealHex);
             const imageIdBytes = hexStringToBytes(imageIdHex);
             const journalBytes = hexStringToBytes(journalHex);
@@ -199,6 +211,7 @@ export const parseGroth16ProofFromObject = (data: any, publicInputsData?: bigint
             return createGroth16ProofFromRisc0(sealBytes, imageIdBytes, journalBytes)
 
         } catch(err){
+            console.log("ERROR AQUI: ", err);
 
         }
 
@@ -217,7 +230,12 @@ export const parseGroth16ProofFromObject = (data: any, publicInputsData?: bigint
                 throw new Error(`Invalid public inputs format: ${publicInputsData}`);
             }
         } else{
-            publicInputs = findItemFromKeyPatterns(data, ['public']);
+
+            try {
+                publicInputs = findItemFromKeyPatterns(data, ['public']);
+            } catch(err){
+                throw new Error(`Error: ${err}`);
+            }
         }
         const a = tryParseG1PointFromKey(proof, ['a'], curveId);
         const b = tryParseG2PointFromKey(proof, ['b'], curveId);
@@ -246,14 +264,17 @@ export const parseGroth16ProofFromObject = (data: any, publicInputsData?: bigint
 export const createGroth16ProofFromRisc0 = (seal: Uint8Array, imageId: Uint8Array, journalBytes: Uint8Array,
     controlRoot: bigint = RISC0_CONTROL_ROOT, bn254ControlId: bigint = RISC0_BN254_CONTROL_ID): Groth16Proof => {
 
-    if(imageId.length <= 32){
+    if(imageId.length > 32){
         throw new Error("imageId must be 32 bytes")
     }
 
     const [constrolRoot0, controlRoot1] = splitDigest(controlRoot);
 
     const proof = seal.slice(4);
+
+
     const journal = createHash("sha256").update(journalBytes).digest();
+
     const claimDigest = digestReceiptClaim(ok(imageId, journal));
 
     const [claim0, claim1] = splitDigest(claimDigest);
@@ -330,6 +351,11 @@ export const checkGroth16VerifyingKey = (vk: Groth16VerifyingKey): boolean => {
 const digestReceiptClaim = (receipt: ReceiptClaim): Uint8Array => {
     const { tagDigest, input, preStateDigest, postStateDigest, output, exitCode } = receipt;
 
+    console.log("receipt: ", receipt);
+    console.log("tagDigest: ", tagDigest);
+
+
+
     // Concatenating all parts into one Buffer
     const data = Buffer.concat([
       tagDigest!,
@@ -362,6 +388,7 @@ function ok(imageId: Uint8Array, journalDigest: Uint8Array): ReceiptClaim {
 
     // Create and return the ReceiptClaim object
     return {
+        tagDigest: createHash('sha256').update(Buffer.from("risc0.ReceiptClaim")).digest(),
         preStateDigest: imageId,
         postStateDigest: SYSTEM_STATE_ZERO_DIGEST,
         exitCode,
@@ -494,7 +521,7 @@ const findItemFromKeyPatterns = (data: { [key: string]: any }, keyPatterns: stri
     if(bestMatch){
         return bestMatch;
     }
-    throw new Error(`No key found with patterns ${keyPatterns}`);
+    throw new KeyPatternNotFoundError(`No key found with patterns ${keyPatterns}`);
 }
 
 export const getPFromCurveId = (curveId: CurveId): bigint => {
diff --git a/tools/npm/garaga_ts/tests/starknet/groth16VkProofParsing.test.ts b/tools/npm/garaga_ts/tests/starknet/groth16VkProofParsing.test.ts
index 15dcfad1..ce400e69 100644
--- a/tools/npm/garaga_ts/tests/starknet/groth16VkProofParsing.test.ts
+++ b/tools/npm/garaga_ts/tests/starknet/groth16VkProofParsing.test.ts
@@ -1,4 +1,4 @@
-import {  parseGroth16ProofFromJson, parseGroth16VerifyingKeyFromJson } from "../../src/node/starknet/groth16ContractGenerator/parsingUtils";
+import {  parseGroth16VerifyingKeyFromJson, parseGroth16ProofFromJson } from "../../src/node/starknet/groth16ContractGenerator/parsingUtils";
 
 const PATH = '../../../hydra/garaga/starknet/groth16_contract_generator/examples';
 
@@ -22,6 +22,7 @@ describe('Groth16 Parsing Tests', () => {
   const proofPaths = [
     `${PATH}/proof_bn254.json`,
     `${PATH}/proof_bls.json`,
+    `${PATH}/proof_risc0.json`,
   ];
 
   test.each(proofPaths)('should parse proof from %s', (proofPath) => {