# chat_voice_client_with_upload.py
import socket
import threading
import time
import zlib
import base64

import numpy as np
import pyaudio

import sounddevice as sd
import opuslib

import requests
import queue
import os

import ttkbootstrap as tb
from ttkbootstrap.constants import *
from tkinter import simpledialog, messagebox, scrolledtext, filedialog, ttk
import tkinter as tk

# ---------------- Config (chat & voice) ----------------
AUDIO_RATE = 8000
AUDIO_CHANNELS = 1
AUDIO_FORMAT = pyaudio.paInt16
CHUNK = 320                 # увеличенный размер пакета (40ms)
ZLIB_LEVEL = 9
VOICE_PREFIX = "!voice "

MAX_CHUNK_SIZE = 400        # base64 одного кусочка
SLEEP_BETWEEN_CHUNKS = 0.05  # 50ms

client_socket = None
stop_event = threading.Event()
ignore_list = []

# Voice (Opus + sounddevice)
VOICE_SERVER_IP = "hoho.ws"
VOICE_SERVER_PORT = 50007
RATE = 16000
CHANNELS = 1
FRAME_SIZE = 960  # Opus frame (approx 60ms at 16kHz)
QUEUE_MAXSIZE = 50

voice_sock = None
encoder = opuslib.Encoder(RATE, CHANNELS, application="voip")
decoder = opuslib.Decoder(RATE, CHANNELS)
audio_queue = queue.Queue(maxsize=QUEUE_MAXSIZE)
send_audio = False
sd_stream = None

# HTTP upload server (DXP-0002)
UPLOAD_SERVER_URL = ""  # пример: "http://spburg.ddns.net:42439" — настраивается через меню

# ---------------- μ-law helpers (оставлены для совместимости, но в этой версии основного голоса мы используем Opus) ----------------
MU = 255
MAX = 32768

def linear2ulaw(pcm):
    pcm = pcm.astype(np.int32)
    sign = np.sign(pcm)
    pcm_abs = np.abs(pcm)
    magnitude = np.log1p(MU * pcm_abs / MAX) / np.log1p(MU)
    ulaw = sign * magnitude
    ulaw_uint = (((ulaw + 1) / 2) * 255).astype(np.uint8)
    return ulaw_uint.tobytes()

def ulaw2linear(ulaw_bytes):
    arr = np.frombuffer(ulaw_bytes, dtype=np.uint8).astype(np.float32)
    f = (arr / 255.0) * 2.0 - 1.0
    sign = np.sign(f)
    mag = np.abs(f)
    pcm = sign * ((np.expm1(np.log1p(MU) * mag)) * (MAX / MU))
    pcm = np.clip(pcm, -32768, 32767).astype(np.int16)
    return pcm.tobytes()

# ---------------- Network & chat ----------------
def connect_to_server():
    global client_socket
    ip = simpledialog.askstring("Подключение", "Введите IP сервера:")
    port = simpledialog.askinteger("Подключение", "Введите порт:")
    if not ip or not port:
        return
    try:
        client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        client_socket.connect((ip, port))
        threading.Thread(target=receive_messages, daemon=True).start()
        threading.Thread(target=send_keepalive, daemon=True).start()
        chat_box.insert(tk.END, f"[+] Подключено к {ip}:{port}\n")
        chat_box.see(tk.END)
    except Exception as e:
        messagebox.showerror("Ошибка подключения", str(e))

def receive_messages():
    global client_socket
    while not stop_event.is_set():
        try:
            data = client_socket.recv(4096)
            if not data:
                break
            try:
                message = data.decode('utf-8', errors='ignore').strip()
            except Exception:
                continue

            if VOICE_PREFIX in message:
                # мы удаляли старую !voice логику; если сервер всё ещё присылает такие пакеты,
                # можно обработать их здесь, но в данной интеграции используем Opus TCP поток отдельно.
                pass

            if any(word in message for word in ignore_list):
                continue

            chat_box.insert(tk.END, message + "\n")
            chat_box.see(tk.END)
        except Exception:
            break

