rebase final

src/agent/index.ts

@@ -35,6 +35,7 @@ import {
   getOwnedAllDomains,
   resolveAllDomains,
   create_gibwork_task,
+  rock_paper_scissor,
 } from "../tools";
 import {
   CollectionDeployment,
@@ -337,4 +338,8 @@ export class SolanaAgentKit {
       payer ? new PublicKey(payer) : undefined,
     );
   }
+
+  async rockPaperScissors(amount: number, choice: "rock" | "paper" | "scissors") {
+    return rock_paper_scissor(this, amount, choice);
+  }
 }

src/langchain/index.ts

@@ -1229,6 +1229,52 @@ export class SolanaCreateGibworkTask extends Tool {
   }
 }
 
+export class SolanaRockPaperScissorsTool extends Tool {
+  name = "rock_paper_scissors_blink";
+  description = `Play rock paper scissors to win the SEND coin.
+  Inputs(convert the input to a json string):
+  choice: string, either "rock", "paper", or "scissors" (required)
+  amount: number, amount of SOL to play the game with, either 0.1, 0.01, or 0.005 SOL (required)`;
+
+  constructor(private solanaKit: SolanaAgentKit) {
+    super();
+  }
+
+  private validateInput(input: any): void {
+    if (input.choice !== undefined) {
+      throw new Error('choice is required.');
+    }
+    if (
+      input.amount !== undefined &&
+      (typeof input.spaceKB !== "number" || input.spaceKB <= 0)
+    ) {
+      throw new Error("amount must be a positive number when provided");
+    }
+  }
+
+  protected async _call(input: string): Promise<string> {
+    try {
+      const parsedInput = toJSON(input);
+      this.validateInput(parsedInput);
+      const result = await this.solanaKit.rockPaperScissors(
+        Number(parsedInput['"amount"']),
+        parsedInput['"choice"'].replace(/^"|"$/g, '') as "rock" | "paper" | "scissors"
+      );
+      
+      return JSON.stringify({
+        status: "success",
+        message: result,
+      });
+    } catch (error: any) {
+      return JSON.stringify({
+        status: "error",
+        message: error.message,
+        code: error.code || "UNKNOWN_ERROR",
+      });
+    }
+  }
+}
+
 export function createSolanaTools(solanaKit: SolanaAgentKit) {
   return [
     new SolanaBalanceTool(solanaKit),
@@ -1263,5 +1309,6 @@ export function createSolanaTools(solanaKit: SolanaAgentKit) {
     new SolanaGetMainDomain(solanaKit),
     new SolanaResolveAllDomainsTool(solanaKit),
     new SolanaCreateGibworkTask(solanaKit),
+    new SolanaRockPaperScissorsTool(solanaKit),
   ];
 }

src/tools/index.ts

@@ -37,3 +37,5 @@ export * from "./openbook_create_market";
 export * from "./pyth_fetch_price";
 
 export * from "./create_gibwork_task";
+
+export * from "./rock_paper_scissor";

src/tools/rock_paper_scissor.ts

@@ -0,0 +1,126 @@
+import { sendAndConfirmTransaction, Transaction } from "@solana/web3.js";
+import { SolanaAgentKit } from "../agent";
+
+export async function rock_paper_scissor(
+    agent: SolanaAgentKit,
+    amount: number,
+    choice: "rock" | "paper" | "scissors",
+) {
+    try {
+        const res = await fetch(
+            `https://rps.sendarcade.fun/api/actions/bot?amount=${amount}&choice=${choice}`,
+            {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                },
+                body: JSON.stringify({
+                    account: agent.wallet.publicKey.toBase58(),
+                }),
+            },
+        );
+
+        const data = await res.json();
+        if (data.transaction) {
+            const txn = Transaction.from(Buffer.from(data.transaction, "base64"));
+            txn.sign(agent.wallet);
+            txn.recentBlockhash = (
+                await agent.connection.getLatestBlockhash()
+            ).blockhash;
+            const sig = await sendAndConfirmTransaction(
+                agent.connection,
+                txn,
+                [agent.wallet],
+                { commitment: 'confirmed' }
+            );
+            let href = data.links?.next?.href;
+            return await outcome(agent, sig, href);
+        } else {
+            return "failed";
+        }
+    } catch (error: any) {
+        console.error(error);
+        throw new Error(`RPS game failed: ${error.message}`);
+    }
+}
+async function outcome(agent: SolanaAgentKit, sig: string, href: string): Promise<string> {
+    try {
+        const res = await fetch(
+            "https://rps.sendarcade.fun" + href, // href = /api/actions/outcome?id=...
+            {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                },
+                body: JSON.stringify({
+                    account: agent.wallet.publicKey.toBase58(),
+                    signature: sig,
+                }),
+            },
+        );
+
+        const data: any = await res.json();
+        const title = data.title;
+        if (title.startsWith("You lost")) {
+            return title;
+        }
+        let next_href = data.links?.actions?.[0]?.href;
+        return title + "\n" + await won(agent, next_href)
+    } catch (error: any) {
+        console.error(error);
+        throw new Error(`RPS outcome failed: ${error.message}`);
+    }
+}
+async function won(agent: SolanaAgentKit, href: string): Promise<string> {
+    try {
+        const res = await fetch(
+            "https://rps.sendarcade.fun" + href, // href = /api/actions/won?id=...
+            {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                },
+                body: JSON.stringify({
+                    account: agent.wallet.publicKey.toBase58(),
+                }),
+            },
+        );
+
+        const data: any = await res.json();
+        if (data.transaction) {
+            const txn = Transaction.from(Buffer.from(data.transaction, "base64"));
+            txn.partialSign(agent.wallet);
+            await agent.connection.sendRawTransaction(txn.serialize(),{ preflightCommitment: 'confirmed' });        }
+        else {
+            return "Failed to claim prize.";
+        }
+        let next_href = data.links?.next?.href;
+        return await postWin(agent, next_href);
+    } catch (error: any) {
+        console.error(error);
+        throw new Error(`RPS outcome failed: ${error.message}`);
+    }
+}
+async function postWin(agent: SolanaAgentKit, href: string): Promise<string> {
+    try {
+        const res = await fetch(
+            "https://rps.sendarcade.fun" + href, // href = /api/actions/postwin?id=...
+            {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                },
+                body: JSON.stringify({
+                    account: agent.wallet.publicKey.toBase58(),
+                }),
+            },
+        );
+
+        const data: any = await res.json();
+        const title = data.title;
+        return "Prize claimed Successfully" + "\n" + title;
+    } catch (error: any) {
+        console.error(error);
+        throw new Error(`RPS outcome failed: ${error.message}`);
+    }
+}