@aztec/bot 0.0.0-test.1 → 0.0.1-commit.001888fc

package/dest/store/bot_store.js

@@ -0,0 +1,138 @@
+import { Fr } from '@aztec/foundation/curves/bn254';
+import { createLogger } from '@aztec/foundation/log';
+import { DateProvider } from '@aztec/foundation/timer';
+/**
+ * Simple data store for the bot to persist L1 bridge claims.
+ */ export class BotStore {
+    store;
+    log;
+    dateProvider;
+    static SCHEMA_VERSION = 1;
+    bridgeClaims;
+    pendingL1ToL2;
+    constructor(store, log = createLogger('bot:store'), dateProvider = new DateProvider()){
+        this.store = store;
+        this.log = log;
+        this.dateProvider = dateProvider;
+        this.bridgeClaims = store.openMap('bridge_claims');
+        this.pendingL1ToL2 = store.openMap('pending_l1_to_l2');
+    }
+    /**
+   * Saves a bridge claim for a recipient.
+   */ async saveBridgeClaim(recipient, claim) {
+        // Convert Fr fields and BigInts to strings for JSON serialization
+        const serializableClaim = {
+            claimAmount: claim.claimAmount.toString(),
+            claimSecret: claim.claimSecret.toString(),
+            claimSecretHash: claim.claimSecretHash.toString(),
+            messageHash: claim.messageHash,
+            messageLeafIndex: claim.messageLeafIndex.toString()
+        };
+        const data = {
+            claim: serializableClaim,
+            timestamp: this.dateProvider.now(),
+            recipient: recipient.toString()
+        };
+        await this.bridgeClaims.set(recipient.toString(), JSON.stringify(data));
+        this.log.info(`Saved bridge claim for ${recipient.toString()}`);
+    }
+    /**
+   * Gets a bridge claim for a recipient if it exists.
+   */ async getBridgeClaim(recipient) {
+        const data = await this.bridgeClaims.getAsync(recipient.toString());
+        if (!data) {
+            return undefined;
+        }
+        const parsed = JSON.parse(data);
+        // Reconstruct L2AmountClaim from serialized data
+        const claim = {
+            claimAmount: BigInt(parsed.claim.claimAmount),
+            claimSecret: Fr.fromString(parsed.claim.claimSecret),
+            claimSecretHash: Fr.fromString(parsed.claim.claimSecretHash),
+            messageHash: parsed.claim.messageHash,
+            messageLeafIndex: BigInt(parsed.claim.messageLeafIndex)
+        };
+        return {
+            claim,
+            timestamp: parsed.timestamp,
+            recipient: parsed.recipient
+        };
+    }
+    /**
+   * Deletes a bridge claim for a recipient.
+   */ async deleteBridgeClaim(recipient) {
+        await this.bridgeClaims.delete(recipient.toString());
+        this.log.info(`Deleted bridge claim for ${recipient.toString()}`);
+    }
+    /**
+   * Gets all stored bridge claims.
+   */ async getAllBridgeClaims() {
+        const claims = [];
+        const entries = this.bridgeClaims.entriesAsync();
+        for await (const [_, data] of entries){
+            const parsed = JSON.parse(data);
+            // Reconstruct L2AmountClaim from serialized data
+            const claim = {
+                claimAmount: BigInt(parsed.claim.claimAmount),
+                claimSecret: Fr.fromString(parsed.claim.claimSecret),
+                claimSecretHash: Fr.fromString(parsed.claim.claimSecretHash),
+                messageHash: parsed.claim.messageHash,
+                messageLeafIndex: BigInt(parsed.claim.messageLeafIndex)
+            };
+            claims.push({
+                claim,
+                timestamp: parsed.timestamp,
+                recipient: parsed.recipient
+            });
+        }
+        return claims;
+    }
+    /**
+   * Cleans up old bridge claims (older than 24 hours).
+   */ async cleanupOldClaims(maxAgeMs = 24 * 60 * 60 * 1000) {
+        const now = this.dateProvider.now();
+        let cleanedCount = 0;
+        const entries = this.bridgeClaims.entriesAsync();
+        for await (const [key, data] of entries){
+            const parsed = JSON.parse(data);
+            if (now - parsed.timestamp > maxAgeMs) {
+                await this.bridgeClaims.delete(key);
+                cleanedCount++;
+                this.log.info(`Cleaned up old bridge claim for ${parsed.recipient}`);
+            }
+        }
+        return cleanedCount;
+    }
+    /** Saves a pending L1→L2 message keyed by msgHash. */ async savePendingL1ToL2Message(msg) {
+        await this.pendingL1ToL2.set(msg.msgHash, JSON.stringify(msg));
+        this.log.info(`Saved pending L1→L2 message ${msg.msgHash}`);
+    }
+    /** Returns all unconsumed pending L1→L2 messages. */ async getUnconsumedL1ToL2Messages() {
+        const messages = [];
+        for await (const [_, data] of this.pendingL1ToL2.entriesAsync()){
+            messages.push(JSON.parse(data));
+        }
+        return messages;
+    }
+    /** Deletes a consumed L1→L2 message from the store. */ async deleteL1ToL2Message(msgHash) {
+        await this.pendingL1ToL2.delete(msgHash);
+        this.log.info(`Deleted consumed L1→L2 message ${msgHash}`);
+    }
+    /** Cleans up pending L1→L2 messages older than maxAgeMs. */ async cleanupOldPendingMessages(maxAgeMs = 24 * 60 * 60 * 1000) {
+        const now = this.dateProvider.now();
+        let cleanedCount = 0;
+        for await (const [key, data] of this.pendingL1ToL2.entriesAsync()){
+            const parsed = JSON.parse(data);
+            if (now - parsed.timestamp > maxAgeMs) {
+                await this.pendingL1ToL2.delete(key);
+                cleanedCount++;
+                this.log.info(`Cleaned up old pending L1→L2 message ${key}`);
+            }
+        }
+        return cleanedCount;
+    }
+    /** Closes the store. */ async close() {
+        await this.store.close();
+        this.log.info('Closed bot data store');
+    }
+}

