Skip to content

Commit 0525c51

Browse files
Dot-LiuDot-Liu
committed
Consumer MCP completed
1 parent 47950d9 commit 0525c51

19 files changed

Lines changed: 424 additions & 286 deletions

File tree

‎controller/mcp/iml.go‎

Lines changed: 100 additions & 154 deletions
Large diffs are not rendered by default.

‎controller/mcp/mcp.go‎

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,12 +9,20 @@ import (
99

1010
type IMcpController interface {
1111
MCPHandle(ctx *gin.Context)
12+
1213
GlobalMCPHandle(ctx *gin.Context)
1314
GlobalHandleSSE(ctx *gin.Context)
1415
GlobalHandleMessage(ctx *gin.Context)
16+
GlobalMCPConfig(ctx *gin.Context) (string, error)
17+
18+
AppMCPHandle(ctx *gin.Context)
19+
AppHandleSSE(ctx *gin.Context)
20+
AppHandleMessage(ctx *gin.Context)
21+
AppMCPConfig(ctx *gin.Context, appId string) (string, error)
22+
1523
ServiceHandleSSE(ctx *gin.Context)
1624
ServiceHandleMessage(ctx *gin.Context)
17-
GlobalMCPConfig(ctx *gin.Context) (string, error)
25+
1826
ServiceHandleStreamHTTP(ctx *gin.Context)
1927
}
2028

‎controller/mcp/tool.go‎

Lines changed: 104 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,104 @@
1+
package mcp
2+
3+
import "github.com/mark3labs/mcp-go/mcp"
4+
5+
const (
6+
ToolServiceList = "service_list"
7+
ToolOpenAPIDocument = "openapi_document"
8+
ToolInvokeAPI = "invoke_api"
9+
10+
languageZhCN = "zh-CN"
11+
languageZhTW = "zh-TW"
12+
languageEnUs = "en-US"
13+
languageJaJp = "ja-JP"
14+
)
15+
16+
var mcpToolsByLanguage = map[string]map[string]mcp.Tool{
17+
languageZhCN: toolsZhCN,
18+
languageZhTW: toolsZhTW,
19+
languageEnUs: toolsEnUs,
20+
languageJaJp: toolsJaJp,
21+
}
22+
23+
var toolsZhTW = map[string]mcp.Tool{
24+
ToolServiceList: mcp.NewTool(
25+
ToolServiceList,
26+
mcp.WithDescription("此工具用於獲取 APIPark 中已註冊服務的列表。每個服務包含其唯一標識（service ID）、名稱、描述及包含的 API 列表等關鍵信息。支持通過關鍵詞進行模糊搜索，以便快速縮小查找範圍。在獲得某個服務的 ID 後，可以調用 openapi_document 工具來獲取該服務的 OpenAPI 文檔，以便後續調用其提供的 API 接口。"),
27+
mcp.WithString("keyword", mcp.Description("關鍵詞，用於模糊搜索服務"))),
28+
ToolOpenAPIDocument: mcp.NewTool(
29+
ToolOpenAPIDocument,
30+
mcp.WithDescription("此工具用於獲取 APIPark 中服務的 OpenAPI 文檔。"),
31+
mcp.WithString("service_id", mcp.Description("服務 ID"))),
32+
ToolInvokeAPI: mcp.NewTool(
33+
ToolInvokeAPI,
34+
mcp.WithDescription("此工具可直接發送 API 請求。在調用此工具之前，需要根據該 API 的 OpenAPI 文檔構造所需的請求參數，如請求路徑、方法、查詢參數、頭部信息、主體內容等。使用此工具時不需要傳遞任何認證信息，例如 Authorization 頭部可以省略。"),
35+
mcp.WithString("path", mcp.Description("API 的請求路徑"), mcp.Required()),
36+
mcp.WithString("method", mcp.Description("API 的請求方法，例如 GET、POST、PUT"), mcp.Required()),
37+
mcp.WithString("content-type", mcp.Description("請求的 Content-Type。若方法為 POST、PUT 或 PATCH，則必須指定")),
38+
mcp.WithObject("query", mcp.Description("請求的查詢參數，類型為 map[string]string")),
39+
mcp.WithObject("header", mcp.Description("請求的頭部，類型為 map[string]string")),
40+
mcp.WithString("body", mcp.Description("請求主體內容，通常為 JSON 字符串")),
41+
),
42+
}
43+
44+
var toolsZhCN = map[string]mcp.Tool{
45+
ToolServiceList: mcp.NewTool(
46+
ToolServiceList,
47+
mcp.WithDescription("此工具用于获取 APIPark 中已注册服务的列表。每个服务包含其唯一标识（service ID）、名称、描述及包含的 API 列表等关键信息。支持通过关键词进行模糊搜索，以便快速缩小查找范围。在获得某个服务的 ID 后，可以调用 openapi_document 工具来获取该服务的 OpenAPI 文档，以便后续调用其提供的 API 接口。"),
48+
mcp.WithString("keyword", mcp.Description("关键词，用于模糊搜索服务"))),
49+
ToolOpenAPIDocument: mcp.NewTool(
50+
ToolOpenAPIDocument, mcp.WithDescription("此工具用于查询指定服务的 OpenAPI 文档。返回的格式支持 OpenAPI v3 和 v2 标准。通过输入服务 ID，可以查看该服务所有 API 的定义、参数结构、请求方式等详细信息，为后续构造 API 调用请求做准备。"),
51+
mcp.WithString("service", mcp.Description("欲查询的服务唯一标识")),
52+
),
53+
ToolInvokeAPI: mcp.NewTool(ToolInvokeAPI,
54+
mcp.WithDescription("此工具可直接发送 API 请求。在调用此工具之前，需要根据该 API 的 OpenAPI 文档构造所需的请求参数，如请求路径、方法、查询参数、头部信息、主体内容等。使用此工具时不需要传递任何认证信息，例如 Authorization 头部可以省略。"),
55+
mcp.WithString("path", mcp.Description("API 的请求路径"), mcp.Required()),
56+
mcp.WithString("method", mcp.Description("API 的请求方法，例如 GET、POST、PUT"), mcp.Required()),
57+
mcp.WithString("content-type", mcp.Description("请求的 Content-Type。若方法为 POST、PUT 或 PATCH，则必须指定")),
58+
mcp.WithObject("query", mcp.Description("请求的查询参数，类型为 map[string]string")),
59+
mcp.WithObject("header", mcp.Description("请求的头部，类型为 map[string]string")),
60+
mcp.WithString("body", mcp.Description("请求主体内容，通常为 JSON 字符串")),
61+
),
62+
}
63+
64+
var toolsEnUs = map[string]mcp.Tool{
65+
ToolServiceList: mcp.NewTool(
66+
ToolServiceList,
67+
mcp.WithDescription("This tool is used to retrieve a list of registered services in APIPark. Each service includes its unique identifier (service ID), name, description, and a list of APIs it contains. It supports fuzzy searching by keyword for quick narrowing down of results. After obtaining a service ID, you can use the openapi_document tool to get the OpenAPI documentation for that service, which is necessary for invoking its APIs."),
68+
mcp.WithString("keyword", mcp.Description("Keyword for fuzzy search of services"))),
69+
ToolOpenAPIDocument: mcp.NewTool(
70+
ToolOpenAPIDocument,
71+
mcp.WithDescription("This tool is used to query the OpenAPI documentation of a specified service. The returned format supports both OpenAPI v3 and v2 standards. By entering the service ID, you can view detailed information about all APIs of that service, including definitions, parameter structures, request methods, etc., which prepares you for subsequent API calls."),
72+
mcp.WithString("service", mcp.Description("Unique identifier of the service to query"))),
73+
ToolInvokeAPI: mcp.NewTool(
74+
ToolInvokeAPI,
75+
mcp.WithDescription("This tool can directly send API requests. Before using this tool, you need to construct the required request parameters based on the OpenAPI documentation of the API, such as request path, method, query parameters, header information, body content, etc. No authentication information like Authorization header is required when using this tool."),
76+
mcp.WithString("path", mcp.Description("API request path"), mcp.Required()),
77+
mcp.WithString("method", mcp.Description("API request method, e.g., GET, POST, PUT"), mcp.Required()),
78+
mcp.WithString("content-type", mcp.Description("Content-Type of the request. Must be specified if method is POST, PUT, or PATCH")),
79+
mcp.WithObject("query", mcp.Description("Query parameters of the request, type map[string]string")),
80+
mcp.WithObject("header", mcp.Description("Header information of the request, type map[string]string")),
81+
mcp.WithString("body", mcp.Description("Body content of the request, usually in JSON string")),
82+
),
83+
}
84+
85+
var toolsJaJp = map[string]mcp.Tool{
86+
ToolServiceList: mcp.NewTool(
87+
ToolServiceList,
88+
mcp.WithDescription("このツールは、APIParkに登録されているサービスのリストを取得するために使用されます。各サービスには、ユニークな識別子（サービスID）、名前、説明、および含まれるAPIのリストが含まれています。キーワードによるあいまい検索をサポートしており、結果を迅速に絞り込むことができます。サービスIDを取得した後は、openapi_documentツールを使用してそのサービスのOpenAPIドキュメントを取得し、そのAPIを呼び出す準備をします。"),
89+
mcp.WithString("keyword", mcp.Description("サービスをあいまい検索するためのキーワード"))),
90+
ToolOpenAPIDocument: mcp.NewTool(
91+
ToolOpenAPIDocument,
92+
mcp.WithDescription("このツールは、指定されたサービスのOpenAPIドキュメントを照会するために使用されます。返される形式は、OpenAPI v3およびv2標準の両方をサポートしています。サービスIDを入力することで、そのサービスのすべてのAPIに関する詳細情報（定義、パラメータ構造、リクエスト方法など）を表示し、後続のAPI呼び出しの準備をします。"),
93+
mcp.WithString("service", mcp.Description("照会するサービスのユニークな識別子"))),
94+
ToolInvokeAPI: mcp.NewTool(
95+
ToolInvokeAPI,
96+
mcp.WithDescription("このツールは、APIリクエストを直接送信できます。このツールを使用する前に、APIのOpenAPIドキュメントに基づいて、必要なリクエストパラメータ（リクエストパス、メソッド、クエリパラメータ、ヘッダー情報、ボディコンテンツなど）を構築する必要があります。このツールを使用する際には、Authorizationヘッダーなどの認証情報は必要ありません。"),
97+
mcp.WithString("path", mcp.Description("APIのリクエストパス"), mcp.Required()),
98+
mcp.WithString("method", mcp.Description("APIのリクエストメソッド（例：GET、POST、PUT）"), mcp.Required()),
99+
mcp.WithString("content-type", mcp.Description("リクエストのContent-Type。メソッドがPOST、PUT、またはPATCHの場合は必須です")),
100+
mcp.WithObject("query", mcp.Description("リクエストのクエリパラメータ、タイプはmap[string]string")),
101+
mcp.WithObject("header", mcp.Description("リクエストのヘッダー、タイプはmap[string]string")),
102+
mcp.WithString("body", mcp.Description("リクエストのボディコンテンツ、通常はJSON文字列")),
103+
),
104+
}

‎controller/system-apikey/apikey.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ type IAPIKeyController interface {
1616
Search(ctx *gin.Context, keyword string) ([]*system_apikey_dto.Item, error)
1717
SimpleList(ctx *gin.Context) ([]*system_apikey_dto.SimpleItem, error)
1818
MyAPIKeys(ctx *gin.Context) ([]*system_apikey_dto.SimpleItem, error)
19-
MyAPIKeysByService(ctx *gin.Context, serviceId string) ([]*system_apikey_dto.AuthorizationItem, error)
19+
MyAPIKeysByService(ctx *gin.Context, serviceId string, appId string) ([]*system_apikey_dto.AuthorizationItem, error)
2020
}
2121

2222
func init() {

‎controller/system-apikey/iml.go‎

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,14 @@ type imlAPIKeyController struct {
1212
apikeyModule system_apikey.IAPIKeyModule `autowired:""`
1313
}
1414

15-
func (i *imlAPIKeyController) MyAPIKeysByService(ctx *gin.Context, serviceId string) ([]*system_apikey_dto.AuthorizationItem, error) {
16-
return i.apikeyModule.MyAPIKeysByService(ctx, serviceId)
15+
func (i *imlAPIKeyController) MyAPIKeysByService(ctx *gin.Context, serviceId string, appId string) ([]*system_apikey_dto.AuthorizationItem, error) {
16+
if serviceId != "" {
17+
return i.apikeyModule.MyAPIKeysByService(ctx, serviceId)
18+
}
19+
if appId != "" {
20+
return i.apikeyModule.MyAPIKeysByApp(ctx, appId)
21+
}
22+
return nil, nil
1723
}
1824

1925
func (i *imlAPIKeyController) MyAPIKeys(ctx *gin.Context) ([]*system_apikey_dto.SimpleItem, error) {

‎mcp-server/server.go‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ var (
1616
mcpServer = NewServer()
1717
ServiceBasePath = "mcp/service"
1818
GlobalBasePath = "mcp/global"
19+
AppBasePath = "mcp/app"
1920
)
2021

2122
func NewServer() *Server {

‎module/application-authorization/authorization.go‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@ type IAuthorizationModule interface {
3333
// Info 获取项目鉴权详情
3434
Info(ctx context.Context, appId string, aid string) (*application_authorization_dto.Authorization, error)
3535

36-
CheckAPIKeyAuthorization(ctx context.Context, serviceId string, apikey string) (bool, error)
36+
CheckAPIKeyAuthorizationByService(ctx context.Context, serviceId string, apikey string) (bool, error)
37+
CheckAPIKeyAuthorizationByApp(ctx context.Context, appId string, apikey string) (bool, error)
3738

3839
//ExportAll(ctx context.Context) ([]*application_authorization_dto.ExportAuthorization, error)
3940
}

‎module/application-authorization/iml.go‎

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,27 @@ type imlAuthorizationModule struct {
4444
transaction store.ITransaction `autowired:""`
4545
}
4646

47-
func (i *imlAuthorizationModule) CheckAPIKeyAuthorization(ctx context.Context, serviceId string, apikey string) (bool, error) {
47+
func (i *imlAuthorizationModule) CheckAPIKeyAuthorizationByApp(ctx context.Context, appId string, apikey string) (bool, error) {
48+
authorizations, err := i.authorizationService.ListByApp(ctx, appId)
49+
if err != nil {
50+
return false, err
51+
}
52+
for _, a := range authorizations {
53+
if a.Type != "apikey" {
54+
continue
55+
}
56+
cfg := make(map[string]interface{})
57+
if a.Config != "" {
58+
json.Unmarshal([]byte(a.Config), &cfg)
59+
}
60+
if cfg["apikey"] == apikey {
61+
return true, nil
62+
}
63+
}
64+
return false, nil
65+
}
66+
67+
func (i *imlAuthorizationModule) CheckAPIKeyAuthorizationByService(ctx context.Context, serviceId string, apikey string) (bool, error) {
4868
list, err := i.subscribeService.ListBySubscribeStatus(ctx, serviceId, subscribe.ApplyStatusSubscribe)
4969
if err != nil {
5070
return false, err

0 commit comments

Comments
 (0)