import { z } from "zod";
import type { Action } from "../../types";
import { withdrawFromDriftUserAccount } from "../../tools";

const withdrawFromDriftAccountAction: Action = {
  name: "WITHDRAW_OR_BORROW_FROM_DRIFT_ACCOUNT",
  description: "Withdraw funds from your drift account",
  similes: [
    "withdraw from drift account",
    "withdraw funds from drift account",
    "withdraw funds from my drift account",
    "borrow from drift account",
    "borrow funds from my drift account",
    "borrow from drift",
    "withdraw from drift",
  ],
  examples: [
    [
      {
        input: {
          amount: 100,
          symbol: "usdc",
        },
        output: {
          status: "success",
          message: "Funds withdrawn successfully",
          signature:
            "2nFeP7taii3wGVgrWk4YiLMPmhtu3Zg9iXCUu4zGBDadwunHw8reXFxRWT7khbFsQ9JT3zK4RYDLNDFDRYvM3wJk",
        },
        explanation: "Withdraw 100 USDC from your drift account",
      },
    ],
  ],
  schema: z.object({
    amount: z
      .number()
      .positive()
      .describe(
        "The amount in tokens you'd like to withdraw from your drift account in normal token amounts, e.g 50 SOL, 100 USDC, etc",
      ),
    symbol: z
      .string()
      .toUpperCase()
      .describe("The symbol of the token you'd like to withdraw"),
    isBorrow: z
      .boolean()
      .optional()
      .default(false)
      .describe(
        "Whether or not to borrow funds based on collateral provided instead of withdrawing",
      ),
  }),
  handler: async (agent, input) => {
    try {
      const tx = await withdrawFromDriftUserAccount(
        agent,
        input.amount,
        input.symbol,
        input.isBorrow,
      );

      return {
        status: "success",
        message: "Funds withdrawn successfully",
        signature: tx,
      };
    } catch (e) {
      return {
        status: "error",
        // @ts-expect-error - error message is a string
        message: `Failed to withdraw funds: ${e.message}`,
      };
    }
  },
};

export default withdrawFromDriftAccountAction;