import tkinter as tk
from tkinter import filedialog, messagebox
import re
import threading

# ---------------- File dialogs ----------------

def browse_input():
    path = filedialog.askopenfilename(
        filetypes=[("ULP files", "*.ulp"), ("All files", "*.*")]
    )
    if path:
        input_entry.delete(0, tk.END)
        input_entry.insert(0, path)

def browse_output():
    path = filedialog.asksaveasfilename(
        defaultextension=".txt",
        filetypes=[("Text files", "*.txt")]
    )
    if path:
        output_entry.delete(0, tk.END)
        output_entry.insert(0, path)

# ---------------- Main logic ----------------

def run_extractor():
    input_file = input_entry.get().strip()
    output_file = output_entry.get().strip()
    pattern = regex_entry.get().strip()

    if not input_file or not output_file or not pattern:
        messagebox.showerror("Fout", "Vul alle velden in.")
        return

    try:
        regex = re.compile(pattern, re.IGNORECASE)
    except re.error:
        messagebox.showerror("Fout", "Ongeldige regex.")
        return

    start_button.config(state=tk.DISABLED)
    status_label.config(text="Bezig...")

    def task():
        seen = set()
        processed = 0
        written = 0

        try:
            with open(input_file, "r", encoding="utf-8", errors="ignore") as f_in, \
                 open(output_file, "w", encoding="utf-8") as f_out:

                for line in f_in:
                    processed += 1

                    if ":" not in line:
                        continue

                    line = line.strip()

                    # CORRECT split: url : user : pass
                    try:
                        url_part, user, password = line.rsplit(":", 2)
                    except ValueError:
                        continue  # ongeldige regel

                    if regex.search(url_part):
                        userpass = f"{user}:{password}"

                        if userpass not in seen:
                            seen.add(userpass)
                            f_out.write(userpass + "\n")
                            written += 1

                    if processed % 100_000 == 0:
                        status_label.config(
                            text=f"Verwerkt: {processed:,} | Uniek: {written:,}"
                        )

            status_label.config(
                text=f"Klaar ✅ Verwerkt: {processed:,} | Uniek: {written:,}"
            )

        except Exception as e:
            messagebox.showerror("Error", str(e))

        start_button.config(state=tk.NORMAL)

    threading.Thread(target=task, daemon=True).start()

# ---------------- GUI ----------------

root = tk.Tk()
root.title("ULP Extractor (Regex + Dedup)")
root.geometry("520x260")
root.resizable(False, False)

tk.Label(root, text="Regex / Domein (bv crunchyroll\\.com):").pack(
    anchor="w", padx=10, pady=(10, 0)
)
regex_entry = tk.Entry(root, width=70)
regex_entry.pack(padx=10)

tk.Label(root, text="Input ULP bestand:").pack(
    anchor="w", padx=10, pady=(10, 0)
)
input_frame = tk.Frame(root)
input_frame.pack(padx=10, fill="x")
input_entry = tk.Entry(input_frame, width=55)
input_entry.pack(side="left")
tk.Button(input_frame, text="Browse", command=browse_input).pack(side="right")

tk.Label(root, text="Output bestand:").pack(
    anchor="w", padx=10, pady=(10, 0)
)
output_frame = tk.Frame(root)
output_frame.pack(padx=10, fill="x")
output_entry = tk.Entry(output_frame, width=55)
output_entry.pack(side="left")
tk.Button(output_frame, text="Browse", command=browse_output).pack(side="right")

start_button = tk.Button(
    root, text="Start Extractie", command=run_extractor
)
start_button.pack(pady=15)

status_label = tk.Label(root, text="Wachten...")
status_label.pack()

root.mainloop()