package com.ard.sample.component.websocket.service.impl;

import org.springframework.stereotype.Component;

import javax.websocket.*;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.util.concurrent.ConcurrentHashMap;

/**
 * WebSocket端点类：处理连接、消息收发、断开等事件
 * 端点路径：ws://localhost:8080/ws/{username}（username为客户端标识）
 */
@ServerEndpoint("/ws/Dock/{username}")
@Component
public class WebSocketDockServer {

    // 静态变量：存储所有在线客户端的会话（线程安全）
    private static final ConcurrentHashMap<String, Session> ONLINE_USERS = new ConcurrentHashMap<>();

    // 当前连接的用户名
    private String username;


    /**
     * 客户端连接成功时触发
     */
    @OnOpen
    public void onOpen(Session session, @PathParam("username") String username) {
        this.username = username;
        // 保存会话
        ONLINE_USERS.put(username, session);
        // 群发"用户上线"通知
        sendToAll("系统通知：用户【" + username + "】已上线，当前在线人数：" + ONLINE_USERS.size());
        System.out.println("用户【" + username + "】连接成功，在线人数：" + ONLINE_USERS.size());
    }


    /**
     * 客户端断开连接时触发
     */
    @OnClose
    public void onClose() {
        // 移除会话
        ONLINE_USERS.remove(username);
        // 群发"用户下线"通知
        sendToAll("系统通知：用户【" + username + "】已下线，当前在线人数：" + ONLINE_USERS.size());
        System.out.println("用户【" + username + "】断开连接，在线人数：" + ONLINE_USERS.size());
    }


    /**
     * 收到客户端消息时触发
     * 消息格式支持："all:消息内容"（群发）、"username:消息内容"（指定用户发送）
     */
    @OnMessage
    public void onMessage(String message) {
        System.out.println("收到【" + username + "】的消息：" + message);

        if (message.startsWith("all:")) {
            // 群发消息（格式：all:你好大家）
            String content = message.substring(4);
            sendToAll("【" + username + "】：" + content);
        } else if (message.contains(":")) {
            // 给指定用户发消息（格式：目标用户名:消息内容）
            String[] parts = message.split(":", 2);
            String targetUser = parts[0];
            String content = parts[1];
            sendToUser(targetUser, "【" + username + "】对你说：" + content);
        } else {
            // 默认群发
            sendToAll("【" + username + "】：" + message);
        }
    }


    /**
     * 连接异常时触发
     */
    @OnError
    public void onError(Session session, Throwable error) {
        System.err.println("WebSocket错误（用户【" + username + "】）：");
        error.printStackTrace();
    }


    // -------------------------- 消息发送工具方法 --------------------------

    /**
     * 向单个用户发送消息
     * @param targetUser 目标用户名
     * @param message 消息内容
     */
    public static void sendToUser(String targetUser, String message) {
        Session session = ONLINE_USERS.get(targetUser);
        if (session != null && session.isOpen()) { // 检查会话是否有效
            try {
                // 同步发送消息
                session.getBasicRemote().sendText(message);
            } catch (IOException e) {
                System.err.println("向【" + targetUser + "】发送消息失败：" + e.getMessage());
            }
        } else {
            System.out.println("用户【" + targetUser + "】不在线");
        }
    }


    /**
     * 向所有在线用户群发消息
     * @param message 消息内容
     */
    public static void sendToAll(String message) {
        for (Session session : ONLINE_USERS.values()) {
            if (session.isOpen()) { // 只给活跃会话发送
                try {
                    session.getBasicRemote().sendText(message);
                } catch (IOException e) {
                    System.err.println("群发消息失败：" + e.getMessage());
                }
            }
        }
    }
}