import tkinter as tk
from tkinter import messagebox
from database import connect
from admin_menu import AdminMenu
from pembeli_menu import PembeliMenu
from kasir import KasirPage
from pemilik import PemilikPage

class LoginPage:
    def __init__(self, root):
        self.root = root
        self.frame = tk.Frame(root)
        self.frame.pack(expand=True)

        tk.Label(self.frame, text="LOGIN CAFE", font=("Arial",20,"bold")).pack(pady=10)

        self.user = tk.Entry(self.frame)
        self.passw = tk.Entry(self.frame, show="*")
        self.user.pack()
        self.passw.pack()

        tk.Button(self.frame, text="Login", command=self.login).pack(pady=5)

    def login(self):
        db = connect()
        cur = db.cursor()
        cur.execute("SELECT role FROM users WHERE username=? AND password=?",
                    (self.user.get(), self.passw.get()))
        res = cur.fetchone()
        db.close()

        if not res:
            messagebox.showerror("Error","Login gagal")
            return

        self.frame.destroy()
        role = res[0]

        if role=="admin":
            AdminMenu(self.root)
        elif role=="pembeli":
            PembeliMenu(self.root)
        elif role=="kasir":
            KasirPage(self.root)
        elif role=="pemilik":
            PemilikPage(self.root)