import { PrivateKey } from "@bsv/sdk";
import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { z } from "zod";

/**
 * Register the tool to get the wallet address
 * @param server The MCP server instance
 */
export function registerGetAddressTool(server: McpServer): void {
	server.tool(
		"wallet_getAddress",
		"Retrieves the current wallet's Bitcoin SV address. This address can be used to receive BSV, ordinals, or tokens, and is derived from the wallet's private key.",
		{
			args: z
				.object({})
				.optional()
				.describe(
					"No parameters required - simply returns the current wallet address",
				),
		},
		async () => {
			try {
				const wif = process.env.PRIVATE_KEY_WIF;
				if (!wif) {
					return {
						content: [
							{
								type: "text",
								text: JSON.stringify({
									error: "No private key available",
									message:
										"Please set PRIVATE_KEY_WIF environment variable with a valid Bitcoin SV private key in WIF format.",
									status: "error",
								}),
							},
						],
						isError: true,
					};
				}

				const privKey = PrivateKey.fromWif(wif);
				const address = privKey.toAddress();
				return {
					content: [
						{
							type: "text",
							text: JSON.stringify({ address, status: "ok" }),
						},
					],
				};
			} catch (err: unknown) {
				const msg = err instanceof Error ? err.message : String(err);
				return {
					content: [
						{
							type: "text",
							text: JSON.stringify({
								error: "Invalid private key",
								message: msg,
								status: "error",
							}),
						},
					],
					isError: true,
				};
			}
		},
	);
}