import sqlite3

def connect():
    return sqlite3.connect("cafe.db")

def setup_database():
    db = connect()
    cur = db.cursor()

    # Table users
    cur.execute("""
    CREATE TABLE IF NOT EXISTS users (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        username TEXT,
        password TEXT,
        role TEXT
    )
    """)

    # Table menu (buat nanti)
    cur.execute("""
    CREATE TABLE IF NOT EXISTS menu (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        nama TEXT,
        kategori TEXT,
        harga REAL,
        stok INTEGER
    )
    """)

    # Tambah user default
    cur.execute("SELECT COUNT(*) FROM users")
    if cur.fetchone()[0] == 0:
        users = [
            ("admin", "admin", "admin"),
            ("kasir", "kasir", "kasir"),
            ("waiter", "waiter", "waiter"),
            ("pembeli", "pembeli", "pembeli"),
            ("pemilik", "pemilik", "pemilik cafe")
        ]
        cur.executemany("INSERT INTO users(username,password,role) VALUES (?,?,?)", users)
        db.commit()

    db.close()