import os

import pygame
from pygame.sprite import Sprite

from gamelib import data
from gamelib.gui_base import *


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 MainMenuButton(TextButton):
  WIDTH = 128
  HEIGHT = 64
  BACKGROUND_IMAGE_NORMAL = pygame.image.load(data.filepath('images/button_normal.png'))
  BACKGROUND_IMAGE_DOWN = pygame.image.load(data.filepath('images/button_down.png'))
  
  def __init__(self, pos, text):
    rect = (0, 0, self.WIDTH, self.HEIGHT)
    drawable_normal = ImageDrawable(rect, self.BACKGROUND_IMAGE_NORMAL)
    drawable_down = ImageDrawable(rect, self.BACKGROUND_IMAGE_DOWN)
    super(MainMenuButton, self).__init__((pos[0],pos[1], self.WIDTH, self.HEIGHT), drawable_normal, drawable_down, text)