fly bun, fly

signorecello · signorecello · commit f62eff7eacac · 2025-01-16T16:55:14.000Z
diff --git a/tooling/sparta/dist/commands/addValidator.js b/tooling/sparta/dist/commands/addValidator.js
@@ -0,0 +1,71 @@
+import { SlashCommandBuilder } from "discord.js";
+import { ValidatorService } from "../services/validator-service.js";
+import { ChainInfoService } from "../services/chaininfo-service.js";
+export default {
+    data: new SlashCommandBuilder()
+        .setName("validator")
+        .setDescription("Manage validator addresses")
+        .addSubcommand((subcommand) => subcommand
+        .setName("add")
+        .setDescription("Add yourself to the validator set")
+        .addStringOption((option) => option
+        .setName("address")
+        .setDescription("Your validator address")))
+        .addSubcommand((subcommand) => subcommand
+        .setName("check")
+        .setDescription("Check if you are a validator")
+        .addStringOption((option) => option
+        .setName("address")
+        .setDescription("The validator address to check"))),
+    execute: async (interaction) => {
+        const address = interaction.options.getString("address");
+        if (!address) {
+            return interaction.reply({
+                content: "Address is required.",
+                ephemeral: true,
+            });
+        }
+        // Basic address validation
+        if (!address.match(/^0x[a-fA-F0-9]{40}$/)) {
+            return interaction.reply({
+                content: "Please provide a valid Ethereum address.",
+                ephemeral: true,
+            });
+        }
+        await interaction.deferReply();
+        if (interaction.options.getSubcommand() === "add") {
+            try {
+                await ValidatorService.addValidator(address);
+                await interaction.editReply({
+                    content: `Successfully added validator address: ${address}`,
+                });
+            }
+            catch (error) {
+                await interaction.editReply({
+                    content: `Failed to add validator address: ${error instanceof Error ? error.message : String(error)}`,
+                });
+            }
+        }
+        else if (interaction.options.getSubcommand() === "check") {
+            try {
+                const info = await ChainInfoService.getInfo();
+                const { validators, committee } = info;
+                let reply = "";
+                if (validators.includes(address)) {
+                    reply += "You are a validator\n";
+                }
+                if (committee.includes(address)) {
+                    reply += "You are a committee member\n";
+                }
+                await interaction.editReply({
+                    content: reply,
+                });
+            }
+            catch (error) {
+                await interaction.editReply({
+                    content: `Failed to check validator address: ${error instanceof Error ? error.message : String(error)}`,
+                });
+            }
+        }
+    },
+};
diff --git a/tooling/sparta/dist/commands/getChainInfo.js b/tooling/sparta/dist/commands/getChainInfo.js
@@ -0,0 +1,22 @@
+import { SlashCommandBuilder } from "discord.js";
+import { ChainInfoService } from "../services/chaininfo-service.js";
+export default {
+    data: new SlashCommandBuilder()
+        .setName("get-info")
+        .setDescription("Get chain info"),
+    execute: async (interaction) => {
+        await interaction.deferReply();
+        try {
+            const { pendingBlockNum, provenBlockNum, currentEpoch, currentSlot, proposerNow, } = await ChainInfoService.getInfo();
+            await interaction.editReply({
+                content: `Pending block: ${pendingBlockNum}\nProven block: ${provenBlockNum}\nCurrent epoch: ${currentEpoch}\nCurrent slot: ${currentSlot}\nProposer now: ${proposerNow}`,
+            });
+        }
+        catch (error) {
+            console.error("Error in get-info command:", error);
+            await interaction.editReply({
+                content: `Failed to get chain info`,
+            });
+        }
+    },
+};
diff --git a/tooling/sparta/dist/commands/index.js b/tooling/sparta/dist/commands/index.js
@@ -0,0 +1,6 @@
+import addValidator from "./addValidator.js";
+import getChainInfo from "./getChainInfo.js";
+export default {
+    addValidator,
+    getChainInfo,
+};
diff --git a/tooling/sparta/dist/deploy-commands.js b/tooling/sparta/dist/deploy-commands.js
@@ -0,0 +1,17 @@
+import { REST, Routes } from "discord.js";
+import commands from "./commands/index.js";
+import { BOT_TOKEN, BOT_CLIENT_ID, GUILD_ID } from "./env.js";
+export const deployCommands = async () => {
+    const rest = new REST({ version: "10" }).setToken(BOT_TOKEN);
+    try {
+        console.log("Started refreshing application (/) commands.");
+        const commandsData = Object.values(commands).map((command) => command.data.toJSON());
+        await rest.put(Routes.applicationGuildCommands(BOT_CLIENT_ID, GUILD_ID), {
+            body: commandsData,
+        });
+        console.log("Successfully reloaded application (/) commands.");
+    }
+    catch (error) {
+        console.error(error);
+    }
+};
diff --git a/tooling/sparta/dist/env.js b/tooling/sparta/dist/env.js
@@ -0,0 +1,3 @@
+import dotenv from "dotenv";
+dotenv.config();
+export const { TOKEN, CLIENT_ID, GUILD_ID, PRODUCTION_CHANNEL_NAME, DEV_CHANNEL_NAME, PRODUCTION_CHANNEL_ID, DEV_CHANNEL_ID, ETHEREUM_HOST, ETHEREUM_ROLLUP_ADDRESS, ETHEREUM_ADMIN_ADDRESS, ETHEREUM_CHAIN_ID, ETHEREUM_MNEMONIC, ETHEREUM_PRIVATE_KEY, ETHEREUM_VALUE, BOT_TOKEN, BOT_CLIENT_ID, ENVIRONMENT, } = process.env;
diff --git a/tooling/sparta/dist/index.js b/tooling/sparta/dist/index.js
@@ -0,0 +1,46 @@
+import { Client, GatewayIntentBits, Collection, MessageFlags, } from "discord.js";
+import { deployCommands } from "./deploy-commands.js";
+import commands from "./commands/index.js";
+import { BOT_TOKEN, PRODUCTION_CHANNEL_ID, DEV_CHANNEL_ID, ENVIRONMENT, PRODUCTION_CHANNEL_NAME, DEV_CHANNEL_NAME, } from "./env.js";
+const client = new Client({
+    intents: [GatewayIntentBits.Guilds, GatewayIntentBits.GuildMessages],
+});
+client.commands = new Collection();
+for (const command of Object.values(commands)) {
+    client.commands.set(command.data.name, command);
+}
+client.once("ready", () => {
+    console.log("Sparta bot is ready!");
+    deployCommands();
+});
+client.on("interactionCreate", async (interaction) => {
+    if (!interaction.isChatInputCommand())
+        return;
+    // Determine which channel to use based on environment
+    const targetChannelId = ENVIRONMENT === "production" ? PRODUCTION_CHANNEL_ID : DEV_CHANNEL_ID;
+    // Check if the command is in the correct channel
+    if (interaction.channelId !== targetChannelId) {
+        const channelName = ENVIRONMENT === "production"
+            ? PRODUCTION_CHANNEL_NAME
+            : DEV_CHANNEL_NAME;
+        return interaction.reply({
+            content: `This command can only be used in the ${channelName} channel.`,
+            flags: MessageFlags.Ephemeral,
+        });
+    }
+    const command = client.commands.get(interaction.commandName);
+    if (!command)
+        return;
+    try {
+        console.log("Executing command:", command.data.name);
+        const response = await command.execute(interaction);
+    }
+    catch (error) {
+        console.error(error);
+        await interaction.reply({
+            content: "There was an error executing this command!",
+            flags: MessageFlags.Ephemeral,
+        });
+    }
+});
+client.login(BOT_TOKEN);
diff --git a/tooling/sparta/dist/services/chaininfo-service.js b/tooling/sparta/dist/services/chaininfo-service.js
@@ -0,0 +1,34 @@
+import { exec } from "child_process";
+import { promisify } from "util";
+import { ETHEREUM_HOST, ETHEREUM_ROLLUP_ADDRESS, ETHEREUM_CHAIN_ID, } from "../env.js";
+const execAsync = promisify(exec);
+export class ChainInfoService {
+    static async getInfo() {
+        try {
+            // Add validator to the set
+            const command = `docker run --rm aztecprotocol/aztec:unhinged-unicorn debug-rollup -u ${ETHEREUM_HOST} --rollup ${ETHEREUM_ROLLUP_ADDRESS} --l1-chain-id ${ETHEREUM_CHAIN_ID} `;
+            const { stdout, stderr } = await execAsync(command);
+            if (stderr) {
+                throw new Error(stderr);
+            }
+            // looks like hell, but it just parses the output of the command
+            // into a key-value object
+            const info = stdout
+                .split("\n")
+                .map((line) => line.trim())
+                .filter(Boolean)
+                .reduce((acc, s) => {
+                const [key, value] = s.split(": ");
+                const sanitizedKey = key
+                    .toLowerCase()
+                    .replace(/\s+(.)/g, (_, c) => c.toUpperCase());
+                return { ...acc, [sanitizedKey]: value };
+            }, {});
+            return info;
+        }
+        catch (error) {
+            console.error("Error getting chain info:", error);
+            throw error;
+        }
+    }
+}
diff --git a/tooling/sparta/dist/services/validator-service.js b/tooling/sparta/dist/services/validator-service.js
@@ -0,0 +1,37 @@
+import { exec } from "child_process";
+import { promisify } from "util";
+import { ETHEREUM_HOST, ETHEREUM_ROLLUP_ADDRESS, ETHEREUM_ADMIN_ADDRESS, ETHEREUM_CHAIN_ID, ETHEREUM_MNEMONIC, ETHEREUM_PRIVATE_KEY, ETHEREUM_VALUE, } from "../env.js";
+const execAsync = promisify(exec);
+export class ValidatorService {
+    static async addValidator(address) {
+        try {
+            // Send ETH to the validator address
+            await this.fundValidator(address);
+            // Add validator to the set
+            const command = `docker run --rm aztecprotocol/aztec:unhinged-unicorn add-l1-validator -u ${ETHEREUM_HOST} --validator ${address} --rollup ${ETHEREUM_ROLLUP_ADDRESS} --withdrawer ${ETHEREUM_ADMIN_ADDRESS} --l1-chain-id ${ETHEREUM_CHAIN_ID} --mnemonic "${ETHEREUM_MNEMONIC}"`;
+            const { stdout, stderr } = await execAsync(command);
+            if (stderr) {
+                throw new Error(stderr);
+            }
+            return stdout;
+        }
+        catch (error) {
+            console.error("Error adding validator:", error);
+            throw error;
+        }
+    }
+    static async fundValidator(address) {
+        try {
+            const command = `cast send --value ${ETHEREUM_VALUE} --rpc-url ${ETHEREUM_HOST} --chain-id ${ETHEREUM_CHAIN_ID} --private-key ${ETHEREUM_PRIVATE_KEY} ${address}`;
+            const { stdout, stderr } = await execAsync(command);
+            if (stderr) {
+                throw new Error(stderr);
+            }
+            return stdout;
+        }
+        catch (error) {
+            console.error("Error funding validator:", error);
+            throw error;
+        }
+    }
+}
diff --git a/tooling/sparta/package-lock.json b/tooling/sparta/package-lock.json
diff --git a/tooling/sparta/package.json b/tooling/sparta/package.json

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+import dotenv from "dotenv";`
	`2`	`+dotenv.config();`
	`3`	`+export const { TOKEN, CLIENT_ID, GUILD_ID, PRODUCTION_CHANNEL_NAME, DEV_CHANNEL_NAME, PRODUCTION_CHANNEL_ID, DEV_CHANNEL_ID, ETHEREUM_HOST, ETHEREUM_ROLLUP_ADDRESS, ETHEREUM_ADMIN_ADDRESS, ETHEREUM_CHAIN_ID, ETHEREUM_MNEMONIC, ETHEREUM_PRIVATE_KEY, ETHEREUM_VALUE, BOT_TOKEN, BOT_CLIENT_ID, ENVIRONMENT, } = process.env;`