import sys
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5 import uic
import threading, json, pika

form = uic.loadUiType("chat.ui")[0]


class Chat(QWidget, form):
    def __init__(self, name):
        super().__init__()
        self.setupUi(self)
        self.setWindowTitle(name)
        self.listView.setEditTriggers(QAbstractItemView.NoEditTriggers)
        self.plainTextEdit.setReadOnly(True)
        self.send.clicked.connect(self.sendMsg)
        self.id = name
        self.dict = {'name': self.id, 'message': []}
        self.conn = pika.BlockingConnection(
            pika.ConnectionParameters('183.99.87.90', 5672, '/', pika.PlainCredentials('guest', 'guest')))
        self.channel = self.conn.channel()
        self.channel.exchange_declare(exchange='topic_logs3', exchange_type='topic', auto_delete = False)

        result = self.channel.queue_declare(queue='', exclusive=True)
        self.queue_name = result.method.queue

        self.channel.queue_bind(exchange='topic_logs3', queue=self.queue_name, routing_key="Join") #auto_delete = false
        self.channel.queue_bind(exchange='topic_logs3', queue=self.queue_name, routing_key="Leave")
        self.channel.queue_bind(exchange='topic_logs3', queue=self.queue_name, routing_key="SendMsg")
        self.channel.queue_bind(exchange='topic_logs3', queue=self.queue_name, routing_key="List")
        self.channel.queue_bind(exchange='topic_logs3', queue=self.queue_name, routing_key="List2")
        data = json.dumps(self.dict)
        self.channel.basic_publish(exchange='topic_logs3', routing_key="List", body=data.encode('utf-8'))
        # self.channel.basic_publish(exchange='topic_logs', routing_key="Join", body=data.encode('utf-8'))

        print("클라이언트 소켓정보 : ", self.conn)
        th = threading.Thread(target=self.worker, name="[스레드 이름 {}]".format(self.conn), args=())
        th.start()

        self.userList=[]

    """
        [ listView 항목 추가 하는 방법 ] 

        model = QStandardItemModel()
        for x in json.loads(datas['message']):
            model.appendRow(QStandardItem(x))
        self.listView.setModel(model)
    """

    def callback(self, ch, method, properties, body):
        # nameList.append(self.id)
        datas = json.loads(body.decode())
        if method.routing_key == "SendMsg":
            self.plainTextEdit.insertPlainText("[ %s ] : %s\n" % (datas['name'], datas['message']))
        elif method.routing_key == "Leave":
            self.plainTextEdit.insertPlainText("[ %s 님이 퇴장 하였습니다. ]\n" % datas['name'])
        elif method.routing_key == "Join":
            self.plainTextEdit.insertPlainText("[ %s 님이 참여 하였습니다. ]\n" % datas['name'])
        elif method.routing_key == "List2":
            diff = list(set(datas) - set(self.userList))
            for b in diff:
                self.userList.insert(0, b)
            model = QStandardItemModel()
            for x in self.userList:
                model.appendRow(QStandardItem(x))
            self.listView.setModel(model)
        elif method.routing_key == "List":
            self.userList.append(datas['name'])
            # print(self.userList)
            data = json.dumps(self.userList)
            self.channel.basic_publish(exchange='topic_logs3', routing_key="List2", body=data.encode('utf-8'))
            # list = datas['message']
            # list.append(datas['name'])
            # new_list = list[:]
            # self.dict = {'name': self.id, 'message': list}
            # print(self.dict)
            # data = json.dumps(list)
            # if not set(datas['message']) - set(list):
            #     self.channel.basic_publish(exchange='topic_logs3', routing_key="List2", body=data.encode('utf-8'))
            # print(list)
            # print(new_list)



    def keyPressEvent(self, e):
        if e.key() == Qt.Key_Return:
            self.sendMsg()

    def sendMsg(self):
        if len(self.lineEdit.text()) == 0:
            return
        self.dict['message'] = self.lineEdit.text()
        data = json.dumps(self.dict)
        self.channel.basic_publish(exchange='topic_logs3', routing_key="SendMsg", body=data.encode('utf-8'))
        self.lineEdit.clear()

    def worker(self):
        self.channel.basic_consume(queue=self.queue_name, on_message_callback=self.callback, auto_ack=True)
        self.channel.start_consuming()

    def closeEvent(self, event):
        data = json.dumps(self.dict)
        self.channel.basic_publish(exchange='topic_logs3', routing_key="Leave", body=data.encode('utf-8'))


class MyWindow(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setWindowTitle('SWH Academy Window.')

        centerGeometry = QDesktopWidget().availableGeometry().center()
        self.resize(300, 100)
        frameGeometry = self.frameGeometry()
        frameGeometry.moveCenter(centerGeometry)

        self.idLabel = QLabel("아이디 : ")
        self.idLineEdit = QLineEdit()
        self.loginButton = QPushButton("로그인")
        self.loginButton.clicked.connect(self.buttonClicked)
        layout = QGridLayout()

        layout.addWidget(self.idLabel, 0, 0)
        layout.addWidget(self.idLineEdit, 0, 1)

        layout.addWidget(self.loginButton, 1, 1)

        self.setLayout(layout)

    def keyPressEvent(self, e):
        if e.key() == Qt.Key_Return:
            self.buttonClicked()

    def buttonClicked(self):
        if len(self.idLineEdit.text()) == 0:
            return
        self.chatting = Chat(self.idLineEdit.text())
        self.chatting.show()
        window.hide()


if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = MyWindow()
    window.show()
    sys.exit(app.exec_())