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
}