package openapi

import (
	"os"
	"testing"

	"go.yaml.in/yaml/v4"
)

type document struct {
	OpenAPI string                    `yaml:"openapi"`
	Paths   map[string]map[string]any `yaml:"paths"`
}

func TestDistributionContract(t *testing.T) {
	spec := readDocument(t, "proxy-pool.yaml")
	if spec.OpenAPI != "3.1.0" {
		t.Fatalf("openapi version = %q, want 3.1.0", spec.OpenAPI)
	}
	extraction, ok := spec.Paths["/api/v1/proxies/extract"]
	if !ok {
		t.Fatal("exclusive extraction path is missing")
	}
	post, ok := extraction["post"]
	if !ok {
		t.Fatal("exclusive extraction must use POST")
	}
	requireResponses(t, post, "200", "400", "409", "413", "415", "422", "429", "500", "503")
	for path := range spec.Paths {
		if path == "/api/v1/leases" || path == "/api/v1/proxies/release" || path == "/api/v1/proxies/renew" {
			t.Fatalf("lease/release path is forbidden: %s", path)
		}
	}
}

func TestAdminContract(t *testing.T) {
	spec := readDocument(t, "admin.yaml")
	for _, path := range []string{
		"/api/v1/status",
		"/api/v1/upstreams/{name}/enable",
		"/api/v1/upstreams/{name}/disable",
		"/api/v1/routing/{name}/switch",
		"/api/v1/config/reload",
	} {
		if _, ok := spec.Paths[path]; !ok {
			t.Errorf("admin path is missing: %s", path)
		}
	}
	requireResponses(t, spec.Paths["/api/v1/routing/{name}/switch"]["post"],
		"200", "400", "401", "403", "404", "405", "409", "413", "415", "422", "500", "503")
}

func requireResponses(t *testing.T, operation any, codes ...string) {
	t.Helper()
	operationMap, ok := operation.(map[string]any)
	if !ok {
		t.Fatalf("operation has type %T, want map", operation)
	}
	responses, ok := operationMap["responses"].(map[string]any)
	if !ok {
		t.Fatalf("responses has type %T, want map", operationMap["responses"])
	}
	for _, code := range codes {
		if _, ok := responses[code]; !ok {
			t.Errorf("response %s is missing", code)
		}
	}
}

func readDocument(t *testing.T, path string) document {
	t.Helper()
	content, err := os.ReadFile(path)
	if err != nil {
		t.Fatalf("read %s: %v", path, err)
	}
	var spec document
	if err := yaml.Unmarshal(content, &spec); err != nil {
		t.Fatalf("parse %s: %v", path, err)
	}
	if spec.OpenAPI != "3.1.0" {
		t.Fatalf("%s openapi version = %q, want 3.1.0", path, spec.OpenAPI)
	}
	return spec
}