add zkapp to store the password tree root

Author: chfanghr

src/pluginInteropServer/plugins/simplePasswordTree.ts

@@ -1,4 +1,4 @@
-import { Experimental, Field, JsonProof, MerkleTree, Poseidon, verify } from "o1js";
+import { Permissions, DeployArgs, Experimental, Field, JsonProof, MerkleTree, Poseidon, PrivateKey, PublicKey, SmartContract, State, method, state, verify, Mina, AccountUpdate } from "o1js";
 import ProvePasswordInTreeProgram, { PASSWORD_TREE_HEIGHT, PasswordTreePublicInput, PasswordTreeWitness } from "../zkPrograms/passwordTreeProof";
 import { PluginType } from "../plugin";
 
@@ -88,4 +88,39 @@ export const SimplePasswordTree: PluginType = {
   prove,
 }
 
-export default SimplePasswordTree;
+export default SimplePasswordTree;
+
+class PasswordTreeRootStorage extends SmartContract {
+  @state(Field) treeRoot = State<Field>();
+
+  deploy(args?: DeployArgs) {
+    super.deploy(args);
+
+    this.account.permissions.set({
+      ...Permissions.allImpossible(),
+      editState: Permissions.signature(),
+      access: Permissions.signature(),
+    });
+  }
+}
+
+export async function updateTreeRootStorage(
+  contractPrivateKey: PrivateKey,
+  feePayerPrivateKey: PrivateKey,
+  treeRoot: Field
+) {
+  const contractPublicKey = contractPrivateKey.toPublicKey();
+  const contract = new PasswordTreeRootStorage(contractPublicKey);
+  const feePayerPublicKey = feePayerPrivateKey.toPublicKey();
+  const txfn = contract.account.isNew.get() ?
+    () => {
+      AccountUpdate.fundNewAccount(feePayerPublicKey);
+      contract.treeRoot.set(treeRoot);
+      contract.deploy();
+    } : () => {
+      contract.treeRoot.set(treeRoot);
+    }
+  const txn = await Mina.transaction(feePayerPublicKey, txfn);
+  await txn.prove();
+  await txn.sign([feePayerPrivateKey, contractPrivateKey]).send();
+}

tsconfig.json

@@ -102,6 +102,11 @@
     // "allowUnusedLabels": true,                        /* Disable error reporting for unused labels. */
     // "allowUnreachableCode": true,                     /* Disable error reporting for unreachable code. */
 
+
+    "experimentalDecorators": true, // needed for decorators
+    "emitDecoratorMetadata": true, // needed for decorators
+    "useDefineForClassFields": false, // ensure correct behaviour of class fields with decorators
+
     /* Completeness */
     // "skipDefaultLibCheck": true,                      /* Skip type checking .d.ts files that are included with TypeScript. */
     "skipLibCheck": true                                 /* Skip type checking all .d.ts files. */