package scripts

import (
	"encoding/json"
	"errors"
	"net"
	"net/http"
	"net/http/httptest"
	"net/url"
	"os"
	"os/exec"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestHealthCheckJSONExitCodes(t *testing.T) {
	tests := []struct {
		name     string
		status   func(string) int
		exitCode int
	}{
		{name: "healthy", status: func(string) int { return http.StatusOK }, exitCode: 0},
		{name: "degraded", status: func(path string) int {
			if path == "/metrics" {
				return http.StatusServiceUnavailable
			}
			return http.StatusOK
		}, exitCode: 1},
		{name: "unhealthy", status: func(path string) int {
			if path == "/live" || path == "/ready" {
				return http.StatusServiceUnavailable
			}
			return http.StatusOK
		}, exitCode: 2},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				w.WriteHeader(test.status(r.URL.Path))
			}))
			defer server.Close()
			output, exitCode := runHealthScript(t, server.URL, "--json", "--full")
			require.Equal(t, test.exitCode, exitCode)
			var payload map[string]any
			require.NoError(t, json.Unmarshal(output, &payload), string(output))
		})
	}
}

func TestHealthCheckJSONInvalidInvocation(t *testing.T) {
	command := exec.Command("./health_check.sh", "--json", "--timeout")
	output, err := command.Output()
	var exitErr *exec.ExitError
	require.True(t, errors.As(err, &exitErr))
	require.Equal(t, 2, exitErr.ExitCode())
	require.JSONEq(t, `{"error":"--timeout requires a positive integer","checks":[],"summary":{"critical_failures":1,"degraded_failures":0}}`, string(output))
}

func runHealthScript(t *testing.T, rawURL string, args ...string) ([]byte, int) {
	t.Helper()
	parsed, err := url.Parse(rawURL)
	require.NoError(t, err)
	host, port, err := net.SplitHostPort(parsed.Host)
	require.NoError(t, err)
	command := exec.Command("./health_check.sh", args...)
	command.Env = append(os.Environ(), "GOCHAT_HOST="+host, "GOCHAT_PORT="+port, "GOCHAT_METRICS_PORT="+port)
	output, err := command.Output()
	if err == nil {
		return output, 0
	}
	var exitErr *exec.ExitError
	require.True(t, errors.As(err, &exitErr), "health script failed to execute: %v", err)
	return output, exitErr.ExitCode()
}