From 9b6edbbc4952f05fead12a04982bca6fce0002cb Mon Sep 17 00:00:00 2001
From: snsd0805 <levi900227@gmail.com>
Date: Fri, 19 Jun 2020 03:37:07 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E6=88=90=E6=88=BF=E8=99=9F=EF=BC=8Cse?=
 =?UTF-8?q?rver=E5=8F=AF=E6=8E=A5=E5=85=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 61 +++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 57 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 00d2296..00d87e9 100644
--- a/main.py
+++ b/main.py
@@ -1,8 +1,9 @@
-import socket,threading,sys
+import socket,threading,sys,random,multiprocessing,time
 
 MAX = 1024
 
 class Server:
+    roomList = []
     def __init__(self,ip,port):
         self.ip = ip
         self.port = port
@@ -18,18 +19,61 @@ class Server:
             setTypeThread.start()            
 
     def selectType(self,sock):
-        print("SelectType() function started")
         data = sock.recv(MAX).decode('utf-8')
 
         print(data)
         if data == "MAIN":
             sock.send("OK.SERVER".encode('utf-8'))
+            self.main(sock)
         elif data == "CLIENT":
             sock.send("OK.CLIENT".encode('utf-8'))
         else:
             sock.send("FAIL".encode('utf-8'))
         
-        print("SelectType() function closed")
+    def main(self,sock):            
+        portNum = random.randint(1025,65535)    #決定port
+        roomNum = chr(random.randint(65,90)) + chr(random.randint(65,90))
+
+        flag = 1
+        while flag:
+            portNum = random.randint(1025,65535)    #決定port
+            roomNum = chr(random.randint(65,90)) + chr(random.randint(65,90))
+            flag = 0
+            for i in self.roomList:
+                if i['portNum'] == portNum or i['roomNum'] == roomNum:
+                    flag =1
+        room = Room(self.ip,portNum)
+        roomProcess = multiprocessing.Process(target=room.start)
+        roomProcess.start()
+        self.roomList.append(
+            {
+                'portNum' : portNum,
+                'roomNum' : roomNum,
+            }
+        )
+        print(self.roomList)
+        sock.send("{} {}".format(roomNum,portNum).encode('utf-8'))
+    # def getPort(self):
+    #     sock
+class Room:
+    startFlag = False
+    def __init__(self,ip,portNum):
+        self.ip = ip
+        self.portNum = portNum
+    def start(self):
+        listensock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+        listensock.bind((self.ip,self.portNum))
+        print("\t{}:{}".format(self.ip,self.portNum))
+        listensock.listen(5)
+        while True:
+            sock,sockname = listensock.accept()
+            print("[ {} ]{} has connected.".format(self.portNum,sockname))
+            sock.send('WELCOME'.encode('utf-8'))
+    def connect(self):
+        time.sleep(0.5)
+        sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+        sock.connect((self.ip,self.portNum))
+        print(sock.recv(MAX).decode('utf-8'))
 
 class Client:
     def __init__(self,ip,port):
@@ -46,7 +90,16 @@ class Client:
         msg = input("> ")
         sock.send(msg.encode('utf-8'))
         receiveMsg = sock.recv(MAX).decode('utf-8')
-        print(receiveMsg)
+        if receiveMsg=="OK.SERVER":
+            receiveMsg = sock.recv(MAX).decode('utf-8')
+            print(receiveMsg)
+            connectData = receiveMsg.split(' ')
+            room = Room(self.ip,int(connectData[1]))
+            room.connect()
+        elif receiveMsg=="OK.CLIENT":
+            print("CLIENT")
+        else:
+            print("ERROR TYPE")
 
 def main():
     ip = sys.argv[2]