✏️ Format cp tool name

issue #64
This commit is contained in:
OpenWaygate committed 2025-07-01 23:39:08 +08:00
1 parent 3bbf6c1312
commit 3e61ad2e74
3 files changed
+39 -23

No files matched your search

+33 -20
View File
@@ -14,39 +14,45 @@ import (
var defaultParts = []string{"id", "snippet", "contentDetails"}
var listTool = mcp.NewTool(
"activity.list",
"activity-list",
mcp.WithTitleAnnotation("List Activities"),
mcp.WithOpenWorldHintAnnotation(true),
mcp.WithDescription(long),
mcp.WithString(
"channelId", mcp.DefaultString(""), mcp.Description(ciUsage),
"channelId", mcp.DefaultString(""), mcp.Description(ciUsage), mcp.Required(),
),
mcp.WithString(
"home", mcp.Enum("true", "false", ""),
mcp.DefaultString(""), mcp.Description(homeUsage),
mcp.DefaultString(""), mcp.Description(homeUsage), mcp.Required(),
),
mcp.WithNumber(
"maxResults", mcp.DefaultNumber(5), mcp.Description(mrUsage),
"maxResults", mcp.DefaultNumber(5), mcp.Description(mrUsage), mcp.Required(),
),
mcp.WithString(
"mine", mcp.Enum("true", "false", ""),
mcp.DefaultString("true"), mcp.Description(mineUsage),
mcp.DefaultString("true"), mcp.Description(mineUsage), mcp.Required(),
),
mcp.WithString(
"publishedAfter", mcp.DefaultString(""),
mcp.Description(paUsage),
mcp.Description(paUsage), mcp.Required(),
),
mcp.WithString(
"publishedBefore", mcp.DefaultString(""),
mcp.Description(pbUsage),
mcp.Description(pbUsage), mcp.Required(),
),
mcp.WithString(
"regionCode", mcp.DefaultString(""), mcp.Description(rcUsage),
"regionCode", mcp.DefaultString(""),
mcp.Description(rcUsage), mcp.Required(),
),
mcp.WithArray(
"parts", mcp.DefaultArray(defaultParts), mcp.Description(partsUsage),
"parts", mcp.DefaultArray(defaultParts),
mcp.Items(map[string]any{"type": "string"}),
mcp.Description(partsUsage), mcp.Required(),
),
mcp.WithString(
"output", mcp.DefaultString(""), mcp.Description(outputUsage),
mcp.Required(),
),
mcp.WithString("output", mcp.DefaultString(""), mcp.Description(outputUsage)),
)
func run(writer io.Writer) error {
@@ -99,18 +105,25 @@ func listHandler(ctx context.Context, request mcp.CallToolRequest) (
*mcp.CallToolResult, error,
) {
args := request.GetArguments()
channelId = args["channelId"].(string)
home = utils.BoolPtr(args["home"].(string))
maxResults = int64(args["maxResults"].(float64))
mine = utils.BoolPtr(args["mine"].(string))
publishedAfter = args["publishedAfter"].(string)
publishedBefore = args["publishedBefore"].(string)
regionCode = args["regionCode"].(string)
parts = make([]string, len(args["parts"].([]interface{})))
for i, part := range args["parts"].([]interface{}) {
channelId, _ = args["channelId"].(string)
homeRaw, _ := args["home"].(string)
home = utils.BoolPtr(homeRaw)
maxResultsRaw, _ := args["maxResults"].(float64)
maxResults = int64(maxResultsRaw)
mineRaw, ok := args["mine"].(string)
if !ok {
mineRaw = "true" // Default to true if not provided
}
mine = utils.BoolPtr(mineRaw)
publishedAfter, _ = args["publishedAfter"].(string)
publishedBefore, _ = args["publishedBefore"].(string)
regionCode, _ = args["regionCode"].(string)
partsRaw, _ := args["parts"].([]interface{})
parts = make([]string, len(partsRaw))
for i, part := range partsRaw {
parts[i] = part.(string)
}
output = args["output"].(string)
output, _ = args["output"].(string)
var writer bytes.Buffer
err := run(&writer)
+1 -1
View File
@@ -59,5 +59,5 @@ func init() {
RootCmd.AddCommand(mcpCmd)
mcpCmd.Flags().StringVarP(&mode, "mode", "m", "stdio", modeUsage)
mcpCmd.Flags().IntVarP(&port, "port", "p", 8080, portUsage)
mcpCmd.Flags().IntVarP(&port, "port", "p", 8216, portUsage)
}
+5 -2
View File
@@ -99,10 +99,13 @@ func (a *activity) List(
tb.SetOutputMirror(writer)
tb.SetStyle(table.StyleLight)
tb.SetAutoIndex(true)
tb.AppendHeader(table.Row{"ID", "Title", "Type"})
tb.AppendHeader(table.Row{"ID", "Title", "Type", "Time"})
for _, activity := range activities {
tb.AppendRow(
table.Row{activity.Id, activity.Snippet.Title, activity.Snippet.Type},
table.Row{
activity.Id, activity.Snippet.Title,
activity.Snippet.Type, activity.Snippet.PublishedAt,
},
)
}
}