Allow nonceKey for simpleAccount preset (#111)

Author: hazim-j

src/preset/builder/simpleAccount.ts

@@ -23,6 +23,7 @@ export class SimpleAccount extends UserOperationBuilder {
   private entryPoint: EntryPoint;
   private factory: SimpleAccountFactory;
   private initCode: string;
+  private nonceKey: number;
   proxy: SimpleAccountImpl;
 
   private constructor(
@@ -44,14 +45,15 @@ export class SimpleAccount extends UserOperationBuilder {
       this.provider
     );
     this.initCode = "0x";
+    this.nonceKey = opts?.nonceKey || 0;
     this.proxy = SimpleAccount__factory.connect(
       ethers.constants.AddressZero,
       this.provider
     );
   }
 
   private resolveAccount: UserOperationMiddlewareFn = async (ctx) => {
-    ctx.op.nonce = await this.entryPoint.getNonce(ctx.op.sender, 0);
+    ctx.op.nonce = await this.entryPoint.getNonce(ctx.op.sender, this.nonceKey);
     ctx.op.initCode = ctx.op.nonce.eq(0) ? this.initCode : "0x";
   };
 

src/types.ts

@@ -107,6 +107,7 @@ export interface IPresetBuilderOpts {
   entryPoint?: string;
   factory?: string;
   salt?: BigNumberish;
+  nonceKey?: number;
   paymasterMiddleware?: UserOperationMiddlewareFn;
   overrideBundlerRpc?: string;
 }