from pygame import image

from gamelib import data
from gamelib.gui_base import Drawable, TextButton, font_auto, Button


class ImageDrawable(Drawable):

    def __init__(self, rect, image):
        super(ImageDrawable, self).__init__(rect)
        self.image = image

    def draw(self, surface):
        surface.blit(self.image, (self.rect[0], self.rect[1]))


class BigButton(TextButton):
    WIDTH = 128
    HEIGHT = 64
    BG_IMAGE_NORMAL = image.load(data.filepath('images/button_normal.png'))
    BG_IMAGE_DOWN = image.load(data.filepath('images/button_down.png'))

    def __init__(self, pos, text, font=font_auto, shadow=True):
        rect1 = (0, 0, self.WIDTH, self.HEIGHT)
        n = ImageDrawable(rect1, self.BG_IMAGE_NORMAL)
        d = ImageDrawable(rect1, self.BG_IMAGE_DOWN)
        rect2 = (pos[0], pos[1], self.WIDTH, self.HEIGHT)
        super(BigButton, self).__init__(rect2, n, d, text, font, shadow)


class IconButton(Button):
    WIDTH = 64
    HEIGHT = 64
    BG_IMAGE_NORMAL = image.load(data.filepath('images/research_normal.png'))
    BG_IMAGE_DOWN = image.load(data.filepath('images/research_down.png'))

    def __init__(self, pos, name):
        rect = (pos[0], pos[1], self.WIDTH, self.HEIGHT)
        n_icon = image.load(data.filepath('images/icons/%s_normal.png' % name))
        d_icon = image.load(data.filepath('images/icons/%s_down.png' % name))
        n = ImageDrawable(rect, self.BG_IMAGE_NORMAL.copy())
        n.image.blit(n_icon, (0, 0))
        d = ImageDrawable(rect, self.BG_IMAGE_DOWN.copy())
        d.image.blit(d_icon, (0, 0))
        rect2 = (pos[0], pos[1], self.WIDTH, self.HEIGHT)
        super(IconButton, self).__init__(rect2, n, d)