Merge pull request #22 from ETHCF/better-init

Compatibility

Author: holgerd77

src/index.ts

@@ -2,6 +2,7 @@ import { hexToBytes } from './util.js'
 import { loadWasmModule } from './loader.cjs'
 import kzgWasm from './kzg.js'
 import mainnetTrustedSetup from './trustedSetup.js'
+
 export type TrustedSetup = {
     g1_monomial: string
     g1_lagrange: string
@@ -28,7 +29,7 @@ const BYTES_PER_CELL = 2048
  *
  * @returns object - the KZG methods required for all 4844 related operations
  */
-export const loadKZG = async (trustedSetup: TrustedSetup = mainnetTrustedSetup) => {
+export const loadKZG = async (trustedSetup: TrustedSetup = mainnetTrustedSetup, precompute: number = 8) => {
     // In Node.js environment, preload the WASM binary to avoid path resolution issues
     let wasmBinary = await loadWasmModule();
 
@@ -84,7 +85,7 @@ export const loadKZG = async (trustedSetup: TrustedSetup = mainnetTrustedSetup)
         return '0x' + result;
     }
 
-    const blobToKzgCommitment = blobToKZGCommitment // Alias with different casing for ethereumjs compatibility
+    
 
     /**
      * 
@@ -100,7 +101,7 @@ export const loadKZG = async (trustedSetup: TrustedSetup = mainnetTrustedSetup)
         return '0x' + result;
     }
 
-    const computeBlobProof = computeBlobKZGProof // Alias with different casing for ethereumjs compatibility
+   
     /**
      * 
      * @param blobs - an array of blobs
@@ -292,10 +293,44 @@ export const loadKZG = async (trustedSetup: TrustedSetup = mainnetTrustedSetup)
         return res === 'true'
     }
 
+    const loadResult = loadTrustedSetup(trustedSetup, precompute);
+    if (loadResult !== 0) {
+        throw new Error(`Failed to load trusted setup, error code: ${loadResult}`);
+    }
+
+
+    // ALIASING
+    const computeBlobProof = computeBlobKZGProof // Alias with different casing for ethereumjs compatibility
+    const blobToKzgCommitment = blobToKZGCommitment // Alias with different casing for ethereumjs compatibility
+    const verifyProof = verifyKZGProof // Alias with different casing for ethereumjs compatibility
+
+    const verifyCellKzgProofBatch = (commitments: string[], cellIndices: number[], cells: string[], proofs: string[]) => {
+        return verifyCellKZGProofBatch(commitments, cellIndices, cells, proofs, cells.length)
+    }
+
+    const recoverCellsAndProofs = (indices: number[], cells: string[]): [string[], string[]] => {
+        const result = recoverCellsFromKZGProofs(indices, cells, cells.length);
+        return [result.cells, result.proofs];
+    }
+
+    const computeCells = (blob: string): string[] => {
+        const result = computeCellsAndKZGProofs(blob);
+        return result.cells;
+    }
+
+    const computeCellsAndProofs = (blob: string): [string[], string[]] => {
+        const result = computeCellsAndKZGProofs(blob);
+        return [result.cells, result.proofs];
+    }
+
+    const verifyBlobProofBatch = verifyBlobKZGProofBatch
 
     return {
         loadTrustedSetup, freeTrustedSetup, blobToKZGCommitment, computeBlobKZGProof, verifyBlobKZGProofBatch, verifyKZGProof, verifyBlobKZGProof,
-        computeCellsAndKZGProofs, recoverCellsFromKZGProofs, verifyCellKZGProof, verifyCellKZGProofBatch, blobToKzgCommitment, computeBlobProof
+        computeCellsAndKZGProofs, recoverCellsFromKZGProofs, verifyCellKZGProof, verifyCellKZGProofBatch, 
+        // ALIASED/KZGJS COMPATIBILITY METHODS
+        verifyCellKzgProofBatch, recoverCellsAndProofs, computeCells, computeCellsAndProofs, verifyBlobProofBatch, 
+        computeBlobProof, blobToKzgCommitment, verifyProof
     }
 }
 

test/index.spec.ts

@@ -31,8 +31,6 @@ describe('kzg API tests', () => {
   let kzg: Awaited<ReturnType<typeof loadKZG>>
   beforeAll(async () => {
     kzg = await loadKZG()
-    const result = kzg.loadTrustedSetup()
-    assert.equal(result, 0, 'loaded trusted setup successfully')
   })
 
   it('should generate kzg commitments and verify proofs', async () => {