Osmosis MCP Server

// Transaction Execution Utilities for Osmosis import { DirectSecp256k1HdWallet } from '@cosmjs/proto-signing'; import { SigningStargateClient, GasPrice, calculateFee } from '@cosmjs/stargate'; import { OfflineSigner } from '@cosmjs/proto-signing'; import { StargateClient } from '@cosmjs/stargate'; import { OSMOSIS_RPC_ENDPOINT, OSMOSIS_CHAIN_ID, OSMOSIS_DEFAULT_DENOM } from '../config/network.js'; export interface ExecutionResult { success: boolean; txHash?: string; error?: string; gasUsed?: string; gasWanted?: string; height?: number; } export interface ExecutionOptions { gas?: string; gasPrice?: string; memo?: string; fee?: { amount: string; denom: string; }[]; } /** * Create a signing client from mnemonic */ export async function createSigningClient(mnemonic: string): Promise<{ client: SigningStargateClient; address: string; }> { try { // Create wallet from mnemonic const wallet = await DirectSecp256k1HdWallet.fromMnemonic(mnemonic, { prefix: 'osmo' }); // Get first account const [firstAccount] = await wallet.getAccounts(); // Create signing client const client = await SigningStargateClient.connectWithSigner( OSMOSIS_RPC_ENDPOINT, wallet as any, { gasPrice: GasPrice.fromString('0.025uosmo'), } ); return { client, address: firstAccount.address }; } catch (error: any) { throw new Error(`Failed to create signing client: ${error.message}`); } } /** * Execute a transaction with automatic gas estimation */ export async function executeTransaction( mnemonic: string, messages: any[], options: ExecutionOptions = {} ): Promise<ExecutionResult> { try { const { client, address } = await createSigningClient(mnemonic); // Fill in sender addresses in messages const processedMessages = messages.map(msg => { const value = { ...msg.value }; // Replace empty sender/address fields with actual address if ('sender' in value && (!value.sender || value.sender === '')) { value.sender = address; } if ('fromAddress' in value && (!value.fromAddress || value.fromAddress === '')) { value.fromAddress = address; } if ('delegatorAddress' in value && (!value.delegatorAddress || value.delegatorAddress === '')) { value.delegatorAddress = address; } if ('voter' in value && (!value.voter || value.voter === '')) { value.voter = address; } if ('depositor' in value && (!value.depositor || value.depositor === '')) { value.depositor = address; } if ('proposer' in value && (!value.proposer || value.proposer === '')) { value.proposer = address; } if ('owner' in value && (!value.owner || value.owner === '')) { value.owner = address; } return { ...msg, value }; }); // Set default options const memo = options.memo || ''; const gasPrice = GasPrice.fromString(options.gasPrice || '0.025uosmo'); // Calculate fee let fee; if (options.fee) { fee = { amount: options.fee, gas: options.gas || '200000' }; } else { // Auto-estimate gas if not provided const gasEstimate = options.gas ? parseInt(options.gas) : 200000; fee = calculateFee(gasEstimate, gasPrice); } // Broadcast transaction const result = await client.signAndBroadcast( address, processedMessages, fee, memo ); // Check if transaction was successful if (result.code !== 0) { return { success: false, error: `Transaction failed with code ${result.code}: ${result.rawLog}`, txHash: result.transactionHash, gasUsed: result.gasUsed?.toString(), gasWanted: result.gasWanted?.toString(), height: result.height }; } return { success: true, txHash: result.transactionHash, gasUsed: result.gasUsed?.toString(), gasWanted: result.gasWanted?.toString(), height: result.height }; } catch (error: any) { return { success: false, error: error.message }; } } /** * Validate mnemonic before execution */ export function validateMnemonicForExecution(mnemonic: string): boolean { if (!mnemonic || typeof mnemonic !== 'string') { return false; } const words = mnemonic.trim().split(/\s+/); return [12, 15, 18, 21, 24].includes(words.length); } /** * Get account info for execution validation */ export async function getAccountInfo(mnemonic: string): Promise<{ address: string; accountNumber: number; sequence: number; }> { try { const { client, address } = await createSigningClient(mnemonic); const account = await client.getAccount(address); if (!account) { throw new Error('Account not found on chain'); } return { address, accountNumber: account.accountNumber, sequence: account.sequence }; } catch (error: any) { throw new Error(`Failed to get account info: ${error.message}`); } } /** * Check account balance before execution */ export async function checkBalance(mnemonic: string, requiredAmount?: string, denom?: string): Promise<{ address: string; balances: Array<{ denom: string; amount: string }>; hasRequiredBalance: boolean; }> { try { const { client, address } = await createSigningClient(mnemonic); const balances = await client.getAllBalances(address); let hasRequiredBalance = true; if (requiredAmount && denom) { const balance = balances.find(b => b.denom === denom); const available = balance ? parseInt(balance.amount) : 0; const required = parseInt(requiredAmount); hasRequiredBalance = available >= required; } return { address, balances: balances.map(b => ({ denom: b.denom, amount: b.amount })), hasRequiredBalance }; } catch (error: any) { throw new Error(`Failed to check balance: ${error.message}`); } } /** * Simulate transaction before execution */ export async function simulateTransaction( mnemonic: string, messages: any[], memo: string = '' ): Promise<{ gasUsed: number; gasWanted: number; success: boolean; error?: string; }> { try { const { client, address } = await createSigningClient(mnemonic); // Use a minimal fee for simulation const gasPrice = GasPrice.fromString('0.025uosmo'); const fee = calculateFee(200000, gasPrice); const result = await client.simulate(address, messages, memo); return { gasUsed: result, gasWanted: Math.ceil(result * 1.3), // Add 30% buffer success: true }; } catch (error: any) { return { gasUsed: 0, gasWanted: 0, success: false, error: error.message }; } }