adding missing actions

2026-06-08 15:11:53 +00:00 · 2024-12-31 17:54:33 +01:00
parent 1b601b839d
commit 2a08017dcd
6 changed files with 272 additions and 79 deletions
--- a/src/actions/compressedAirdrop.ts
+++ b/src/actions/compressedAirdrop.ts
@@ -0,0 +1,105 @@
+import { Action } from "../types/action";
+import { SolanaAgentKit } from "../agent";
+import { z } from "zod";
+import { sendCompressedAirdrop } from "../tools";
+
+const compressedAirdropAction: Action = {
+    name: "solana_compressed_airdrop",
+    similes: [
+        "ZK Compressed airdrop",
+        "Airdrop tokens with compression",
+        "Send compressed SPL airdrop",
+        "Airdrop to multiple recipients",
+    ],
+    description:
+        "Airdrop SPL tokens with ZK Compression (also known as airdropping tokens) to multiple recipients",
+    examples: [
+        [
+            {
+                input: {
+                    mintAddress: "JUPyiwrYJFskUPiHa7hkeR8VUtAeFoSYbKedZNsDvCN",
+                    amount: 42,
+                    decimals: 6,
+                    recipients: [
+                        "1nc1nerator11111111111111111111111111111111",
+                        "BrFndAe111111111111111111111111111111111",
+                    ],
+                    priorityFeeInLamports: 30000,
+                    shouldLog: true,
+                },
+                output: {
+                    status: "success",
+                    message:
+                        "Airdropped 42 tokens to 2 recipients.",
+                    transactionHashes: ["4uyfBN...", "9XsF2N..."],
+                },
+                explanation:
+                    "Airdrops 42 tokens (with 6 decimals) to 2 recipients, optionally logging progress to stdout.",
+            },
+        ],
+    ],
+    // Validate inputs with zod
+    schema: z.object({
+        mintAddress: z
+            .string()
+            .min(1)
+            .describe("Mint address of the token, e.g., 'JUPy...'"),
+        amount: z
+            .number()
+            .positive()
+            .describe("Number of tokens to airdrop per recipient, e.g., 42"),
+        decimals: z
+            .number()
+            .nonnegative()
+            .int()
+            .describe("Decimals of the token, e.g., 6"),
+        recipients: z
+            .array(z.string())
+            .nonempty()
+            .describe("Array of recipient addresses, e.g., ['1nc1n...']"),
+        priorityFeeInLamports: z
+            .number()
+            .optional()
+            .describe("Priority fee in lamports (default is 30_000)"),
+        shouldLog: z
+            .boolean()
+            .optional()
+            .describe("Whether to log progress to stdout (default is false)"),
+    }),
+    handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
+        try {
+            const {
+                mintAddress,
+                amount,
+                decimals,
+                recipients,
+                priorityFeeInLamports,
+                shouldLog,
+            } = input;
+
+            // Call your airdrop method on the SolanaAgentKit
+            const txs = await sendCompressedAirdrop(
+                mintAddress,
+                amount,
+                decimals,
+                recipients,
+                priorityFeeInLamports || 30_000,
+                shouldLog || false,
+            );
+
+            return {
+                status: "success",
+                message: `Airdropped ${amount} tokens to ${recipients.length} recipients.`,
+                transactionHashes: txs,
+            };
+        } catch (error: any) {
+            return {
+                status: "error",
+                message: `Failed to airdrop tokens: ${error.message}`,
+                code: error.code || "UNKNOWN_ERROR",
+            };
+        }
+    },
+};
+
+export default compressedAirdropAction;