build flagset from inputschema to parse args

Author: burmudar

cmd/src/mcp_args.go

@@ -0,0 +1,56 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"strings"
+)
+
+var _ flag.Value = (*strSliceFlag)(nil)
+
+type strSliceFlag struct {
+	vals []string
+}
+
+func (s *strSliceFlag) Set(v string) error {
+	s.vals = append(s.vals, v)
+	return nil
+}
+
+func (s *strSliceFlag) String() string {
+	return strings.Join(s.vals, ",")
+}
+
+func buildArgFlagSet(tool *MCPToolDef) (*flag.FlagSet, map[string]any, error) {
+	fs := flag.NewFlagSet(tool.Name(), flag.ContinueOnError)
+	flagVars := map[string]any{}
+
+	for name, pVal := range tool.InputSchema.Properties {
+		switch pv := pVal.(type) {
+		case *SchemaPrimitive:
+			switch pv.Kind {
+			case "integer":
+				dst := fs.Int(name, 0, pv.Description)
+				flagVars[name] = dst
+
+			case "boolean":
+				dst := fs.Bool(name, false, pv.Description)
+				flagVars[name] = dst
+			case "string":
+				dst := fs.String(name, "", pv.Description)
+				flagVars[name] = dst
+			default:
+				return nil, nil, fmt.Errorf("unknown schema primitive kind %q", pv.Kind)
+
+			}
+		case *SchemaArray:
+			strSlice := new(strSliceFlag)
+			fs.Var(strSlice, name, pv.Description)
+			flagVars[name] = strSlice
+		case *SchemaObject:
+			// not supported yet
+		}
+	}
+
+	return fs, flagVars, nil
+}

cmd/src/mcp_args_test.go

@@ -0,0 +1,97 @@
+package main
+
+import (
+	"testing"
+)
+
+func TestFlagSetParse(t *testing.T) {
+	toolJSON := []byte(`{
+	  "tools": [
+		{
+		  "name": "sg_test_tool",
+		  "description": "test description",
+		  "inputSchema": {
+			"type": "object",
+			"$schema": "https://localhost/schema-draft/2025-07",
+			"required": ["values"],
+			"properties": {
+			  "repos": {
+				"type": "array",
+				"items": {
+				  "type": "string"
+				}
+			  },
+			  "tag": {
+				"type": "string",
+				"items": true
+			  },
+			  "count": {
+				"type": "integer"
+	          },
+	          "boolFlag": {
+				"type": "boolean"
+	          }
+			}
+		  },
+		  "outputSchema": {
+			"type": "object",
+			"$schema": "https://localhost/schema-draft/2025-07",
+			"properties": {
+			  "result": { "type": "string" }
+			}
+		  }
+		}
+	  ]
+	}`)
+
+	defs, err := LoadMCPToolDefinitions(toolJSON)
+	if err != nil {
+		t.Fatalf("failed to load tool json: %v", err)
+	}
+
+	flagSet, vars, err := buildArgFlagSet(defs["sg_test_tool"])
+	if err != nil {
+		t.Fatalf("failed to build flagset from mcp tool definition: %v", err)
+	}
+
+	if len(vars) == 0 {
+		t.Fatalf("vars from buildArgFlagSet should not be empty")
+	}
+
+	args := []string{"-repos=A", "-repos=B", "-count=10", "-boolFlag", "-tag=testTag"}
+
+	if err := flagSet.Parse(args); err != nil {
+		t.Fatalf("flagset parsing failed: %v", err)
+	}
+	derefFlagValues(vars)
+
+	if v, ok := vars["repos"].([]string); ok {
+		if len(v) != 2 {
+			t.Fatalf("expected flag 'repos' values to have length %d but got %d", 2, len(v))
+		}
+	} else {
+		t.Fatalf("expected flag 'repos' to have type of []string but got %T", v)
+	}
+	if v, ok := vars["tag"].(string); ok {
+		if v != "testTag" {
+			t.Fatalf("expected flag 'tag' values to have value %q but got %q", "testTag", v)
+		}
+	} else {
+		t.Fatalf("expected flag 'tag' to have type of string but got %T", v)
+	}
+	if v, ok := vars["count"].(int); ok {
+		if v != 10 {
+			t.Fatalf("expected flag 'count' values to have value %d but got %d", 10, v)
+		}
+	} else {
+		t.Fatalf("expected flag 'count' to have type of int but got %T", v)
+	}
+	if v, ok := vars["boolFlag"].(bool); ok {
+		if v != true {
+			t.Fatalf("expected flag 'boolFlag' values to have value %v but got %v", true, v)
+		}
+	} else {
+		t.Fatalf("expected flag 'boolFlag' to have type of bool but got %T", v)
+	}
+
+}