Refactor

lib/room.go

@@ -1,6 +1,8 @@
 package lib
 
 import (
+	"fmt"
+	"strings"
 	"time"
 
 	"github.com/google/uuid"
@@ -22,6 +24,11 @@ type RoomMember struct {
 	JoinedAt time.Time
 }
 
+type RoomSummary struct {
+	Room
+	MemberCount int
+}
+
 // generate unique room code
 func generateRoomCode() string {
 	code := uuid.New().String()[:8]
@@ -31,19 +38,42 @@ func generateRoomCode() string {
 // create room
 func CreateRoom(name string, ownerID int, scale string) (*Room, error) {
 	code := generateRoomCode()
-	_, err := DB.Exec("INSERT INTO rooms (name, code, scale, owner_id, created_at, active_story_id) VALUES (?, ?, ?, ?, ?, NULL)", name, code, scale, ownerID, time.Now())
+
+	tx, err := DB.Begin()
 	if err != nil {
 		return nil, err
 	}
-	room := &Room{Name: name, Code: code, Scale: scale, OwnerID: ownerID, ActiveStoryID: nil}
-	row := DB.QueryRow("SELECT id, name, code, scale, owner_id, created_at, active_story_id FROM rooms WHERE code = ?", code)
-	err = row.Scan(&room.ID, &room.Name, &room.Code, &room.Scale, &room.OwnerID, &room.CreatedAt, &room.ActiveStoryID)
+	defer tx.Rollback()
+
+	createdAt := time.Now()
+	res, err := tx.Exec("INSERT INTO rooms (name, code, scale, owner_id, created_at, active_story_id) VALUES (?, ?, ?, ?, ?, NULL)", name, code, scale, ownerID, createdAt)
 	if err != nil {
 		return nil, err
 	}
-	// add owner as member
-	DB.Exec("INSERT INTO room_members (room_id, user_id, joined_at) VALUES (?, ?, ?)", room.ID, ownerID, time.Now())
-	return room, nil
+
+	roomID64, err := res.LastInsertId()
+	if err != nil {
+		return nil, err
+	}
+	roomID := int(roomID64)
+
+	if _, err := tx.Exec("INSERT INTO room_members (room_id, user_id, joined_at) VALUES (?, ?, ?)", roomID, ownerID, createdAt); err != nil {
+		return nil, err
+	}
+
+	if err := tx.Commit(); err != nil {
+		return nil, err
+	}
+
+	return &Room{
+		ID:            roomID,
+		Name:          name,
+		Code:          code,
+		Scale:         scale,
+		OwnerID:       ownerID,
+		CreatedAt:     createdAt,
+		ActiveStoryID: nil,
+	}, nil
 }
 
 // get room by id
@@ -70,6 +100,18 @@ func GetRoomByCode(code string) (*Room, error) {
 
 // get rooms for user
 func GetRoomsForUser(userID int) ([]Room, error) {
+	summaries, err := GetRoomSummariesForUser(userID)
+	if err != nil {
+		return nil, err
+	}
+	rooms := make([]Room, 0, len(summaries))
+	for _, s := range summaries {
+		rooms = append(rooms, s.Room)
+	}
+	return rooms, nil
+}
+
+func GetRoomSummariesForUser(userID int) ([]RoomSummary, error) {
 	rows, err := DB.Query(`
 		SELECT r.id, r.name, r.code, r.scale, r.owner_id, r.created_at, r.active_story_id, COUNT(rm.user_id) as member_count
 		FROM rooms r
@@ -81,14 +123,15 @@ func GetRoomsForUser(userID int) ([]Room, error) {
 		return nil, err
 	}
 	defer rows.Close()
-	var rooms []Room
+	var rooms []RoomSummary
 	for rows.Next() {
-		var r Room
+		var r RoomSummary
 		var memberCount int
 		err := rows.Scan(&r.ID, &r.Name, &r.Code, &r.Scale, &r.OwnerID, &r.CreatedAt, &r.ActiveStoryID, &memberCount)
 		if err != nil {
 			return nil, err
 		}
+		r.MemberCount = memberCount
 		rooms = append(rooms, r)
 	}
 	return rooms, nil
@@ -135,3 +178,71 @@ func DeleteRoom(id int) error {
 	_, err := DB.Exec("DELETE FROM rooms WHERE id = ?", id)
 	return err
 }
+
+func GetVotesForStories(storyIDs []int) (map[int][]Vote, error) {
+	votesByStory := make(map[int][]Vote)
+	if len(storyIDs) == 0 {
+		return votesByStory, nil
+	}
+
+	placeholders := strings.TrimSuffix(strings.Repeat("?,", len(storyIDs)), ",")
+	args := make([]interface{}, 0, len(storyIDs))
+	for _, id := range storyIDs {
+		args = append(args, id)
+	}
+
+	query := fmt.Sprintf("SELECT story_id, user_id, value FROM votes WHERE story_id IN (%s)", placeholders)
+	rows, err := DB.Query(query, args...)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	for rows.Next() {
+		var v Vote
+		if err := rows.Scan(&v.StoryID, &v.UserID, &v.Value); err != nil {
+			return nil, err
+		}
+		votesByStory[v.StoryID] = append(votesByStory[v.StoryID], v)
+	}
+
+	return votesByStory, nil
+}
+
+func GetVoteViewsForStories(storyIDs []int) (map[int][]VoteView, error) {
+	voteViewsByStory := make(map[int][]VoteView)
+	if len(storyIDs) == 0 {
+		return voteViewsByStory, nil
+	}
+
+	placeholders := strings.TrimSuffix(strings.Repeat("?,", len(storyIDs)), ",")
+	args := make([]interface{}, 0, len(storyIDs))
+	for _, id := range storyIDs {
+		args = append(args, id)
+	}
+
+	query := fmt.Sprintf(`
+		SELECT v.story_id, u.username, v.value
+		FROM votes v
+		JOIN users u ON v.user_id = u.id
+		WHERE v.story_id IN (%s)
+	`, placeholders)
+	rows, err := DB.Query(query, args...)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	for rows.Next() {
+		var (
+			storyID int
+			vv      VoteView
+		)
+		if err := rows.Scan(&storyID, &vv.Username, &vv.Value); err != nil {
+			return nil, err
+		}
+		voteViewsByStory[storyID] = append(voteViewsByStory[storyID], vv)
+	}
+
+	return voteViewsByStory, nil
+}