import sys
from PyQt5.QtWidgets import QApplication, QMainWindow,  QTableWidgetItem, QFileDialog
from PyQt5 import QtWidgets
from newfile import Ui_MainWindow # 导入生成的UI类
from tcp_client import TCPClient
from tcp_server import TCPServer
from isIP import isIP,isPORT
from AES import *
from RSA_Module import *
from Zip_Module import *
import os
import threading
import time
import shutil

class MainWindow(QMainWindow, Ui_MainWindow):
    def __init__(self):
        super().__init__()
        self.setupUi(self)  # 设置UI
        #初始化TCP服务器
        self.client_conn = None
        self.running = True
        self.run_client = True
        self.server =TCPServer()
        self.conn =None
        self.client = TCPClient()
        self.client_conn = None
        #client文件信号
        # self.client_recv = TCPClient_recv()
        self.client.receive_status.connect(self.add_list)
        self.client.list_received.connect(self.add_list)
        self.client.receive_progress.connect(self.update_receive_bar)

        #连接按钮点击事件自动跳转到自定义槽函数
        self.send_file.clicked.connect(self.show_send_page)
        self.receive_file.clicked.connect(self.show_receive_page)
        self.add_file.clicked.connect(self.read_file)
        self.handler.clicked.connect(self.toggle_server)
        self.receive.clicked.connect(self.toggle_client)
        self.send.clicked.connect(self.send_files)
        self.delete_file.clicked.connect(self.delete_filex)
        self.empty_file.clicked.connect(self.delete_all_filex)
        self.server.progress_updated.connect(self.update_current_bar)


    def mk_temp_files(self):
        """创建临时文件夹"""
        temp_dir = './.tempfile'
        if not os.path.exists(temp_dir):
            os.makedirs(temp_dir)

    def del_temp_files(self):
        """删除临时文件夹及其内容"""
        temp_dir = './.tempfile'
        if os.path.exists(temp_dir):
            try:
                shutil.rmtree(temp_dir)
                print(f"Temporary directory deleted: {temp_dir}")
            except Exception as e:
                print(f"Failed to delete temporary directory: {e}")
    def closeEvent(self, event):
        """重写 closeEvent 方法，在窗口关闭时删除临时目录"""
        self.del_temp_files()
        # 调用父类的 closeEvent 方法，确保窗口正常关闭
        super().closeEvent(event)

#-------------------------------------------------------------------------------------------------
    #左上角三个按钮
    #读文件
    def read_file(self):
        filename, _ =QFileDialog.getOpenFileName(self, "选取文件", "C:/", "All Files(*)")
        if not filename:
            return  #如果用户取消选择=文件，直接返回
        # 获取文件的字节大小
        file_size_bytes = os.path.getsize(filename)
        file_size_str = f"{file_size_bytes}字节"
        # 找到当前表格的最后一行（非空行）
        row_count = self.tableWidget.rowCount()
        for row in range(row_count):
            item = self.tableWidget.item(row, 0)
            if not item or item.text().strip() == "":  # 检查第一列是否为空或仅包含空白字符
                break
        else:
            # 如果所有行都已填满，增加新行
            self.tableWidget.insertRow(row_count)
            row = row_count
        self.tableWidget.setItem(row, 0, QTableWidgetItem(filename))
        self.tableWidget.setItem(row, 1, QTableWidgetItem(file_size_str))
    #删除文件
    def delete_filex(self):
        row_count = self.tableWidget.rowCount()
        if row_count > 0:
            for row in range(row_count):
                item = self.tableWidget.item(row, 0)
                if not item or item.text().strip() == "":  # 检查第一列是否为空或仅包含空白字符
                    break
            self.tableWidget.setItem(row-1, 0, QTableWidgetItem(""))
            self.tableWidget.setItem(row-1, 1, QTableWidgetItem(""))
        else:
            return
    #删除全部文件
    def delete_all_filex(self):
        self.tableWidget.clearContents()

    def read_table(self):
        row_count = self.tableWidget.rowCount()
        filenames = []
        filesizes = []
        newfilenames = []
        newfilesizes = []
        exts = []
        for row in range(row_count):
            filename_item = self.tableWidget.item(row, 0)
            filetype_item = self.tableWidget.item(row, 1)

            if filename_item is not None:
                filenames.append(filename_item.text())

            if filetype_item is not None:
                filesizex = int(filetype_item.text().strip('字节'))
                filesizes.append(filesizex)
        for i in range(len(filenames)):
            basename, ext = os.path.splitext(os.path.basename(filenames[i])) #不带扩展名的文件
            aeskey = create_key(basename)    #生成对称密钥
            sign_message(filenames[i], './.tempfile/serverpri.pem', f'./.tempfile/{basename}_sig') #计算摘要加签名

            with open(filenames[i],'rb') as f:  #使用对称密钥对文件进行加密
                data = f.read()
                with open(f'./.tempfile/{basename}{ext}', 'wb') as s:
                    endata = AESencrypt(data, aeskey)
                    s.write(endata)
            RSAencrypt('./.tempfile/new_clientpub.pem', aeskey, f'./.tempfile/{basename}_key')#对对称密钥进行加密
            files_to_zip = [
                f'./.tempfile/{basename}_sig',
                f'./.tempfile/{basename}{ext}',
                f'./.tempfile/{basename}_key'
            ]
            zip_filename = f'./.tempfile/{basename}'
            zip_files(zip_filename, files_to_zip)
            newfilenames.append(zip_filename)
            #获取文件大小
            file_size_bytes = os.path.getsize(zip_filename)
            newfilesizes.append(file_size_bytes)
            #文件后缀
            exts.append(ext)



        return newfilenames,newfilesizes,exts
