@aztec/bot 0.0.0-test.1 → 0.0.1-fake-ceab37513c

package/dest/interface.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"interface.d.ts","sourceRoot":"","sources":["../src/interface.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAI1D,OAAO,EAAE,KAAK,SAAS,EAAmB,MAAM,aAAa,CAAC;AAE9D,MAAM,WAAW,YAAY;IAC3B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACrB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,MAAM,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1C;AAED,eAAO,MAAM,kBAAkB,EAAE,YAAY,CAAC,YAAY,CAOzD,CAAC"}
+{"version":3,"file":"interface.d.ts","sourceRoot":"","sources":["../src/interface.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAE1D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,KAAK,SAAS,EAAmB,MAAM,aAAa,CAAC;AAE9D,eAAO,MAAM,aAAa;;;;;;EAExB,CAAC;AAEH,MAAM,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAEpD,MAAM,WAAW,YAAY;IAC3B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACrB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5B,MAAM,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1C;AAED,eAAO,MAAM,kBAAkB,EAAE,YAAY,CAAC,YAAY,CAQzD,CAAC"}

package/dest/interface.js

@@ -1,10 +1,15 @@
+import { AztecAddress } from '@aztec/aztec.js';
 import { z } from 'zod';
 import { BotConfigSchema } from './config.js';
+export const BotInfoSchema = z.object({
+    botAddress: AztecAddress.schema
+});
 export const BotRunnerApiSchema = {
     start: z.function().args().returns(z.void()),
     stop: z.function().args().returns(z.void()),
     run: z.function().args().returns(z.void()),
     setup: z.function().args().returns(z.void()),
+    getInfo: z.function().args().returns(BotInfoSchema),
     getConfig: z.function().args().returns(BotConfigSchema),
     update: z.function().args(BotConfigSchema).returns(z.void())
 };

package/dest/rpc.d.ts

@@ -1,10 +1,4 @@
 import type { ApiHandler } from '@aztec/foundation/json-rpc/server';
 import type { BotRunner } from './runner.js';
-/**
- * Wraps a bot runner with a JSON RPC HTTP server.
- * @param botRunner - The BotRunner.
- * @returns An JSON-RPC HTTP server
- */
-export declare function createBotRunnerRpcServer(botRunner: BotRunner): void;
 export declare function getBotRunnerApiHandler(botRunner: BotRunner): ApiHandler;
 //# sourceMappingURL=rpc.d.ts.map

package/dest/rpc.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"rpc.d.ts","sourceRoot":"","sources":["../src/rpc.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mCAAmC,CAAC;AAIpE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAE7C;;;;GAIG;AACH,wBAAgB,wBAAwB,CAAC,SAAS,EAAE,SAAS,QAK5D;AAED,wBAAgB,sBAAsB,CAAC,SAAS,EAAE,SAAS,GAAG,UAAU,CAEvE"}
+{"version":3,"file":"rpc.d.ts","sourceRoot":"","sources":["../src/rpc.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mCAAmC,CAAC;AAGpE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAE7C,wBAAgB,sBAAsB,CAAC,SAAS,EAAE,SAAS,GAAG,UAAU,CAEvE"}

package/dest/rpc.js

@@ -1,15 +1,4 @@
-import { createTracedJsonRpcServer } from '@aztec/telemetry-client';
 import { BotRunnerApiSchema } from './interface.js';
-/**
- * Wraps a bot runner with a JSON RPC HTTP server.
- * @param botRunner - The BotRunner.
- * @returns An JSON-RPC HTTP server
- */ export function createBotRunnerRpcServer(botRunner) {
-    createTracedJsonRpcServer(botRunner, BotRunnerApiSchema, {
-        http200OnError: false,
-        healthCheck: botRunner.isHealthy.bind(botRunner)
-    });
-}
 export function getBotRunnerApiHandler(botRunner) {
     return [
         botRunner,

package/dest/runner.d.ts

@@ -1,7 +1,8 @@
 import { type AztecNode, type PXE } from '@aztec/aztec.js';
+import { type AztecNodeAdmin } from '@aztec/stdlib/interfaces/client';
 import { type TelemetryClient, type Traceable, type Tracer } from '@aztec/telemetry-client';
 import { type BotConfig } from './config.js';
-import type { BotRunnerApi } from './interface.js';
+import type { BotInfo, BotRunnerApi } from './interface.js';
 export declare class BotRunner implements BotRunnerApi, Traceable {
     #private;
     private config;
@@ -9,6 +10,7 @@ export declare class BotRunner implements BotRunnerApi, Traceable {
     private bot?;
     private pxe?;
     private node;
+    private nodeAdmin?;
     private runningPromise;
     private consecutiveErrors;
     private healthy;
@@ -16,6 +18,7 @@ export declare class BotRunner implements BotRunnerApi, Traceable {
     constructor(config: BotConfig, dependencies: {
         pxe?: PXE;
         node?: AztecNode;
+        nodeAdmin?: AztecNodeAdmin;
         telemetry: TelemetryClient;
     });
     /** Initializes the bot if needed. Blocks until the bot setup is finished. */
@@ -45,5 +48,7 @@ export declare class BotRunner implements BotRunnerApi, Traceable {
     run(): Promise<void>;
     /** Returns the current configuration for the bot. */
     getConfig(): Promise<BotConfig>;
+    /** Returns the bot sender address. */
+    getInfo(): Promise<BotInfo>;
 }
 //# sourceMappingURL=runner.d.ts.map

package/dest/runner.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runner.d.ts","sourceRoot":"","sources":["../src/runner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,SAAS,EAAE,KAAK,GAAG,EAAuC,MAAM,iBAAiB,CAAC;AAEhG,OAAO,EAAE,KAAK,eAAe,EAAE,KAAK,SAAS,EAAE,KAAK,MAAM,EAA8B,MAAM,yBAAyB,CAAC;AAGxH,OAAO,EAAE,KAAK,SAAS,EAAe,MAAM,aAAa,CAAC;AAC1D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAEnD,qBAAa,SAAU,YAAW,YAAY,EAAE,SAAS;;IAYrD,OAAO,CAAC,MAAM;IAXhB,OAAO,CAAC,GAAG,CAAuB;IAClC,OAAO,CAAC,GAAG,CAAC,CAAe;IAC3B,OAAO,CAAC,GAAG,CAAC,CAAM;IAClB,OAAO,CAAC,IAAI,CAAY;IACxB,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,OAAO,CAAQ;IAEvB,SAAgB,MAAM,EAAE,MAAM,CAAC;gBAGrB,MAAM,EAAE,SAAS,EACzB,YAAY,EAAE;QAAE,GAAG,CAAC,EAAE,GAAG,CAAC;QAAC,IAAI,CAAC,EAAE,SAAS,CAAC;QAAC,SAAS,EAAE,eAAe,CAAA;KAAE;IAY3E,6EAA6E;IAChE,KAAK;YAOJ,OAAO;IAMrB;;;OAGG;IACU,KAAK;IAQlB;;OAEG;IACU,IAAI;IAQV,SAAS;IAIhB,0CAA0C;IACnC,SAAS;IAIhB;;;OAGG;IACU,MAAM,CAAC,MAAM,EAAE,SAAS;IAerC;;;OAGG;IACU,GAAG;IAwBhB,qDAAqD;IAC9C,SAAS;CAuCjB"}
+{"version":3,"file":"runner.d.ts","sourceRoot":"","sources":["../src/runner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,SAAS,EAAE,KAAK,GAAG,EAAuC,MAAM,iBAAiB,CAAC;AAGhG,OAAO,EAAE,KAAK,cAAc,EAA8B,MAAM,iCAAiC,CAAC;AAClG,OAAO,EAAE,KAAK,eAAe,EAAE,KAAK,SAAS,EAAE,KAAK,MAAM,EAA8B,MAAM,yBAAyB,CAAC;AAKxH,OAAO,EAAE,KAAK,SAAS,EAAe,MAAM,aAAa,CAAC;AAC1D,OAAO,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE5D,qBAAa,SAAU,YAAW,YAAY,EAAE,SAAS;;IAarD,OAAO,CAAC,MAAM;IAZhB,OAAO,CAAC,GAAG,CAAuB;IAClC,OAAO,CAAC,GAAG,CAAC,CAAmB;IAC/B,OAAO,CAAC,GAAG,CAAC,CAAM;IAClB,OAAO,CAAC,IAAI,CAAY;IACxB,OAAO,CAAC,SAAS,CAAC,CAAiB;IACnC,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,OAAO,CAAQ;IAEvB,SAAgB,MAAM,EAAE,MAAM,CAAC;gBAGrB,MAAM,EAAE,SAAS,EACzB,YAAY,EAAE;QAAE,GAAG,CAAC,EAAE,GAAG,CAAC;QAAC,IAAI,CAAC,EAAE,SAAS,CAAC;QAAC,SAAS,CAAC,EAAE,cAAc,CAAC;QAAC,SAAS,EAAE,eAAe,CAAA;KAAE;IAgBvG,6EAA6E;IAChE,KAAK;YAOJ,OAAO;IAMrB;;;OAGG;IACU,KAAK;IAQlB;;OAEG;IACU,IAAI;IAQV,SAAS;IAIhB,0CAA0C;IACnC,SAAS;IAIhB;;;OAGG;IACU,MAAM,CAAC,MAAM,EAAE,SAAS;IAerC;;;OAGG;IACU,GAAG;IAwBhB,qDAAqD;IAC9C,SAAS;IAKhB,sCAAsC;IACzB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;CA6CzC"}

package/dest/runner.js

@@ -5,8 +5,11 @@ function _ts_decorate(decorators, target, key, desc) {
     return c > 3 && r && Object.defineProperty(target, key, r), r;
 }
 import { createAztecNodeClient, createLogger } from '@aztec/aztec.js';
+import { omit } from '@aztec/foundation/collection';
 import { RunningPromise } from '@aztec/foundation/running-promise';
+import { createAztecNodeAdminClient } from '@aztec/stdlib/interfaces/client';
 import { makeTracedFetch, trackSpan } from '@aztec/telemetry-client';
+import { AmmBot } from './amm_bot.js';
 import { Bot } from './bot.js';
 import { getVersions } from './config.js';
 export class BotRunner {
@@ -15,6 +18,7 @@ export class BotRunner {
     bot;
     pxe;
     node;
+    nodeAdmin;
     runningPromise;
     consecutiveErrors;
     healthy;
@@ -29,11 +33,14 @@ export class BotRunner {
         if (!dependencies.node && !config.nodeUrl) {
             throw new Error(`Missing node URL in config or dependencies`);
         }
-        this.node = dependencies.node ?? createAztecNodeClient(config.nodeUrl, getVersions(), makeTracedFetch([
+        const versions = getVersions();
+        const fetch = makeTracedFetch([
             1,
             2,
             3
-        ], true));
+        ], true);
+        this.node = dependencies.node ?? createAztecNodeClient(config.nodeUrl, versions, fetch);
+        this.nodeAdmin = dependencies.nodeAdmin ?? (config.nodeAdminUrl ? createAztecNodeAdminClient(config.nodeAdminUrl, versions, fetch) : undefined);
         this.runningPromise = new RunningPromise(()=>this.#work(), this.log, config.txIntervalSeconds * 1000);
     }
     /** Initializes the bot if needed. Blocks until the bot setup is finished. */ async setup() {
@@ -116,13 +123,28 @@ export class BotRunner {
         }
     }
     /** Returns the current configuration for the bot. */ getConfig() {
-        return Promise.resolve(this.config);
+        const redacted = omit(this.config, 'l1Mnemonic', 'l1PrivateKey', 'senderPrivateKey');
+        return Promise.resolve(redacted);
+    }
+    /** Returns the bot sender address. */ async getInfo() {
+        if (!this.bot) {
+            throw new Error(`Bot is not initialized`);
+        }
+        const botAddress = await this.bot.then((b)=>b.wallet.getAddress());
+        return {
+            botAddress
+        };
     }
     async #createBot() {
         try {
-            this.bot = Bot.create(this.config, {
+            this.bot = this.config.ammTxs ? AmmBot.create(this.config, {
+                pxe: this.pxe,
+                node: this.node,
+                nodeAdmin: this.nodeAdmin
+            }) : Bot.create(this.config, {
                 pxe: this.pxe,
-                node: this.node
+                node: this.node,
+                nodeAdmin: this.nodeAdmin
             });
             await this.bot;
         } catch (err) {
@@ -132,15 +154,15 @@ export class BotRunner {
     }
     async #work() {
         if (this.config.maxPendingTxs > 0) {
-            const pendingTxs = await this.node.getPendingTxs();
-            if (pendingTxs.length >= this.config.maxPendingTxs) {
-                this.log.verbose(`Not sending bot tx since node has ${pendingTxs.length} pending txs`);
+            const pendingTxCount = await this.node.getPendingTxCount();
+            if (pendingTxCount >= this.config.maxPendingTxs) {
+                this.log.verbose(`Not sending bot tx since node has ${pendingTxCount} pending txs`);
                 return;
             }
         }
         try {
             await this.run();
-        } catch (err) {
+        } catch  {
             // Already logged in run()
             if (this.config.maxConsecutiveErrors > 0 && this.consecutiveErrors >= this.config.maxConsecutiveErrors) {
                 this.log.error(`Too many errors bot is unhealthy`);

package/dest/utils.d.ts

@@ -1,4 +1,6 @@
-import type { EasyPrivateTokenContract } from '@aztec/noir-contracts.js/EasyPrivateToken';
+import type { ContractBase } from '@aztec/aztec.js';
+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';
 /**
@@ -11,6 +13,7 @@ export declare function getBalances(token: TokenContract, who: AztecAddress): Pr
     privateBalance: bigint;
     publicBalance: bigint;
 }>;
-export declare function getPrivateBalance(token: EasyPrivateTokenContract, who: AztecAddress): Promise<bigint>;
-export declare function isStandardTokenContract(token: TokenContract | EasyPrivateTokenContract): token is TokenContract;
+export declare function getPrivateBalance(token: PrivateTokenContract, who: AztecAddress): Promise<bigint>;
+export declare function isStandardTokenContract(token: ContractBase): token is TokenContract;
+export declare function isAMMContract(contract: ContractBase): contract is AMMContract;
 //# sourceMappingURL=utils.d.ts.map

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,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,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,GAChB,OAAO,CAAC;IAAE,cAAc,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAE,CAAC,CAI5D;AAED,wBAAsB,iBAAiB,CAAC,KAAK,EAAE,oBAAoB,EAAE,GAAG,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,CAGvG;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

@@ -4,17 +4,26 @@
  * @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();
+    const privateBalance = await token.methods.balance_of_private(who).simulate({
+        from: who
+    });
+    const publicBalance = await token.methods.balance_of_public(who).simulate({
+        from: who
+    });
     return {
         privateBalance,
         publicBalance
     };
 }
 export async function getPrivateBalance(token, who) {
-    const privateBalance = await token.methods.get_balance(who).simulate();
+    const privateBalance = await token.methods.get_balance(who).simulate({
+        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-fake-ceab37513c",
   "type": "module",
   "exports": {
     ".": "./dest/index.js",
@@ -13,8 +13,6 @@
     "build": "yarn clean && tsc -b",
     "build:dev": "tsc -b --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,36 @@
     "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-fake-ceab37513c",
+    "@aztec/aztec.js": "0.0.1-fake-ceab37513c",
+    "@aztec/entrypoints": "0.0.1-fake-ceab37513c",
+    "@aztec/ethereum": "0.0.1-fake-ceab37513c",
+    "@aztec/foundation": "0.0.1-fake-ceab37513c",
+    "@aztec/noir-contracts.js": "0.0.1-fake-ceab37513c",
+    "@aztec/noir-protocol-circuits-types": "0.0.1-fake-ceab37513c",
+    "@aztec/protocol-contracts": "0.0.1-fake-ceab37513c",
+    "@aztec/stdlib": "0.0.1-fake-ceab37513c",
+    "@aztec/telemetry-client": "0.0.1-fake-ceab37513c",
     "source-map-support": "^0.5.21",
     "tslib": "^2.4.0",
     "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",
+    "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 +85,6 @@
   ],
   "types": "./dest/index.d.ts",
   "engines": {
-    "node": ">=18"
+    "node": ">=20.10"
   }
 }

package/src/amm_bot.ts

@@ -0,0 +1,116 @@
+import { AztecAddress, Fr, SentTx, TxReceipt, type Wallet } from '@aztec/aztec.js';
+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, PXE } from '@aztec/stdlib/interfaces/client';
+
+import { BaseBot } from './base_bot.js';
+import type { BotConfig } from './config.js';
+import { BotFactory } from './factory.js';
+
+const TRANSFER_BASE_AMOUNT = 1_000;
+const TRANSFER_VARIANCE = 200;
+
+type Balances = { token0: bigint; token1: bigint };
+
+export class AmmBot extends BaseBot {
+  protected constructor(
+    pxe: PXE,
+    wallet: Wallet,
+    defaultAccountAddress: AztecAddress,
+    public readonly amm: AMMContract,
+    public readonly token0: TokenContract,
+    public readonly token1: TokenContract,
+    config: BotConfig,
+  ) {
+    super(pxe, wallet, defaultAccountAddress, config);
+  }
+
+  static async create(
+    config: BotConfig,
+    dependencies: { pxe?: PXE; node?: AztecNode; nodeAdmin?: AztecNodeAdmin },
+  ): Promise<AmmBot> {
+    const { pxe, wallet, defaultAccountAddress, token0, token1, amm } = await new BotFactory(
+      config,
+      dependencies,
+    ).setupAmm();
+    return new AmmBot(pxe, wallet, defaultAccountAddress, amm, token0, token1, config);
+  }
+
+  protected async createAndSendTx(logCtx: object): Promise<SentTx> {
+    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({
+      caller: amm.address,
+      action: tokenIn.methods.transfer_to_public(this.defaultAccountAddress, amm.address, amountIn, authwitNonce),
+    });
+
+    const amountOutMin = await amm.methods
+      .get_amount_out_for_exact_in(
+        await tokenIn.methods.balance_of_public(amm.address).simulate({ from: this.defaultAccountAddress }),
+        await tokenOut.methods.balance_of_public(amm.address).simulate({ from: this.defaultAccountAddress }),
+        amountIn,
+      )
+      .simulate({ from: this.defaultAccountAddress });
+
+    const swapExactTokensInteraction = amm.methods.swap_exact_tokens_for_tokens(
+      tokenIn.address,
+      tokenOut.address,
+      amountIn,
+      amountOutMin,
+      authwitNonce,
+    );
+
+    const opts = this.getSendMethodOpts(swapAuthwit);
+
+    this.log.verbose(`Proving transaction`, logCtx);
+    const tx = await swapExactTokensInteraction.prove(opts);
+
+    this.log.info(`Tx. Balances: ${jsonStringify(balances)}`, { ...logCtx, balances });
+
+    return tx.send();
+  }
+
+  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),
+    };
+  }
+
+  private async getPublicBalanceFor(address: AztecAddress): Promise<Balances> {
+    return {
+      token0: await this.token0.methods.balance_of_public(address).simulate({ from: address }),
+      token1: await this.token1.methods.balance_of_public(address).simulate({ from: address }),
+    };
+  }
+  private async getPrivateBalanceFor(address: AztecAddress): Promise<Balances> {
+    return {
+      token0: await this.token0.methods.balance_of_private(address).simulate({ from: address }),
+      token1: await this.token1.methods.balance_of_private(address).simulate({ from: address }),
+    };
+  }
+}

package/src/base_bot.ts

@@ -0,0 +1,94 @@
+import {
+  AuthWitness,
+  AztecAddress,
+  FeeJuicePaymentMethod,
+  type SendMethodOptions,
+  SentTx,
+  TxHash,
+  TxReceipt,
+  type Wallet,
+  createLogger,
+  waitForProven,
+} from '@aztec/aztec.js';
+import { Gas } from '@aztec/stdlib/gas';
+import type { PXE } from '@aztec/stdlib/interfaces/client';
+
+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 pxe: PXE,
+    public readonly wallet: Wallet,
+    public readonly defaultAccountAddress: AztecAddress,
+    public config: BotConfig,
+  ) {}
+
+  public async run(): Promise<TxReceipt | TxHash> {
+    this.attempts++;
+    const logCtx = { runId: Date.now() * 1000 + Math.floor(Math.random() * 1000) };
+    const { followChain, txMinedWaitSeconds } = this.config;
+
+    this.log.verbose(`Creating tx`, logCtx);
+    const tx = await this.createAndSendTx(logCtx);
+
+    const txHash = await tx.getTxHash();
+
+    if (followChain === 'NONE') {
+      this.log.info(`Transaction ${txHash.toString()} sent, not waiting for it to be mined`);
+      return txHash;
+    }
+
+    this.log.verbose(
+      `Awaiting tx ${txHash.toString()} to be on the ${followChain} chain (timeout ${txMinedWaitSeconds}s)`,
+      logCtx,
+    );
+    const receipt = await tx.wait({
+      timeout: txMinedWaitSeconds,
+    });
+    if (followChain === 'PROVEN') {
+      await waitForProven(this.pxe, receipt, { provenTimeout: txMinedWaitSeconds });
+    }
+    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<SentTx>;
+
+  protected onTxMined(_receipt: TxReceipt, _logCtx: object): Promise<void> {
+    // no-op
+    return Promise.resolve();
+  }
+
+  protected getSendMethodOpts(...authWitnesses: AuthWitness[]): SendMethodOptions {
+    const { l2GasLimit, daGasLimit } = this.config;
+    const paymentMethod = new FeeJuicePaymentMethod(this.defaultAccountAddress);
+
+    let gasSettings, estimateGas;
+    if (l2GasLimit !== undefined && l2GasLimit > 0 && daGasLimit !== undefined && daGasLimit > 0) {
+      gasSettings = { gasLimits: Gas.from({ l2Gas: l2GasLimit, daGas: daGasLimit }) };
+      estimateGas = false;
+      this.log.verbose(`Using gas limits ${l2GasLimit} L2 gas ${daGasLimit} DA gas`);
+    } else {
+      estimateGas = true;
+      this.log.verbose(`Estimating gas for transaction`);
+    }
+    const baseFeePadding = 2; // Send 3x the current base fee
+    return {
+      from: this.defaultAccountAddress,
+      fee: { estimateGas, paymentMethod, gasSettings, baseFeePadding },
+      authWitnesses,
+    };
+  }
+}