feat/cache-srs tmp changes

axiom-crypto · Jan 16, 2024 · 262b66d · 262b66d
1 parent 78677fd
commit 262b66d
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 22 deletions.
diff --git a/halo2-wasm/js/js/kzg.ts b/halo2-wasm/js/js/kzg.ts
@@ -4,30 +4,31 @@ import * as fs from "fs";
 import { fetchAndConvertToUint8Array } from "../shared/utils";
 
 export const getKzgParams = async (k: number): Promise<Uint8Array> => {
-  const home = os.homedir();
-  const axiomSrsPath = path.join(
-    home,
-    ".axiom",
-    "srs",
-    "challenge_0085",
-    `kzg_bn254_${k}.srs`
-  );
-  const exists = fs.existsSync(axiomSrsPath);
-  if (exists) {
-    const buffer = fs.readFileSync(axiomSrsPath);
-    return new Uint8Array(buffer);
-  }
-  const folderPath = path.dirname(axiomSrsPath);
-  if (!fs.existsSync(folderPath)) {
-    fs.mkdirSync(folderPath, { recursive: true });
-  }
-  if (k < 6 || k > 19) {
-    throw new Error(`k=${k} is not supported`);
-  }
+  console.log("KZG: js");
+  // const home = os.homedir();
+  // const axiomSrsPath = path.join(
+  //   home,
+  //   ".axiom",
+  //   "srs",
+  //   "challenge_0085",
+  //   `kzg_bn254_${k}.srs`
+  // );
+  // const exists = fs.existsSync(axiomSrsPath);
+  // if (exists) {
+  //   const buffer = fs.readFileSync(axiomSrsPath);
+  //   return new Uint8Array(buffer);
+  // }
+  // const folderPath = path.dirname(axiomSrsPath);
+  // if (!fs.existsSync(folderPath)) {
+  //   fs.mkdirSync(folderPath, { recursive: true });
+  // }
+  // if (k < 6 || k > 19) {
+  //   throw new Error(`k=${k} is not supported`);
+  // }
   const srs = await fetchAndConvertToUint8Array(
     `https://axiom-crypto.s3.amazonaws.com/challenge_0085/kzg_bn254_${k}.srs`
   );
-  fs.writeFileSync(axiomSrsPath, srs);
+  // fs.writeFileSync(axiomSrsPath, srs);
 
   return srs;
 };
diff --git a/halo2-wasm/js/shared/utils.ts b/halo2-wasm/js/shared/utils.ts
@@ -19,9 +19,11 @@ export function fetchAndConvertToUint8Array(url: string): Promise<Uint8Array> {
     }
     // Check if running in browser or web worker environment
     else if (typeof window !== "undefined" || typeof self !== "undefined") {
+      console.log("fetching SRS...");
       fetch(url)
         .then((response) => response.arrayBuffer())
         .then((buffer) => {
+          console.log("got SRS successfully");
           resolve(new Uint8Array(buffer));
         })
         .catch(reject);

diff --git a/halo2-wasm/js/web/kzg.ts b/halo2-wasm/js/web/kzg.ts
@@ -1,8 +1,9 @@
 import { fetchAndConvertToUint8Array } from "../shared/utils";
 
 export const getKzgParams = async (k: number): Promise<Uint8Array> => {
+  console.log("KZG: web");
   if (k < 6 || k > 19) {
     throw new Error(`k=${k} is not supported`);
   }
-  return fetchAndConvertToUint8Array(`https://axiom-crypto.s3.amazonaws.com/challenge_0085/kzg_bn254_${k}.srs`);
+  return await fetchAndConvertToUint8Array(`https://axiom-crypto.s3.amazonaws.com/challenge_0085/kzg_bn254_${k}.srs`);
 };