package/dest/store/index.d.ts

@@ -0,0 +1,2 @@
+export { BotStore, type BridgeClaimData, type PendingL1ToL2Message } from './bot_store.js';
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9zdG9yZS9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEVBQUUsUUFBUSxFQUFFLEtBQUssZUFBZSxFQUFFLEtBQUssb0JBQW9CLEVBQUUsTUFBTSxnQkFBZ0IsQ0FBQyJ9

package/dest/store/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/store/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,KAAK,eAAe,EAAE,KAAK,oBAAoB,EAAE,MAAM,gBAAgB,CAAC"}

package/dest/store/index.js

@@ -0,0 +1 @@
+export { BotStore } from './bot_store.js';

package/dest/utils.d.ts

@@ -1,4 +1,6 @@
-import type { EasyPrivateTokenContract } from '@aztec/noir-contracts.js/EasyPrivateToken';
+import { ContractBase } from '@aztec/aztec.js/contracts';
+import type { AMMContract } from '@aztec/noir-contracts.js/AMM';
+import type { PrivateTokenContract } from '@aztec/noir-contracts.js/PrivateToken';
 import type { TokenContract } from '@aztec/noir-contracts.js/Token';
 import type { AztecAddress } from '@aztec/stdlib/aztec-address';
 /**
@@ -7,10 +9,11 @@ import type { AztecAddress } from '@aztec/stdlib/aztec-address';
  * @param who - Address to get the balance for.
  * @returns - Private and public token balances as bigints.
  */
