"""
chat room  chat_server服务端
socket udp & fork
"""
from socket import *
import os,sys
"""
全局变量: 很多封装模块都要用或者有一定的固定含义
"""
# 服务器地址
ADDR = ('0.0.0.0',8888)
# 存储用户 {name:address}
user = {}
# 登录
def do_login(s,name,addr):
    if name in user or '管理员' in name:
        s.sendto('该用户存在'.encode(),addr)
        return
    s.sendto(b'OK',addr) # 可以进入聊天室
    # 通知其他人
    msg = "\n欢迎'%s'进入聊天室"%name
    for i in user:
        s.sendto(msg.encode(),user[i])
    user[name] = addr # 插入字典
# 聊天
def do_chat(s,name,text):
    msg = "\n%s: %s"%(name,text)
    for i in user:
        # 刨除其本人
        if i != name:
            s.sendto(msg.encode(),user[i])
# 退出
def do_quit(s,name):
    msg = "\n%s 退出聊天室"%name
    for i in user:
        if i != name: # 其他人
            s.sendto(msg.encode(),user[i])
        else:
            s.sendto(b'EXIT',user[i])
    del user[name] # 删除用户
#　处理请求
def do_request(s):
    while True:
        data,addr = s.recvfrom(1024)
        tmp = data.decode().split(' ') #　拆分请求
        # 根据不同的请求类型具体执行不同的事情
        # L 进入　　C 聊天　　Q 退出
        if tmp[0] == 'L':
            do_login(s,tmp[1],addr)  # 执行具体工作
        elif tmp[0] == 'C':
            text = ' '.join(tmp[2:])
            do_chat(s,tmp[1],text)
        elif tmp[0] == 'Q':
            do_quit(s,tmp[1])
# 搭建网络
def main():
    # udp服务端
    s = socket(AF_INET,SOCK_DGRAM)
    s.bind(ADDR)
    pid = os.fork()
    if pid == 0: # 子进程处理管理员消息
        while True:
            msg = input("管理员消息：")
            msg = "C 管理员 "+msg
            s.sendto(msg.encode(),ADDR)
    else:
        #　请求处理函数
        do_request(s)
main()

"""
chat room  客户端
发送请求，展示结果
"""
from socket import *
import os,sys

# 服务器地址
ADDR = ('127.0.0.1',8888)

# 发送消息
def send_msg(s,name):
    while True:
        try:
            text = input("发言:")
        except KeyboardInterrupt:
            text = 'quit'
        if text.strip() == 'quit':
            msg = "Q " + name
            s.sendto(msg.encode(),ADDR)
            sys.exit("退出聊天室")
        msg = 'C %s %s'%(name,text)
        s.sendto(msg.encode(),ADDR)

# 接收消息
def recv_msg(s):
    while True:
        try:
            data,addr = s.recvfrom(4096)
        except KeyboardInterrupt:
            sys.exit()
        # 从服务器收到EXIT退出
        if data.decode() == 'EXIT':
            sys.exit()
        print(data.decode() + '\n发言:',end='')

#客户端启动函数
def main():
    s = socket(AF_INET,SOCK_DGRAM)

    # 进入聊天室
    while True:
        name = input("请输入姓名:")
        msg = 'L ' + name
        s.sendto(msg.encode(),ADDR)
        #　接收反馈
        data,addr = s.recvfrom(128)
        if  data.decode() == 'OK':
            print("您已进入聊天室")
            break
        else:
            print(data.decode())
    # 已经进入聊天室
    pid = os.fork()
    if pid < 0:
        sys.exit("Error!")
    elif pid == 0:
        send_msg(s,name)  # 子进程负责消息发送
    else:
        recv_msg(s)  # 父进程负责消息接收

main()