added: to context list a marker of current context

strowk · Dec 15, 2024 · e92f1bc · e92f1bc
1 parent 0d9df27
commit e92f1bc
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/internal/tools/contexts.go b/internal/tools/contexts.go
@@ -40,21 +40,22 @@ func NewListContextsTool() fxctx.Tool {
 
 			return &mcp.CallToolResult{
 				Meta:    map[string]interface{}{},
-				Content: getListContextsToolContent(cfg),
+				Content: getListContextsToolContent(cfg, cfg.CurrentContext),
 				IsError: utils.Ptr(false),
 			}
 		},
 	)
 }
 
-func getListContextsToolContent(cfg api.Config) []interface{} {
+func getListContextsToolContent(cfg api.Config, current string) []interface{} {
 	var contents []interface{} = make([]interface{}, len(cfg.Contexts))
 	i := 0
 
-	for _, c := range cfg.Contexts {
+	for name, c := range cfg.Contexts {
 		marshalled, err := json.Marshal(ContextJsonEncoded{
 			Context: c,
 			Name:    c.Cluster,
+			Current: name == current,
 		})
 		if err != nil {
 			log.Printf("failed to marshal context: %v", err)
@@ -73,4 +74,5 @@ func getListContextsToolContent(cfg api.Config) []interface{} {
 type ContextJsonEncoded struct {
 	Context *api.Context `json:"context"`
 	Name    string       `json:"name"`
+	Current bool         `json:"current"`
 }
diff --git a/testdata/k8s_contexts/list_k8s_contexts_test.yaml b/testdata/k8s_contexts/list_k8s_contexts_test.yaml
@@ -15,7 +15,7 @@ out:
           [
             {
               "type": "text",
-              "text": '{"context":{"cluster":"test-cluster","user":"test-user"},"name":"test-cluster"}',
+              "text": '{"context":{"cluster":"test-cluster","user":"test-user"},"name":"test-cluster","current":true}',
             },
           ],
         "isError": false,

diff --git a/testdata/with_k3d/list_k8s_contexts_test.yaml b/testdata/with_k3d/list_k8s_contexts_test.yaml
@@ -15,7 +15,7 @@ out:
           [
             {
               "type": "text",
-              "text": '{"context":{"cluster":"k3d-mcp-k8s-integration-test","user":"admin@k3d-mcp-k8s-integration-test"},"name":"k3d-mcp-k8s-integration-test"}',
+              "text": '{"context":{"cluster":"k3d-mcp-k8s-integration-test","user":"admin@k3d-mcp-k8s-integration-test"},"name":"k3d-mcp-k8s-integration-test","current":true}',
             },
           ],
         "isError": false,