import tkinter as tk
from tkinter import messagebox
from PIL import Image, ImageTk
from database import connect
import datetime

class PembeliMenu:
    def __init__(self, parent):
        self.parent = parent
        self.frame = tk.Frame(parent)
        self.frame.pack(fill="both", expand=True)
        
        self.cart = []
        self.images = []

        # Header
        tk.Label(self.frame, text="SELAMAT DATANG", font=("Arial", 18, "bold")).pack(pady=10)
        
        # --- INPUT DATA PELANGGAN (Penting karena tanpa login) ---
        info_frame = tk.Frame(self.frame, bd=2, relief="groove", padx=10, pady=10)
        info_frame.pack(pady=5)

        tk.Label(info_frame, text="Nama Anda:").grid(row=0, column=0, sticky="e")
        self.entry_nama = tk.Entry(info_frame)
        self.entry_nama.grid(row=0, column=1, padx=5)

        tk.Label(info_frame, text="No. Meja:").grid(row=0, column=2, sticky="e")
        self.entry_meja = tk.Entry(info_frame, width=5)
        self.entry_meja.grid(row=0, column=3, padx=5)

        # Tombol Kembali
        tk.Button(self.frame, text="Kembali ke Utama", bg="#f9e79f", command=self.logout).pack(pady=5)

        # Area Menu
        self.menu_frame = tk.Frame(self.frame)
        self.menu_frame.pack(pady=10, fill="both", expand=True)
        self.load_menu()

        # Keranjang
        tk.Label(self.frame, text="Keranjang Pesanan:").pack()
        self.listbox = tk.Listbox(self.frame, width=50, height=6)
        self.listbox.pack()

        self.total_lbl = tk.Label(self.frame, text="Total: Rp 0", font=("Arial", 12, "bold"))
        self.total_lbl.pack(pady=5)

        tk.Button(self.frame, text="KIRIM PESANAN", bg="#d1e7dd", font=("Arial", 11, "bold"), command=self.checkout).pack(pady=10)

    def load_menu(self):
        # (SAMA SEPERTI KODE SEBELUMNYA)
        db = connect()
        cur = db.cursor()
        cur.execute("SELECT nama, harga, gambar FROM menu")
        data = cur.fetchall()
        db.close()

        row_frame = None
        for i, (nama, harga, gambar) in enumerate(data):
            if i % 3 == 0:
                row_frame = tk.Frame(self.menu_frame)
                row_frame.pack()
            
            f = tk.Frame(row_frame, bd=2, relief="ridge", padx=5, pady=5)
            f.pack(side=tk.LEFT, padx=5)

            try:
                img = Image.open(gambar).resize((100, 80))
                photo = ImageTk.PhotoImage(img)
                self.images.append(photo)
                tk.Label(f, image=photo).pack()
            except:
                tk.Label(f, text="[No Image]").pack()

            tk.Label(f, text=nama, font=("Arial",10,"bold")).pack()
            tk.Label(f, text=f"Rp {harga:,.0f}").pack()
            tk.Button(f, text="Tambah", command=lambda n=nama, h=harga: self.add_to_cart(n, h)).pack()

    def add_to_cart(self, nama, harga):
        self.cart.append({'nama': nama, 'harga': harga})
        self.refresh_cart()

    def refresh_cart(self):
        self.listbox.delete(0, tk.END)
        total = 0
        for item in self.cart:
            self.listbox.insert(tk.END, f"{item['nama']} - Rp {item['harga']:,.0f}")
            total += item['harga']
        self.total_lbl.config(text=f"Total: Rp {total:,.0f}")

    def checkout(self):
        nama_pemesan = self.entry_nama.get()
        meja = self.entry_meja.get()

        if not nama_pemesan or not meja:
            messagebox.showwarning("Data Kurang", "Mohon isi Nama dan Nomor Meja!")
            return
        if not self.cart:
            messagebox.showwarning("Kosong", "Pilih menu dulu!")
            return

        total_belanja = sum(item['harga'] for item in self.cart)
        tanggal = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")

        db = connect()
        cur = db.cursor()
        
        # Simpan Transaksi dengan Nama Pemesan (Saya perlu update database sedikit untuk simpan nama pemesan di tabel transaksi, atau tempel di header struk)
        # Untuk simplifikasi, kita simpan nama pemesan di kolom 'status' sementara atau buat kolom baru. 
        # TAPI, agar tidak mengubah database lagi, kita anggap Nama Pemesan hanya untuk display Waiter/Kasir.
        
        # Opsi terbaik: Masukkan nama pemesan ke tabel transaksi (Perlu update database.py sedikit lagi agar sempurna)
        # Jika malas ubah database.py, kita gabung di tanggal atau status.
        # Mari kita lakukan yang benar: Update query INSERT untuk mendukung kolom nama_pelanggan jika ada, 
        # TAPI karena kamu sudah setup DB, kita tempel di note saja.
        
        # Agar simple dan tidak error di DB yang sudah kamu buat, kita kirim pesan sukses saja dulu.
        cur.execute("INSERT INTO transaksi (tanggal, total, meja_id, status) VALUES (?, ?, ?, ?)",
                    (f"{tanggal} ({nama_pemesan})", total_belanja, meja, "Pending"))
        
        transaksi_id = cur.lastrowid

        for item in self.cart:
            cur.execute("INSERT INTO detail_transaksi (transaksi_id, menu_nama, harga, jumlah, subtotal) VALUES (?, ?, ?, ?, ?)",
                        (transaksi_id, item['nama'], item['harga'], 1, item['harga']))
        
        db.commit()
        db.close()

        messagebox.showinfo("Berhasil", f"Pesanan a.n {nama_pemesan} berhasil dikirim!")
        self.cart.clear()
        self.refresh_cart()
        self.entry_nama.delete(0, tk.END)
        self.entry_meja.delete(0, tk.END)

    def logout(self):
        self.frame.destroy()
        from main import LoginScreen
        LoginScreen(self.parent)