138 lines
3.6 KiB
Go
138 lines
3.6 KiB
Go
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
|
|
}
|
|
|
|
// delete room by id
|
|
func DeleteRoom(id int) error {
|
|
_, err := DB.Exec("DELETE FROM rooms WHERE id = ?", id)
|
|
return err
|
|
}
|