import { Tool } from "langchain/tools"
import { create_image } from "../../tools/agent"

export class SolanaCreateImageTool extends Tool {
  name = "solana_create_image"
  description =
    "Create an image using OpenAI's DALL-E. Input should be a string prompt for the image."

  constructor(solanaKit) {
    super()
    this.solanaKit = solanaKit
  }

  validateInput(input) {
    if (typeof input !== "string" || input.trim().length === 0) {
      throw new Error("Input must be a non-empty string prompt")
    }
  }

  async _call(input) {
    try {
      this.validateInput(input)
      const result = await create_image(this.solanaKit, input.trim())

      return JSON.stringify({
        status: "success",
        message: "Image created successfully",
        ...result
      })
    } catch (error) {
      return JSON.stringify({
        status: "error",
        message: error.message,
        code: error.code || "UNKNOWN_ERROR"
      })
    }
  }
}