package httpapi

import (
	"context"
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"testing"

	"data-terminal-cloud/internal/device"
	"data-terminal-cloud/internal/deviceauth"
)

type fakeDeviceService struct {
	bootstrap device.Bootstrap
	config    device.Configuration
}

func (s fakeDeviceService) ReportBootstrap(context.Context, deviceauth.Identity, device.BootstrapReport) (device.Bootstrap, error) {
	return s.bootstrap, nil
}

func (s fakeDeviceService) GetConfiguration(context.Context, deviceauth.Identity) (device.Configuration, error) {
	return s.config, nil
}

func TestBootstrapValidationAndResponse(t *testing.T) {
	service := fakeDeviceService{bootstrap: device.Bootstrap{DeviceID: "device-1", Status: "registered", ConfigRevision: 3}}
	handler := DeviceBootstrapHandler(service)
	badRequest := httptest.NewRequest(http.MethodGet, "/device/v2/bootstrap", nil)
	badRequest = badRequest.WithContext(deviceauth.WithIdentity(badRequest.Context(), deviceauth.Identity{PublicID: "device-1"}))
	badRecorder := httptest.NewRecorder()
	handler.ServeHTTP(badRecorder, badRequest)
	if badRecorder.Code != http.StatusBadRequest {
		t.Fatalf("validation status = %d", badRecorder.Code)
	}

	request := httptest.NewRequest(http.MethodGet, "/device/v2/bootstrap?firmware_version=1.0&device_time=2026-08-23T12%3A00%3A00Z&sample_rate=16000", nil)
	request = request.WithContext(deviceauth.WithIdentity(request.Context(), deviceauth.Identity{PublicID: "device-1"}))
	recorder := httptest.NewRecorder()
	handler.ServeHTTP(recorder, request)
	if recorder.Code != http.StatusOK || !json.Valid(recorder.Body.Bytes()) || !containsText(recorder.Body.String(), `"device_id":"device-1"`) {
		t.Fatalf("unexpected response: %d %s", recorder.Code, recorder.Body.String())
	}
}

func TestConfigUsesETagAndConditionalGet(t *testing.T) {
	service := fakeDeviceService{config: device.Configuration{DeviceID: "device-1", Revision: 3, DesiredConfig: json.RawMessage(`{"codec":"opus"}`)}}
	handler := DeviceConfigHandler(service)
	for _, test := range []struct {
		ifNoneMatch string
		status      int
	}{{"", http.StatusOK}, {`"v3"`, http.StatusNotModified}} {
		request := httptest.NewRequest(http.MethodGet, "/device/v2/config", nil)
		request.Header.Set("If-None-Match", test.ifNoneMatch)
		request = request.WithContext(deviceauth.WithIdentity(request.Context(), deviceauth.Identity{PublicID: "device-1"}))
		recorder := httptest.NewRecorder()
		handler.ServeHTTP(recorder, request)
		if recorder.Code != test.status || recorder.Header().Get("ETag") != `"v3"` {
			t.Fatalf("If-None-Match=%q status=%d etag=%q", test.ifNoneMatch, recorder.Code, recorder.Header().Get("ETag"))
		}
	}
}