package handler

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

	"aiaa-notification-service/internal/adapter"
	"aiaa-notification-service/internal/model"
	"aiaa-notification-service/internal/safew"

	"github.com/gin-gonic/gin"
)

func TestListSafewChatsPOSTMissingToken(t *testing.T) {
	gin.SetMode(gin.TestMode)
	h := &ChannelHandler{chats: safew.NewWatcher(safew.NewMemStore(), func(string, int64, int) ([]adapter.SafewChat, int64, error) {
		return nil, 0, nil
	})}
	r := gin.New()
	r.POST("/api/v1/channels/safew/chats", h.ListSafewChats)
	req := httptest.NewRequest(http.MethodPost, "/api/v1/channels/safew/chats", bytes.NewReader([]byte(`{}`)))
	req.Header.Set("Content-Type", "application/json")
	w := httptest.NewRecorder()
	r.ServeHTTP(w, req)
	if w.Code != http.StatusBadRequest {
		t.Fatalf("code=%d body=%s", w.Code, w.Body.String())
	}
}

func TestListSafewChatsPOSTOk(t *testing.T) {
	gin.SetMode(gin.TestMode)
	st := safew.NewMemStore()
	_ = st.MergeSafewChats(nil, "tok", []adapter.SafewChat{{ID: "10000778141", Type: "group", Title: "测试AI"}})
	h := &ChannelHandler{chats: safew.NewWatcher(st, func(string, int64, int) ([]adapter.SafewChat, int64, error) {
		return nil, 0, nil
	})}
	r := gin.New()
	r.POST("/api/v1/channels/safew/chats", h.ListSafewChats)
	req := httptest.NewRequest(http.MethodPost, "/api/v1/channels/safew/chats", bytes.NewReader([]byte(`{"token":"tok","q":"测试"}`)))
	req.Header.Set("Content-Type", "application/json")
	w := httptest.NewRecorder()
	r.ServeHTTP(w, req)
	if w.Code != http.StatusOK {
		t.Fatalf("code=%d body=%s", w.Code, w.Body.String())
	}
	var resp struct {
		Data  []adapter.SafewChat `json:"data"`
		Total int                 `json:"total"`
	}
	if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil {
		t.Fatal(err)
	}
	if resp.Total != 1 || resp.Data[0].ID != "10000778141" {
		t.Fatalf("%#v", resp)
	}
}

func TestSafewTokenFromChannel(t *testing.T) {
	raw := json.RawMessage(`{"token":"abc","chat_id":"1"}`)
	ch := &model.Channel{Type: "safew", Config: &raw}
	tok, err := safewTokenFromChannel(ch)
	if err != nil || tok != "abc" {
		t.Fatalf("%q %v", tok, err)
	}
	ch.Type = "bark"
	if _, err := safewTokenFromChannel(ch); err == nil {
		t.Fatal("expected not safew")
	}
}