feat: Configure bot to run a local PXE (#7700)

This PR configures the devnet bot to run all in one, with an integrated PXE
AztecProtocol · Jul 31, 2024 · eabafa2 · eabafa2
1 parent 26cec69
commit eabafa2
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 5 deletions.
diff --git a/yarn-project/aztec/terraform/bot/main.tf b/yarn-project/aztec/terraform/bot/main.tf
@@ -76,8 +76,8 @@ locals {
 resource "aws_ecs_task_definition" "aztec-bot" {
   family                   = "${var.DEPLOY_TAG}-aztec-bot"
   network_mode             = "awsvpc"
-  cpu                      = 2048
-  memory                   = 4096
+  cpu                      = 16384
+  memory                   = 32768
   requires_compatibilities = ["FARGATE"]
   execution_role_arn       = data.terraform_remote_state.setup_iac.outputs.ecs_task_execution_role_arn
   task_role_arn            = data.terraform_remote_state.aztec2_iac.outputs.cloudwatch_logging_ecs_role_arn
@@ -86,7 +86,7 @@ resource "aws_ecs_task_definition" "aztec-bot" {
     {
       name      = "${var.DEPLOY_TAG}-aztec-bot"
       image     = "${var.DOCKERHUB_ACCOUNT}/aztec:${var.DEPLOY_TAG}"
-      command   = ["start", "--bot"]
+      command   = ["start", "--bot", "--pxe"]
       essential = true
       portMappings = [
         {
@@ -97,14 +97,15 @@ resource "aws_ecs_task_definition" "aztec-bot" {
       environment = [
         { name = "BOT_PRIVATE_KEY", value = var.BOT_PRIVATE_KEY },
         { name = "BOT_NO_START", value = var.BOT_NO_START },
-        { name = "BOT_PXE_URL", value = "http://${var.DEPLOY_TAG}-aztec-pxe.local/${var.DEPLOY_TAG}/aztec-pxe/${var.API_KEY}" },
         { name = "BOT_TX_INTERVAL_SECONDS", value = var.BOT_TX_INTERVAL_SECONDS },
         { name = "LOG_LEVEL", value = var.LOG_LEVEL },
         { name = "AZTEC_PORT", value = "80" },
         { name = "API_PREFIX", value = local.api_prefix },
         { name = "BOT_PRIVATE_TRANSFERS_PER_TX", value = var.BOT_PRIVATE_TRANSFERS_PER_TX },
         { name = "BOT_PUBLIC_TRANSFERS_PER_TX", value = var.BOT_PUBLIC_TRANSFERS_PER_TX },
         { name = "BOT_TX_MINED_WAIT_SECONDS", value = var.BOT_TX_MINED_WAIT_SECONDS },
+        { name = "AZTEC_NODE_URL", value = "http://${var.DEPLOY_TAG}-aztec-node-1.local/${var.DEPLOY_TAG}/aztec-node-1/${var.API_KEY}" },
+        { name = "PXE_PROVER_ENABLED", value = tostring(var.PROVING_ENABLED) }
       ]
       logConfiguration = {
         logDriver = "awslogs"

diff --git a/yarn-project/aztec/terraform/bot/variables.tf b/yarn-project/aztec/terraform/bot/variables.tf
@@ -42,3 +42,8 @@ variable "BOT_TX_INTERVAL_SECONDS" {
 variable "BOT_TX_MINED_WAIT_SECONDS" {
   type = string
 }
+
+variable "PROVING_ENABLED" {
+  type    = bool
+  default = false
+}
diff --git a/yarn-project/bot/src/factory.ts b/yarn-project/bot/src/factory.ts
@@ -19,7 +19,13 @@ export class BotFactory {
       throw new Error(`Either a PXE client or a PXE URL must be provided`);
     }
 
-    this.pxe = dependencies.pxe ?? createPXEClient(config.pxeUrl!);
+    if (dependencies.pxe) {
+      this.log.info(`Using local PXE`);
+      this.pxe = dependencies.pxe;
+      return;
+    }
+    this.log.info(`Using remote PXE at ${config.pxeUrl!}`);
+    this.pxe = createPXEClient(config.pxeUrl!);
   }
 
   /**