Add actionState to the wrapping request

zeko-labs · Aug 10, 2023 · fd2f903 · fd2f903
1 parent f455f49
commit fd2f903
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/src/ZekoBridge.test.ts b/src/ZekoBridge.test.ts
@@ -160,6 +160,8 @@ describe('ZekoBridge', () => {
       requests.push(
         new WrappingRequest({
           id: counter,
+          actionState:
+            requests.at(-1)?.getNextActionState() ?? Reducer.initialActionState,
           ...action,
         })
       );
@@ -328,6 +330,8 @@ describe('ZekoBridge', () => {
       requests.push(
         new WrappingRequest({
           id: counter,
+          actionState:
+            requests.at(-1)?.getNextActionState() ?? Reducer.initialActionState,
           ...action,
         })
       );
@@ -428,6 +432,8 @@ describe('ZekoBridge', () => {
       requests.push(
         new WrappingRequest({
           id: counter,
+          actionState:
+            requests.at(-1)?.getNextActionState() ?? Reducer.initialActionState,
           ...action,
         })
       );

diff --git a/src/ZekoBridge.ts b/src/ZekoBridge.ts
@@ -27,6 +27,7 @@ export const TREE_CAPACITY = new MerkleTree(TREE_HEIGHT).leafCount;
 
 export class WrappingRequest extends Struct({
   id: Field,
+  actionState: Field,
   amount: UInt64,
   tokenId: Field,
   receiver: PublicKey,
@@ -39,6 +40,18 @@ export class WrappingRequest extends Struct({
     return this.id.toBigInt() % TREE_CAPACITY;
   }
 
+  toAction() {
+    return new WrappingRequestAction({
+      amount: this.amount,
+      tokenId: this.tokenId,
+      receiver: this.receiver,
+    });
+  }
+
+  getNextActionState() {
+    return updateActionsState(this.actionState, this.toAction().hash());
+  }
+
   static buildMerkleTree(requests: WrappingRequest[]): MerkleTree {
     const tree = new MerkleTree(TREE_HEIGHT);
 
@@ -85,6 +98,7 @@ export class WrappingRequestAction extends Struct({
     const witnesses = actions.map((action) => {
       const request = new WrappingRequest({
         id: currentState.counter,
+        actionState: currentState.actionState,
         amount: action.amount,
         tokenId: action.tokenId,
         receiver: action.receiver,
@@ -145,7 +159,7 @@ export class WrappingRequestAction extends Struct({
 
       proof =
         proof === undefined
-          ? await ActionsRecursiveReducer.first(publicInput, witnessesBatch)
+          ? await ActionsRecursiveReducer.init(publicInput, witnessesBatch)
           : await ActionsRecursiveReducer.step(
               publicInput,
               proof,
@@ -205,6 +219,7 @@ export class ActionsBatch extends Struct({
       );
       const request = new WrappingRequest({
         id: stateAux.counter,
+        actionState: stateAux.actionState,
         amount: action.amount,
         tokenId: action.tokenId,
         receiver: action.receiver,
@@ -265,7 +280,7 @@ export const ActionsRecursiveReducer = Experimental.ZkProgram({
   publicInput: ReducerPublicInput,
 
   methods: {
-    first: {
+    init: {
       privateInputs: [ActionsBatch],
 
       method(publicInput: ReducerPublicInput, batch: ActionsBatch) {