import sqlite3

DB_NAME = "calendar.db"

conn = sqlite3.connect(DB_NAME)

# SQLite foreign keys activeren
conn.execute("PRAGMA foreign_keys = ON;")

cur = conn.cursor()


cur.execute("""
CREATE TABLE IF NOT EXISTS users (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    username TEXT NOT NULL UNIQUE,
    password_hash TEXT NOT NULL
);
""")


cur.execute("""
CREATE TABLE IF NOT EXISTS rooms (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    name TEXT NOT NULL,
    owner_id INTEGER NOT NULL,
    invite_code TEXT NOT NULL UNIQUE,

    FOREIGN KEY (owner_id) REFERENCES users(id)
);
""")


cur.execute("""
CREATE TABLE IF NOT EXISTS room_members (
    room_id INTEGER NOT NULL,
    user_id INTEGER NOT NULL,
    role TEXT NOT NULL DEFAULT 'member'
        CHECK(role IN ('owner', 'admin', 'member')),

    PRIMARY KEY (room_id, user_id),

    FOREIGN KEY (room_id)
        REFERENCES rooms(id)
        ON DELETE CASCADE,

    FOREIGN KEY (user_id)
        REFERENCES users(id)
        ON DELETE CASCADE
);
""")


cur.execute("""
CREATE TABLE IF NOT EXISTS events (
    id INTEGER PRIMARY KEY AUTOINCREMENT,

    room_id INTEGER NOT NULL,
    creator_id INTEGER NOT NULL,

    type TEXT NOT NULL
        CHECK(type IN ('private', 'busy', 'proposal')),

    title TEXT NOT NULL,
    description TEXT,

    start_time TEXT NOT NULL,
    end_time TEXT NOT NULL,

    visibility TEXT NOT NULL
        CHECK(visibility IN ('private', 'busy_only', 'room')),

    status TEXT NOT NULL DEFAULT 'draft'
        CHECK(status IN ('draft', 'proposed', 'confirmed', 'cancelled')),

    min_people INTEGER DEFAULT 0,

    FOREIGN KEY (room_id)
        REFERENCES rooms(id)
        ON DELETE CASCADE,

    FOREIGN KEY (creator_id)
        REFERENCES users(id)
);
""")


cur.execute("""
CREATE TABLE IF NOT EXISTS event_signups (
    event_id INTEGER NOT NULL,
    user_id INTEGER NOT NULL,

    status TEXT NOT NULL DEFAULT 'going'
        CHECK(status IN ('going', 'maybe', 'declined')),

    PRIMARY KEY(event_id, user_id),

    FOREIGN KEY(event_id)
        REFERENCES events(id)
        ON DELETE CASCADE,

    FOREIGN KEY(user_id)
        REFERENCES users(id)
        ON DELETE CASCADE
);
""")


# Snellere zoekopdrachten
cur.execute("""
CREATE INDEX IF NOT EXISTS idx_room_members_user
ON room_members(user_id);
""")

cur.execute("""
CREATE INDEX IF NOT EXISTS idx_events_room
ON events(room_id);
""")


conn.commit()


print("Database aangemaakt.")
print("Tabellen:")

for row in cur.execute(
    "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name;"
):
    print(" -", row[0])


conn.close()