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
99
100
101
102
103
104
105
106
107
108
109
| from rich.console import Console
import socket
import json
import threading
import time
console = Console()
threads = {}
messages = [
{"from":"Server","msg":"Server Started","time":time.time()}
]
threadList = []
users = {}
def handle(sock, addr):
global console, messages, threads, threadList, users
msgid = 0
username = ""
while True:
resp_data = {
"code":200,
"msg":"OK",
"data":{}
}
recv_data = sock.recv(1024)
try:
recv_data = recv_data.decode("utf-8")
recv_data = json.loads(recv_data)
if username != "":
if recv_data["mode"] == "getMsg":
try:
resp_data["data"]["messages"] = messages[msgid:]
msgid = messages.__len__()
except:
resp_data["data"]["messages"] = []
elif recv_data["mode"] == "sendMsg":
messages += [
{
"from":username,
"msg":recv_data["data"]["msg"],
"time":time.time()
}
]
console.log(f"[CHAT] <{username}> {recv_data['data']['msg']}")
elif recv_data["mode"] == "exit":
sock.close()
return 0
elif recv_data["mode"] == "getList":
userList = []
for t in threadList:
if threads[t].is_alive():
userList += [users[t].getName()]
resp_data["data"]["list"] = userList
else:
resp_data["code"] = 404
resp_data["msg"] = "Unkown mode"
elif recv_data["mode"] == "login":
username = recv_data["data"]["username"]
users[addr[1]] = username
else:
resp_data["code"] = 403
resp_data["msg"] = "Please Login First"
except Exception as e:
console.print_exception(show_locals=True)
resp_data["code"] = 500
resp_data["msg"] = "Internal Server Error"
resp_data["data"]["exception"] = str(e)
resp_data = json.dumps(resp_data)
sock.send(resp_data.encode("utf-8"))
if __name__ == "__main__":
port = int(console.input("[blue]Port: "))
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(
(
"",
port
)
)
sock.listen(128)
console.log(f"Server started on 0.0.0.0:{port}.")
while True:
s, addr = sock.accept()
console.log(f"{addr[1]} connected to this server.")
threadList += [addr[1]]
users[addr[1]] = ""
threads[addr[1]] = threading.Thread(None, lambda: handle(s, addr))
threads[addr[1]].setDaemon(True)
threads[addr[1]].start()
|