#----------------------------------------------------------------------------------
    #服务端
    def send_files(self):
        filenames, filesizes, exts = self.read_table()
        self.send_fileth(filenames, filesizes, exts)
    def send_fileth(self, filenames, filesizes, exts):
        if not self.conn:
            self.update_status(f'服务器未连接')
            return
        self.thread_send = threading.Thread(target=self.send_file_all, kwargs={'filenames': filenames, 'filesizes': filesizes, 'exts': exts})
        self.thread_send.start()

    def send_file_all(self, filenames, filesizes, exts):
        send_file = 0
        i = 0
        while i < len(filenames):
            self.update_status(f"{filenames[i]}文件发送中")
            if i + 1 == len(filenames):
                index = -1
            else:
                index = i + 1
            self.server.send_file(self.conn, filenames[i], filesizes[i], index, exts[i])
            send_file += 1
            state, fileindex = self.server.judge_send_status(self.conn)
            i = i + 1
            progress = int((send_file / len(filenames)) * 100)
            self.update_total_bar(progress)
            if not state:
                i = fileindex - 1
                self.update_status(f'{filenames[fileindex]}传输错误，重传中...')
                time.sleep(3)

        self.update_status(f"文件全部发送完成")
        self.server.wait_for_send = False


    def toggle_server(self):
        if self.running:
            self.thread = threading.Thread(target=self.start_server)
            self.thread.start()
        else:
            self.running = True
            if self.thread and self.thread.is_alive():
                self.thread.join()
            self.handler.setText('启动服务器')

    def start_server(self):
        ip_address = self.IP_text.text().strip()  # 读取并去除可能存在的空白字符
        port = self.PORT_text.text().strip()
        if not (isIP(ip_address) and isPORT(port)):
            # 如果不是有效的 IP 地址
            self.update_status('ip is error.')
            return
        self.handler.setText('停止服务器')
        self.running = False
        self.update_status(f'等待客户端连接...')
        self.server.ip = ip_address
        self.server.port = int(port)

        self.server.run_server()
        conn, addr = self.server.tcp_server.accept()
        self.conn = conn  # 保存客户端连接
        if conn:
            # self.server.client_conn = True
            self.update_status(f'客户端连接成功')
        else:
            self.update_status(f'客户端连接失败')
        state, pubname = self.server.recv_clientpub(self.conn)
        if state:
            self.update_status(f'{pubname}接收成功')
        else:
            self.update_status(f'{pubname}接收失败')
        self.server.send_serverpub(self.conn)


#--------------------------------------------------------------------------------
    #客户端

    def toggle_client(self):
        if self.run_client:
            self.thread_client = threading.Thread(target=self.start_client)
            self.thread_client.start()
        else:
            self.run_client = True
            if self.thread_client and self.thread_client.is_alive():
                self.thread_client.join()
            self.receive.setText('接收')
    def start_client(self):
        ip = self.IP_receive_text.text().strip()  # 读取并去除可能存在的空白字符
        port = self.PORT_receive_text.text().strip()

        if not (isIP(ip) and isPORT(port)):
            # 如果不是有效的 IP 地址
            self.update_receive_play('ip or port is error.')
            return
        port = int(port)
        self.receive.setText('接收中')
        self.run_client = False
        self.update_receive_play(f'服务端连接中')
        time.sleep(2)
        self.client_conn = self.client.connect(ip, port)
        if self.client_conn:
            self.receive.setEnabled(False)
            self.update_receive_play(f'连接服务端成功')
        else:
            self.update_receive_play(f'连接服务端失败')
            return
        self.client.send_clientpub(self.client_conn)
        state, pubname = self.client.recv_serverpub(self.client_conn)
        if state:
            self.update_receive_play(f'{pubname}接收成功')
        else:
            self.update_receive_play(f'{pubname}接收失败')
        self.client.receive(self.client_conn)
        self.receive.setEnabled(True)







#--------------------------------------------------------------------------------
    #更新显示常用函数。
    def update_current_bar(self, value):
        self.current_status_bar.setValue(value)

    def update_total_bar(self, value):
        self.total_status_bar.setValue(value)

    def update_receive_bar(self, value):
        self.receive_bar.setValue(value)

    def add_list(self, text):
        if text:
            item = QtWidgets.QListWidgetItem(text)  # 创建一个新的 QListWidgetItem
            self.receive_widget.addItem(item)  # 将新项目添加到 QListWidget
            self.receive_widget.scrollToItem(item)  # 滚动到最新添加的项目

    def update_status(self, message):
        self.status_show.setText(message)
    def update_receive_play(self, message):
        self.receive_play.setText(message)
    def show_send_page(self):
        """切换到发送文件页面"""
        self.stackedWidget.setCurrentIndex(0)

    def show_receive_page(self):
        """切换到接收文件页面"""
        self.stackedWidget.setCurrentIndex(1)


if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()  # 显示窗口
    window.mk_temp_files()
    sys.exit(app.exec_())