package reverseproxyapi

import (
	"context"
	"errors"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"github.com/itworx/pulse/internal/auth"
	"github.com/itworx/pulse/internal/reverseproxy"
)

type staticProvider struct {
	snapshot reverseproxy.Snapshot
	err      error
}

func (p staticProvider) Snapshot(context.Context) (reverseproxy.Snapshot, error) {
	return p.snapshot, p.err
}

func authenticatedRequest(method, path string) *http.Request {
	request := httptest.NewRequest(method, path, nil)
	return request.WithContext(auth.WithPrincipal(request.Context(), auth.Principal{Subject: "viewer", Role: auth.RoleViewer}))
}

func TestHandlerRequiresAuthAndReturnsDisabledSnapshot(t *testing.T) {
	handler := Handler{}
	unauthorized := httptest.NewRecorder()
	handler.ServeHTTP(unauthorized, httptest.NewRequest(http.MethodGet, "/api/v1/reverse-proxy", nil))
	if unauthorized.Code != http.StatusUnauthorized {
		t.Fatalf("unauthorized status=%d", unauthorized.Code)
	}
	response := httptest.NewRecorder()
	handler.ServeHTTP(response, authenticatedRequest(http.MethodGet, "/api/v1/reverse-proxy"))
	if response.Code != http.StatusOK || !strings.Contains(response.Body.String(), "\"state\":\"disabled\"") || !strings.Contains(response.Body.String(), "\"routes\":[]") {
		t.Fatalf("disabled response status=%d body=%s", response.Code, response.Body.String())
	}
}

func TestHandlerPreservesRouteSourceAndMapsProviderError(t *testing.T) {
	handler := Handler{Provider: staticProvider{snapshot: reverseproxy.Snapshot{Source: reverseproxy.Source{ID: "npm", State: reverseproxy.StateEnabled}, Routes: []reverseproxy.Route{{ID: "route-1", SourceID: "npm", Hostname: "pulse.example.test", URL: "https://pulse.example.test", TargetServiceID: "svc", Enabled: true}}, Total: 1}}}
	response := httptest.NewRecorder()
	handler.ServeHTTP(response, authenticatedRequest(http.MethodGet, "/api/v1/reverse-proxy"))
	if response.Code != http.StatusOK || !strings.Contains(response.Body.String(), "\"sourceId\":\"npm\"") {
		t.Fatalf("route response status=%d body=%s", response.Code, response.Body.String())
	}
	failed := Handler{Provider: staticProvider{err: errors.New("backend unavailable")}}
	failure := httptest.NewRecorder()
	failed.ServeHTTP(failure, authenticatedRequest(http.MethodGet, "/api/v1/reverse-proxy"))
	if failure.Code != http.StatusServiceUnavailable {
		t.Fatalf("failure status=%d", failure.Code)
	}
}