def send_keepalive():
    while not stop_event.is_set():
        try:
            if client_socket:
                client_socket.send(b"/")
        except:
            break
        time.sleep(5)

def send_message(event=None):
    if not client_socket:
        chat_box.insert(tk.END, "[!] Не подключено к серверу\n")
        chat_box.see(tk.END)
        return
    msg = msg_entry.get()
    if msg:
        try:
            client_socket.send(msg.encode('utf-8'))
            msg_entry.delete(0, tk.END)
        except Exception:
            chat_box.insert(tk.END, "[!] Ошибка отправки\n")
            chat_box.see(tk.END)

# ---------------- Menu actions (chat) ----------------
def menu_login():
    user = simpledialog.askstring("Логин", "Введите логин:")
    password = simpledialog.askstring("Логин", "Введите пароль:", show="*")
    if user and password and client_socket:
        client_socket.send(f"/login {user} {password}".encode('utf-8'))

def menu_register():
    user = simpledialog.askstring("Регистрация", "Введите логин:")
    password = simpledialog.askstring("Регистрация", "Введите пароль:", show="*")
    if user and password and client_socket:
        client_socket.send(f"/register {user} {password}".encode('utf-8'))

def menu_pm():
    recipient = simpledialog.askstring("Приватное сообщение", "Кому отправить:")
    text = simpledialog.askstring("Приватное сообщение", "Текст сообщения:")
    if recipient and text and client_socket:
        client_socket.send(f"/pm {recipient} {text}".encode('utf-8'))

def menu_join():
    channel = simpledialog.askstring("Вход на канал", "Введите название канала:")
    if channel and client_socket:
        client_socket.send(f"/join_server {channel}".encode('utf-8'))

def menu_ignore():
    global ignore_list
    ignore_text = simpledialog.askstring("Игнор-лист", "Введите слова через запятую:")
    if ignore_text:
        ignore_list = [w.strip() for w in ignore_text.split(",") if w.strip()]
        chat_box.insert(tk.END, f"[+] Игнор обновлён: {ignore_list}\n")
        chat_box.see(tk.END)

def menu_theme(theme_name: str):
    root.style.theme_use(theme_name)

# ---------------- Voice (Opus) functions ----------------
def audio_callback(indata, frames, time_info, status):
    if send_audio:
        try:
            audio_queue.put_nowait(indata[:,0].copy())
        except queue.Full:
            pass

def audio_send_thread():
    global voice_sock
    while True:
        try:
            frame = audio_queue.get(timeout=0.1)
            if len(frame) < FRAME_SIZE:
                frame = np.pad(frame, (0, FRAME_SIZE - len(frame)), 'constant')
            pcm16 = (frame * 32767).astype(np.int16)
            packet = encoder.encode(pcm16.tobytes(), FRAME_SIZE)
            voice_sock.sendall(len(packet).to_bytes(4, 'big') + packet)
        except queue.Empty:
            continue
        except Exception as e:
            print("Send error:", e)
            break

def audio_receive_thread():
    global voice_sock
    try:
        with sd.OutputStream(samplerate=RATE, channels=CHANNELS, blocksize=FRAME_SIZE) as out_stream:
            buffer = b''
            while True:
                try:
                    data = voice_sock.recv(4096)
                    if not data:
                        break
                    buffer += data
                    while len(buffer) >= 4:
                        length = int.from_bytes(buffer[:4], 'big')
                        if len(buffer) < 4 + length:
                            break
                        packet = buffer[4:4+length]
                        buffer = buffer[4+length:]
                        try:
                            pcm = decoder.decode(packet, FRAME_SIZE)
                            audio = np.frombuffer(pcm, dtype=np.int16).astype(np.float32)/32768
                            out_stream.write(audio)
                        except opuslib.OpusError:
                            continue
                except Exception as e:
                    print("Receive error:", e)
                    break
    except Exception as e:
        print("Output stream error:", e)

