Merge pull request #101 from RealTeamRocket/98-chat-room

98 chat room

Author: Pazl27

rocket-backend/go.mod

@@ -57,6 +57,7 @@ require (
 	github.com/goccy/go-json v0.10.5 // indirect
 	github.com/gogo/protobuf v1.3.2 // indirect
 	github.com/google/uuid v1.6.0
+	github.com/gorilla/websocket v1.5.3
 	github.com/jackc/pgpassfile v1.0.0 // indirect
 	github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
 	github.com/jackc/puddle/v2 v2.2.2 // indirect

rocket-backend/go.sum

@@ -82,6 +82,8 @@ github.com/google/pprof v0.0.0-20250403155104-27863c87afa6 h1:BHT72Gu3keYf3ZEu2J
 github.com/google/pprof v0.0.0-20250403155104-27863c87afa6/go.mod h1:boTsfXsheKC2y+lKOCMpSfarhxDeIzfZG1jqGcPl3cA=
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
+github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
 github.com/grpc-ecosystem/grpc-gateway/v2 v2.16.0 h1:YBftPWNWd4WwGqtY2yeZL2ef8rHAxPBD8KFhJpmcqms=
 github.com/grpc-ecosystem/grpc-gateway/v2 v2.16.0/go.mod h1:YN5jB8ie0yfIUg6VvR9Kz84aCaG7AsGZnLjhHbUqwPg=
 github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=

rocket-backend/integration-tests/mocks/database_mock.go

@@ -52,6 +52,11 @@ type MockDB struct {
 	GetChallengeByIDFunc               func(challengeID uuid.UUID) (*types.Challenge, error)
 	SaveActivityFunc                   func(userID uuid.UUID, message string) error
 	GetActivitiesForUserAndFriendsFunc func(userID uuid.UUID) ([]types.ActivityWithUser, error)
+	SaveChatMessageFunc                func(userID uuid.UUID, message string, timestamp string) (uuid.UUID, error)
+	GetChatMessagesFunc                func(userID uuid.UUID) ([]types.ChatMessage, error)
+	AddReactionToChatMessageFunc       func(userID uuid.UUID, messageID uuid.UUID) error
+	CountReactionsForMessageFunc       func(messageID uuid.UUID) (int, error)
+	GetIDByMessageIDFunc               func(messageID uuid.UUID) (uuid.UUID, error)
 }
 
 func (m *MockDB) ExecuteRawSQL(query string) (sql.Result, error) {
@@ -347,3 +352,38 @@ func (m *MockDB) DeleteRun(runID uuid.UUID) error {
 	}
 	return nil
 }
+
+func (m *MockDB) SaveChatMessage(userID uuid.UUID, message string, timestamp string) (uuid.UUID, error) {
+	if m.SaveChatMessageFunc != nil {
+		return m.SaveChatMessageFunc(userID, message, timestamp)
+	}
+	return uuid.Nil, nil
+}
+
+func (m *MockDB) GetChatMessages(userID uuid.UUID) ([]types.ChatMessage, error) {
+	if m.GetChatMessagesFunc != nil {
+		return m.GetChatMessagesFunc(userID)
+	}
+	return nil, nil
+}
+
+func (m *MockDB) AddReactionToChatMessage(userID uuid.UUID, messageID uuid.UUID) error {
+	if m.AddReactionToChatMessageFunc != nil {
+		return m.AddReactionToChatMessageFunc(userID, messageID)
+	}
+	return nil
+}
+
+func (m *MockDB) CountReactionsForMessage(messageID uuid.UUID) (int, error) {
+	if m.CountReactionsForMessageFunc != nil {
+		return m.CountReactionsForMessageFunc(messageID)
+	}
+	return 0, nil
+}
+
+func (m *MockDB) GetIDByMessageID(messageID uuid.UUID) (uuid.UUID, error) {
+	if m.GetIDByMessageIDFunc != nil {
+		return m.GetIDByMessageIDFunc(messageID)
+	}
+	return uuid.Nil, nil
+}

rocket-backend/internal/custom_error/database_error.go

@@ -12,5 +12,6 @@ var (
 	ErrChallengeNotFound    = errors.New("challenge not found")
 	ErrImageNotFound        = errors.New("image not found")
 	ErrSettingsNotFound     = errors.New("settings not found")
-	ErrFailedToDelete = errors.New("failed to delete data")
+	ErrFailedToDelete       = errors.New("failed to delete data")
+	ErrFailedToLoad         = errors.New("failed to load data")
 )

rocket-backend/internal/database/chat_messages_table.go

@@ -0,0 +1,104 @@
+package database
+
+import (
+	"fmt"
+	"rocket-backend/internal/custom_error"
+	"rocket-backend/internal/types"
+	"rocket-backend/pkg/logger"
+
+	"github.com/google/uuid"
+)
+
+func (s *service) SaveChatMessage(userID uuid.UUID, message string, timestamp string) (uuid.UUID, error) {
+	query := `
+		INSERT INTO chat_messages (user_id, message, timestamp)
+		VALUES ($1, $2, $3)
+		RETURNING id
+	`
+	var id uuid.UUID
+	err := s.db.QueryRow(query, userID, message, timestamp).Scan(&id)
+	if err != nil {
+		logger.Error("Failed to save chat message and return ID", err)
+		return uuid.Nil, fmt.Errorf("%w: %v", custom_error.ErrFailedToSave, err)
+	}
+	return id, nil
+}
+
+func (s *service) GetChatMessages(userID uuid.UUID) ([]types.ChatMessage, error) {
+	query := `
+		SELECT cm.id, cm.user_id, u.username, cm.message, cm.timestamp
+		FROM chat_messages cm
+		JOIN users u ON cm.user_id = u.id
+		ORDER BY cm.timestamp ASC
+	`
+	rows, err := s.db.Query(query)
+	if err != nil {
+		logger.Error("Failed to load chat messages", err)
+		return nil, fmt.Errorf("%w: %v", custom_error.ErrFailedToLoad, err)
+	}
+	defer rows.Close()
+
+	var messages []types.ChatMessage
+	for rows.Next() {
+		var msg types.ChatMessage
+		var dbUserID uuid.UUID
+		if err := rows.Scan(&msg.ID, &dbUserID, &msg.Username, &msg.Message, &msg.Timestamp); err != nil {
+			logger.Error("Failed to scan chat message", err)
+			continue
+		}
+		if dbUserID == userID {
+			msg.Username = "You"
+		}
+		// Count reactions for this message
+		count, err := s.CountReactionsForMessage(msg.ID)
+		if err != nil {
+			logger.Error("Failed to count reactions for message", err)
+			msg.Reactions = 0
+		} else {
+			msg.Reactions = count
+		}
+		// Check if the current user has reacted to this message
+		hasReacted := false
+		checkQuery := `SELECT 1 FROM chat_messages_reactions WHERE message_id = $1 AND user_id = $2 LIMIT 1`
+		row := s.db.QueryRow(checkQuery, msg.ID, userID)
+		var dummy int
+		if err := row.Scan(&dummy); err == nil {
+			hasReacted = true
+		}
+		msg.HasReacted = hasReacted
+		messages = append(messages, msg)
+	}
+	if err := rows.Err(); err != nil {
+		logger.Error("Error iterating chat messages", err)
+		return nil, fmt.Errorf("%w: %v", custom_error.ErrFailedToLoad, err)
+	}
+	return messages, nil
+}
+
+func (s *service) AddReactionToChatMessage(userID uuid.UUID, messageID uuid.UUID) error {
+	query := `
+        INSERT INTO chat_messages_reactions (message_id, user_id)
+        VALUES ($1, $2)
+        ON CONFLICT DO NOTHING
+    `
+    _, err := s.db.Exec(query, messageID, userID)
+    return err
+}
+
+func (s *service) CountReactionsForMessage(messageID uuid.UUID) (int, error) {
+    var count int
+    query := `SELECT COUNT(*) FROM chat_messages_reactions WHERE message_id = $1`
+    err := s.db.QueryRow(query, messageID).Scan(&count)
+    return count, err
+}
+
+func (s *service) GetIDByMessageID(messageID uuid.UUID) (uuid.UUID, error) {
+	query := `SELECT user_id FROM chat_messages WHERE id = $1`
+	var userID uuid.UUID
+	err := s.db.QueryRow(query, messageID).Scan(&userID)
+	if err != nil {
+		logger.Error("Failed to get user ID by message ID", err)
+		return uuid.Nil, fmt.Errorf("%w: %v", custom_error.ErrFailedToRetrieveData, err)
+	}
+	return userID, nil
+}

rocket-backend/internal/database/daily_steps_table.go

@@ -9,32 +9,37 @@ import (
 	"github.com/google/uuid"
 )
 
+func (s *service) saveStepMilestoneActivities(userID uuid.UUID, oldSteps, newSteps int) {
+	milestone := 2000
+	for threshold := ((oldSteps/milestone)+1)*milestone; threshold <= newSteps; threshold += milestone {
+		message := fmt.Sprintf("🎉 Has reached %d steps today! 🚀", threshold)
+		_ = s.SaveActivity(userID, message)
+	}
+}
+
 func (s *service) UpdateDailySteps(userID uuid.UUID, steps int) error {
 	currentDate := time.Now().Format("2006-01-02")
 
-	// Check if the user already has a record for today
 	var existingSteps int
 	queryCheck := `SELECT steps_taken FROM daily_steps WHERE user_id = $1 AND date = $2`
 	err := s.db.QueryRow(queryCheck, userID, currentDate).Scan(&existingSteps)
 
 	if err != nil {
-		// No record for today, insert new record and award rocket points
 		id := uuid.New()
 		queryInsert := `INSERT INTO daily_steps (id, user_id, steps_taken, date) VALUES ($1, $2, $3, $4)`
 		_, err := s.db.Exec(queryInsert, id, userID, steps, currentDate)
 		if err != nil {
-			// Log and return error if the insert fails
 			logger.Error("Error inserting daily steps: %v\n", err)
 			return fmt.Errorf("failed to insert daily steps: %w", err)
 		}
-		// Award rocket points for all steps (steps/10)
 		rocketPoints := steps / 10
 		if rocketPoints > 0 {
 			if err := s.UpdateRocketPoints(userID, rocketPoints); err != nil {
 				logger.Error("Error updating rocket points: %v\n", err)
 				return fmt.Errorf("failed to update rocket points: %w", err)
 			}
 		}
+		s.saveStepMilestoneActivities(userID, 0, steps)
 	} else {
 		// If an entry exists, update the steps and rocket points only if the new steps are greater than the existing steps
 		if steps > existingSteps {
@@ -43,18 +48,17 @@ func (s *service) UpdateDailySteps(userID uuid.UUID, steps int) error {
 				queryUpdate := `UPDATE daily_steps SET steps_taken = $1 WHERE user_id = $2 AND date = $3`
 				_, err := s.db.Exec(queryUpdate, steps, userID, currentDate)
 				if err != nil {
-					// Log and return error if the update fails
 					logger.Error("Error updating daily steps: %v\n", err)
 					return fmt.Errorf("failed to update daily steps: %w", err)
 				}
-				// Award rocket points for the difference in steps (stepDiff/10)
 				rocketPoints := stepDiff / 10
 				if rocketPoints > 0 {
 					if err := s.UpdateRocketPoints(userID, rocketPoints); err != nil {
 						logger.Error("Error updating rocket points: %v\n", err)
 						return fmt.Errorf("failed to update rocket points: %w", err)
 					}
 				}
+				s.saveStepMilestoneActivities(userID, existingSteps, steps)
 			}
 		}
 	}

rocket-backend/internal/database/database.go

@@ -84,6 +84,13 @@ type Service interface {
 	// activities
 	SaveActivity(userID uuid.UUID, message string) error
 	GetActivitiesForUserAndFriends(userID uuid.UUID) ([]types.ActivityWithUser, error)
+
+	// chat
+	SaveChatMessage(userID uuid.UUID, message string, timestamp string) (uuid.UUID, error)
+	GetChatMessages(userID uuid.UUID) ([]types.ChatMessage, error)
+	AddReactionToChatMessage(userID uuid.UUID, messageID uuid.UUID) error
+	CountReactionsForMessage(messageID uuid.UUID) (int, error)
+	GetIDByMessageID(messageID uuid.UUID) (uuid.UUID, error)
 }
 
 type service struct {

rocket-backend/internal/server/chat_handlers.go

@@ -0,0 +1,30 @@
+package server
+
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
+)
+
+func (s *Server) GetChatHistoryHandler(c *gin.Context) {
+	userID, exists := c.Get("userID")
+	if !exists {
+		c.JSON(http.StatusUnauthorized, gin.H{"error": "User not authenticated"})
+		return
+	}
+
+	userUUID, err := uuid.Parse(userID.(string))
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid user ID format"})
+		return
+	}
+
+	messages, err := s.db.GetChatMessages(userUUID)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch chat messages"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"messages": messages})
+}