from typing import Optional

from highlight import Highlight
from highlighted_range import HighlightedRange
from line import Line
from PyQt6.QtCore import *
from PyQt6.QtGui import *
from PyQt6.QtWidgets import *

from settings import Settings
from typing import List
import re


class HighlightRegex(Highlight):

    def __init__(self, query: str, ignore_case: bool, is_regex: bool, hit_background_color: str = "None",
                 line_background_color: str = "None"):
        self.query = query
        self.ignore_case = ignore_case
        self.is_regex = is_regex
        self.regex = self._get_regex()
        self.hit_background_color = hit_background_color
        self.line_background_color = line_background_color

    def _get_regex(self):
        flags = re.IGNORECASE if self.ignore_case else 0
        if self.is_regex:
            return re.compile(self.query, flags=flags)
        else:
            return re.compile(re.escape(self.query), flags=flags)

    def compute_highlight(self, line: Line) -> Optional[List[HighlightedRange]]:
        result = []
        # print("execute regex: %s in %s" % (self.regex, line.line()))
        match_iter = re.finditer(self.regex, line.line())
        for match in match_iter:
            # print("%s" % match)
            start = match.start(0)
            end = match.end(0)
            result.append(HighlightedRange(
                start,
                end - start,
                highlight_full_line=True,
                brush=self.brush(self.hit_background_color),
                brush_full_line=self.brush(self.line_background_color)
            ))

        return result

    def hit_background_brush(self):
        return self.brush(self.hit_background_color)

    @staticmethod
    def brush(color: str) -> QBrush:
        if re.match("[0-9a-f]{6}", color, flags=re.IGNORECASE):
            red = int(color[0:2], 16)
            green = int(color[2:4], 16)
            blue = int(color[4:6], 16)
            return QBrush(QColor(red, green, blue))
        return QBrush()