Tools

func NewGreatTool() fxctx.Tool {
	return fxctx.NewTool(
		// This information about the tool would be used when it is listed:
		&mcp.Tool{
			Name:        "my-great-tool",
			Description: Ptr("The great tool"),
			InputSchema: mcp.ToolInputSchema{ // here we tell client what we expect as input
				Type:       "object",
				Properties: map[string]map[string]interface{}{},
				Required:   []string{},
			},
		},

		// This is the callback that would be executed when the tool is called:
		func(_ context.Context, args map[string]interface{}) *mcp.CallToolResult {
			// here we can do anything we want
			return &mcp.CallToolResult{
				Content: []interface{}{
					mcp.TextContent{
						Type: "text",
						Text: fmt.Sprintf("Sup"),
					},
				},
			}
		},
	)
}

func main() {
	app.
		NewBuilder().
		// adding the tool to the app
		WithTool(NewGreatTool).
		// setting up server
		WithName("great-tool-server").
		WithVersion("0.0.1").
		WithTransport(stdio.NewTransport()).
		// Configuring fx logging to only show errors
		WithFxOptions(
			fx.Provide(func() *zap.Logger {
				cfg := zap.NewDevelopmentConfig()
				cfg.Level.SetLevel(zap.ErrorLevel)
				logger, _ := cfg.Build()
				return logger
			}),
			fx.Option(fx.WithLogger(
				func(logger *zap.Logger) fxevent.Logger {
					return &fxevent.ZapLogger{Logger: logger}
				},
			)),
		).Run()
}

	schema := toolinput.NewToolInputSchema(
		toolinput.WithString("kubeconfig", "Path to kubeconfig file"),
	)

	// Here we define a tool that lists k8s contexts using client-go
	return fxctx.NewTool(
		// This information about the tool would be used when it is listed:
		&mcp.Tool{
			Name:        "list-k8s-contexts",
			Description: Ptr("List Kubernetes contexts from configuration files such as kubeconfig"),
			InputSchema: schema.GetMcpToolInputSchema(),
		},

		// This is the callback that would be executed when the tool is called:
		func(_ context.Context, args map[string]interface{}) *mcp.CallToolResult {
			input, err := schema.Validate(args)
			if err != nil {
				log.Printf("failed to validate input: %v", err)
				return &mcp.CallToolResult{
					IsError: Ptr(true),
					Content: []interface{}{
						mcp.TextContent{
							Type: "text",
							Text: fmt.Sprintf("failed to validate input: %v", err),
						},
					},
				}
			}

			path := input.StringOr("kubeconfig", "")
			if path != "" {
				os.Setenv("KUBECONFIG", path)
			}