<?xml version="1.0" encoding="UTF-8" ?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>3.2.3</version>
        <relativePath/>
        <!-- lookup parent from repository -->
    </parent>
    <groupId>com.demo</groupId>
    <artifactId>mcp-demo</artifactId>
    <version>1.0-SNAPSHOT</version>
    <packaging>pom</packaging>
    <properties>
        <java.version>17</java.version>
        <guava.version>32.1.2-jre</guava.version>
        <commons-collections4.version>4.4</commons-collections4.version>
        <spring-ai.version>1.0.0-SNAPSHOT</spring-ai.version>
    </properties>
    <modules>
        <module>mcp-demo-launcher</module>
    </modules>
    <dependencyManagement>
        <dependencies>
            <dependency>
                <groupId>org.springframework.ai</groupId>
                <artifactId>spring-ai-bom</artifactId>
                <version>${spring-ai.version}</version>
                <type>pom</type>
                <scope>import</scope>
            </dependency>
        </dependencies>
    </dependencyManagement>
    <repositories>
        <!-- Spring Milestone 倉庫 -->
        <repository>
            <id>spring-milestones</id>
            <name>Spring Milestones</name>
            <url>https://repo.spring.io/milestone</url>
            <snapshots>
                <enabled>false</enabled>
            </snapshots>
        </repository>
        <repository>
            <id>spring-snapshots</id>
            <name>Spring Snapshots</name>
            <url>https://repo.spring.io/snapshot</url>
            <releases>
                <enabled>false</enabled>
            </releases>
        </repository>
    </repositories>
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter</artifactId>
            <exclusions>
                <exclusion>
                    <artifactId>spring-boot-starter-logging</artifactId>
                    <groupId>org.springframework.boot</groupId>
                </exclusion>
            </exclusions>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
        <!-- https://mvnrepository.com/artifact/org.springframework.ai/spring-ai-core -->
        <dependency>
            <groupId>org.springframework.ai</groupId>
            <artifactId>spring-ai-core</artifactId>
            <version>${spring-ai.version}</version>
        </dependency>
        <!-- MCP 服務(wù)器支持 - WebMVC版本 -->
        <dependency>
            <groupId>org.springframework.ai</groupId>
            <artifactId>spring-ai-starter-mcp-server-webmvc</artifactId>
            <version>${spring-ai.version}</version>
        </dependency>
        <!-- Spring Web -->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
            <exclusions>
                <exclusion>
                    <groupId>org.springframework.boot</groupId>
                    <artifactId>spring-boot-starter-logging</artifactId>
                </exclusion>
            </exclusions>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-context</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-log4j2</artifactId>
        </dependency>
    </dependencies>
</project>

server.port=8080
spring.application.name=mcp-demo
spring.main.banner-mode=off
# MCP
spring.ai.mcp.server.enabled=true
spring.ai.mcp.server.resource-change-notification=true
spring.ai.mcp.server.prompt-change-notification=true
spring.ai.mcp.server.tool-change-notification=true
spring.ai.mcp.server.name=mcp-demo-service
spring.ai.mcp.server.version=1.0.0
spring.ai.mcp.server.type=SYNC
spring.ai.mcp.server.sse-message-endpoint=/mcp/messages

@Configuration
public class McpServerConfig {
	@Autowired
	private ApplicationContext applicationContext;
	@Bean
	public ToolCallbackProvider autoRegisterTools() {
		// 獲取所有帶有 @Component 注解且類名以 Facade 結(jié)尾的 bean
		String[] beanNames = applicationContext.getBeanNamesForAnnotation(Component.class);
		List<Object> facadeBeans = new ArrayList<>();
		for (String beanName : beanNames) {
			if (beanName.endsWith("Facade")) {
				facadeBeans.add(applicationContext.getBean(beanName));
			}
		}
		// 一次性構(gòu)建所有 Facade
		return MethodToolCallbackProvider.builder()
		.toolObjects(facadeBeans.toArray())
		.build();
	}
}

@MCPService(packageName = "com.demo.mcp.web.facade")
@Service
public class WeatherService {
	/**
* Get weather information by city name
* @return
*/
	public String getWeather(String cityName) {
		// Implementation
		return String.format(
		"{"success":true,"data":{"city":"%s","temperature":"25°C","condition":"Sunny"},"message":"Success","code":"200"}",
		cityName
		);
	}
}

public class WeatherServiceFacade {
	@Autowired
	private WeatherService service;
	@Tool(description = "Get weather information by city name")
	public MCPResponse getWeather(MCPRequest request) {
		try {
			// 解析請求參數(shù)
			java.lang.String cityName = request.getParameter("cityName", java.lang.String.class);
			Object result = service.getWeather(cityName);
			return MCPResponse.success(result);
		}
		catch (Exception e) {
			return MCPResponse.error(e.getMessage());
		}
	}
}

public class ClientSse {
	public static void main(String[] args) {
		var transport = new HttpClientSseClientTransport("http://localhost:19401");
		new SampleClient(transport).run();
	}
}
public class SampleClient {
	private final McpClientTransport transport;
	public SampleClient(McpClientTransport transport) {
		this.transport = transport;
	}
	public void run() {
		try {
			var client = McpClient.sync(this.transport).build();
			client.initialize();
			client.ping();
			// List and demonstrate tools
			McpSchema.ListToolsResult toolsList = client.listTools();
			log.info("Available Tools = " + toolsList);
			// 調(diào)用 getWeather 接口
			McpSchema.CallToolResult weatherResult = client.callTool(new McpSchema.CallToolRequest(
			"getWeather",
			Map.of("cityName", "北京")
			));
			// 打印完整的響應(yīng)對象
			log.info("Weather Result: {}", weatherResult);
			// 獲取響應(yīng)內(nèi)容
			for (McpSchema.Content content : weatherResult.content()) {
				if (content instanceof McpSchema.TextContent) {
					McpSchema.TextContent textContent = (McpSchema.TextContent) content;
					log.info("Weather Info: {}", textContent.text());
				}
			}
			client.closeGracefully();
		}
		catch (Exception e) {
			log.error("Error calling weather service", e);
		}
	}
}