-export declare function getBalances(token: TokenContract, who: AztecAddress): Promise<{
+export declare function getBalances(token: TokenContract, who: AztecAddress, from?: AztecAddress): Promise<{
     privateBalance: bigint;
     publicBalance: bigint;
 }>;
-export declare function getPrivateBalance(token: EasyPrivateTokenContract, who: AztecAddress): Promise<bigint>;
-export declare function isStandardTokenContract(token: TokenContract | EasyPrivateTokenContract): token is TokenContract;
-//# sourceMappingURL=utils.d.ts.map
+export declare function getPrivateBalance(token: PrivateTokenContract, who: AztecAddress, from?: AztecAddress): Promise<bigint>;
+export declare function isStandardTokenContract(token: ContractBase): token is TokenContract;
+export declare function isAMMContract(contract: ContractBase): contract is AMMContract;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidXRpbHMuZC50cyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy91dGlscy50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEVBQUUsWUFBWSxFQUFFLE1BQU0sMkJBQTJCLENBQUM7QUFDekQsT0FBTyxLQUFLLEVBQUUsV0FBVyxFQUFFLE1BQU0sOEJBQThCLENBQUM7QUFDaEUsT0FBTyxLQUFLLEVBQUUsb0JBQW9CLEVBQUUsTUFBTSx1Q0FBdUMsQ0FBQztBQUNsRixPQUFPLEtBQUssRUFBRSxhQUFhLEVBQUUsTUFBTSxnQ0FBZ0MsQ0FBQztBQUNwRSxPQUFPLEtBQUssRUFBRSxZQUFZLEVBQUUsTUFBTSw2QkFBNkIsQ0FBQztBQUVoRTs7Ozs7R0FLRztBQUNILHdCQUFzQixXQUFXLENBQy9CLEtBQUssRUFBRSxhQUFhLEVBQ3BCLEdBQUcsRUFBRSxZQUFZLEVBQ2pCLElBQUksQ0FBQyxFQUFFLFlBQVksR0FDbEIsT0FBTyxDQUFDO0lBQUUsY0FBYyxFQUFFLE1BQU0sQ0FBQztJQUFDLGFBQWEsRUFBRSxNQUFNLENBQUE7Q0FBRSxDQUFDLENBSTVEO0FBRUQsd0JBQXNCLGlCQUFpQixDQUNyQyxLQUFLLEVBQUUsb0JBQW9CLEVBQzNCLEdBQUcsRUFBRSxZQUFZLEVBQ2pCLElBQUksQ0FBQyxFQUFFLFlBQVksR0FDbEIsT0FBTyxDQUFDLE1BQU0sQ0FBQyxDQUdqQjtBQUVELHdCQUFnQix1QkFBdUIsQ0FBQyxLQUFLLEVBQUUsWUFBWSxHQUFHLEtBQUssSUFBSSxhQUFhLENBRW5GO0FBRUQsd0JBQWdCLGFBQWEsQ0FBQyxRQUFRLEVBQUUsWUFBWSxHQUFHLFFBQVEsSUFBSSxXQUFXLENBRTdFIn0=

package/dest/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,2CAA2C,CAAC;AAC1F,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AACpE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAEhE;;;;;GAKG;AACH,wBAAsB,WAAW,CAC/B,KAAK,EAAE,aAAa,EACpB,GAAG,EAAE,YAAY,GAChB,OAAO,CAAC;IAAE,cAAc,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAE,CAAC,CAI5D;AAED,wBAAsB,iBAAiB,CAAC,KAAK,EAAE,wBAAwB,EAAE,GAAG,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,CAG3G;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,aAAa,GAAG,wBAAwB,GAAG,KAAK,IAAI,aAAa,CAE/G"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACzD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uCAAuC,CAAC;AAClF,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AACpE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAEhE;;;;;GAKG;AACH,wBAAsB,WAAW,CAC/B,KAAK,EAAE,aAAa,EACpB,GAAG,EAAE,YAAY,EACjB,IAAI,CAAC,EAAE,YAAY,GAClB,OAAO,CAAC;IAAE,cAAc,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAE,CAAC,CAI5D;AAED,wBAAsB,iBAAiB,CACrC,KAAK,EAAE,oBAAoB,EAC3B,GAAG,EAAE,YAAY,EACjB,IAAI,CAAC,EAAE,YAAY,GAClB,OAAO,CAAC,MAAM,CAAC,CAGjB;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,YAAY,GAAG,KAAK,IAAI,aAAa,CAEnF;AAED,wBAAgB,aAAa,CAAC,QAAQ,EAAE,YAAY,GAAG,QAAQ,IAAI,WAAW,CAE7E"}

