package v1

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

	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/suite"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"

	"github.com/gochat/gochat/internal/model"
	"github.com/gochat/gochat/internal/repository"
	"github.com/gochat/gochat/internal/service"
)

type PortalMemberHandlerTestSuite struct {
	suite.Suite
	db      *gorm.DB
	handler *PortalMemberHandler
	router  *gin.Engine
	account *model.Account
	portal  *model.Portal
	user    *model.User
}

func (s *PortalMemberHandlerTestSuite) SetupSuite() {
	s.db, _ = gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
	s.db.AutoMigrate(&model.PortalMember{}, &model.Portal{}, &model.Account{}, &model.User{})

	repo := repository.NewPortalMemberRepo(s.db)
	svc := service.NewPortalMemberService(repo)
	s.handler = NewPortalMemberHandler(svc)

	gin.SetMode(gin.TestMode)
	r := gin.New()

	accountGroup := r.Group("/api/v1/accounts/:account_id")
	portalGroup := accountGroup.Group("/portals/:portal_id/members")
	portalGroup.POST("", s.handler.Create)
	portalGroup.GET("/:member_id", s.handler.Get)
	portalGroup.PUT("/:member_id", s.handler.Update)
	portalGroup.DELETE("/:member_id", s.handler.Delete)
	portalGroup.GET("", s.handler.List)

	s.router = r

	s.account = &model.Account{Name: "TestAccount"}
	s.db.Create(s.account)

	s.portal = &model.Portal{AccountID: s.account.ID, Name: "TestPortal", Slug: "test-portal"}
	s.db.Create(s.portal)

	s.user = &model.User{Name: "TestUser", Email: "test@test.com"}
	s.db.Create(s.user)
}

func (s *PortalMemberHandlerTestSuite) TearDownSuite() {
	s.db.Exec("DELETE FROM portal_members")
	s.db.Exec("DELETE FROM portals")
	s.db.Exec("DELETE FROM accounts")
	s.db.Exec("DELETE FROM users")
}

func (s *PortalMemberHandlerTestSuite) SetupTest() {
	s.db.Exec("DELETE FROM portal_members")
}

func TestPortalMemberHandlerTestSuite(t *testing.T) {
	suite.Run(t, new(PortalMemberHandlerTestSuite))
}

func (s *PortalMemberHandlerTestSuite) TestCreate_Success() {
	body := fmt.Sprintf(`{"user_id":%d,"role":"reader"}`, s.user.ID)
	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodPost, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members", s.account.ID, s.portal.ID), bytes.NewBufferString(body))
	req.Header.Set("Content-Type", "application/json")
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusCreated, w.Code)

	var resp map[string]interface{}
	s.NoError(json.Unmarshal(w.Body.Bytes(), &resp))
	data := resp["data"].(map[string]interface{})
	s.Equal("reader", data["role"])
}

func (s *PortalMemberHandlerTestSuite) TestCreate_InvalidPortalID() {
	body := fmt.Sprintf(`{"user_id":%d,"role":"reader"}`, s.user.ID)
	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodPost, "/api/v1/accounts/1/portals/abc/members", bytes.NewBufferString(body))
	req.Header.Set("Content-Type", "application/json")
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusBadRequest, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestCreate_InvalidJSON() {
	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodPost, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members", s.account.ID, s.portal.ID), bytes.NewBufferString(`{invalid`))
	req.Header.Set("Content-Type", "application/json")
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusBadRequest, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestGet_Success() {
	pm := &model.PortalMember{PortalID: s.portal.ID, UserID: s.user.ID, Role: "editor"}
	s.db.Create(pm)

	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members/%d", s.account.ID, s.portal.ID, pm.ID), nil)
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusOK, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestGet_InvalidMemberID() {
	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodGet, "/api/v1/accounts/1/portals/1/members/abc", nil)
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusBadRequest, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestGet_NotFound() {
	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members/99999", s.account.ID, s.portal.ID), nil)
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusNotFound, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestUpdate_Success() {
	pm := &model.PortalMember{PortalID: s.portal.ID, UserID: s.user.ID, Role: "reader"}
	s.db.Create(pm)

	body := `{"role":"editor"}`
	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodPut, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members/%d", s.account.ID, s.portal.ID, pm.ID), bytes.NewBufferString(body))
	req.Header.Set("Content-Type", "application/json")
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusOK, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestUpdate_InvalidMemberID() {
	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodPut, "/api/v1/accounts/1/portals/1/members/abc", bytes.NewBufferString(`{"role":"editor"}`))
	req.Header.Set("Content-Type", "application/json")
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusBadRequest, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestDelete_Success() {
	pm := &model.PortalMember{PortalID: s.portal.ID, UserID: s.user.ID, Role: "reader"}
	s.db.Create(pm)

	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodDelete, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members/%d", s.account.ID, s.portal.ID, pm.ID), nil)
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusNoContent, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestDelete_InvalidMemberID() {
	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodDelete, "/api/v1/accounts/1/portals/1/members/abc", nil)
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusBadRequest, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestList_Success() {
	s.db.Create(&model.PortalMember{PortalID: s.portal.ID, UserID: s.user.ID, Role: "reader"})

	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members", s.account.ID, s.portal.ID), nil)
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusOK, w.Code)
}

func (s *PortalMemberHandlerTestSuite) TestList_Empty() {
	w := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodGet, fmt.Sprintf("/api/v1/accounts/%d/portals/%d/members", s.account.ID, s.portal.ID), nil)
	s.router.ServeHTTP(w, req)
	s.Equal(http.StatusOK, w.Code)
}