140 lines
3.6 KiB
Go
140 lines
3.6 KiB
Go
package params
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
)
|
|
|
|
// GetString extracts a required string parameter from MCP tool arguments.
|
|
func GetString(args map[string]any, key string) (string, error) {
|
|
val, ok := args[key].(string)
|
|
if !ok {
|
|
return "", fmt.Errorf("%s is required", key)
|
|
}
|
|
return val, nil
|
|
}
|
|
|
|
// GetOptionalString extracts an optional string parameter with a default value.
|
|
func GetOptionalString(args map[string]any, key, defaultVal string) string {
|
|
if val, ok := args[key].(string); ok {
|
|
return val
|
|
}
|
|
return defaultVal
|
|
}
|
|
|
|
// GetStringSlice extracts an optional string slice parameter from MCP tool arguments.
|
|
func GetStringSlice(args map[string]any, key string) []string {
|
|
val, ok := args[key]
|
|
if !ok {
|
|
return nil
|
|
}
|
|
sliceVal, ok := val.([]any)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
out := make([]string, 0, len(sliceVal))
|
|
for _, item := range sliceVal {
|
|
if s, ok := item.(string); ok {
|
|
out = append(out, s)
|
|
}
|
|
}
|
|
return out
|
|
}
|
|
|
|
// GetPagination extracts page and perPage parameters, returning them as ints.
|
|
func GetPagination(args map[string]any, defaultPageSize int64) (page, pageSize int) {
|
|
return int(GetOptionalInt(args, "page", 1)), int(GetOptionalInt(args, "perPage", defaultPageSize))
|
|
}
|
|
|
|
// ToInt64 converts a value to int64, accepting both float64 (JSON number) and
|
|
// string representations. Returns false if the value cannot be converted.
|
|
func ToInt64(val any) (int64, bool) {
|
|
switch v := val.(type) {
|
|
case float64:
|
|
return int64(v), true
|
|
case string:
|
|
i, err := strconv.ParseInt(v, 10, 64)
|
|
if err != nil {
|
|
return 0, false
|
|
}
|
|
return i, true
|
|
default:
|
|
return 0, false
|
|
}
|
|
}
|
|
|
|
// GetIndex extracts a required integer parameter from MCP tool arguments.
|
|
// It accepts both numeric (float64 from JSON) and string representations.
|
|
// This provides better UX for LLM callers that may naturally use strings
|
|
// for identifiers like issue/PR numbers.
|
|
func GetIndex(args map[string]any, key string) (int64, error) {
|
|
val, exists := args[key]
|
|
if !exists {
|
|
return 0, fmt.Errorf("%s is required", key)
|
|
}
|
|
|
|
if i, ok := ToInt64(val); ok {
|
|
return i, nil
|
|
}
|
|
|
|
if s, ok := val.(string); ok {
|
|
return 0, fmt.Errorf("%s must be a valid integer (got %q)", key, s)
|
|
}
|
|
|
|
return 0, fmt.Errorf("%s must be a number or numeric string", key)
|
|
}
|
|
|
|
// GetInt64Slice extracts a required int64 slice parameter from MCP tool arguments.
|
|
func GetInt64Slice(args map[string]any, key string) ([]int64, error) {
|
|
raw, ok := args[key].([]any)
|
|
if !ok {
|
|
return nil, fmt.Errorf("%s (array of IDs) is required", key)
|
|
}
|
|
out := make([]int64, 0, len(raw))
|
|
for _, v := range raw {
|
|
id, ok := ToInt64(v)
|
|
if !ok {
|
|
return nil, fmt.Errorf("invalid ID in %s array", key)
|
|
}
|
|
out = append(out, id)
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
// GetOptionalInt extracts an optional integer parameter from MCP tool arguments.
|
|
// Returns defaultVal if the key is missing or the value cannot be parsed.
|
|
// Accepts both float64 (JSON number) and string representations.
|
|
func GetOptionalInt(args map[string]any, key string, defaultVal int64) int64 {
|
|
val, exists := args[key]
|
|
if !exists {
|
|
return defaultVal
|
|
}
|
|
if i, ok := ToInt64(val); ok {
|
|
return i
|
|
}
|
|
return defaultVal
|
|
}
|
|
|
|
// GetOptionalBool extracts an optional boolean parameter from MCP tool arguments.
|
|
// Returns defaultVal if the key is missing or the value cannot be parsed.
|
|
// Accepts bool, float64 (1=true, 0=false), and string representations.
|
|
func GetOptionalBool(args map[string]any, key string, defaultVal bool) bool {
|
|
val, exists := args[key]
|
|
if !exists {
|
|
return defaultVal
|
|
}
|
|
|
|
switch v := val.(type) {
|
|
case bool:
|
|
return v
|
|
case float64:
|
|
return v != 0
|
|
case string:
|
|
if b, err := strconv.ParseBool(v); err == nil {
|
|
return b
|
|
}
|
|
}
|
|
|
|
return defaultVal
|
|
}
|