'''
本模块用于创建登录界面并保存用户信息
'''
import sys
from PyQt5.QtWidgets import QApplication, QWidget, QLineEdit, QPushButton, QVBoxLayout, QHBoxLayout, QLabel
from PyQt5.QtCore import Qt,QTimer
from PyQt5.QtGui import QPixmap, QPainter
from PyQt5.QtWidgets import QCheckBox
import platform
from solve_path import b2_path, b2txt_path
import json
from show_welcome import show_image
platf = platform.platform()
class MyApp(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        # 加载背景图片
        if 'Windows' in platf:
            b2_path.replace('\b','\\b')
        self.background = QPixmap(b2_path)
        # 创建一个垂直布局
        mainLayout = QVBoxLayout()

        self.showPasswordCheckbox = QCheckBox("显示密码", self)
        self.showPasswordCheckbox.stateChanged.connect(self.togglePasswordVisibility)
        # 创建三个文本框并设置占位符
        self.nameEdit = QLineEdit()
        self.pwdEdit = QLineEdit()
        self.urlEdit = QLineEdit()

        self.pwdEdit.setEchoMode(QLineEdit.Password)
        # 创建一个按钮并连接信号
        btn = QPushButton('提交')
        btn.clicked.connect(self.onSubmit)

        mainLayout.addWidget(self.showPasswordCheckbox, 0, Qt.AlignCenter)

        # 创建一个用于显示错误信息的标签
        self.errorLabel = QLabel('', self)
        self.errorLabel.hide()  # 初始时隐藏该标签

        self.setWindowTitle('头歌助手登录')

        # 对于每个输入框，创建一个水平布局以保持居中
        for label_text, edit_widget in [("用户名", self.nameEdit), 
                                        ("密   码", self.pwdEdit), 
                                        ("作业网址", self.urlEdit)]:
            hbox = QHBoxLayout()
            hbox.addStretch()
            vbox_inner = QVBoxLayout()
            vbox_inner.addWidget(QLabel(label_text))
            vbox_inner.addWidget(edit_widget)
            hbox.addLayout(vbox_inner)
            hbox.addStretch()
            mainLayout.addLayout(hbox)

        # 添加按钮和错误信息标签到布局
        mainLayout.addWidget(btn, 0, Qt.AlignCenter)
        mainLayout.addWidget(self.errorLabel, 0, Qt.AlignCenter)

        # 设置窗口的布局
        self.setLayout(mainLayout)
        self.setGeometry(350, 350, 1200, 700)
        self.centerWindow()

        # 应用样式
        self.applyStyles()

    def togglePasswordVisibility(self):
        if self.showPasswordCheckbox.isChecked():
            self.pwdEdit.setEchoMode(QLineEdit.Normal)
        else:
            self.pwdEdit.setEchoMode(QLineEdit.Password)

    def paintEvent(self, event):
        painter = QPainter(self)
        painter.drawPixmap(self.rect(), self.background)

    def centerWindow(self):
        qr = self.frameGeometry()
        cp = QApplication.desktop().availableGeometry().center()
        qr.moveCenter(cp)
        self.move(qr.topLeft())

    def applyStyles(self):
        self.setStyleSheet("""
            QLineEdit {
                border: 1px solid gray; 
                border-radius: 10px; 
                padding: 5px; 
                background: rgba(255, 255, 255, 100);  
                color: black;
                width: 300px;
                height: 50px;
                font-size: 14pt; /* 设置字号为14磅 */
                font-family: 'Microsoft YaHei'; /* 设置字体为微软雅黑 */
            }
            QLineEdit:focus {
                border: 3px solid black;
            }
            QLabel {
                color: black;
                font-size: 20pt; 
                font-family: "Microsoft YaHei";
                margin-top: 50px; /* 顶部外边距 */
                margin-left: 200px; /* 左侧外边距 */
            }
            QPushButton {
                border: 1px solid gray;
                border-radius: 10px;
                background: rgba(255, 255, 255, 100);
                color: black;
                width: 100px;
                height: 27px;
            }
            /* 样式用于错误信息标签 */
            QLabel#errorLabel {
                color: red;
                font-size: 16pt;
            }
        """)

    def onSubmit(self):
        name = self.nameEdit.text()
        pwd = self.pwdEdit.text()
        url = self.urlEdit.text()
        
        if name == '' or pwd == '' or url == '':
            self.showError("非法输入！")
        else:
            with open('userinfo.json', 'w') as f:
                json.dump({'name': name, 'pwd': pwd, 'url': url}, f)
            self.close()

    def showError(self, message):
        self.errorLabel.setText(message)
        self.errorLabel.show()
        QTimer.singleShot(1200, self.hideError)

    def hideError(self):
        self.errorLabel.hide()
        # 关闭窗口


def show_login():
    app = QApplication(sys.argv)
    ex = MyApp()
    ex.show()
    app.exec_()
if __name__ == '__main__':
    show_image()
    show_login()