from PyQt5 import uic
import sys
import re
from PyQt5.QtWidgets import QApplication, QWidget, QFileDialog, QMessageBox
from PyQt5.QtGui import QSyntaxHighlighter, QTextCharFormat, QColor, QFont
from PyQt5.QtCore import QRegularExpression

class MyWindow(QWidget):

    def __init__(self):
        super().__init__()
        self.init_ui()

    def init_ui(self):
        self.ui = uic.loadUi("./文本編輯器.ui")
        print(self.ui.__dict__)  
        self.ui.setWindowTitle("文本編輯器")  # 修改標(biāo)題的名稱

        self.textEdit = self.ui.textEdit
        self.syntaxHighlighter = SyntaxHighlighter(self.textEdit.document())

        self.statusBar = self.ui.statusBar
        self.statusBar().showMessage('狀態(tài)欄')
        # 加載好ui文件之后再進(jìn)行設(shè)置菜單欄
        menu = self.ui.menubar
        file_menu = menu.addMenu("文件")
        # 清空文本
        self.create_file = file_menu.addAction("新建")
        self.create_file.setShortcut('Ctrl+N')
        self.create_file.triggered.connect(self.newFile)

        # 打開文件動作
        self.open = file_menu.addAction("打開")
        self.open.setShortcut("Ctrl+O")
        self.open.triggered.connect(self.openFile)


        self.save = file_menu.addAction("保存")
        self.save.setShortcut("Ctrl+S")
        self.save.triggered.connect(self.saveFile)

        self.exit = file_menu.addAction("退出")
        self.exit.setShortcut("Ctrl+E")
        self.exit.triggered.connect(self.closeEvent)


        edit_menu = menu.addMenu("編輯")
        edit_menu.addAction("復(fù)制")
        edit_menu.addAction("粘貼")
        edit_menu.addAction("剪切")


    def newFile(self):
        # 清空文本編輯區(qū)域
        self.textEdit.clear()
        self.statusBar().showMessage('新文件')

    def openFile(self):
        # 打開文件對話框,讀取文件內(nèi)容，將其顯示在textEdit中
        fileName, _ = QFileDialog.getOpenFileName(self, '打開文件', '', '文本文件 (*.txt);;所有文件 (*)')
        if fileName:
            with open(fileName, 'r', encoding='utf-8') as file:
                self.textEdit.setText(file.read())
        self.statusBar().showMessage('打開新文件')

    def saveFile(self):
        # 保存文件對話框，打開一個(gè)文件，將textEdit中的內(nèi)容寫進(jìn)去
        fileName, _ = QFileDialog.getSaveFileName(self, '保存文件', '', '文本文件 (*.txt);;所有文件 (*)')
        if fileName:
            with open(fileName, 'w', encoding='utf-8') as file:
                file.write(self.textEdit.toPlainText())

    def closeEvent(self, event):
        # 關(guān)閉窗口時(shí)提示保存
        reply = QMessageBox.question(self, '退出', '你是否想保存文件？',
                                     QMessageBox.Yes | QMessageBox.No | QMessageBox.Cancel, QMessageBox.Cancel)

        if reply == QMessageBox.Yes:
            self.saveFile()
            event.accept()
        elif reply == QMessageBox.No:
            event.accept()
        else:
            event.ignore()


class SyntaxHighlighter(QSyntaxHighlighter):
    def __init__(self, parent=None):
        super().__init__(parent)

        keywordFormat = QTextCharFormat()
        keywordFormat.setForeground(QColor("blue"))
        keywordFormat.setFontWeight(QFont.Bold)

        commentFormat = QTextCharFormat()
        commentFormat.setForeground(QColor("green"))
        commentFormat.setFontItalic(True)

        stringFormat = QTextCharFormat()
        stringFormat.setForeground(QColor("red"))
        stringFormat.setFontWeight(QFont.Bold)

        keywords = ["def", "class", "import", "from", "if", "else", "elif", "for", "while", "return", "try", "except"]

        self.highlightingRules = []

        for word in keywords:
            escaped_word = re.escape(word)
            rule = QRegularExpression(rf"\b{escaped_word}\b")
            self.highlightingRules.append((rule, keywordFormat))

        rule = QRegularExpression(r"#[^\n]*")
        self.highlightingRules.append((rule, commentFormat))

        rule = QRegularExpression(r'"([^"\\]|\\.)*"')
        self.highlightingRules.append((rule, stringFormat))

    def highlightBlock(self, text):
        for pattern, format in self.highlightingRules:
            matchIterator = pattern.globalMatch(text)
            while matchIterator.hasNext():
                match = matchIterator.next()
                self.setFormat(match.capturedStart(), match.capturedLength(), format)


if __name__=="__main__":
    app = QApplication(sys.argv)

    w = MyWindow()
    # 展示窗口
    w.ui.show()

    app.exec()