// Copyright (c) 2025 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 services

import (
	"fmt"
)

func HandleAddServiceSource(client *HigressClient, body interface{}) ([]byte, error) {
	data, ok := body.(map[string]interface{})
	// fmt.Printf("request body: %v\n", data)
	if !ok {
		return nil, fmt.Errorf("failed to parse request body")
	}
	// Validate
	if _, ok := data["name"]; !ok {
		return nil, fmt.Errorf("missing required field 'name' in body")
	}
	if _, ok := data["type"]; !ok {
		return nil, fmt.Errorf("missing required field 'type' in body")
	}
	if _, ok := data["domain"]; !ok {
		return nil, fmt.Errorf("missing required field 'domain' in body")
	}
	if _, ok := data["port"]; !ok {
		return nil, fmt.Errorf("missing required field 'port' in body")
	}

	resp, err := client.Post("/v1/service-sources", data)
	if err != nil {
		return nil, fmt.Errorf("failed to add service source: %w", err)
	}
	// res := make(map[string]interface{})

	return resp, nil
}

// add MCP server to higress console, example request body as followed:
//
//	{
//	  "name": "mcp-deepwiki",
//	  "description": "",
//	  "type": "DIRECT_ROUTE", // or OPEN_API
//	  "service": "hgctl-deepwiki.dns:443",
//	  "upstreamPathPrefix": "/mcp",
//	  "services": [
//	    {
//	      "name": "hgctl-deepwiki.dns",
//	      "port": 443,
//	      "version": "1.0",
//	      "weight": 100
//	    }
//	  ]
//	}
func HandleAddMCPServer(client *HigressClient, body interface{}) ([]byte, error) {
	data, ok := body.(map[string]interface{})
	if !ok {
		return nil, fmt.Errorf("failed to parse request body")
	}
	// Validate
	if _, ok := data["name"]; !ok {
		return nil, fmt.Errorf("missing required field 'name' in body")
	}
	if _, ok := data["type"]; !ok {
		return nil, fmt.Errorf("missing required field 'type' in body")
	}
	if _, ok := data["service"]; !ok {
		return nil, fmt.Errorf("missing required field 'service' in body")
	}

	// if _, ok := data["upstreamPathPrefix"]; !ok {
	// 	return nil, fmt.Errorf("missing required field 'upstreamPathPrefix' in body")
	// }

	_, ok = data["services"]
	if !ok {
		return nil, fmt.Errorf("missing required field 'port' in body")
	}

	resp, err := client.Put("/v1/mcpServer", data)
	if err != nil {
		return nil, fmt.Errorf("failed to add mcp server: %w", err)
	}

	return resp, nil
}

// add OpenAPI MCP tools to higress console, example request body:
//
//	{
//	  "id": null,
//	  "name": "openapi-name",
//	  "description": "123",
//	  "domains": [],
//	  "services": [
//	    {
//	      "name": "kubernetes.default.svc.cluster.local",
//	      "port": 443,
//	      "version": null,
//	      "weight": 100
//	    }
//	  ],
//	  "type": "OPEN_API",
//	  "consumerAuthInfo": {
//	    "type": "key-auth",
//	    "enable": false,
//	    "allowedConsumers": []
//	  },
//	  "rawConfigurations": "", // MCP configuration str
//	  "dsn": null,
//	  "dbType": null,
//	  "upstreamPathPrefix": null,
//	  "mcpServerName": "openapi-name"
//	}
func HandleAddOpenAPITool(client *HigressClient, body interface{}) ([]byte, error) {
	return client.Put("/v1/mcpServer", body)
}