SolarWebSocket/MainWebSocket.java at master · cyhcyh2/SolarWebSocket · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package me.cyh2.bungee.solarwebsocket.websocket;

import com.google.gson.Gson;
import com.google.gson.JsonObject;
import me.cyh2.solarwebsocket.utils.MainUtil;
import net.md_5.bungee.api.chat.TextComponent;
import org.java_websocket.WebSocket;
import org.java_websocket.handshake.ClientHandshake;
import org.java_websocket.server.WebSocketServer;

import java.net.InetSocketAddress;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.concurrent.atomic.AtomicReference;

import static me.cyh2.bungee.solarwebsocket.SolarWebSocketBungee.*;
import static me.cyh2.bungee.solarwebsocket.utils.textutils.ColorUtils.ReColor;


public class MainWebSocket extends WebSocketServer{
    @Deprecated
    public static Map<String, WebSocket> WSMap = new HashMap<>();
    public static Map<String, Map<String, WebSocket>> WSMapS = new HashMap<>();
    public int Clients = 0;

    public MainWebSocket(String hostName, int port) {
        super(new InetSocketAddress(hostName, port));
        proxyServer.getServers().keySet().forEach( str -> WSMapS.put(proxyServer.getServers().get(str).getName(), new HashMap<>()));
    }
    @Override
    public void onClose(WebSocket ws, int arg1, String arg2, boolean arg3) {
        AtomicReference<String> remove = new AtomicReference<>("");
        AtomicReference<String> rserver = new AtomicReference<>("");
        if(ws != null) {
            Clients --;
            WSMapS.keySet().forEach( str -> WSMapS.get(str).keySet().forEach(str1 -> {
                if (WSMapS.get(str).get(str1).equals(ws)) {
                    remove.set(str1);
                    rserver.set(str);
                }
            }));
            WSMapS.get(rserver.get()).remove(remove.get());
        }
        logger.info(ReColor("&b[Web Socket] &a编号为 " + remove.get() + " 的连接已断开"));
    }
    @Override
    public void onError(WebSocket ws, Exception e) {
        logger.severe("WebSocket服务器发生异常！请检查您的配置文件中的配置端口是否已被占用！");
        AtomicReference<String> remove = new AtomicReference<>("");
        AtomicReference<String> rserver = new AtomicReference<>("");
        if(ws != null) {
            Clients --;
            WSMapS.keySet().forEach( str -> WSMapS.get(str).keySet().forEach(str1 -> {
                if (WSMapS.get(str).get(str1).equals(ws)) {
                    remove.set(str1);
                    rserver.set(str);
                }
            }));
            WSMapS.get(rserver.get()).remove(remove.get());
        }
        e.printStackTrace();
        logger.severe(Arrays.toString(e.getStackTrace()));
    }
    @Override
    public void onMessage(WebSocket ws, String msg) {
        Gson gson = new Gson();
        JsonObject jsonObject = gson.fromJson(msg, JsonObject.class);
        if (jsonObject.get("msg").getAsString() != null) {
            logger.info(ReColor("&b[WEB SOCKET CHAT]&r " + jsonObject.get("user").getAsString() + "：" + jsonObject.get("msg").getAsString()));
            WSMap.keySet().forEach(str -> WSMap.get(str).send(jsonObject.get("user").getAsString() + "：" + jsonObject.get("msg").getAsString()));
            proxyServer.getPlayers().forEach(p -> p.sendMessage(new TextComponent(ReColor("[ Web Chat ] " + jsonObject.get("user").getAsString() + "：" + jsonObject.get("msg").getAsString()))));
        } else if (jsonObject.get("login_user").getAsString() != null && jsonObject.get("is_anonymous").getAsString() != null && jsonObject.get("server").getAsString() != null) {
            boolean anonymous = jsonObject.get("is_anonymous").getAsBoolean();
            String user = jsonObject.get("login_user").getAsString();
            String server = jsonObject.get("server").getAsString();
            if (WSMapS.get(server) != null) {
                WSMapS.get(server).put(user, ws);
            }
        }
    }
    @Override
    public void onOpen(WebSocket ws, ClientHandshake shake) {
        logger.info(ReColor("&b[Web Socket 连接事件]&a\n"+ws.isOpen()+"--"+ws.getReadyState()+"--"+ws.getAttachment()));
        for(Iterator<String> it=shake.iterateHttpFields();it.hasNext();) {
            String key = it.next();
            logger.info(ReColor(key+":"+shake.getFieldValue(key)));
        }
        ws.send(msgConfig.getString("WebSocketWelcomeMessage"));
        Clients ++;
        WSMap.put("WS" + Clients, ws);
    }
    @Override
    public void onStart() {
        logger.info(ReColor("&a&lSolar&rWebSocket&r服务器已启动！"));
    }
}