import re
import string
from kaa import cursor, keyboard
from kaa.filetype.default import modebase
from kaa.theme import Theme, Style
from kaa.exceptions import KaaError

DialogThemes = {
    'basic': [
        Style('default', 'Base3', 'Base02', False, False),
        Style('caption', 'Orange', 'Base02', nowrap=True),
        Style('blank-line-header', 'Blue', None),

        Style('activemark', 'Base02', 'Yellow', nowrap=True),
        Style('nonactivemark', 'Yellow', 'Base02', nowrap=True),

        Style('button', 'Base3', 'Base01', nowrap=True),
        Style('button.shortcut', 'Base3', 'Base01', underline=True,
              nowrap=True),

        # todo: right-button with rjust should be defined in another name.
        Style('right-button', 'Base3', 'Base01', rjust=True, nowrap=True),
        Style('right-button.checked', 'Base3', 'Orange', rjust=True,
              nowrap=True),
        Style('right-button.shortcut', 'Base3', 'Base01', underline=True,
              rjust=True, nowrap=True),
        Style('right-button.shortcut.checked', 'Base3', 'Orange',
              underline=True, rjust=True, nowrap=True),

        Style('selectitem', 'Cyan', 'Base02', nowrap=True),
        Style('selectitem2', 'Yellow', 'Base02', nowrap=True),
        Style('selectitem-active', 'Base02', 'Yellow', nowrap=True),

        Style('selectphrase', 'Black', 'Cyan', nowrap=True),
        Style('selectphrase2', 'Yellow', 'Base02', nowrap=True),
        Style('selectphrase-active', 'Base02', 'Yellow', nowrap=True),

        Style('dialog-border', 'Blue', 'Base03'),
    ],
}


class DialogCursor(cursor.Cursor):

    def __init__(self, wnd, ranges):
        super().__init__(wnd)
        self.ranges = ranges

    def _findrange(self, curpos):
        for func in self.ranges:
            f, t = func(self)
            if f <= curpos <= t:
                return f, t
        return self.ranges[0](self)

    def adjust_nextpos(self, curpos, nextpos):
        f, t = self._findrange(curpos)
        if nextpos >= t:
            return t
        if nextpos <= f:
            return f
        return nextpos


def _MarkRange(mark, cursor):
    return cursor.wnd.document.marks[mark]


def MarkRange(mark):
    return lambda cursor: _MarkRange(mark, cursor)


class DialogMode(modebase.ModeBase):
    NO_WRAPINDENT = False
    DEFAULT_STATUS_MSG = ''

    autoshrink = False
    suspend_autoshrink = False
    stack_upper = True
    border = False
    min_height = 1

    def init_themes(self):
        super().init_themes()
        self.themes.append(DialogThemes)

    def init_tokenizers(self):
        self.tokenizers = []

    def on_add_window(self, wnd):
        super().on_add_window(wnd)

    def calc_width(self, wnd):
        ret = wnd.mainframe.getsize()[0]
        if self.border:
            ret -= wnd.mainframe.BORDER_WIDTH * 2
        return ret

    def calc_height(self, wnd):
        maxw, maxh = wnd.mainframe.getsize()
        height = wnd.screen.get_total_height(maxh)
        ret = min(maxh, height)
        if height < self.min_height:
            return self.min_height
        return ret

    def calc_position(self, wnd):
        w = self.calc_width(wnd)

        # temporary set screen size
        # should be resized later soon
        wnd.screen.setsize(w, wnd.screen.height)

        height = self.calc_height(wnd)

        
        if not self.border:
            height = min(height, wnd.mainframe.messagebar.rect[1])
            top = wnd.mainframe.messagebar.rect[1] - height
            return 0, top, wnd.mainframe.width, top + height
        else:
            border_width = wnd.mainframe.BORDER_WIDTH
            height = min(
                height,
                wnd.mainframe.messagebar.rect[1] - border_width * 2)
            top = wnd.mainframe.messagebar.rect[1] - height - border_width
            return (
                border_width, top, wnd.mainframe.width - border_width, top + height
            )

    def resize_inputs(self, wnd, inputs):
        rects = [w.document.mode.calc_position(w) for w in inputs]
        rc_input = self.calc_position(wnd)

        heights = [b - t for (l, t, r, b) in rects]

        l, t, r, b = rc_input

        if self.stack_upper:
            t = t - sum(heights)
        else:
            t = b
        if t < 0:
            raise KaaError(
                'Invalid window position {!r}:{!r}/{!r}'.format(self, rects, rc_input))

        positions = []
        for wl, wt, wr, wb in rects:
            b = t + (wb - wt)
            positions.append((l, t, r, b))
            t = b + 1

        return rc_input, positions

    def on_start(self, wnd):
        pass

    def run_autoshrink(self):
        for wnd in self.document.wnds:
            w, h = wnd.getsize()
            l, t, r, b = self.calc_position(wnd)
            newh = b - t
            # resize window if number of rows changed
            if newh != h:
                wnd.get_label('popup').on_console_resized()
                wnd.mainframe.on_console_resized()
                f, t = wnd.screen.get_visible_range()
                if t == self.document.endpos():
                    # display as many rows as we can
                    wnd.screen.locate(t, bottom=True, align_always=True)
                    wnd.cursor.setpos(wnd.cursor.pos)

    def on_document_updated(self, pos, inslen, dellen):
        super().on_document_updated(pos, inslen, dellen)

        if self.autoshrink:
            if not self.suspend_autoshrink:
                self.run_autoshrink()


