import { CircuitId, ZkpGen, ZkpOperators } from '@rarimo/zkp-gen-iden3';
import { type Identity } from '@rarimo/identity-gen-iden3';
import { type VerifiableCredentials } from '@rarimo/auth-zkp-iden3';

/**
 * define type with variable to proof,
 * verifiableCredentials response will keep this variable in credentialSubject
 */
type QueryVariableName = { isNatural: number };

const getZkProof = async (
  identity: Identity,
  verifiableCredentials: VerifiableCredentials,
  providerAddress: string,
) => {
  ZkpGen.setConfig({
    // rpc url where statev2 contract is deployed
    RPC_URL: 'https://matic-mumbai.chainstacklabs.com',
    // statev2 contract address
    STATE_V2_ADDRESS: '0x134B1BE34911E39A8397ec6289782989729807a4',
    // api url of issuer svc
    ISSUER_API_URL: 'http://127.0.0.1:8000/',
  });

  const zkProof = new ZkpGen<QueryVariableName>({
    /**
     * In zkp environment, every prove request should have a unique requestId
     * by using this requestId, issuer svc can identify the request
     */
    requestId: '1',
    /**
     * Identity generated by @rarimo/identity-gen-iden3
     */
    identity: identity,
    /**
     * Verifiable credentials generated by @rarimo/auth-zkp-iden3
     */
    verifiableCredentials: verifiableCredentials,

    /**
     * Challenge is a random string for example user metamask address without '0x'
     */
    challenge: String(providerAddress).substring(2),

    /**
     * Query is a json object with variable to proof
     */
    query: {
      /**
       * variableName is the name of variable to proof
       */
      variableName: 'isNatural',
      /**
       * operator is the operator to compare the variable with value
       */
      operator: ZkpOperators.Equals,
      /**
       * value is the value to compare with variable
       */
      value: ['1'],

      circuitId: CircuitId.AtomicQueryMTPV2OnChain,

      issuerId: config.ISSUER_ID,
    },
  });

  await zkProof.generateProof();

  setIsNaturalProof(zkProof);
};

import { makeRarimoQuerier } from '@rarimo/client';
import { getTransitStateTxBody } from '@rarimo/shared-zkp-iden3';
import { FetcherError, HTTP_STATUS_CODES } from '@distributedlab/jac';
import { sleep } from '@rarimo/shared';
import { config } from '@/config';

const querier = makeRarimoQuerier({
  apiUrl: config.RARIMO_CORE_RPC_API_URL,
});

await Promise.all([
  zkProof.loadStatesDetails(querier),
  zkProof.loadMerkleProof(querier, config.ISSUER_ID),
]);

await zkProof.loadOperation(
  querier,
  zkProof.coreStateDetails.lastUpdateOperationIndex,
);

do {
  try {
    await zkProof.loadOperationProof(
      querier,
      zkProof.coreStateDetails.lastUpdateOperationIndex,
    );
  } catch (error) {
    if (
      error instanceof FetcherError &&
      error.response.status === HTTP_STATUS_CODES.BAD_REQUEST
    ) {
      await sleep(30 * 1000);
    } else {
      throw error;
    }
  }
} while (!zkProof.operationProof);

if (zkProof.isStatesActual()) {
  const transitParams = await isNaturalZkp?.loadParamsForTransitState(querier);

  // use
  const txBody = getTransitStateTxBody(
    config.LIGHTWEIGHT_STATE_V2_CONTRACT_ADDRESS,
    transitParams.newIdentitiesStatesRoot,
    transitParams.gistData,
    transitParams.proof,
  );

  // send tx with txBody ...
}