Raindrops-editor/syntax.py at editor · JuneSunAt7/Raindrops-editor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
from PyQt5.QtWidgets import QApplication, QMainWindow, QTextEdit

from PyQt5.QtGui import QTextCharFormat, QColor, QFont, QSyntaxHighlighter
from PyQt5.QtCore import  QRegExp


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

        self.highlighting_rules = []

        # Ключевые слова Golang
        keyword_format = QTextCharFormat()
        keyword_format.setForeground(QColor(255, 179, 255))
        keywords = ['break', 'default', 'func', 'interface', 'select',
                    'case', 'defer', 'go', 'map', 'struct',
                    'chan', 'else', 'goto', 'package', 'switch',
                    'const', 'fallthrough', 'if', 'range', 'type',
                    'continue', 'for', 'import', 'return', 'var']
        data_format = QTextCharFormat()
        data_format.setForeground(QColor(255, 255, 179))

        datatypes = ['string', 'int', 'int64', 'bool', 'uint64',
                     'byte', 'rune', 'float32', 'float64',
                     'uint8', 'complex64', 'type', 'nil',
                     'err']

        text_format = QTextCharFormat()
        text_format.setForeground(QColor(153, 179, 255))
        textypes = ['Println', 'Printlf', 'Sprintf', 'Sscanf', 'Sscanln',
                     'Appendf', 'Append', 'FormatString', 'Fprint',
                     'Scan', 'Print', 'Scanln', 'Fscan',
                     'Sprintln']


        for keyword in keywords:
            pattern = QRegExp(r'\b' + keyword + r'\b')
            rule = (pattern, keyword_format)
            self.highlighting_rules.append(rule)

        for data in datatypes:
            pattern = QRegExp(r'\b' + data + r'\b')
            rule = (pattern, data_format)
            self.highlighting_rules.append(rule)

        for text in textypes:
            pattern = QRegExp(r'\b' + text + r'\b')
            rule = (pattern, text_format)
            self.highlighting_rules.append(rule)


        # Комментарии
        comment_format = QTextCharFormat()
        comment_format.setForeground(QColor(179, 255, 179))
        self.highlighting_rules.append((QRegExp(r'//.*'), comment_format))
        # vars

        var_format = QTextCharFormat()
        var_format.setForeground(QColor(255, 179, 179))
        self.highlighting_rules.append((QRegExp(r'var.*\s'),var_format))

        func_format = QTextCharFormat()
        func_format.setForeground(QColor(153, 153, 255))
        self.highlighting_rules.append((QRegExp(r'func.*\s'), func_format))


    def highlightBlock(self, text):
        for rule in self.highlighting_rules:
            pattern = rule[0]
            format = rule[1]
            expression = QRegExp(pattern)
            index = expression.indexIn(text)

            while index >= 0:
                length = expression.matchedLength()
                self.setFormat(index, length, format)
                index = expression.indexIn(text, index + length)

        self.setCurrentBlockState(0)