# -*- coding: utf-8 -*-

class EditBox(Widget):

    def __init__(self, width=20, height=20, text=''):

        Widget.__init__(self, width, height)

        self.xofs = 0

        self.yofs = 0

        # cursor

        self.cline = 0

        self.cpos = 0

        # selection

        self.sline = 0

        self.spos = 0

        self.connect('draw', self.on_draw)

        self.connect('keypress', self.on_keypress)

        self.connect('mousewheel', self.on_mousewheel)

        self.addevents('scroll', 'areasize')

        self.set_text(text)

    def on_draw(self, screen, x, y):

        for j in range(self.height):

            if self.yofs + j >= len(self.lines):

                break

            line = self.lines[self.yofs + j]

            #if len(line) < self.width:

                #line += ' ' * (self.width - len(line))

            #else:

                #line = line[:self.width]

            screen.puts(x, y + j, line)

        screen.showcursor(x + self.get_cpos() - self.xofs, y + self.cline - self.yofs)

    def on_keypress(self, keyname, char):

        if keyname:

            if keyname == 'left':

                self.move_left()

            if keyname == 'right':

                self.move_right()

            if keyname == 'home':

                self.move_home()

            if keyname == 'end':

                self.move_end()

            if keyname == 'up':

                self.move_up()

            if keyname == 'down':

                self.move_down()

            if keyname == 'pageup':

                self.move_pageup()

            if keyname == 'pagedown':

                self.move_pagedown()

            if keyname == 'backspace':

                if self.cline > 0 or self.cpos > 0:

                    self.move_left()

                    self.del_char()

            if keyname == 'delete':

                self.del_char()

            if keyname == 'enter':

                self.add_newline()

                self.move_right()

        if char:

            self.add_char(char)

            self.move_right()

        self.redraw()

    def on_mousewheel(self, ev):

        # up

        if ev.button == 4:

            self.move_up()

        # down

        if ev.button == 5:

            self.move_down()

        self.redraw()

    def set_text(self, text):

        self.lines = text.split('\n')

        self.handle('areasize')

    def get_text(self):

        return '\n'.join(self.lines)

    def get_linelen(self):

        return len(self.lines[self.cline])

    def get_cpos(self):

        if self.cpos > self.get_linelen():

            return self.get_linelen()

        return self.cpos

    def set_yofs(self, yofs):

        if yofs > len(self.lines) - self.height:

            yofs = len(self.lines) - self.height

        if yofs < 0:

            yofs = 0

        self.yofs = yofs

        self.handle('scroll')

    def move_left(self):

        if self.cpos > 0:

            self.cpos = self.get_cpos() - 1

        else:

            if self.move_up():

                self.cpos = self.get_linelen()

    def move_right(self):

        if self.cpos < self.get_linelen():

            self.cpos += 1

        else:

            if self.move_down():

                self.cpos = 0

    def move_home(self):

        self.cpos = 0

    def move_end(self):

        self.cpos = self.get_linelen()

    def move_up(self):

        if self.cline > 0:

            self.cline -= 1

            if self.cline < self.yofs:

                self.set_yofs(self.cline)

            return True

        return False

    def move_down(self):

        if self.cline < len(self.lines) - 1:

            self.cline += 1

            if self.cline > self.yofs + self.height - 1:

                self.set_yofs(self.cline - (self.height - 1))

            return True

        return False

    def move_pageup(self):

        if self.cline >= self.height - 1:

            self.cline -= self.height - 1

            self.set_yofs(self.yofs - (self.height - 1))

        else:

            self.cline = 0

            self.set_yofs(0)

    def move_pagedown(self):

        if self.cline <= len(self.lines) - (self.height - 1):

            self.cline += self.height - 1

            self.set_yofs(self.yofs + (self.height - 1))

        else:

            self.cline = len(self.lines) - 1

            self.set_yofs(self.cline)

    def move_pagefirst(self):

        self.cline = 0

        self.set_yofs(0)

    def move_pagelast(self):

        self.cline = len(self.lines) - 1

        self.set_yofs(self.cline)

    def add_char(self, c):

        ln = self.lines[self.cline]

        cpos = self.get_cpos()

        self.lines[self.cline] = ln[:cpos] + c + ln[cpos:]

        self.cpos = cpos

    def add_newline(self):

        ln = self.lines[self.cline]

        cpos = self.get_cpos()

        self.lines[self.cline] = ln[cpos:]

        self.lines.insert(self.cline, ln[:cpos])

        self.handle('areasize')

    def del_char(self):

        ln = self.lines[self.cline]

        cpos = self.get_cpos()

        if cpos == self.get_linelen():

            if self.cline + 1 < len(self.lines):

                self.lines[self.cline] = self.lines[self.cline] + self.lines[self.cline+1]

                del self.lines[self.cline+1]

                self.handle('areasize')

        else:

            self.lines[self.cline] = ln[:cpos] + ln[cpos+1:]