def switch_device(device_name):
    global sd_stream
    if sd_stream:
        try:
            sd_stream.stop(); sd_stream.close()
        except:
            pass
    try:
        sd_stream = sd.InputStream(
            device=device_name,
            channels=CHANNELS,
            samplerate=RATE,
            blocksize=FRAME_SIZE,
            dtype='float32',
            latency='low',
            callback=audio_callback
        )
        sd_stream.start()
    except Exception as e:
        print(f"Cannot open microphone '{device_name}': {e}")

def toggle_send():
    global send_audio
    send_audio = var.get()

def select_device(event):
    threading.Thread(
        target=switch_device,
        args=(input_devices[devices_combo.current()]["name"],),
        daemon=True
    ).start()

def connect_voice_server():
    global voice_sock
    try:
        voice_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        voice_sock.connect((VOICE_SERVER_IP, VOICE_SERVER_PORT))
        threading.Thread(target=audio_send_thread, daemon=True).start()
        threading.Thread(target=audio_receive_thread, daemon=True).start()
        chat_box.insert(tk.END, f"[+] Подключено к голосовому серверу {VOICE_SERVER_IP}:{VOICE_SERVER_PORT}\n")
        chat_box.see(tk.END)
    except Exception as e:
        messagebox.showerror("Голос", f"Ошибка подключения к голосовому серверу: {e}")

def set_voice_server():
    global VOICE_SERVER_IP, VOICE_SERVER_PORT
    ip = simpledialog.askstring("Голосовой сервер", "Введите IP сервера:", initialvalue=VOICE_SERVER_IP)
    port = simpledialog.askinteger("Голосовой сервер", "Введите порт:", initialvalue=VOICE_SERVER_PORT)
    if ip and port:
        VOICE_SERVER_IP = ip
        VOICE_SERVER_PORT = port
        chat_box.insert(tk.END, f"[+] Голосовой сервер изменён на {ip}:{port}\n")
        chat_box.see(tk.END)

# ---------------- HTTP File Upload (DXP-0002) ----------------
def set_upload_server():
    global UPLOAD_SERVER_URL
    new_url = simpledialog.askstring("HTTP Upload сервер",
                                     "Введите базовый адрес (например http://example.com:42439):",
                                     initialvalue=UPLOAD_SERVER_URL or "http://")
    if new_url:
        UPLOAD_SERVER_URL = new_url.rstrip('/')
        chat_box.insert(tk.END, f"[+] Сервер загрузки изменён на {UPLOAD_SERVER_URL}\n")
        chat_box.see(tk.END)

def upload_file():
    if not UPLOAD_SERVER_URL:
        messagebox.showwarning("Upload", "Сначала настройте адрес HTTP Upload сервера в меню.")
        return
    file_path = filedialog.askopenfilename(title="Выберите файл для отправки")
    if not file_path:
        return

    chat_box.insert(tk.END, f"[⏫] Загружаю файл: {os.path.basename(file_path)}\n")
    chat_box.see(tk.END)

    def worker(path):
        try:
            with open(path, "rb") as f:
                files = {"file": (os.path.basename(path), f)}
                r = requests.post(f"{UPLOAD_SERVER_URL}/upload", files=files, timeout=60)
            if r.status_code == 200:
                try:
                    data = r.json()
                    url = data.get("url")
                    if url:
                        # Отправляем ссылку в чат как обычное сообщение
                        if client_socket:
                            try:
                                client_socket.send(url.encode('utf-8'))
                            except Exception:
                                pass
                        chat_box.insert(tk.END, f"[✅] Файл загружен: {url}\n")
                    else:
                        chat_box.insert(tk.END, f"[!] Сервер вернул ответ без url: {r.text[:200]}\n")
                except Exception:
                    chat_box.insert(tk.END, f"[!] Не удалось разобрать JSON: HTTP {r.status_code}\n")
            else:
                chat_box.insert(tk.END, f"[!] Ошибка загрузки: HTTP {r.status_code}\n")
        except Exception as e:
            chat_box.insert(tk.END, f"[!] Ошибка загрузки: {e}\n")
        chat_box.see(tk.END)

    threading.Thread(target=worker, args=(file_path,), daemon=True).start()

