import locale

from tuikit.core.widget import Widget


class TextField(Widget):

    def __init__(self, value=''):
        Widget.__init__(self)
        self.sizereq.update(10, 1)

        self.allow_focus = True

        self.code = locale.getpreferredencoding()
        if not isinstance(value, str):
            value = str(value, self.code)

        self.value = value
        self.maxlen = None   # unlimited

        self.tw = 0       # real width of text field (minus space for arrows)
        self.curspos = 0  # position of cursor in value
        self.ofs = 0      # position of value beginning on screen

        self.move_end()

    def resize(self, w, h):
        self.tw = self.width - 2
        if self.curspos >= self.tw:
            self.ofs = self.curspos - self.tw

    def set_theme(self, theme):
        self.color = theme.normal

    def draw(self, buffer):
        with buffer.attr(self.color):
            # draw value
            val = self.value + ' ' * self.tw         # add spaces to fill rest of field
            val = val[self.ofs : self.ofs + self.tw]  # cut value - begin from ofs, limit to tw chars
            buffer.puts(val.encode(self.code), 1)

            # draw arrows if content overflows
            c = ' '
            if self.ofs > 0:
                c = '<'
            buffer.putch(c)

            c = ' '
            if len(self.value[self.ofs:]) > self.tw:
                c = '>'
            buffer.putch(c, self.width-1, 0)

            self._cursor = (1 + self.curspos - self.ofs, 0)

    def keypress(self, keyname, char, mod=0):
        Widget.keypress(self, keyname, char, mod)
        map_keyname_to_func = {
            'left':       self.move_left,
            'right':      self.move_right,
            'home':       self.move_home,
            'end':        self.move_end,
            'backspace':  self.move_left_and_del,
            'delete':     self.del_char,
        }
        accepted = False
        if keyname in map_keyname_to_func:
            map_keyname_to_func[keyname]()
            accepted = True
        if char:
            self.add_char(char)
            self.move_right()
            accepted = True
        #if accepted:
            #self.redraw()
        return accepted

    def move_left(self):
        if self.curspos - self.ofs > 1 or (self.ofs == 0 and self.curspos == 1):
            # move cursor
            self.curspos -= 1
        else:
            # move content in field
            if self.ofs > 0:
                self.curspos -= 1
                self.ofs -= 1

    def move_right(self):
        if self.curspos < len(self.value):
            if self.curspos - self.ofs < self.tw - 1 \
            or (self.curspos - self.ofs == self.tw - 1 and self.curspos == len(self.value)-1):
                # move cursor
                self.curspos += 1
            else:
                # move content in field
                self.curspos += 1
                self.ofs += 1

    def move_home(self):
        self.curspos = 0
        self.ofs = 0

    def move_end(self):
        self.curspos = len(self.value)
        if self.curspos >= self.tw:
            self.ofs = self.curspos - self.tw

    def add_char(self, c):
        self.value = self.value[:self.curspos] + c + self.value[self.curspos:]

    def del_char(self):
        self.value = self.value[:self.curspos] + self.value[self.curspos+1:]

    def move_left_and_del(self):
        if self.curspos > 0:
            self.move_left()
            self.del_char()