package internalhttp

import (
	"context"
	"encoding/json"
	"io"
	"net"
	"net/http"
	"testing"
	"time"

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

func TestNewServerRejectsInvalidConfiguration(t *testing.T) {
	t.Parallel()

	cfg := Config{
		ReadHeaderTimeout: time.Second,
		ReadTimeout:       time.Second,
		IdleTimeout:       time.Second,
	}

	_, err := NewServer(cfg, Dependencies{})
	require.Error(t, err)
	assert.Contains(t, err.Error(), "addr")
}

func TestServerRunAndShutdown(t *testing.T) {
	t.Parallel()

	cfg := testConfig(t)
	server, err := NewServer(cfg, Dependencies{})
	require.NoError(t, err)

	runErr := make(chan error, 1)
	go func() {
		runErr <- server.Run(context.Background())
	}()

	client := newTestHTTPClient(t)
	waitForReservedRouteReady(t, client, cfg.Addr)

	shutdownCtx, cancel := context.WithTimeout(context.Background(), time.Second)
	defer cancel()
	require.NoError(t, server.Shutdown(shutdownCtx))
	waitForServerRunResult(t, runErr)
}

func TestReservedRoutesReturnStableServiceUnavailable(t *testing.T) {
	t.Parallel()

	cfg := testConfig(t)
	server, err := NewServer(cfg, Dependencies{})
	require.NoError(t, err)

	runErr := make(chan error, 1)
	go func() {
		runErr <- server.Run(context.Background())
	}()

	client := newTestHTTPClient(t)
	waitForReservedRouteReady(t, client, cfg.Addr)

	tests := []struct {
		method string
		path   string
	}{
		{method: http.MethodPost, path: LoginCodeDeliveriesPath},
		{method: http.MethodGet, path: DeliveriesPath},
		{method: http.MethodGet, path: "/api/v1/internal/deliveries/delivery-123"},
		{method: http.MethodGet, path: "/api/v1/internal/deliveries/delivery-123/attempts"},
		{method: http.MethodPost, path: "/api/v1/internal/deliveries/delivery-123/resend"},
	}

	for _, tt := range tests {
		tt := tt

		t.Run(tt.method+" "+tt.path, func(t *testing.T) {
			request, err := http.NewRequest(tt.method, "http://"+cfg.Addr+tt.path, nil)
			require.NoError(t, err)

			response, err := client.Do(request)
			require.NoError(t, err)
			defer response.Body.Close()

			require.Equal(t, http.StatusServiceUnavailable, response.StatusCode)
			require.Equal(t, "application/json", response.Header.Get("Content-Type"))

			var payload ErrorResponse
			require.NoError(t, json.NewDecoder(response.Body).Decode(&payload))
			require.Equal(t, ErrorCodeServiceUnavailable, payload.Error.Code)
			require.Equal(t, "service is unavailable", payload.Error.Message)
		})
	}

	shutdownCtx, cancel := context.WithTimeout(context.Background(), time.Second)
	defer cancel()
	require.NoError(t, server.Shutdown(shutdownCtx))
	waitForServerRunResult(t, runErr)
}

func TestServerDoesNotExposeProbeOrUnknownRoutes(t *testing.T) {
	t.Parallel()

	cfg := testConfig(t)
	server, err := NewServer(cfg, Dependencies{})
	require.NoError(t, err)

	runErr := make(chan error, 1)
	go func() {
		runErr <- server.Run(context.Background())
	}()

	client := newTestHTTPClient(t)
	waitForReservedRouteReady(t, client, cfg.Addr)

	for _, path := range []string{"/healthz", "/readyz", "/metrics", "/unknown"} {
		request, err := http.NewRequest(http.MethodGet, "http://"+cfg.Addr+path, nil)
		require.NoError(t, err)

		response, err := client.Do(request)
		require.NoError(t, err)
		_, _ = io.ReadAll(response.Body)
		response.Body.Close()

		assert.Equalf(t, http.StatusNotFound, response.StatusCode, "path %s", path)
	}

	shutdownCtx, cancel := context.WithTimeout(context.Background(), time.Second)
	defer cancel()
	require.NoError(t, server.Shutdown(shutdownCtx))
	waitForServerRunResult(t, runErr)
}

func testConfig(t *testing.T) Config {
	t.Helper()

	return Config{
		Addr:              mustFreeAddr(t),
		ReadHeaderTimeout: time.Second,
		ReadTimeout:       2 * time.Second,
		IdleTimeout:       time.Minute,
	}
}

func newTestHTTPClient(t *testing.T) *http.Client {
	t.Helper()

	transport := &http.Transport{DisableKeepAlives: true}
	t.Cleanup(transport.CloseIdleConnections)

	return &http.Client{
		Timeout:   250 * time.Millisecond,
		Transport: transport,
	}
}

func waitForReservedRouteReady(t *testing.T, client *http.Client, addr string) {
	t.Helper()

	require.Eventually(t, func() bool {
		request, err := http.NewRequest(http.MethodPost, "http://"+addr+LoginCodeDeliveriesPath, nil)
		if err != nil {
			return false
		}

		response, err := client.Do(request)
		if err != nil {
			return false
		}
		defer response.Body.Close()
		_, _ = io.ReadAll(response.Body)

		return response.StatusCode == http.StatusServiceUnavailable
	}, 5*time.Second, 25*time.Millisecond, "internal HTTP server did not become reachable")
}

func waitForServerRunResult(t *testing.T, runErr <-chan error) {
	t.Helper()

	var err error
	require.Eventually(t, func() bool {
		select {
		case err = <-runErr:
			return true
		default:
			return false
		}
	}, 5*time.Second, 10*time.Millisecond, "internal HTTP server did not stop")
	require.NoError(t, err)
}

func mustFreeAddr(t *testing.T) string {
	t.Helper()

	listener, err := net.Listen("tcp", "127.0.0.1:0")
	require.NoError(t, err)
	defer func() {
		assert.NoError(t, listener.Close())
	}()

	return listener.Addr().String()
}