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

from tuikit.widget import Widget

class Button(Widget):

    '''Clickable button.'''

    def __init__(self, label=''):

        '''Create button with given label, size according to label.'''

        w = len(label) + 4

        h = 1

        Widget.__init__(self, w, h)

        #: Button label.

        self.label = label

        #: Text or graphics to be added before label

        self.prefix = '['

        #: Text or graphics to be added after label

        self.suffix = ']'

        #: How should label be aligned if button has excess space - center | left | right

        self.align = 'center'

        self.bg = 'button'

        self.bghi = 'button-active'

        self.highlight = False

        # size

        self.sizereq.w = w

        self.sizereq.h = h

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

        self.connect('mousedown', self.on_mousedown)

        self.connect('mouseup', self.on_mouseup)

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

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

        pad = self.width - len(self.label) - len(self.prefix) - len(self.suffix)

        lpad, rpad = self._divide_padding(pad)

        screen.pushcolor(self.getcolor())

        # prefix

        screen.puts(x, y, self.prefix)

        pos = len(self.prefix)

        # left pad

        screen.puts(x + pos, y, ' ' * lpad)

        pos += lpad

        # label

        screen.puts(x + pos, y, self.label)

        pos += len(self.label)

        # right pad

        screen.puts(x + pos, y, ' ' * rpad)

        pos += rpad

        # suffix

        screen.puts(x + pos, y, self.suffix)

        screen.popcolor()

    def on_mousedown(self, ev):

        self.highlight = True

        self.redraw()

    def on_mouseup(self, ev):

        self.highlight = False

        self.redraw()

        if self.enclose(ev.px, ev.py):

            self.handle('click')

    def on_keypress(self, keyname, char):

        if keyname == 'enter':

            self.handle('click')

    def getcolor(self):

        if self.highlight or self.hasfocus():

            return self.bghi

        return self.bg

    def _divide_padding(self, pad):

        # default is 'left'

        lpad, rpad = 0, pad

        if self.align == 'center':

            lpad = pad // 2

            rpad = pad - lpad

        elif self.align == 'right':

            lpad, rpad = pad, 0

        return lpad, rpad