Basic features done

lib/room.go

@@ -0,0 +1,131 @@
+package lib
+
+import (
+	"time"
+
+	"github.com/google/uuid"
+)
+
+type Room struct {
+	ID            int
+	Name          string
+	Code          string
+	Scale         string
+	OwnerID       int
+	CreatedAt     time.Time
+	ActiveStoryID *int
+}
+
+type RoomMember struct {
+	RoomID   int
+	UserID   int
+	JoinedAt time.Time
+}
+
+// generate unique room code
+func generateRoomCode() string {
+	code := uuid.New().String()[:8]
+	return code
+}
+
+// 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())
+	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)
+	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
+}
+
+// get room by id
+func GetRoomByID(id int) (*Room, error) {
+	row := DB.QueryRow("SELECT id, name, code, scale, owner_id, created_at, active_story_id FROM rooms WHERE id = ?", id)
+	r := &Room{}
+	err := row.Scan(&r.ID, &r.Name, &r.Code, &r.Scale, &r.OwnerID, &r.CreatedAt, &r.ActiveStoryID)
+	if err != nil {
+		return nil, err
+	}
+	return r, nil
+}
+
+// get room by code
+func GetRoomByCode(code string) (*Room, error) {
+	row := DB.QueryRow("SELECT id, name, code, scale, owner_id, created_at, active_story_id FROM rooms WHERE code = ?", code)
+	r := &Room{}
+	err := row.Scan(&r.ID, &r.Name, &r.Code, &r.Scale, &r.OwnerID, &r.CreatedAt, &r.ActiveStoryID)
+	if err != nil {
+		return nil, err
+	}
+	return r, nil
+}
+
+// get rooms for user
+func GetRoomsForUser(userID int) ([]Room, 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
+		LEFT JOIN room_members rm ON r.id = rm.room_id
+		WHERE r.owner_id = ?
+		GROUP BY r.id
+	`, userID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var rooms []Room
+	for rows.Next() {
+		var r Room
+		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
+		}
+		rooms = append(rooms, r)
+	}
+	return rooms, nil
+}
+
+// set active story for room
+func SetActiveStory(roomID, storyID int) error {
+	_, err := DB.Exec("UPDATE rooms SET active_story_id = ? WHERE id = ?", storyID, roomID)
+	return err
+}
+
+// add user to room
+func AddUserToRoom(roomID, userID int) error {
+	_, err := DB.Exec("INSERT OR IGNORE INTO room_members (room_id, user_id, joined_at) VALUES (?, ?, ?)", roomID, userID, time.Now())
+	return err
+}
+
+// get room members
+func GetRoomMembers(roomID int) ([]User, error) {
+	rows, err := DB.Query(`
+		SELECT u.id, u.username, u.password_hash
+		FROM users u
+		JOIN room_members rm ON u.id = rm.user_id
+		WHERE rm.room_id = ?
+	`, roomID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var members []User
+	for rows.Next() {
+		var u User
+		err := rows.Scan(&u.ID, &u.Username, &u.PasswordHash)
+		if err != nil {
+			return nil, err
+		}
+		members = append(members, u)
+	}
+	return members, nil
+}