# ---------------- GUI Setup ----------------
root = tb.Window(themename="cyborg")
root.title("Chat Client (Voice + Upload)")
root.geometry("920x700")

# Menus
menubar = tk.Menu(root)
root.config(menu=menubar)

connection_menu = tk.Menu(menubar, tearoff=0)
menubar.add_cascade(label="Подключение", menu=connection_menu)
connection_menu.add_command(label="Подключиться к серверу", command=connect_to_server)

actions_menu = tk.Menu(menubar, tearoff=0)
menubar.add_cascade(label="Действия", menu=actions_menu)
actions_menu.add_command(label="Логин", command=menu_login)
actions_menu.add_command(label="Регистрация", command=menu_register)
actions_menu.add_command(label="Приватное сообщение", command=menu_pm)
actions_menu.add_command(label="Войти на канал", command=menu_join)
actions_menu.add_command(label="Игнор-лист", command=menu_ignore)

theme_menu = tk.Menu(menubar, tearoff=0)
menubar.add_cascade(label="Темы", menu=theme_menu)
for theme in ["cyborg","darkly","flatly","superhero","morph","solar"]:
    theme_menu.add_command(label=theme, command=lambda t=theme: menu_theme(t))

voice_menu = tk.Menu(menubar, tearoff=0)
menubar.add_cascade(label="Голосовой сервер", menu=voice_menu)
voice_menu.add_command(label="Настроить адрес", command=set_voice_server)
voice_menu.add_command(label="Подключиться", command=connect_voice_server)

upload_menu = tk.Menu(menubar, tearoff=0)
menubar.add_cascade(label="HTTP Upload сервер", menu=upload_menu)
upload_menu.add_command(label="Настроить адрес", command=set_upload_server)

# Chat box
chat_box = scrolledtext.ScrolledText(root, wrap="word", width=120, height=30,
                                     bg="#1e1e1e", fg="white",
                                     insertbackground="white", font=("Consolas", 11))
chat_box.pack(padx=10, pady=10, fill="both", expand=True)

# Message entry and buttons
frame_bottom = tb.Frame(root)
frame_bottom.pack(fill="x", padx=10, pady=5)

msg_entry = tb.Entry(frame_bottom, width=80)
msg_entry.pack(side="left", padx=5, fill="x", expand=True)
msg_entry.bind("<Return>", send_message)

send_btn = tb.Button(frame_bottom, text="Отправить", bootstyle=SUCCESS, command=send_message)
send_btn.pack(side="left", padx=5)

file_btn = tb.Button(frame_bottom, text="📎", bootstyle="secondary-outline", command=upload_file)
file_btn.pack(side="left", padx=5)

# Voice controls
voice_frame = tb.Frame(root)
voice_frame.pack(padx=10, pady=5, fill="x")

var = tk.BooleanVar()
ttk.Checkbutton(voice_frame, text="Говорить", variable=var, command=toggle_send).pack(side="left", padx=5)
ttk.Label(voice_frame, text="Выберите микрофон:").pack(side="left", padx=5)

devices = sd.query_devices()
input_devices = [d for d in devices if d["max_input_channels"] > 0]
device_names = [d["name"] for d in input_devices] if input_devices else []

devices_combo = ttk.Combobox(voice_frame, values=device_names, width=50)
if device_names:
    devices_combo.current(0)
devices_combo.bind("<<ComboboxSelected>>", select_device)
devices_combo.pack(side="left", padx=5)

# Select first device automatically
if input_devices:
    threading.Thread(target=switch_device, args=(input_devices[0]["name"],), daemon=True).start()

# ---------------- Closing ----------------
def on_closing():
    global send_audio, stop_event
    send_audio = False
    stop_event.set()
    try:
        if client_socket:
            client_socket.close()
    except: pass
    try:
        if voice_sock:
            voice_sock.close()
    except: pass
    try:
        if sd_stream:
            sd_stream.stop(); sd_stream.close()
    except: pass
    root.destroy()

root.protocol("WM_DELETE_WINDOW", on_closing)
root.mainloop()