import { Client, Wallet, AMMVote } from "xrpl";
import { z } from "zod";
import { server } from "../../server/server.js";
import { getXrplClient } from "../../core/services/clients.js";
import { MAINNET_URL, TESTNET_URL } from "../../core/constants.js";
import { connectedWallet, isConnectedToTestnet } from "../../core/state.js";
// Register amm-vote tool
server.tool(
"amm-vote",
"Vote on parameters for an Automated Market Maker (AMM)",
{
fromSeed: z
.string()
.optional()
.describe(
"Optional seed of the wallet to use. If not provided, the connected wallet will be used."
),
asset1: z
.object({
currency: z
.string()
.describe("Currency code of the first asset"),
issuer: z
.string()
.optional()
.describe(
"Issuer address of the first asset (not needed for XRP)"
),
})
.describe("First asset in the AMM's pool"),
asset2: z
.object({
currency: z
.string()
.describe("Currency code of the second asset"),
issuer: z
.string()
.optional()
.describe(
"Issuer address of the second asset (not needed for XRP)"
),
})
.describe("Second asset in the AMM's pool"),
tradingFee: z
.number()
.min(0)
.max(1000)
.optional()
.describe("Trading fee in basis points (0-1000, where 100 = 1%)"),
fee: z.string().optional().describe("Transaction fee in XRP"),
useTestnet: z
.boolean()
.optional()
.describe(
"Whether to use the testnet (true) or mainnet (false). If not provided, uses the network from the connected wallet."
),
},
async ({ fromSeed, asset1, asset2, tradingFee, fee, useTestnet }) => {
let client: Client | null = null;
try {
// Determine which network to use
const useTestnetNetwork =
useTestnet !== undefined ? useTestnet : isConnectedToTestnet;
client = await getXrplClient(useTestnetNetwork);
// Use provided seed or connected wallet
let wallet: Wallet;
if (fromSeed) {
wallet = Wallet.fromSeed(fromSeed);
} else if (connectedWallet) {
wallet = connectedWallet;
} else {
throw new Error(
"No wallet connected. Please connect first using connect-to-xrpl tool or provide a fromSeed."
);
}
// Format assets for the transaction
const formatAsset = (asset: {
currency: string;
issuer?: string;
}) => {
if (asset.currency === "XRP") {
return { currency: "XRP" };
} else {
if (!asset.issuer) {
throw new Error(
`Issuer is required for non-XRP currency ${asset.currency}`
);
}
return {
currency: asset.currency,
issuer: asset.issuer,
};
}
};
// Create AMMVote transaction
const ammVoteTx: any = {
TransactionType: "AMMVote",
Account: wallet.address,
Asset: formatAsset(asset1),
Asset2: formatAsset(asset2),
};
// Add optional fields if provided
if (tradingFee !== undefined) {
ammVoteTx.TradingFee = tradingFee;
}
if (fee) {
ammVoteTx.Fee = fee;
}
// Submit transaction
const prepared = await client.autofill(ammVoteTx);
const signed = wallet.sign(prepared);
const result = await client.submitAndWait(signed.tx_blob);
let status = "unknown";
if (typeof result.result.meta !== "string" && result.result.meta) {
status =
result.result.meta.TransactionResult === "tesSUCCESS"
? "success"
: "failed";
}
return {
content: [
{
type: "text",
text: JSON.stringify(
{
status,
hash: result.result.hash,
account: wallet.address,
asset1,
asset2,
tradingFee:
tradingFee !== undefined
? tradingFee
: "Not voted on",
network: useTestnetNetwork
? TESTNET_URL
: MAINNET_URL,
networkType: useTestnetNetwork
? "testnet"
: "mainnet",
result: result.result,
},
null,
2
),
},
],
};
} catch (error) {
return {
content: [
{
type: "text",
text: `Error voting on AMM: ${
error instanceof Error
? error.message
: String(error)
}`,
},
],
};
} finally {
if (client) {
await client.disconnect();
}
}
}
);