import com.github.codeboyzhou.mcp.declarative.McpServers;

// You can use this annotation to specify the base package
// to scan for MCP resources, prompts, tools, but it's optional.
// If not specified, it will scan the package where the main method is located.
@McpComponentScan(basePackage = "com.github.codeboyzhou.mcp.server.examples")
public class MyMcpServer {

    public static void main(String[] args) {
        // Start a STDIO MCP server
        McpServers.run(MyMcpServer.class, args).startSyncStdioServer(
            McpServerInfo.builder().name("mcp-server").version("1.0.0").build()
        );
        // or a HTTP SSE MCP server
        McpServers.run(MyMcpServer.class, args).startSyncSseServer(
            McpSseServerInfo.builder().name("mcp-server").version("1.0.0").port(8080).build()
        );
        // or start with yaml configuration file (compatible with the Spring AI framework)
        McpServers.run(MyMcpServer.class, args).startServer();
        // or start with a specific configuration file (compatible with the Spring AI framework)
        McpServers.run(MyMcpServer.class, args).startServer("my-mcp-server.yml");
    }

}

@McpResources
public class MyMcpResources {

    // This method defines a MCP resource to expose the OS env variables
    @McpResource(uri = "env://variables", description = "OS env variables")
    public String getSystemEnv() {
        // Just put your logic code here, forget about the MCP SDK details.
        return System.getenv().toString();
    }

    // Your other MCP resources here...
}

@McpPrompts
public class MyMcpPrompts {

    @McpPrompt(description = "A simple prompt to read a file")
    public String readFile(
        @McpPromptParam(name = "path", description = "filepath", required = true) String path) {
        // Just put your logic code here, forget about the MCP SDK details.
        return String.format("What is the complete contents of the file: %s", path);
    }

}

@McpTools
public class MyMcpTools {

    // This method defines a MCP tool to read a file
    @McpTool(description = "Read complete file contents with UTF-8 encoding")
    public String readFile(
        @McpToolParam(name = "path", description = "filepath", required = true) String path) {
        // Just put your logic code here, forget about the MCP SDK details.
        return Files.readString(Path.of(path));
    }

    // Your other MCP tools here...
}