refactor: migrate MCP SDK to main repo (#3516)

plugins/wasm-go/pkg/mcp/validator/config_validator.go

@@ -0,0 +1,129 @@
+// Copyright (c) 2022 Alibaba Group Holding Ltd.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//	http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package validator
+
+import (
+	"encoding/json"
+	"fmt"
+	"os"
+
+	"github.com/tidwall/gjson"
+	"gopkg.in/yaml.v3"
+
+	"github.com/higress-group/wasm-go/pkg/log"
+	"github.com/alibaba/higress/plugins/wasm-go/pkg/mcp/server"
+)
+
+// validatorLogger is a simple logger implementation for validation mode
+type validatorLogger struct{}
+
+func (l *validatorLogger) Trace(msg string) { fmt.Fprintf(os.Stderr, "[TRACE] %s\n", msg) }
+func (l *validatorLogger) Tracef(format string, args ...interface{}) {
+	fmt.Fprintf(os.Stderr, "[TRACE] "+format+"\n", args...)
+}
+func (l *validatorLogger) Debug(msg string) { fmt.Fprintf(os.Stderr, "[DEBUG] %s\n", msg) }
+func (l *validatorLogger) Debugf(format string, args ...interface{}) {
+	fmt.Fprintf(os.Stderr, "[DEBUG] "+format+"\n", args...)
+}
+func (l *validatorLogger) Info(msg string) { fmt.Fprintf(os.Stderr, "[INFO] %s\n", msg) }
+func (l *validatorLogger) Infof(format string, args ...interface{}) {
+	fmt.Fprintf(os.Stderr, "[INFO] "+format+"\n", args...)
+}
+func (l *validatorLogger) Warn(msg string) { fmt.Fprintf(os.Stderr, "[WARN] %s\n", msg) }
+func (l *validatorLogger) Warnf(format string, args ...interface{}) {
+	fmt.Fprintf(os.Stderr, "[WARN] "+format+"\n", args...)
+}
+func (l *validatorLogger) Error(msg string) { fmt.Fprintf(os.Stderr, "[ERROR] %s\n", msg) }
+func (l *validatorLogger) Errorf(format string, args ...interface{}) {
+	fmt.Fprintf(os.Stderr, "[ERROR] "+format+"\n", args...)
+}
+func (l *validatorLogger) Critical(msg string) { fmt.Fprintf(os.Stderr, "[CRITICAL] %s\n", msg) }
+func (l *validatorLogger) Criticalf(format string, args ...interface{}) {
+	fmt.Fprintf(os.Stderr, "[CRITICAL] "+format+"\n", args...)
+}
+func (l *validatorLogger) ResetID(pluginID string) {}
+
+// init initializes the validator package
+func init() {
+	// Set a custom logger for validation mode to prevent panics
+	log.SetPluginLog(&validatorLogger{})
+}
+
+// ValidationResult contains the result of configuration validation
+type ValidationResult struct {
+	IsValid    bool   `json:"isValid"`
+	Error      error  `json:"error,omitempty"`
+	ServerName string `json:"serverName,omitempty"`
+	IsComposed bool   `json:"isComposed"`
+}
+
+// ValidateConfig validates MCP configuration
+// This function focuses on validating REST tools and toolSet configurations
+// It skips validation for pre-registered Go-based servers
+func ValidateConfig(configJSON string) (*ValidationResult, error) {
+	// Create empty dependencies for validation mode
+	// We skip pre-registered servers validation by setting SkipPreRegisteredServers to true
+	toolRegistry := &server.GlobalToolRegistry{}
+	toolRegistry.Initialize() // Initialize the registry to prevent nil map assignment panic
+
+	deps := &server.ConfigOptions{
+		Servers:                  make(map[string]server.Server), // Empty servers map
+		ToolRegistry:             toolRegistry,                   // Initialized registry
+		SkipPreRegisteredServers: true,                           // Skip pre-registered servers
+	}
+
+	// Call core parsing logic for validation
+	configGjson := gjson.Parse(configJSON)
+	mockConfig := &server.McpServerConfig{}
+
+	err := server.ParseConfigCore(configGjson, mockConfig, deps)
+
+	result := &ValidationResult{
+		IsValid: err == nil,
+		Error:   err,
+	}
+
+	if err == nil {
+		result.ServerName = mockConfig.GetServerName()
+		result.IsComposed = mockConfig.GetIsComposed()
+	}
+
+	return result, nil
+}
+
+// ValidateConfigYAML validates MCP configuration from YAML format
+// This function converts YAML to JSON first, then validates using the same logic
+func ValidateConfigYAML(configYAML string) (*ValidationResult, error) {
+	// Parse YAML into a generic interface
+	var yamlData interface{}
+	if err := yaml.Unmarshal([]byte(configYAML), &yamlData); err != nil {
+		return &ValidationResult{
+			IsValid: false,
+			Error:   fmt.Errorf("failed to parse YAML: %v", err),
+		}, nil
+	}
+
+	// Convert to JSON
+	jsonBytes, err := json.Marshal(yamlData)
+	if err != nil {
+		return &ValidationResult{
+			IsValid: false,
+			Error:   fmt.Errorf("failed to convert YAML to JSON: %v", err),
+		}, nil
+	}
+
+	// Use the existing JSON validation logic
+	return ValidateConfig(string(jsonBytes))
+}