package com.inmind.s_test_11.chatroom1;

import java.io.*;
import java.net.*;
import java.util.*;
import java.io.*;
import java.net.*;
import java.util.*;

public class ChatServer {
    // 服务器监听的端口号
    private static final int PORT = 12345;
    // 存储所有客户端输出流的集合，用于广播消息
    private static Set<PrintWriter> clientWriters = new HashSet<>();

    public static void main(String[] args) {
        System.out.println("多人聊天服务器启动，监听端口: " + PORT);

        try (ServerSocket serverSocket = new ServerSocket(PORT)) {
            // 循环接受客户端连接
            while (true) {
                // 阻塞等待客户端连接
                Socket clientSocket = serverSocket.accept();
                System.out.println("新客户端连接: " + clientSocket);

                // 创建PrintWriter用于向客户端发送消息
                PrintWriter out = new PrintWriter(clientSocket.getOutputStream(), true);

                // 将新客户端的输出流添加到集合中，使用同步块确保线程安全
                synchronized (clientWriters) {
                    clientWriters.add(out);
                }

                // 为每个客户端创建并启动一个新的处理线程
                new Thread(new ClientHandler(clientSocket, out)).start();
            }
        } catch (IOException e) {
            System.err.println("服务器错误: " + e.getMessage());
        }
    }

    // 客户端处理线程类，负责接收客户端消息并广播
    private static class ClientHandler implements Runnable {
        private Socket clientSocket;    // 客户端Socket
        private PrintWriter out;        // 向客户端输出的流
        private String username;        // 客户端用户名

        public ClientHandler(Socket socket, PrintWriter out) {
            this.clientSocket = socket;
            this.out = out;
        }

        public void run() {
            try {
                // 读取客户端发送的数据
                BufferedReader in = new BufferedReader(
                        new InputStreamReader(clientSocket.getInputStream()));

                // 读取客户端发送的用户名
                username = in.readLine();
                System.out.println("用户 " + username + " 加入聊天");

                // 广播新用户加入的消息
                broadcast(username + " 加入了聊天！");

                // 循环读取客户端发送的消息
                String input;
                while ((input = in.readLine()) != null) {
                    // 如果用户输入exit，退出循环
                    if (input.equalsIgnoreCase("exit")) {
                        break;
                    }

                    // 广播用户发送的消息
                    broadcast(username + ": " + input);
                }

                // 用户退出时的处理
                System.out.println("用户 " + username + " 离开聊天");
                broadcast(username + " 离开了聊天！");

            } catch (IOException e) {
                System.err.println("与客户端通信错误: " + e.getMessage());
            } finally {
                // 清理资源
                if (out != null) {
                    // 从客户端集合中移除当前客户端的输出流
                    synchronized (clientWriters) {
                        clientWriters.remove(out);
                    }
                }
                try {
                    // 关闭客户端连接
                    clientSocket.close();
                } catch (IOException e) {
                    System.err.println("关闭客户端连接错误: " + e.getMessage());
                }
            }
        }

        // 广播消息给所有连接的客户端
        private void broadcast(String message) {
            // 同步遍历客户端输出流集合，确保线程安全
            synchronized (clientWriters) {
                for (PrintWriter writer : clientWriters) {
                    writer.println(message);
                }
            }
        }
    }
}