feat: add public key override to verify

tlsnotary · Feb 1, 2024 · 3d82cb9 · 3d82cb9
1 parent e7f46fc
commit 3d82cb9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "tlsn-js",
-  "version": "v0.1.0-alpha.3-rc2",
+  "version": "v0.1.0-alpha.3",
   "description": "",
   "repository": "https://github.com/tlsnotary/tlsn-js",
   "main": "build/index.js",

diff --git a/src/index.ts b/src/index.ts
@@ -63,19 +63,26 @@ export const prove = async (
 
 export const verify = async (
   proof: Proof,
+  publicKeyOverride?: string,
 ): Promise<{
   time: number;
   sent: string;
   recv: string;
   notaryUrl: string;
 }> => {
-  const res = await fetch(proof.notaryUrl + '/info');
-  const json: any = await res.json();
-  const publicKey = json.publicKey as string;
+  const publicKey =
+    publicKeyOverride || (await fetchPublicKeyFromNotary(proof.notaryUrl));
   const tlsn = await getTLSN();
   const result = await tlsn.verify(proof, publicKey);
   return {
     ...result,
     notaryUrl: proof.notaryUrl,
   };
 };
+
+async function fetchPublicKeyFromNotary(notaryUrl: string) {
+  const res = await fetch(notaryUrl + '/info');
+  const json: any = await res.json();
+  if (!json.publicKey) throw new Error('invalid response');
+  return json.publicKey;
+}