All files / src/plugins badge.ts

97.87% Statements 46/47
100% Branches 0/0
85.71% Functions 6/7
97.56% Lines 40/41

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 16727x   27x                                 27x 27x 27x 27x                                   2x         63x 63x 63x 63x 63x   63x             14x 14x                             27x                                             18x       18x         18x       18x   18x           18x 18x     18x 18x       18x           18x     27x 12x 12x   12x 12x           12x     12x     27x 12x 12x         12x 12x   12x      
export { MinaNFTVerifierBadgeEvent, MinaNFTVerifierBadge };
 
import {
  method,
  DeployArgs,
  Permissions,
  TokenContract,
  AccountUpdateForest,
  PublicKey,
  Field,
  Signature,
  State,
  state,
  Struct,
  Account,
  UInt64,
  AccountUpdate,
} from "o1js";
 
import { RedactedMinaNFTMapStateProof } from "./redactedmap";
import { MinaNFTContract } from "../contract/nft";
import { Metadata } from "../contract/metadata";
import { MinaNFTBadgeProof } from "./badgeproof";
 
class MinaNFTVerifierBadgeEvent extends Struct({
  address: PublicKey,
  owner: Field,
  name: Field,
  version: UInt64,
  data: Metadata,
  key: Field,
}) {
  constructor(args: {
    address: PublicKey;
    owner: Field;
    name: Field;
    version: UInt64;
    data: Metadata;
    key: Field;
  }) {
    super(args);
  }
}
 
class MinaNFTVerifierBadge extends TokenContract {
  @state(Field) name = State<Field>();
  @state(Field) owner = State<Field>();
  @state(Field) verifiedKey = State<Field>();
  @state(Field) verifiedKind = State<Field>();
  @state(PublicKey) oracle = State<PublicKey>();
 
  events = {
    deploy: Field,
    issue: MinaNFTVerifierBadgeEvent,
    revoke: PublicKey,
  };
 
  async deploy(args: DeployArgs) {
    super.deploy(args);
    this.account.permissions.set({
      ...Permissions.default(),
      editState: Permissions.proof(),
    });
  }
 
  async approveBase(forest: AccountUpdateForest) {
    // https://discord.com/channels/484437221055922177/1215258350577647616
    // this.checkZeroBalanceChange(forest);
    //forest.isEmpty().assertEquals(Bool(true));
    throw Error(
      "transfers of tokens are not allowed, change the owner instead"
    );
  }
 
  @method async issueBadge(
    nft: PublicKey,
    nftTokenId: Field,
    badgeEvent: MinaNFTVerifierBadgeEvent,
    signature: Signature,
    proof: RedactedMinaNFTMapStateProof,
    badgeProof: MinaNFTBadgeProof
  ) {
    /*
    Excluded pending resolution of the issue
    https://github.com/o1-labs/o1js/issues/1245
   
    const minanft = new MinaNFTContract(nft, nftTokenId);
    badgeEvent.owner.assertEquals(minanft.owner.getAndRequireEquals());
    badgeEvent.address.assertEquals(nft);
    badgeEvent.name.assertEquals(minanft.name.getAndRequireEquals());
    badgeEvent.version.assertEquals(minanft.version.getAndRequireEquals());
    Metadata.assertEquals(
      minanft.metadata.getAndRequireEquals(),
      proof.publicInput.originalRoot
    );
    */
 
    badgeProof.publicInput.data.kind.assertEquals(
      this.verifiedKind.getAndRequireEquals(),
      "Kind mismatch"
    );
    badgeProof.publicInput.key.assertEquals(
      this.verifiedKey.getAndRequireEquals(),
      "Key mismatch"
    );
 
    Metadata.assertEquals(
      badgeProof.publicInput.root,
      proof.publicInput.redactedRoot
    );
    Metadata.assertEquals(badgeProof.publicInput.data, badgeEvent.data);
 
    signature
      .verify(
        this.oracle.getAndRequireEquals(),
        MinaNFTVerifierBadgeEvent.toFields(badgeEvent)
      )
      .assertEquals(true);
    proof.verify();
    badgeProof.verify();
 
    // Issue verification badge
    const tokenId = this.deriveTokenId();
    const tokenBalance = AccountUpdate.create(
      nft,
      tokenId
    ).account.balance.getAndRequireEquals();
    this.internal.mint({
      address: nft,
      amount: badgeEvent.version.sub(tokenBalance),
    });
 
    // Emit event
    this.emitEvent("issue", badgeEvent);
  }
 
  @method async revokeBadge(nft: PublicKey, signature: Signature) {
    const oracle = this.oracle.getAndRequireEquals();
    signature.verify(oracle, nft.toFields());
 
    const tokenId = this.deriveTokenId();
    const tokenBalance = AccountUpdate.create(
      nft,
      tokenId
    ).account.balance.getAndRequireEquals();
 
    // Revoke verification badge
    this.internal.burn({ address: nft, amount: tokenBalance });
 
    // Emit event
    this.emitEvent("revoke", nft);
  }
 
  @method async verifyBadge(nft: PublicKey, nftTokenId: Field) {
    const tokenId = this.deriveTokenId();
    const tokenBalance = AccountUpdate.create(
      nft,
      tokenId
    ).account.balance.getAndRequireEquals();
 
    const minanft = new MinaNFTContract(nft, nftTokenId);
    const version = minanft.version.getAndRequireEquals();
 
    version.assertEquals(tokenBalance);
  }
}