web3.ts

import {
    ACCOUNTS_CONTRACT,
    ACCOUNTS_PROXY_ADDRESS,
    ALFAJORES_CUSD_ADDRESS,
    FA_CONTRACT,
    FA_PROXY_ADDRESS,
    ODIS_PAYMENTS_CONTRACT,
    ODIS_PAYMENTS_PROXY_ADDRESS,
    STABLE_TOKEN_CONTRACT,
} from "./constants";
import Web3 from "web3";
import { OdisUtils } from '@celo/identity'
import { AuthSigner, OdisContextName } from "@celo/identity/lib/odis/query";

const ISSUER_PRIVATE_KEY = "0x726e53db4f0a79dfd63f58b19874896fce3748fcb80874665e0c147369c04a37";
const DEK_PUBLIC_KEY = "0x026063780c81991c032fb4fa7485c6607b7542e048ef85d08516fe5c4482360e4b";
const DEK_PRIVATE_KEY = "0xc2bbdabb440141efed205497a41d5fb6114e0435fd541e368dc628a8e086bfee";

const ALFAJORES_RPC = "https://alfajores-forno.celo-testnet.org";

class ASv2 {
    web3 = new Web3(ALFAJORES_RPC);
    issuer = this.web3.eth.accounts.privateKeyToAccount(ISSUER_PRIVATE_KEY);
    authSigner: AuthSigner = {
        authenticationMethod: OdisUtils.Query.AuthenticationMethod.ENCRYPTION_KEY,
        rawKey: DEK_PRIVATE_KEY
    }
    serviceContext = OdisUtils.Query.getServiceContext(OdisContextName.ALFAJORES);
    
    /** Contracts **/
    accountsContract = new this.web3.eth.Contract(
        ACCOUNTS_CONTRACT.abi,
        ACCOUNTS_PROXY_ADDRESS
    );
    federatedAttestationsContract = new this.web3.eth.Contract(
        FA_CONTRACT.abi,
        FA_PROXY_ADDRESS
    );
    odisPaymentsContract = new this.web3.eth.Contract(
        ODIS_PAYMENTS_CONTRACT.abi,
        ODIS_PAYMENTS_PROXY_ADDRESS
    );
    stableTokenContract = new this.web3.eth.Contract(
        STABLE_TOKEN_CONTRACT.abi,
        ALFAJORES_CUSD_ADDRESS
    );

    constructor(){
        this.web3.eth.accounts.wallet.add(this.issuer);
        this.web3.eth.defaultAccount = this.issuer.address
        this.accountsContract.methods.setAccountDataEncryptionKey(DEK_PUBLIC_KEY).send({from: this.issuer.address, gas: 50000})
    }

    async registerAttestation(phoneNumber: string, account: string, attestationIssuedTime: number) {
        await this.checkAndTopUpODISQuota();
    
        // get identifier from phone number using ODIS
        const identifier = (await OdisUtils.Identifier.getObfuscatedIdentifier(
            phoneNumber,
            OdisUtils.Identifier.IdentifierPrefix.PHONE_NUMBER,
            this.issuer.address,
            this.authSigner,
            this.serviceContext
          )).obfuscatedIdentifier
    
        // upload identifier <-> address mapping to onchain registry
        await this.federatedAttestationsContract.methods
            .registerAttestationAsIssuer(
                identifier,
                account,
                attestationIssuedTime
            )
            .send({from: this.issuer.address, gas: 50000});
    }

    async lookupAddresses(phoneNumber: string){
        // get identifier from phone number using ODIS
        const identifier = (await OdisUtils.Identifier.getObfuscatedIdentifier(
          phoneNumber,
          OdisUtils.Identifier.IdentifierPrefix.PHONE_NUMBER,
          this.issuer.address,
          this.authSigner,
          this.serviceContext
        )).obfuscatedIdentifier    
    
        // query onchain mappings
        const attestations = await this.federatedAttestationsContract.methods
            .lookupAttestations(identifier, [this.issuer.address])
            .call();

        return attestations.accounts
    }

    private async checkAndTopUpODISQuota() {
        //check remaining quota
        const { remainingQuota } = await OdisUtils.Quota.getPnpQuotaStatus(
          this.issuer.address,
          this.authSigner,
          this.serviceContext
        );
    
        console.log("remaining ODIS quota", remainingQuota);
        if (remainingQuota < 1) {
          // give odis payment contract permission to use cUSD
          const currentAllowance = await this.stableTokenContract.methods.allowance(
            this.issuer.address,
            this.odisPaymentsContract.options.address
          );
          console.log("current allowance:", currentAllowance.toString());
          let enoughAllowance: boolean = false;
    
          const ONE_CENT_CUSD_WEI = this.web3.utils.toWei("0.01", "ether");
    
          if (currentAllowance < ONE_CENT_CUSD_WEI) {
            const approvalTxReceipt = await this.stableTokenContract.methods
              .increaseAllowance(
                this.odisPaymentsContract.options.address,
                ONE_CENT_CUSD_WEI
              )
              .sendAndWaitForReceipt();
            console.log("approval status", approvalTxReceipt.status);
            enoughAllowance = approvalTxReceipt.status;
          } else {
            enoughAllowance = true;
          }
    
          // increase quota
          if (enoughAllowance) {
            const odisPayment = await this.odisPaymentsContract.methods
              .payInCUSD(this.issuer.address, ONE_CENT_CUSD_WEI)
              .sendAndWaitForReceipt();
            console.log("odis payment tx status:", odisPayment.status);
            console.log("odis payment tx hash:", odisPayment.transactionHash);
          } else {
            throw "cUSD approval failed";
          }
        }
    }
}

(async () => {
    const asv2 = new ASv2()
    const userAccount = "0xf14790BAdd2638cECB5e885fc7fAD1b6660AAc34";
    const userPhoneNumber = "+18009099999";
    const timeAttestationWasVerified = Math.floor(new Date().getTime() / 1000);
    try{
        await asv2.registerAttestation(userPhoneNumber, userAccount, timeAttestationWasVerified)
        console.log("attestation registered")
    } catch(err){
        // mostly likely reason registering would fail is if this issuer has already
        // registered a mapping between this number and account
    }
    console.log(await asv2.lookupAddresses(userPhoneNumber))
})()