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 = [] # List untuk menyimpan belanjaan sementara
        self.images = [] # List untuk reference gambar agar tidak hilang

        # --- Header ---
        tk.Label(self.frame, text="SELAMAT DATANG", font=("Arial", 18, "bold")).pack(pady=10)
        
        # --- INPUT DATA PELANGGAN ---
        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 Belanja ---
        tk.Label(self.frame, text="Keranjang Pesanan:", font=("Arial", 10, "bold")).pack()
        self.listbox = tk.Listbox(self.frame, width=50, height=6)
        self.listbox.pack()

        # [BARU] Tombol Hapus Item
        tk.Button(self.frame, text="❌ Hapus Item Terpilih", fg="red", font=("Arial", 9), command=self.hapus_item_cart).pack(pady=2)

        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):
        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", bg="#cfe2ff", command=lambda n=nama, h=harga: self.add_to_cart(n, h)).pack(pady=2)

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

    # [BARU] Fungsi Logic Hapus
    def hapus_item_cart(self):
        # Ambil index item yang dipilih di listbox
        idx = self.listbox.curselection()
        
        if not idx:
            messagebox.showwarning("Peringatan", "Pilih menu di keranjang yang ingin dihapus!")
            return

        index = idx[0]
        # Hapus data dari list cart
        item_removed = self.cart.pop(index)
        
        # Update tampilan listbox dan total harga
        self.refresh_cart()
        # Opsional: Tampilkan pesan kecil di console atau status bar (disini saya skip biar tidak ganggu user)

    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()
        
        # Menyimpan nama pemesan di dalam string tanggal (Trik agar tidak ubah struktur DB)
        # Format: "2023-12-15 10:00:00 (Budi)"
        tanggal_custom = f"{tanggal} ({nama_pemesan})"

        cur.execute("INSERT INTO transaksi (tanggal, total, meja_id, status) VALUES (?, ?, ?, ?)",
                    (tanggal_custom, 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 ke Dapur!")
        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)