import { CallToolResult } from "@modelcontextprotocol/sdk/types.js"
import { RecraftServer } from "../RecraftServer"

const ALMOST_ZERO_CREDITS = 50

export const getUserTool = {
  name: "get_user",
  description: "Get information about the current Recraft API user (their email, name, and credit balance).",
  inputSchema: {
    type: "object",
    properties: {
    },
    required: []
  }
}

export const getUserHandler = async (server: RecraftServer, _args: Record<string, unknown>): Promise<CallToolResult> => {
  try {
    const result = await server.api.userApi.getCurrentUser()

    return {
      content: [
        {
          type: 'text',
          text: `User email: ${result.email}, name: ${result.name}.\nYou have ${result.credits} credits left.` + 
          (
            result.credits >= ALMOST_ZERO_CREDITS ? 
            "" :
            `\nYou are ${result.credits > 0 ? "almost" : ""} out of credits. Please top up your account on https://www.recraft.ai/profile/api.`
          )
        },
      ]
    }
  } catch (error) {
    return {
      content: [
        {
          type: 'text',
          text: `Get user error: ${error}`
        }
      ],
      isError: true
    }
  }
}