import { PublicKey } from "@solana/web3.js";
import { Action } from "../types/action";
import { SolanaAgentKit } from "../agent";
import { z } from "zod";
import { mintCollectionNFT } from "../tools";

const mintNFTAction: Action = {
  name: "solana_mint_nft",
  similes: [
    "mint nft",
    "create nft",
    "mint token",
    "create token",
    "add nft to collection"
  ],
  description: `Mint a new NFT in a collection on Solana blockchain.`,
  examples: [
    [
      {
        input: {
          collectionMint: "J1S9H3QjnRtBbbuD4HjPV6RpRhwuk4zKbxsnCHuTgh9w",
          name: "My NFT",
          uri: "https://example.com/nft.json"
        },
        output: {
          status: "success",
          message: "NFT minted successfully",
          mintAddress: "7nE9GvcwsqzYxmJLSrYmSB1V1YoJWVK1KWzAcWAzjXkN",
          metadata: {
            name: "My NFT",
            uri: "https://example.com/nft.json"
          },
          recipient: "7nE9GvcwsqzYxmJLSrYmSB1V1YoJWVK1KWzAcWAzjXkN"
        },
        explanation: "Mint an NFT to the default wallet"
      }
    ],
    [
      {
        input: {
          collectionMint: "J1S9H3QjnRtBbbuD4HjPV6RpRhwuk4zKbxsnCHuTgh9w",
          name: "Gift NFT",
          uri: "https://example.com/gift.json",
          recipient: "9aUn5swQzUTRanaaTwmszxiv89cvFwUCjEBv1vZCoT1u"
        },
        output: {
          status: "success",
          message: "NFT minted successfully",
          mintAddress: "8nE9GvcwsqzYxmJLSrYmSB1V1YoJWVK1KWzAcWAzjXkM",
          metadata: {
            name: "Gift NFT",
            uri: "https://example.com/gift.json"
          },
          recipient: "9aUn5swQzUTRanaaTwmszxiv89cvFwUCjEBv1vZCoT1u"
        },
        explanation: "Mint an NFT to a specific recipient"
      }
    ]
  ],
  schema: z.object({
    collectionMint: z.string().min(32, "Invalid collection mint address"),
    name: z.string().min(1, "Name is required"),
    uri: z.string().url("URI must be a valid URL"),
    recipient: z.string().min(32, "Invalid recipient address").optional()
  }),
  handler: async (agent: SolanaAgentKit, input: Record<string, any>) => {
    const result = await mintCollectionNFT(
      agent,
      new PublicKey(input.collectionMint),
      {
        name: input.name,
        uri: input.uri
      },
      input.recipient ? new PublicKey(input.recipient) : undefined
    );

    return {
      status: "success",
      message: "NFT minted successfully",
      mintAddress: result.mint.toString(),
      metadata: {
        name: input.name,
        uri: input.uri
      },
      recipient: input.recipient || result.mint.toString()
    };
  }
};

export default mintNFTAction;