Midnight MCP Server

/* istanbul ignore file */ import { Request, Response, NextFunction } from 'express'; import { WalletServiceMCP } from '../mcp/index.js'; import { createLogger } from '../logger/index.js'; export class WalletController { private logger = createLogger('wallet-controller'); constructor(private readonly walletService: WalletServiceMCP) {} async getStatus(req: Request, res: Response, next: NextFunction): Promise<void> { try { const status = this.walletService.getWalletStatus(); res.json(status); } catch (error) { this.logger.error('Error getting wallet status:', error); next(error); } } async getAddress(req: Request, res: Response, next: NextFunction): Promise<void> { try { const address = this.walletService.getAddress(); res.json({ address }); } catch (error) { this.logger.error('Error getting wallet address:', error); next(error); } } // ==================== TOKEN OPERATIONS ==================== async registerToken(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { name, symbol, contractAddress, domainSeparator, description, decimals } = req.body; if (!name || !symbol || !contractAddress) { res.status(400).json({ error: 'Missing required parameters: name, symbol, and contractAddress' }); return; } const result = this.walletService.registerToken( name, symbol, contractAddress, domainSeparator || 'custom_token', description, decimals ); if (result.success) { res.json(result); } else { res.status(400).json(result); } } catch (error) { this.logger.error('Error registering token:', error); next(error); } } async getTokenBalance(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { tokenName } = req.params; if (!tokenName) { res.status(400).json({ error: 'Missing required parameter: tokenName' }); return; } const balance = this.walletService.getTokenBalance(tokenName); res.json({ tokenName, balance }); } catch (error) { this.logger.error('Error getting token balance:', error); next(error); } } async sendToken(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { tokenName, toAddress, amount } = req.body; if (!tokenName || !toAddress || !amount) { res.status(400).json({ error: 'Missing required parameters: tokenName, toAddress, and amount' }); return; } const result = await this.walletService.sendToken(tokenName, toAddress, amount); res.json(result); } catch (error) { this.logger.error('Error sending token:', error); next(error); } } async listTokens(req: Request, res: Response, next: NextFunction): Promise<void> { try { const tokens = this.walletService.listWalletTokens(); res.json({ tokens }); } catch (error) { this.logger.error('Error listing tokens:', error); next(error); } } async registerTokensBatch(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { tokens } = req.body; if (!tokens || !Array.isArray(tokens)) { res.status(400).json({ error: 'Missing or invalid parameter: tokens (must be an array)' }); return; } const result = this.walletService.registerTokensBatch(tokens); res.json(result); } catch (error) { this.logger.error('Error batch registering tokens:', error); next(error); } } async registerTokensFromEnv(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { envValue } = req.body; if (!envValue) { res.status(400).json({ error: 'Missing required parameter: envValue' }); return; } const result = this.walletService.registerTokensFromEnvString(envValue); res.json(result); } catch (error) { this.logger.error('Error registering tokens from env string:', error); next(error); } } async getTokenEnvConfigTemplate(req: Request, res: Response, next: NextFunction): Promise<void> { try { const template = this.walletService.getTokenEnvConfigTemplate(); res.json({ template }); } catch (error) { this.logger.error('Error getting token env config template:', error); next(error); } } async getTokenRegistryStats(req: Request, res: Response, next: NextFunction): Promise<void> { try { const stats = this.walletService.getTokenRegistryStats(); res.json(stats); } catch (error) { this.logger.error('Error getting token registry stats:', error); next(error); } } async getBalance(req: Request, res: Response, next: NextFunction): Promise<void> { try { const balance = this.walletService.getBalance(); res.json(balance); } catch (error) { this.logger.error('Error getting wallet balance:', error); next(error); } } async sendFunds(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { destinationAddress, amount } = req.body; if (!destinationAddress || !amount) { res.status(400).json({ error: 'Missing required parameters: destinationAddress and amount' }); return; } const result = await this.walletService.sendFunds(destinationAddress, amount); res.json(result); } catch (error) { this.logger.error('Error sending funds:', error); next(error); } } async send(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { destinationAddress, amount, token } = req.body; if (!destinationAddress || !amount) { res.status(400).json({ error: 'Missing required parameters: destinationAddress and amount' }); return; } // Determine if this is a native token or shielded token const isNativeToken = !token || token.toLowerCase() === 'native' || token.toLowerCase() === 'tdust' || token.toLowerCase() === 'dust'; let result; if (isNativeToken) { // Send native tokens result = await this.walletService.sendFunds(destinationAddress, amount); } else { // Send shielded tokens result = await this.walletService.sendToken(token, destinationAddress, amount); } res.json(result); } catch (error) { this.logger.error('Error sending funds/tokens:', error); next(error); } } async verifyTransaction(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { identifier } = req.body; if (!identifier) { res.status(400).json({ error: 'Missing required parameter: identifier' }); return; } const result = this.walletService.confirmTransactionHasBeenReceived(identifier); res.json(result); } catch (error) { this.logger.error('Error verifying transaction:', error); next(error); } } async getTransactionStatus(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { transactionId } = req.params; if (!transactionId) { res.status(400).json({ error: 'Missing required parameter: transactionId' }); return; } const status = this.walletService.getTransactionStatus(transactionId); res.json(status); } catch (error) { this.logger.error('Error getting transaction status:', error); next(error); } } async getTransactions(req: Request, res: Response, next: NextFunction): Promise<void> { try { const transactions = this.walletService.getTransactions(); res.json(transactions); } catch (error) { this.logger.error('Error getting transactions:', error); next(error); } } async getPendingTransactions(req: Request, res: Response, next: NextFunction): Promise<void> { try { const transactions = this.walletService.getPendingTransactions(); res.json(transactions); } catch (error) { this.logger.error('Error getting pending transactions:', error); next(error); } } async getWalletConfig(req: Request, res: Response, next: NextFunction): Promise<void> { try { const config = this.walletService.getWalletConfig(); res.json(config); } catch (error) { this.logger.error('Error getting wallet config:', error); next(error); } } async healthCheck(req: Request, res: Response): Promise<void> { res.json({ status: 'ok' }); } async registerInMarketplace(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { userId, userData } = req.body; if (!userId || !userData) { res.status(400).json({ error: 'Missing required parameters: userId and userData' }); return; } const result = await this.walletService.registerInMarketplace(userId, userData); res.json(result); } catch (error) { this.logger.error('Error registering in marketplace:', error); next(error); } } async verifyUserInMarketplace(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { userId, verificationData } = req.body; if (!userId || !verificationData || !verificationData.pubkey) { res.status(400).json({ error: 'Missing required parameters: userId, verificationData and pubkey' }); return; } const result = await this.walletService.verifyUserInMarketplace(userId, verificationData); res.json(result); } catch (error) { this.logger.error('Error verifying user in marketplace:', error); next(error); } } // ==================== DAO OPERATIONS ==================== async openDaoElection(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { electionId } = req.body; if (!electionId) { res.status(400).json({ error: 'Missing required parameter: electionId' }); return; } const result = await this.walletService.openDaoElection(electionId); res.json(result); } catch (error) { this.logger.error('Error opening DAO election:', error); next(error); } } async closeDaoElection(req: Request, res: Response, next: NextFunction): Promise<void> { try { const result = await this.walletService.closeDaoElection(); res.json(result); } catch (error) { this.logger.error('Error closing DAO election:', error); next(error); } } async castDaoVote(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { voteType } = req.body; if (!voteType) { res.status(400).json({ error: 'Missing required parameter: voteType (yes, no, or absence)' }); return; } const result = await this.walletService.castDaoVote(voteType); res.json(result); } catch (error) { this.logger.error('Error casting DAO vote:', error); next(error); } } async fundDaoTreasury(req: Request, res: Response, next: NextFunction): Promise<void> { try { const { amount } = req.body; if (!amount) { res.status(400).json({ error: 'Missing required parameter: amount' }); return; } const result = await this.walletService.fundDaoTreasury(amount); res.json(result); } catch (error) { this.logger.error('Error funding DAO treasury:', error); next(error); } } async payoutDaoProposal(req: Request, res: Response, next: NextFunction): Promise<void> { try { const result = await this.walletService.payoutDaoProposal(); res.json(result); } catch (error) { this.logger.error('Error paying out DAO proposal:', error); next(error); } } async getDaoElectionStatus(req: Request, res: Response, next: NextFunction): Promise<void> { try { const result = await this.walletService.getDaoElectionStatus(); res.json(result); } catch (error) { this.logger.error('Error getting DAO election status:', error); next(error); } } async getDaoState(req: Request, res: Response, next: NextFunction): Promise<void> { try { const result = await this.walletService.getDaoState(); res.json(result); } catch (error) { this.logger.error('Error getting DAO state:', error); next(error); } } async getDaoConfigTemplate(req: Request, res: Response, next: NextFunction): Promise<void> { try { const template = this.walletService.getDaoConfigTemplate(); res.json({ template }); } catch (error) { this.logger.error('Error getting DAO config template:', error); next(error); } } }