server.ts•3.2 kB
import "dotenv/config";
import { logger } from "./logger.js";
import { HttpMCPServer, StdioMCPServer } from "./servers/index.js";
import { createTailscaleAPI } from "./tailscale/index.js";
import { TailscaleCLI } from "./tailscale/tailscale-cli.js";
import { UnifiedTailscaleClient } from "./tailscale/unified-client.js";
import { ToolRegistry } from "./tools/index.js";
export type ServerMode = "stdio" | "http";
export class TailscaleMCPServer {
private toolRegistry!: ToolRegistry;
private stdioServer?: StdioMCPServer;
private httpServer?: HttpMCPServer;
private unifiedClient!: UnifiedTailscaleClient;
async initialize(mode: ServerMode = "stdio"): Promise<void> {
logger.info("Initializing Tailscale MCP Server...");
// Initialize Tailscale integrations
const api = createTailscaleAPI();
const cli = new TailscaleCLI();
// Create unified client based on transport mode
this.unifiedClient = new UnifiedTailscaleClient({
transportMode: mode,
apiKey: process.env.TAILSCALE_API_KEY,
tailnet: process.env.TAILSCALE_TAILNET,
preferAPI: mode === "http", // HTTP mode prefers API, stdio prefers CLI
});
// Initialize the unified client
await this.unifiedClient.initialize();
// Create tool registry and register tool modules
this.toolRegistry = new ToolRegistry({
api,
cli,
client: this.unifiedClient,
});
await this.toolRegistry.loadTools();
// Log capabilities
const capabilities = this.unifiedClient.getCapabilities();
logger.debug(`Loaded ${this.toolRegistry.getTools().length} tools`);
logger.debug(
`Client capabilities - API: ${capabilities.api}, CLI: ${capabilities.cli}`,
);
logger.debug("Available features:", capabilities.features);
}
async start(mode: ServerMode = "stdio", port?: number): Promise<void> {
await this.initialize(mode);
// Log server configuration
logger.info(`Tailscale MCP Server starting in ${mode} mode...`);
if (process.env.MCP_SERVER_LOG_FILE) {
logger.debug("File logging enabled");
} else {
logger.debug("File logging disabled (set MCP_SERVER_LOG_FILE to enable)");
}
if (mode === "http") {
await this.startHttpServer(port);
} else {
await this.startStdioServer();
}
}
private async startStdioServer(): Promise<void> {
this.stdioServer = new StdioMCPServer(this.toolRegistry);
await this.stdioServer.start();
}
private async startHttpServer(port = 3000): Promise<void> {
this.httpServer = new HttpMCPServer(this.toolRegistry);
await this.httpServer.start(port);
}
async stop(): Promise<void> {
logger.info("Stopping Tailscale MCP Server...");
if (this.stdioServer) {
await this.stdioServer.stop();
this.stdioServer = undefined;
} else if (this.httpServer) {
await this.httpServer.stop();
this.httpServer = undefined;
}
// Dispose of ToolRegistry resources
if (this.toolRegistry) {
await this.toolRegistry.dispose();
}
logger.info("Tailscale MCP Server stopped");
// Close logger last to ensure all log messages are written
await logger.close();
}
}