package v1

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

	"github.com/gin-gonic/gin"
	"github.com/gochat/gochat/internal/model"
	"github.com/gochat/gochat/internal/repository"
	"github.com/gochat/gochat/internal/service"
	"github.com/stretchr/testify/require"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

func setupNotificationSubscriptionHandlerTest(t *testing.T, userID uint) (*gin.Engine, *gorm.DB) {
	t.Helper()
	gin.SetMode(gin.TestMode)
	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
	require.NoError(t, err)
	require.NoError(t, db.AutoMigrate(&model.User{}, &model.NotificationSubscription{}))

	handler := NewNotificationSubscriptionHandler(service.NewNotificationSubscriptionService(repository.NewNotificationSubscriptionRepo(db)))
	router := gin.New()
	router.Use(func(c *gin.Context) {
		c.Set("user_id", userID)
		c.Next()
	})
	router.POST("/api/v1/notification_subscriptions", handler.Create)
	router.DELETE("/api/v1/notification_subscriptions", handler.Destroy)
	router.DELETE("/api/v1/notification_subscriptions/:identifier", handler.Destroy)
	return router, db
}

func TestNotificationSubscriptionCreateAcceptsFrontendPayload(t *testing.T) {
	router, db := setupNotificationSubscriptionHandlerTest(t, 7)
	body := `{"subscription_type":"browser_push","subscription_attributes":{"endpoint":"https://push.example/sub","p256dh":"key","auth":"secret"}}`

	req := httptest.NewRequest(http.MethodPost, "/api/v1/notification_subscriptions", bytes.NewBufferString(body))
	req.Header.Set("Content-Type", "application/json")
	w := httptest.NewRecorder()
	router.ServeHTTP(w, req)

	require.Equal(t, http.StatusOK, w.Code)
	var payload map[string]any
	require.NoError(t, json.Unmarshal(w.Body.Bytes(), &payload), w.Body.String())
	require.NotContains(t, payload, "success")
	require.NotContains(t, payload, "data")
	require.Equal(t, "https://push.example/sub", payload["identifier"])
	require.Equal(t, "browser_push", payload["subscription_type"])
	require.Equal(t, float64(7), payload["user_id"])

	var sub model.NotificationSubscription
	require.NoError(t, db.First(&sub).Error)
	require.Equal(t, uint(7), sub.UserID)
	require.Equal(t, "https://push.example/sub", sub.Identifier)
}

func TestNotificationSubscriptionCreateAcceptsRailsWrapperAndUpdatesExisting(t *testing.T) {
	router, db := setupNotificationSubscriptionHandlerTest(t, 11)
	existing := model.NotificationSubscription{
		Identifier:             "device-1",
		UserID:                 2,
		SubscriptionType:       model.NotificationSubFCM,
		SubscriptionAttributes: json.RawMessage(`{"device_id":"device-1","push_token":"old"}`),
	}
	require.NoError(t, db.Create(&existing).Error)
	body := `{"notification_subscription":{"subscription_type":"fcm","subscription_attributes":{"device_id":"device-1","push_token":"new"}}}`

	req := httptest.NewRequest(http.MethodPost, "/api/v1/notification_subscriptions", bytes.NewBufferString(body))
	req.Header.Set("Content-Type", "application/json")
	w := httptest.NewRecorder()
	router.ServeHTTP(w, req)

	require.Equal(t, http.StatusOK, w.Code)
	var sub model.NotificationSubscription
	require.NoError(t, db.First(&sub, existing.ID).Error)
	require.Equal(t, uint(11), sub.UserID)
	require.JSONEq(t, `{"device_id":"device-1","push_token":"new"}`, string(sub.SubscriptionAttributes))
}

func TestNotificationSubscriptionDestroyUsesPushTokenAndReturnsEmptyOK(t *testing.T) {
	router, db := setupNotificationSubscriptionHandlerTest(t, 7)
	sub := model.NotificationSubscription{
		Identifier:             "https://push.example/sub",
		UserID:                 7,
		SubscriptionType:       model.NotificationSubBrowserPush,
		SubscriptionAttributes: json.RawMessage(`{"endpoint":"https://push.example/sub","p256dh":"key","auth":"secret"}`),
	}
	require.NoError(t, db.Create(&sub).Error)

	req := httptest.NewRequest(http.MethodDelete, "/api/v1/notification_subscriptions?push_token=https%3A%2F%2Fpush.example%2Fsub", nil)
	w := httptest.NewRecorder()
	router.ServeHTTP(w, req)

	require.Equal(t, http.StatusOK, w.Code)
	require.Empty(t, w.Body.String())
	var count int64
	require.NoError(t, db.Model(&model.NotificationSubscription{}).Where("id = ?", sub.ID).Count(&count).Error)
	require.Equal(t, int64(0), count)
}

func TestNotificationSubscriptionDestroyMissingTokenStillOK(t *testing.T) {
	router, _ := setupNotificationSubscriptionHandlerTest(t, 7)

	req := httptest.NewRequest(http.MethodDelete, "/api/v1/notification_subscriptions", nil)
	w := httptest.NewRecorder()
	router.ServeHTTP(w, req)

	require.Equal(t, http.StatusOK, w.Code)
	require.Empty(t, w.Body.String())
}