from textwrap import wrap

import pygame
from pygame.constants import SRCALPHA

from pyntnclick.widgets.base import Widget, Button, convert_color


class TextWidget(Widget):
    def __init__(self, pos, gd, text, size=None, fontname=None, fontsize=None,
                 color=None):
        super(TextWidget, self).__init__(pos, gd, size)
        self.text = text
        constants = self.gd.constants
        self.fontname = fontname or constants.font
        self.fontsize = fontsize or constants.font_size
        self.color = color or constants.text_color

    def prepare(self):
        self.font = self.resource.get_font(self.fontname, self.fontsize)
        self.color = convert_color(self.color)
        self.surface = self.font.render(self.text, True, self.color)
        self.text_rect = self.surface.get_rect()
        if not self.size:
            self.rect.size = self.text_rect.size

    def draw(self, surface):
        if self.visible:
            self.do_prepare()
            surface.blit(self.surface, self.rect)


class LabelWidget(TextWidget):
    def __init__(self, pos, gd, *args, **kwargs):
        constants = gd.constants
        self.padding = kwargs.pop('padding', constants.label_padding)
        self.border = kwargs.pop('border', constants.label_border)
        self.bg_color = convert_color(
                kwargs.pop('bg_color', constants.label_bg_color))
        self.border_color = convert_color(
                kwargs.pop('border_color', constants.label_border_color))
        super(LabelWidget, self).__init__(pos, gd, *args, **kwargs)

    def prepare(self):
        super(LabelWidget, self).prepare()
        if not self.size:
            self.rect.width += 2 * self.padding
            self.rect.height += 2 * self.padding
        new_surface = pygame.Surface(self.rect.size)
        new_surface = new_surface.convert_alpha()
        new_surface.fill(self.bg_color)
        new_surface.blit(self.surface, self.surface.get_rect().move(
                (self.padding, self.padding)))
        if self.border:
            pygame.draw.rect(new_surface, self.border_color,
                             new_surface.get_rect(),
                             self.border)
        self.surface = new_surface

    def draw(self, surface):
        if self.visible:
            self.do_prepare()
            surface.blit(self.surface, self.rect)


class TextButton(Button, TextWidget):
    def __init__(self, pos, gd, *args, **kwargs):
        constants = gd.constants
        self.padding = kwargs.pop('padding', constants.label_padding)
        self.border = kwargs.pop('border', constants.label_border)

        kwargs['color'] = convert_color(
                kwargs.pop('color', constants.button_color))
        self.disabled_color = convert_color(
                kwargs.pop('disabled_color', constants.button_disabled_color))
        self.bg_color = convert_color(
                kwargs.pop('bg_color', constants.button_bg_color))

        super(TextButton, self).__init__(pos, gd, *args, **kwargs)

    def prepare(self):
        super(TextButton, self).prepare()
        text = self.surface
        text_rect = self.text_rect
        color = self.disabled_color if self.disabled else self.color

        width = text_rect.width + self.padding * 2
        height = text_rect.height + self.padding * 2
        if not self.size:
            self.rect.width = max(self.rect.width, width)
            self.rect.height = max(self.rect.height, height)
        self.surface = pygame.Surface(self.rect.size, SRCALPHA)
        self.surface.fill(self.bg_color)
        offset = (
            (self.rect.width - width) / 2 + self.padding,
            (self.rect.height - height) / 2 + self.padding)
        self.surface.blit(text, text.get_rect().move(offset))

        if self.border:
            pygame.draw.rect(self.surface, color, self.surface.get_rect(),
                             self.border)

    def draw(self, surface):
        super(TextButton, self).draw(surface)


class WrappedTextLabel(LabelWidget):
    """A Label Widget that wraps the text to a given maximum width"""

    def __init__(self, pos, gd, *args, **kwargs):
        self.max_width = kwargs.pop('max_width', gd.constants.screen[0] - 50)
        self._wrap_width = None
        self._text_lines = None
        super(WrappedTextLabel, self).__init__(pos, gd, *args, **kwargs)

    def prepare(self):
        if self._wrap_width is None:
            # Start without wrapping
            self._wrap_width = len(self.text) + 1
            self._text_lines = [self.text]

        self.font = self.resource.get_font(self.fontname, self.fontsize)
        self.color = convert_color(self.color)
        self._render()
        self.text_rect = self.surface.get_rect()
        width, height = self.surface.get_rect().size
        while width > self.max_width:
            # Very simplistic approach
            self._wrap_width = self._wrap_width / 2
            self._text_lines = wrap(self.text, self._wrap_width)
            self._render()
            width, height = self.surface.get_rect().size
        if not self.size:
            self.rect.width = max(self.rect.width, width)
            self.rect.height = max(self.rect.height, height)

        if self.border:
            pygame.draw.rect(self.surface, self.border_color,
                             self.surface.get_rect(),
                             self.border)

    def _render(self):
        surfaces = []
        width = 0
        height = 0
        for line in self._text_lines:
            line_surf = self.font.render(line, True, self.color)
            surfaces.append(line_surf)
            width = max(line_surf.get_rect().width, width)
            height += line_surf.get_rect().height

        width += 2 * self.padding
        height += 2 * self.padding

        self.surface = pygame.Surface((width, height))
        self.surface = self.surface.convert_alpha()
        self.surface.fill(self.bg_color)
        height = self.padding
        for line_surf in surfaces:
            rect = pygame.Rect((self.padding, height),
                               (line_surf.get_rect().size))
            self.surface.blit(line_surf, rect)
            height += line_surf.get_rect().height