#!/usr/bin/env node
import { spawn } from 'child_process';
import { fileURLToPath } from 'url';
import { dirname, join } from 'path';
const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);
// Test the MCP server
async function testServer() {
console.log('Testing fal-ai/hidream-i1-full MCP Server...\n');
// Check if FAL_KEY is set
if (!process.env.FAL_KEY) {
console.error('ā FAL_KEY environment variable is not set');
console.error('Please set your fal.ai API key: export FAL_KEY=your_api_key_here');
process.exit(1);
}
console.log('ā
FAL_KEY environment variable is set');
// Build the project first
console.log('š¦ Building project...');
const buildProcess = spawn('npm', ['run', 'build'], {
stdio: 'inherit',
shell: true,
cwd: __dirname
});
buildProcess.on('close', (code) => {
if (code !== 0) {
console.error('ā Build failed');
process.exit(1);
}
console.log('ā
Build successful');
// Start the server
console.log('š Starting MCP server...');
const serverPath = join(__dirname, 'build', 'index.js');
const serverProcess = spawn('node', [serverPath], {
stdio: ['pipe', 'pipe', 'pipe'],
env: { ...process.env, FAL_KEY: process.env.FAL_KEY }
});
let serverOutput = '';
let serverError = '';
serverProcess.stdout.on('data', (data) => {
serverOutput += data.toString();
});
serverProcess.stderr.on('data', (data) => {
serverError += data.toString();
console.log('Server:', data.toString().trim());
});
// Wait a moment for server to start
setTimeout(() => {
console.log('ā
Server appears to be running');
console.log('š Available tools:');
console.log(' - hidream_i1_full_generate: Generate images synchronously');
console.log(' - hidream_i1_full_generate_stream: Generate images with streaming');
console.log(' - hidream_i1_full_generate_queue: Submit to queue for long-running requests');
console.log(' - hidream_i1_full_queue_status: Check queue status');
console.log(' - hidream_i1_full_queue_result: Get queue results');
console.log('\nšÆ Example usage in MCP client:');
console.log(' Tool: hidream_i1_full_generate');
console.log(' Parameters: {');
console.log(' "prompt": "a cat holding a skateboard which has \'fal\' written on it in red spray paint",');
console.log(' "image_size": {"width": 1024, "height": 1024},');
console.log(' "num_inference_steps": 50,');
console.log(' "guidance_scale": 7.5');
console.log(' }');
console.log('\nā
Test completed successfully!');
console.log('š” The server is ready to be used with your MCP client.');
// Terminate the server
serverProcess.kill('SIGTERM');
process.exit(0);
}, 2000);
serverProcess.on('error', (error) => {
console.error('ā Failed to start server:', error.message);
process.exit(1);
});
serverProcess.on('close', (code) => {
if (code !== 0 && code !== null) {
console.error('ā Server exited with code:', code);
if (serverError) {
console.error('Server errors:', serverError);
}
process.exit(1);
}
});
});
buildProcess.on('error', (error) => {
console.error('ā Failed to build:', error.message);
process.exit(1);
});
}
// Handle graceful shutdown
process.on('SIGINT', () => {
console.log('\nš Test interrupted');
process.exit(0);
});
process.on('SIGTERM', () => {
console.log('\nš Test terminated');
process.exit(0);
});
// Run the test
testServer().catch((error) => {
console.error('ā Test failed:', error.message);
process.exit(1);
});
