miladyai 2.0.0-alpha.27

package/dist/api/wallet-routes.js

@@ -0,0 +1,266 @@
+import { fetchEvmBalances, fetchEvmNfts, fetchSolanaBalances, fetchSolanaNfts, generateWalletForChain, getWalletAddresses, importWallet, validatePrivateKey } from "./wallet.js";
+import { createIntegrationTelemetrySpan } from "../diagnostics/integration-observability.js";
+import { logger } from "@elizaos/core";
+
+//#region src/api/wallet-routes.ts
+const DEFAULT_WALLET_ROUTE_DEPENDENCIES = {
+	getWalletAddresses,
+	fetchEvmBalances,
+	fetchSolanaBalances,
+	fetchEvmNfts,
+	fetchSolanaNfts,
+	validatePrivateKey,
+	importWallet,
+	generateWalletForChain
+};
+async function handleWalletRoutes(ctx) {
+	const { req, res, method, pathname, config, saveConfig, ensureWalletKeysInEnvAndConfig, resolveWalletExportRejection, readJsonBody, json, error } = ctx;
+	const deps = ctx.deps ?? DEFAULT_WALLET_ROUTE_DEPENDENCIES;
+	if (method === "GET" && pathname === "/api/wallet/addresses") {
+		json(res, deps.getWalletAddresses());
+		return true;
+	}
+	if (method === "GET" && pathname === "/api/wallet/balances") {
+		const addresses = deps.getWalletAddresses();
+		const alchemyKey = process.env.ALCHEMY_API_KEY;
+		const heliusKey = process.env.HELIUS_API_KEY;
+		const result = {
+			evm: null,
+			solana: null
+		};
+		if (addresses.evmAddress && alchemyKey) {
+			const evmBalancesSpan = createIntegrationTelemetrySpan({
+				boundary: "wallet",
+				operation: "fetch_evm_balances"
+			});
+			try {
+				const chains = await deps.fetchEvmBalances(addresses.evmAddress, alchemyKey);
+				result.evm = {
+					address: addresses.evmAddress,
+					chains
+				};
+				evmBalancesSpan.success();
+			} catch (err) {
+				evmBalancesSpan.failure({ error: err });
+				logger.warn(`[wallet] EVM balance fetch failed: ${err}`);
+			}
+		}
+		if (addresses.solanaAddress && heliusKey) {
+			const solanaBalancesSpan = createIntegrationTelemetrySpan({
+				boundary: "wallet",
+				operation: "fetch_solana_balances"
+			});
+			try {
+				const solanaData = await deps.fetchSolanaBalances(addresses.solanaAddress, heliusKey);
+				result.solana = {
+					address: addresses.solanaAddress,
+					...solanaData
+				};
+				solanaBalancesSpan.success();
+			} catch (err) {
+				solanaBalancesSpan.failure({ error: err });
+				logger.warn(`[wallet] Solana balance fetch failed: ${err}`);
+			}
+		}
+		json(res, result);
+		return true;
+	}
+	if (method === "GET" && pathname === "/api/wallet/nfts") {
+		const addresses = deps.getWalletAddresses();
+		const alchemyKey = process.env.ALCHEMY_API_KEY;
+		const heliusKey = process.env.HELIUS_API_KEY;
+		const result = {
+			evm: [],
+			solana: null
+		};
+		if (addresses.evmAddress && alchemyKey) {
+			const evmNftsSpan = createIntegrationTelemetrySpan({
+				boundary: "wallet",
+				operation: "fetch_evm_nfts"
+			});
+			try {
+				result.evm = await deps.fetchEvmNfts(addresses.evmAddress, alchemyKey);
+				evmNftsSpan.success();
+			} catch (err) {
+				evmNftsSpan.failure({ error: err });
+				logger.warn(`[wallet] EVM NFT fetch failed: ${err}`);
+			}
+		}
+		if (addresses.solanaAddress && heliusKey) {
+			const solanaNftsSpan = createIntegrationTelemetrySpan({
+				boundary: "wallet",
+				operation: "fetch_solana_nfts"
+			});
+			try {
+				result.solana = { nfts: await deps.fetchSolanaNfts(addresses.solanaAddress, heliusKey) };
+				solanaNftsSpan.success();
+			} catch (err) {
+				solanaNftsSpan.failure({ error: err });
+				logger.warn(`[wallet] Solana NFT fetch failed: ${err}`);
+			}
+		}
+		json(res, result);
+		return true;
+	}
+	if (method === "POST" && pathname === "/api/wallet/import") {
+		const body = await readJsonBody(req, res);
+		if (!body) return true;
+		if (!body.privateKey?.trim()) {
+			error(res, "privateKey is required");
+			return true;
+		}
+		let chain;
+		if (body.chain === "evm" || body.chain === "solana") chain = body.chain;
+		else if (body.chain) {
+			error(res, `Unsupported chain: ${body.chain}. Must be "evm" or "solana".`);
+			return true;
+		} else chain = deps.validatePrivateKey(body.privateKey.trim()).chain;
+		const result = deps.importWallet(chain, body.privateKey.trim());
+		if (!result.success) {
+			error(res, result.error ?? "Import failed", 422);
+			return true;
+		}
+		if (!config.env) config.env = {};
+		const envKey = chain === "evm" ? "EVM_PRIVATE_KEY" : "SOLANA_PRIVATE_KEY";
+		config.env[envKey] = process.env[envKey] ?? "";
+		try {
+			saveConfig(config);
+		} catch (err) {
+			logger.warn(`[api] Config save failed: ${err instanceof Error ? err.message : err}`);
+		}
+		json(res, {
+			ok: true,
+			chain,
+			address: result.address
+		});
+		return true;
+	}
+	if (method === "POST" && pathname === "/api/wallet/generate") {
+		const body = await readJsonBody(req, res);
+		if (!body) return true;
+		const chain = body.chain;
+		if (chain && ![
+			"evm",
+			"solana",
+			"both"
+		].includes(chain)) {
+			error(res, `Unsupported chain: ${chain}. Must be "evm", "solana", or "both".`);
+			return true;
+		}
+		const targetChain = chain ?? "both";
+		if (!config.env) config.env = {};
+		const generated = [];
+		if (targetChain === "both" || targetChain === "evm") {
+			const result = deps.generateWalletForChain("evm");
+			process.env.EVM_PRIVATE_KEY = result.privateKey;
+			config.env.EVM_PRIVATE_KEY = result.privateKey;
+			generated.push({
+				chain: "evm",
+				address: result.address
+			});
+			logger.info(`[milady-api] Generated EVM wallet: ${result.address}`);
+		}
+		if (targetChain === "both" || targetChain === "solana") {
+			const result = deps.generateWalletForChain("solana");
+			process.env.SOLANA_PRIVATE_KEY = result.privateKey;
+			config.env.SOLANA_PRIVATE_KEY = result.privateKey;
+			generated.push({
+				chain: "solana",
+				address: result.address
+			});
+			logger.info(`[milady-api] Generated Solana wallet: ${result.address}`);
+		}
+		try {
+			saveConfig(config);
+		} catch (err) {
+			logger.warn(`[api] Config save failed: ${err instanceof Error ? err.message : err}`);
+		}
+		json(res, {
+			ok: true,
+			wallets: generated
+		});
+		return true;
+	}
+	if (method === "GET" && pathname === "/api/wallet/config") {
+		const addresses = deps.getWalletAddresses();
+		json(res, {
+			alchemyKeySet: Boolean(process.env.ALCHEMY_API_KEY),
+			infuraKeySet: Boolean(process.env.INFURA_API_KEY),
+			ankrKeySet: Boolean(process.env.ANKR_API_KEY),
+			heliusKeySet: Boolean(process.env.HELIUS_API_KEY),
+			birdeyeKeySet: Boolean(process.env.BIRDEYE_API_KEY),
+			evmChains: [
+				"Ethereum",
+				"Base",
+				"Arbitrum",
+				"Optimism",
+				"Polygon"
+			],
+			evmAddress: addresses.evmAddress,
+			solanaAddress: addresses.solanaAddress
+		});
+		return true;
+	}
+	if (method === "PUT" && pathname === "/api/wallet/config") {
+		const body = await readJsonBody(req, res);
+		if (!body) return true;
+		const allowedKeys = [
+			"ALCHEMY_API_KEY",
+			"INFURA_API_KEY",
+			"ANKR_API_KEY",
+			"HELIUS_API_KEY",
+			"BIRDEYE_API_KEY"
+		];
+		if (!config.env) config.env = {};
+		for (const key of allowedKeys) {
+			const value = body[key];
+			if (typeof value === "string" && value.trim()) {
+				process.env[key] = value.trim();
+				config.env[key] = value.trim();
+			}
+		}
+		const heliusValue = body.HELIUS_API_KEY;
+		if (typeof heliusValue === "string" && heliusValue.trim()) {
+			const rpcUrl = `https://mainnet.helius-rpc.com/?api-key=${heliusValue.trim()}`;
+			process.env.SOLANA_RPC_URL = rpcUrl;
+			config.env.SOLANA_RPC_URL = rpcUrl;
+		}
+		ensureWalletKeysInEnvAndConfig(config);
+		try {
+			saveConfig(config);
+		} catch (err) {
+			logger.warn(`[api] Config save failed: ${err instanceof Error ? err.message : err}`);
+		}
+		json(res, { ok: true });
+		ctx.scheduleRuntimeRestart?.("Wallet configuration updated");
+		return true;
+	}
+	if (method === "POST" && pathname === "/api/wallet/export") {
+		const body = await readJsonBody(req, res);
+		if (!body) return true;
+		const rejection = resolveWalletExportRejection(req, body);
+		if (rejection) {
+			error(res, rejection.reason, rejection.status);
+			return true;
+		}
+		const evmKey = process.env.EVM_PRIVATE_KEY ?? null;
+		const solanaKey = process.env.SOLANA_PRIVATE_KEY ?? null;
+		const addresses = deps.getWalletAddresses();
+		logger.warn("[wallet] Private keys exported via API");
+		json(res, {
+			evm: evmKey ? {
+				privateKey: evmKey,
+				address: addresses.evmAddress
+			} : null,
+			solana: solanaKey ? {
+				privateKey: solanaKey,
+				address: addresses.solanaAddress
+			} : null
+		});
+		return true;
+	}
+	return false;
+}
+
+//#endregion
+export { handleWalletRoutes };