package/dest/utils.js

@@ -3,18 +3,27 @@
  * @param token - Token contract.
  * @param who - Address to get the balance for.
  * @returns - Private and public token balances as bigints.
- */ export async function getBalances(token, who) {
-    const privateBalance = await token.methods.balance_of_private(who).simulate();
-    const publicBalance = await token.methods.balance_of_public(who).simulate();
+ */ export async function getBalances(token, who, from) {
+    const { result: privateBalance } = await token.methods.balance_of_private(who).simulate({
+        from: from ?? who
+    });
+    const { result: publicBalance } = await token.methods.balance_of_public(who).simulate({
+        from: from ?? who
+    });
     return {
         privateBalance,
         publicBalance
     };
 }
-export async function getPrivateBalance(token, who) {
-    const privateBalance = await token.methods.get_balance(who).simulate();
+export async function getPrivateBalance(token, who, from) {
+    const { result: privateBalance } = await token.methods.get_balance(who).simulate({
+        from: from ?? who
+    });
     return privateBalance;
 }
 export function isStandardTokenContract(token) {
     return 'mint_to_public' in token.methods;
 }
+export function isAMMContract(contract) {
+    return 'add_liquidity' in contract.methods;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aztec/bot",
-  "version": "0.0.0-test.1",
+  "version": "0.0.1-commit.001888fc",
   "type": "module",
   "exports": {
     ".": "./dest/index.js",
@@ -10,11 +10,9 @@
     "../package.common.json"
   ],
   "scripts": {
-    "build": "yarn clean && tsc -b",
-    "build:dev": "tsc -b --watch",
+    "build": "yarn clean && ../scripts/tsc.sh",
+    "build:dev": "../scripts/tsc.sh --watch",
     "clean": "rm -rf ./dest .tsbuildinfo",
-    "formatting": "run -T prettier --check ./src && run -T eslint ./src",
-    "formatting:fix": "run -T eslint --fix ./src && run -T prettier -w ./src",
     "bb": "node --no-warnings ./dest/bb/index.js",
     "test": "NODE_NO_WARNINGS=1 node --experimental-vm-modules ../node_modules/.bin/jest --passWithNoTests --maxWorkers=${JEST_MAX_WORKERS:-8}"
   },
@@ -49,33 +47,42 @@
     "testTimeout": 120000,
     "setupFiles": [
       "../../foundation/src/jest/setup.mjs"
+    ],
+    "testEnvironment": "../../foundation/src/jest/env.mjs",
+    "setupFilesAfterEnv": [
+      "../../foundation/src/jest/setupAfterEnv.mjs"
     ]
   },
   "dependencies": {
-    "@aztec/accounts": "0.0.0-test.1",
-    "@aztec/aztec.js": "0.0.0-test.1",
-    "@aztec/entrypoints": "0.0.0-test.1",
-    "@aztec/ethereum": "0.0.0-test.1",
-    "@aztec/foundation": "0.0.0-test.1",
-    "@aztec/noir-contracts.js": "0.0.0-test.1",
-    "@aztec/noir-protocol-circuits-types": "0.0.0-test.1",
-    "@aztec/protocol-contracts": "0.0.0-test.1",
-    "@aztec/stdlib": "0.0.0-test.1",
-    "@aztec/telemetry-client": "0.0.0-test.1",
+    "@aztec/accounts": "0.0.1-commit.001888fc",
+    "@aztec/aztec.js": "0.0.1-commit.001888fc",
+    "@aztec/entrypoints": "0.0.1-commit.001888fc",
+    "@aztec/ethereum": "0.0.1-commit.001888fc",
+    "@aztec/foundation": "0.0.1-commit.001888fc",
+    "@aztec/kv-store": "0.0.1-commit.001888fc",
+    "@aztec/l1-artifacts": "0.0.1-commit.001888fc",
+    "@aztec/noir-contracts.js": "0.0.1-commit.001888fc",
+    "@aztec/noir-protocol-circuits-types": "0.0.1-commit.001888fc",
+    "@aztec/noir-test-contracts.js": "0.0.1-commit.001888fc",
+    "@aztec/protocol-contracts": "0.0.1-commit.001888fc",
+    "@aztec/stdlib": "0.0.1-commit.001888fc",
+    "@aztec/telemetry-client": "0.0.1-commit.001888fc",
+    "@aztec/wallets": "0.0.1-commit.001888fc",
     "source-map-support": "^0.5.21",
     "tslib": "^2.4.0",
+    "viem": "npm:@aztec/viem@2.38.2",
     "zod": "^3.23.8"
   },
   "devDependencies": {
-    "@jest/globals": "^29.5.0",
-    "@types/jest": "^29.5.0",
-    "@types/memdown": "^3.0.0",
-    "@types/node": "^18.7.23",
+    "@jest/globals": "^30.0.0",
+    "@types/jest": "^30.0.0",
+    "@types/node": "^22.15.17",
     "@types/source-map-support": "^0.5.10",
-    "jest": "^29.5.0",
-    "jest-mock-extended": "^3.0.3",
+    "@typescript/native-preview": "7.0.0-dev.20260113.1",
+    "jest": "^30.0.0",
+    "jest-mock-extended": "^4.0.0",
     "ts-node": "^10.9.1",
-    "typescript": "^5.0.4"
+    "typescript": "^5.3.3"
   },
   "files": [
     "dest",
@@ -84,6 +91,6 @@
   ],
   "types": "./dest/index.d.ts",
   "engines": {
-    "node": ">=18"
+    "node": ">=20.10"
   }
 }

package/src/amm_bot.ts

@@ -0,0 +1,129 @@
+import { AztecAddress } from '@aztec/aztec.js/addresses';
+import { NO_WAIT } from '@aztec/aztec.js/contracts';
+import { Fr } from '@aztec/aztec.js/fields';
+import { TxHash, TxReceipt } from '@aztec/aztec.js/tx';
+import { jsonStringify } from '@aztec/foundation/json-rpc';
+import type { AMMContract } from '@aztec/noir-contracts.js/AMM';
+import type { TokenContract } from '@aztec/noir-contracts.js/Token';
+import type { AztecNode, AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
+import type { EmbeddedWallet } from '@aztec/wallets/embedded';
+
+import { BaseBot } from './base_bot.js';
+import type { BotConfig } from './config.js';
+import { BotFactory } from './factory.js';
+import type { BotStore } from './store/index.js';
+
+const TRANSFER_BASE_AMOUNT = 1_000;
+const TRANSFER_VARIANCE = 200;
+
+type Balances = { token0: bigint; token1: bigint };
+
+export class AmmBot extends BaseBot {
+  protected constructor(
+    node: AztecNode,
+    wallet: EmbeddedWallet,
+    defaultAccountAddress: AztecAddress,
+    public readonly amm: AMMContract,
+    public readonly token0: TokenContract,
+    public readonly token1: TokenContract,
+    config: BotConfig,
+  ) {
+    super(node, wallet, defaultAccountAddress, config);
+  }
+
+  static async create(
+    config: BotConfig,
+    wallet: EmbeddedWallet,
+    aztecNode: AztecNode,
+    aztecNodeAdmin: AztecNodeAdmin | undefined,
+    store: BotStore,
+  ): Promise<AmmBot> {
+    const { defaultAccountAddress, token0, token1, amm } = await new BotFactory(
+      config,
+      wallet,
+      store,
+      aztecNode,
+      aztecNodeAdmin,
+    ).setupAmm();
+    return new AmmBot(aztecNode, wallet, defaultAccountAddress, amm, token0, token1, config);
+  }
+
+  protected async createAndSendTx(logCtx: object): Promise<TxHash> {
+    const { feePaymentMethod } = this.config;
+    const { wallet, amm, token0, token1 } = this;
+
+    const balances = this.getBalances();
+    this.log.info(`Preparing tx with ${feePaymentMethod} fee to swap tokens. Balances: ${jsonStringify(balances)}`, {
+      ...logCtx,
+      balances,
+    });
+
+    // 1000 ± 200
+    const amountIn = Math.floor(TRANSFER_BASE_AMOUNT + (Math.random() - 0.5) * TRANSFER_VARIANCE);
+    const authwitNonce = Fr.random();
+
+    const [tokenIn, tokenOut] = Math.random() < 0.5 ? [token0, token1] : [token1, token0];
+
+    const swapAuthwit = await wallet.createAuthWit(this.defaultAccountAddress, {
+      caller: amm.address,
+      call: await tokenIn.methods
+        .transfer_to_public(this.defaultAccountAddress, amm.address, amountIn, authwitNonce)
+        .getFunctionCall(),
+    });
+
+    const { result: tokenInBalance } = await tokenIn.methods
+      .balance_of_public(amm.address)
+      .simulate({ from: this.defaultAccountAddress });
+    const { result: tokenOutBalance } = await tokenOut.methods
+      .balance_of_public(amm.address)
+      .simulate({ from: this.defaultAccountAddress });
+    const { result: amountOutMin } = await amm.methods
+      .get_amount_out_for_exact_in(tokenInBalance, tokenOutBalance, amountIn)
+      .simulate({ from: this.defaultAccountAddress });
+
+    const swapExactTokensInteraction = amm.methods
+      .swap_exact_tokens_for_tokens(tokenIn.address, tokenOut.address, amountIn, amountOutMin, authwitNonce)
+      .with({
+        authWitnesses: [swapAuthwit],
+      });
+
+    const opts = await this.getSendMethodOpts(swapExactTokensInteraction);
+
+    this.log.verbose(`Sending transaction`, logCtx);
+    this.log.info(`Tx. Balances: ${jsonStringify(balances)}`, { ...logCtx, balances });
+    const { txHash } = await swapExactTokensInteraction.send({ ...opts, wait: NO_WAIT });
+    return txHash;
+  }
+
+  protected override async onTxMined(receipt: TxReceipt, logCtx: object): Promise<void> {
+    const balances = await this.getBalances();
+    this.log.info(`Balances after swap in tx ${receipt.txHash}: ${jsonStringify(balances)}`, { ...logCtx, balances });
+  }
+
+  public getAmmBalances(): Promise<Balances> {
+    return this.getPublicBalanceFor(this.amm.address);
+  }
+
+  public async getBalances(): Promise<{ senderPublic: Balances; senderPrivate: Balances; amm: Balances }> {
+    return {
+      senderPublic: await this.getPublicBalanceFor(this.defaultAccountAddress),
+      senderPrivate: await this.getPrivateBalanceFor(this.defaultAccountAddress),
+      amm: await this.getPublicBalanceFor(this.amm.address, this.defaultAccountAddress),
+    };
+  }
+
+  private async getPublicBalanceFor(address: AztecAddress, from?: AztecAddress): Promise<Balances> {
+    const { result: token0 } = await this.token0.methods.balance_of_public(address).simulate({ from: from ?? address });
+    const { result: token1 } = await this.token1.methods.balance_of_public(address).simulate({ from: from ?? address });
+    return { token0, token1 };
+  }
+  private async getPrivateBalanceFor(address: AztecAddress, from?: AztecAddress): Promise<Balances> {
+    const { result: token0 } = await this.token0.methods
+      .balance_of_private(address)
+      .simulate({ from: from ?? address });
+    const { result: token1 } = await this.token1.methods
+      .balance_of_private(address)
+      .simulate({ from: from ?? address });
+    return { token0, token1 };
+  }
+}

package/src/base_bot.ts

@@ -0,0 +1,82 @@
+import { AztecAddress } from '@aztec/aztec.js/addresses';
+import { BatchCall, ContractFunctionInteraction, type SendInteractionOptions } from '@aztec/aztec.js/contracts';
+import { createLogger } from '@aztec/aztec.js/log';
+import { waitForTx } from '@aztec/aztec.js/node';
+import { TxHash, TxReceipt, TxStatus } from '@aztec/aztec.js/tx';
+import { Gas } from '@aztec/stdlib/gas';
+import type { AztecNode } from '@aztec/stdlib/interfaces/client';
+import type { EmbeddedWallet } from '@aztec/wallets/embedded';
+
+import type { BotConfig } from './config.js';
+
+export abstract class BaseBot {
+  protected log = createLogger('bot');
+
+  protected attempts: number = 0;
+  protected successes: number = 0;
+
+  protected constructor(
+    public readonly node: AztecNode,
+    public readonly wallet: EmbeddedWallet,
+    public readonly defaultAccountAddress: AztecAddress,
+    public config: BotConfig,
+  ) {}
+
+  public async run(): Promise<TxReceipt | TxHash> {
+    this.attempts++;
+    const { followChain, txMinedWaitSeconds } = this.config;
+    const logCtx = { runId: Date.now() * 1000 + Math.floor(Math.random() * 1000), followChain, txMinedWaitSeconds };
+
+    this.log.verbose(`Creating tx`, logCtx);
+    const txHash = await this.createAndSendTx(logCtx);
+
+    if (followChain === 'NONE') {
+      this.log.info(`Transaction ${txHash.toString()} sent, not waiting for it to be mined`);
+      return txHash;
+    }
+
+    const waitForStatus = TxStatus[followChain];
+    this.log.verbose(`Awaiting tx ${txHash.toString()} to be on the ${followChain} chain`, logCtx);
+    const receipt = await waitForTx(this.node, txHash, { timeout: txMinedWaitSeconds, waitForStatus });
+    this.successes++;
+    this.log.info(
+      `Tx #${this.attempts} ${receipt.txHash} successfully mined in block ${receipt.blockNumber} (stats: ${this.successes}/${this.attempts} success)`,
+      logCtx,
+    );
+
+    await this.onTxMined(receipt, logCtx);
+
+    return receipt;
+  }
+
+  protected abstract createAndSendTx(logCtx: object): Promise<TxHash>;
+
+  protected onTxMined(_receipt: TxReceipt, _logCtx: object): Promise<void> {
+    // no-op
+    return Promise.resolve();
+  }
+
+  protected async getSendMethodOpts(
+    interaction: ContractFunctionInteraction | BatchCall,
+  ): Promise<SendInteractionOptions> {
+    const { l2GasLimit, daGasLimit, minFeePadding } = this.config;
+
+    this.wallet.setMinFeePadding(minFeePadding);
+
+    let gasSettings;
+    if (l2GasLimit !== undefined && l2GasLimit > 0 && daGasLimit !== undefined && daGasLimit > 0) {
+      gasSettings = { gasLimits: Gas.from({ l2Gas: l2GasLimit, daGas: daGasLimit }) };
+      this.log.verbose(`Using gas limits ${l2GasLimit} L2 gas ${daGasLimit} DA gas`);
+    } else {
+      this.log.verbose(`Estimating gas for transaction`);
+      ({ estimatedGas: gasSettings } = await interaction.simulate({
+        fee: { estimateGas: true },
+        from: this.defaultAccountAddress,
+      }));
+    }
+    return {
+      from: this.defaultAccountAddress,
+      fee: { gasSettings },
+    };
+  }
+}