class FormBuilder:
    re_accel = re.compile(r'&.')

    def __init__(self, document):
        self.document = document
        self.doc_autoshrink = getattr(self.document.mode, 'autoshrink', None)

    def __enter__(self):
        if self.doc_autoshrink:
            self.document.mode.suspend_autoshrink = True
        return self

    def __exit__(self, type, value, traceback):
        if self.document.mode.suspend_autoshrink:
            self.document.mode.suspend_autoshrink = False
            self.document.mode.run_autoshrink()

    def append_text(
            self, stylename, text, mark_start=None, mark_end=None,
            mark_pair=None, on_shortcut=None, shortcut_style=None,
            shortcut_mark=None, shortcut_need_alt=True):

        # marks should not be moved until the document completed.
        self.document.marks.locked = True
        try:
            start = self.document.endpos()

            if shortcut_style is None:
                shortcut_style = stylename

            if stylename:
                style_id = self.document.mode.get_styleid(stylename)

            if shortcut_style:
                shortcut_style_id = self.document.mode.get_styleid(
                    shortcut_style)

            # convert & escape to char and build shortcut dict
            lastpos = 0
            for m in self.re_accel.finditer(text):
                span = m.span()
                if span[0] != lastpos:
                    f = self.document.endpos()
                    self.document.append(text[lastpos:span[0]])
                    if stylename:
                        self.document.setstyles(
                            f, self.document.endpos(), style_id, update=False)

                f = self.document.endpos()
                c = m.group()[1]
                self.document.append(c)
                if c != '&':
                    if on_shortcut:
                        self.document.mode.keybind.add_keybind(
                            {(keyboard.alt, c.lower()): on_shortcut})
                        if not shortcut_need_alt:
                            self.document.mode.keybind.add_keybind(
                                {(c.lower()): on_shortcut})
                            self.document.mode.keybind.add_keybind(
                                {(c.upper()): on_shortcut})

                    if shortcut_mark:
                        self.document.marks[shortcut_mark] = f

                    if shortcut_style:
                        self.document.setstyles(
                            f,
                            f + 1,
                            shortcut_style_id,
                            update=False)

                lastpos = m.end()

            else:
                if lastpos != len(text):
                    f = self.document.endpos()
                    self.document.append(text[lastpos:])
                    if stylename:
                        self.document.setstyles(f, self.document.endpos(),
                                                style_id, update=False)

            if mark_start is not None:
                self.document.marks[mark_start] = start

            if mark_end is not None:
                self.document.marks[mark_end] = self.document.endpos()

            if mark_pair is not None:
                self.document.marks[mark_pair] = (
                    start, self.document.endpos())

            # Dialog texts are not undoable
            if self.document.undo:
                self.document.undo.clear()

        finally:
            self.document.marks.locked = False
