Enhance GetAllChats method to filter chats by user ID and add token v…

…alidation in GetChats handler

pkg/query/httphandler/controller/chat_controller.go

@@ -6,6 +6,7 @@ import (
 	"github.com/PRYVT/chats/pkg/models/query"
 	"github.com/PRYVT/chats/pkg/query/store/repository"
 	"github.com/PRYVT/chats/pkg/query/utils"
+	"github.com/PRYVT/utils/pkg/auth"
 	"github.com/PRYVT/utils/pkg/interfaces"
 	"github.com/gin-gonic/gin"
 )
@@ -41,7 +42,14 @@ func (ctrl *ChatController) GetChats(c *gin.Context) {
 	limit := utils.GetLimit(c)
 	offset := utils.GetOffset(c)
 
-	Chats, err := ctrl.ChatRepo.GetAllChats(limit, offset)
+	token := auth.GetTokenFromHeader(c)
+	userUuid, err := auth.GetUserUuidFromToken(token)
+	if err != nil {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": err.Error()})
+		return
+	}
+
+	Chats, err := ctrl.ChatRepo.GetAllChats(limit, offset, userUuid)
 	if err != nil {
 		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
 		return

pkg/query/store/repository/chats_repository.go

@@ -110,18 +110,19 @@ func (repo *ChatRepository) GetChatById(chatId uuid.UUID) (*models.Chat, error)
 	return &chat, nil
 }
 
-func (repo *ChatRepository) GetAllChats(limit, offset int) ([]models.ChatReduced, error) {
+func (repo *ChatRepository) GetAllChats(limit, offset int, userId uuid.UUID) ([]models.ChatReduced, error) {
 	stmt, err := repo.db.Prepare(`
 		SELECT id, name
 		FROM Chats
+		WHERE id IN (SELECT chat_id FROM Users WHERE user_id = ?)
 		LIMIT ? OFFSET ?
 	`)
 	if err != nil {
 		return nil, err
 	}
 	defer stmt.Close()
 
-	rows, err := stmt.Query(limit, offset)
+	rows, err := stmt.Query(limit, offset, userId.String())
 	if err != nil {
 		return nil, err
 	}