from pygame.sprite import Sprite

from mamba.data import load_image
from mamba.constants import TILE_SIZE


def tile_sizify(pos):
    ts_x, ts_y = TILE_SIZE
    p_x, p_y = pos
    return (ts_x * p_x, ts_y * p_y)


class BaseSprite(Sprite):
    tileset = 'common'

    def __init__(self, image_name, *groups):
        super(BaseSprite, self).__init__(*groups)
        self.image_name = image_name
        self.load_image(image_name, self.tileset)

    def load_image(self, image_name, tileset):
        if tileset is None:
            tileset = self.tileset
        self.image = load_image('tiles/%s/%s.png' % (tileset, image_name))

    def set_tile_pos(self, tile_pos):
        self.rect = self.image.get_rect().move(tile_sizify(tile_pos))


class TileSprite(BaseSprite):
    def __init__(self, tileset_name, image_name, *groups):
        self.tileset = tileset_name
        super(TileSprite, self).__init__(image_name, *groups)


class SingleImageTileSprite(TileSprite):
    def __init__(self, tileset, *groups):
        super(SingleImageTileSprite, self).__init__(tileset, self.image_name)


class DoorSprite(TileSprite):
    def __init__(self, tileset_name, colour, *groups):
        self.colour = colour
        image_name = 'door_%s' % (colour,)
        super(DoorSprite, self).__init__(tileset_name, image_name, *groups)


class EntrySprite(SingleImageTileSprite):
    # image_name = 'entry'
    image_name = 'puddle-r'


class ExitSprite(SingleImageTileSprite):
    # image_name = 'exit'
    image_name = 'puddle-tl'


class PuddleSprite(SingleImageTileSprite):
    image_name = 'puddle'