@x402storage/mcp 1.0.1 → 1.0.3

package/dist/config.d.ts

@@ -1,6 +1,11 @@
 /**
  * Environment validation for x402store MCP server
  */
+export type WalletType = 'evm' | 'solana';
+/**
+ * Detects wallet type from private key format
+ */
+export declare function detectWalletType(key: string): WalletType;
 /**
  * Validates required environment variables.
  * Returns error message if X402_PRIVATE_KEY is missing, null if valid.
@@ -10,4 +15,16 @@ export declare function validateEnvironment(): string | null;
  * Gets the private key from environment.
  * Should only be called after validateEnvironment() returns null.
  */
-export declare function getPrivateKey(): `0x${string}`;
+export declare function getPrivateKey(): string;
+/**
+ * Wallet info returned by getWalletInfo
+ */
+export interface WalletInfo {
+    address: string;
+    balance: string;
+    chain: string;
+}
+/**
+ * Gets wallet address and balance from the configured private key
+ */
+export declare function getWalletInfo(): Promise<WalletInfo>;

package/dist/config.js

@@ -1,13 +1,33 @@
 /**
  * Environment validation for x402store MCP server
  */
+import { privateKeyToAccount } from 'viem/accounts';
+import { createPublicClient, http, formatUnits } from 'viem';
+import { base } from 'viem/chains';
+import { Connection, Keypair, LAMPORTS_PER_SOL } from '@solana/web3.js';
+/**
+ * Detects wallet type from private key format
+ */
+export function detectWalletType(key) {
+    const trimmed = key.trim();
+    // EVM keys start with 0x
+    if (trimmed.startsWith('0x')) {
+        return 'evm';
+    }
+    // Solana keys are JSON arrays
+    if (trimmed.startsWith('[')) {
+        return 'solana';
+    }
+    // Default to EVM for other formats
+    return 'evm';
+}
 /**
  * Validates required environment variables.
  * Returns error message if X402_PRIVATE_KEY is missing, null if valid.
  */
 export function validateEnvironment() {
     if (!process.env.X402_PRIVATE_KEY) {
-        return 'X402_PRIVATE_KEY not set. Export your Base wallet private key: export X402_PRIVATE_KEY=0x... Need a wallet? Create one at https://x402.storage';
+        return 'X402_PRIVATE_KEY not set. Run /x402storage:setup to configure.';
     }
     return null;
 }
@@ -18,3 +38,66 @@ export function validateEnvironment() {
 export function getPrivateKey() {
     return process.env.X402_PRIVATE_KEY;
 }
+/**
+ * Gets wallet address and balance from the configured private key
+ */
+export async function getWalletInfo() {
+    const privateKey = getPrivateKey();
+    const walletType = detectWalletType(privateKey);
+    if (walletType === 'solana') {
+        return getSolanaWalletInfo(privateKey);
+    }
+    else {
+        return getEvmWalletInfo(privateKey);
+    }
+}
+/**
+ * Gets Solana wallet info
+ */
+async function getSolanaWalletInfo(privateKey) {
+    const secretKey = Uint8Array.from(JSON.parse(privateKey));
+    const keypair = Keypair.fromSecretKey(secretKey);
+    const address = keypair.publicKey.toBase58();
+    const connection = new Connection('https://api.mainnet-beta.solana.com', 'confirmed');
+    const balance = await connection.getBalance(keypair.publicKey);
+    const solBalance = (balance / LAMPORTS_PER_SOL).toFixed(4);
+    return {
+        address,
+        balance: `${solBalance} SOL`,
+        chain: 'Solana',
+    };
+}
+/**
+ * Gets EVM wallet info (Base network)
+ */
+async function getEvmWalletInfo(privateKey) {
+    const account = privateKeyToAccount(privateKey);
+    const client = createPublicClient({
+        chain: base,
+        transport: http(),
+    });
+    const ethBalance = await client.getBalance({ address: account.address });
+    // USDC contract on Base
+    const USDC_ADDRESS = '0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913';
+    const usdcBalance = await client.readContract({
+        address: USDC_ADDRESS,
+        abi: [
+            {
+                name: 'balanceOf',
+                type: 'function',
+                stateMutability: 'view',
+                inputs: [{ name: 'account', type: 'address' }],
+                outputs: [{ name: '', type: 'uint256' }],
+            },
+        ],
+        functionName: 'balanceOf',
+        args: [account.address],
+    });
+    const ethFormatted = parseFloat(formatUnits(ethBalance, 18)).toFixed(4);
+    const usdcFormatted = parseFloat(formatUnits(usdcBalance, 6)).toFixed(2);
+    return {
+        address: account.address,
+        balance: `${usdcFormatted} USDC, ${ethFormatted} ETH`,
+        chain: 'Base',
+    };
+}

package/dist/index.js

@@ -3,8 +3,21 @@ import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 import { z } from 'zod';
 import { resolve } from 'node:path';
-import { validateEnvironment, getPrivateKey } from './config.js';
+import { randomBytes } from 'node:crypto';
+import { writeFileSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { privateKeyToAccount } from 'viem/accounts';
+import { validateEnvironment, getPrivateKey, getWalletInfo } from './config.js';
 import { uploadFile, FileNotFoundError, UploadError } from './upload.js';
+// Handle --generate-wallet flag before MCP server setup
+if (process.argv.includes('--generate-wallet')) {
+    const privateKey = `0x${randomBytes(32).toString('hex')}`;
+    const walletPath = `${homedir()}/.x402-wallet`;
+    writeFileSync(walletPath, privateKey, { mode: 0o600 });
+    const account = privateKeyToAccount(privateKey);
+    console.log(account.address);
+    process.exit(0);
+}
 // Create MCP server
 const server = new McpServer({
     name: 'x402store',
@@ -70,6 +83,32 @@ server.tool('store_file', 'Store a file permanently on IPFS via x402.storage. Re
         };
     }
 });
+// Register wallet tool
+server.tool('wallet', 'Show your x402.storage wallet address and balance', {}, async () => {
+    const envError = validateEnvironment();
+    if (envError) {
+        return {
+            content: [{ type: 'text', text: `Error: ${envError}` }],
+        };
+    }
+    try {
+        const info = await getWalletInfo();
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `Address: ${info.address}\nBalance: ${info.balance}`,
+                },
+            ],
+        };
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : 'Unknown error';
+        return {
+            content: [{ type: 'text', text: `Error: ${message}` }],
+        };
+    }
+});
 // Run server with stdio transport
 async function main() {
     const transport = new StdioServerTransport();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@x402storage/mcp",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "MCP server for x402.storage file uploads",
   "type": "module",
   "license": "MIT",
@@ -9,6 +9,7 @@
     "url": "https://github.com/rawgroundbeef/x402.storage"
   },
   "bin": {
+    "mcp": "./dist/index.js",
     "x402-mcp": "./dist/index.js"
   },
   "files": [
@@ -30,9 +31,11 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.25.0",
+    "@solana/web3.js": "^1.98.4",
     "@x402/core": "^2.2.0",
     "@x402/evm": "latest",
     "@x402/fetch": "latest",
+    "bs58": "^6.0.0",
     "viem": "^2.21.0",
     "zod": "^3.24.0"
   },