context.ts

/* eslint-disable require-atomic-updates */
import bls from "bls-eth-wasm";
import {NotInitializedError} from "../errors.js";

type Bls = typeof bls;
let blsGlobal: Bls | null = null;
let blsGlobalPromise: Promise<void> | null = null;

// Patch to fix multiVerify() calls on a browser with polyfilled NodeJS crypto
declare global {
  interface Window {
    msCrypto: typeof window["crypto"];
  }
}

export async function setupBls(): Promise<void> {
  if (!blsGlobal) {
    await bls.init(bls.BLS12_381);

    // Patch to fix multiVerify() calls on a browser with polyfilled NodeJS crypto
    if (typeof window === "object") {
      const crypto = window.crypto || window.msCrypto;
      // getRandomValues is not typed in `bls-eth-wasm` because it's not meant to be exposed
      // eslint-disable-next-line @typescript-eslint/ban-ts-comment
      // @ts-ignore
      bls.getRandomValues = (x) => crypto.getRandomValues(x);
    }

    blsGlobal = bls;
  }
}

// Cache a promise for Bls instead of Bls to make sure it is initialized only once
export async function init(): Promise<void> {
  if (!blsGlobalPromise) {
    blsGlobalPromise = setupBls();
  }
  return blsGlobalPromise;
}

export function destroy(): void {
  blsGlobal = null;
  blsGlobalPromise = null;
}

export function getContext(): Bls {
  if (!blsGlobal) {
    throw new NotInitializedError("herumi");
  }
  return blsGlobal;
}