import fs from "node:fs/promises";
import { MCPServer } from "@mastra/mcp";
import { prepare } from "./prepare-docs/prepare.js";
import { listServicesTool } from "./tools/compute/list-services.js";
import { docsTool } from "./tools/docs.js";
import { examplesTool } from "./tools/examples.js";
import { storageDownloadTool } from "./tools/storage/download.js";
import { kvGetTool } from "./tools/storage/kv-get.js";
import { kvSetTool } from "./tools/storage/kv-set.js";
import { storageNodesTool } from "./tools/storage/nodes.js";
import { storageUploadTool } from "./tools/storage/upload.js";
import { fromProjectRoot } from "./utils/file-utils.js";
import { createLogger, logger } from "./utils/logger.js";
let server: MCPServer;
if (process.env.REBUILD_DOCS_ON_START === "true") {
void logger.info("Rebuilding docs on start");
try {
await prepare();
void logger.info("Docs rebuilt successfully");
} catch (error) {
void logger.error("Failed to rebuild docs", error);
}
}
server = new MCPServer({
name: "0g Documentation Server",
version: JSON.parse(
await fs.readFile(fromProjectRoot(`package.json`), "utf8"),
).version,
tools: {
"0gDocs": docsTool,
"0gExamples": examplesTool,
"0gStorageUpload": storageUploadTool,
"0gStorageDownload": storageDownloadTool,
"0gStorageNodes": storageNodesTool,
"0gKvSet": kvSetTool,
"0gKvGet": kvGetTool,
"0gComputeListServices": listServicesTool,
},
});
// Update logger with server instance
Object.assign(logger, createLogger(server));
async function runServer() {
try {
await server.startStdio();
void logger.info("Started 0g Docs MCP Server");
} catch (error) {
void logger.error("Failed to start server", error);
process.exit(1);
}
}
export { runServer, server };
