import { ToolAnnotations } from "@modelcontextprotocol/sdk/types.js";
import { getCurrentAccountKeys, getNetwork } from "../shared/account.js";
import { getDefaultProvider, Wallet } from "@coti-io/coti-ethers";
import { z } from "zod";
export const DECRYPT_VALUE: ToolAnnotations = {
title: "Decrypt Value",
name: "decrypt_value",
description:
"Decrypt a value using the COTI AES key. " +
"Requires a ciphertext as input. " +
"Returns the decrypted value.",
inputSchema: {
ciphertext: z.string().describe("Ciphertext to decrypt"),
},
};
/**
* Checks if the provided arguments are valid for the decrypt_value tool.
* @param args The arguments to check.
* @returns True if the arguments are valid, false otherwise.
*/
export function isDecryptValueArgs(args: unknown): args is { ciphertext: bigint } {
return (
typeof args === "object" &&
args !== null &&
"ciphertext" in args &&
typeof (args as { ciphertext: bigint }).ciphertext === "bigint"
);
}
/**
* Decrypts a value using the COTI AES key.
* @param ciphertext The ciphertext to decrypt.
* @returns The decrypted value.
*/
export async function performDecryptValue(ciphertext: bigint) {
try {
const currentAccountKeys = getCurrentAccountKeys();
const provider = getDefaultProvider(getNetwork());
const wallet = new Wallet(currentAccountKeys.privateKey, provider);
const decryptedMessage = await wallet.decryptValue(ciphertext);
return `Decrypted Message: ${decryptedMessage}`;
} catch (error) {
console.error('Error decrypting message:', error);
throw new Error(`Failed to decrypt message: ${error instanceof Error ? error.message : String(error)}`);
}
}
/**
* Handler for the decryptValue tool
* @param args The arguments for the tool
* @returns The tool response
*/
export async function decryptValueHandler(args: Record<string, unknown> | undefined): Promise<any> {
if (!isDecryptValueArgs(args)) {
throw new Error("Invalid arguments for decrypt_value");
}
const { ciphertext } = args;
const results = await performDecryptValue(BigInt(ciphertext));
return {
content: [{ type: "text", text: results }],
isError: false,
};
}
