import pygame as pg
from game.util import load_image, load_images
from game.tilemap import Tilemap

RENDER_SCALE = 2.0
WINDOW_SIZE=(640, 480)

class Editor():
    def __init__(self):
        self.screen = pg.display.set_mode(WINDOW_SIZE)
        self.clock = pg.time.Clock()

        self.width, self.height = pg.display.get_surface().get_size()       
        self.tilemap = Tilemap(self)
        self.display = pg.Surface((self.width/4*3, self.height/6*5))

        self.map = 'test'
        self.path = 'assets/maps/' + str(self.map) + '.json'

        try:
            self.tilemap.load(self.path)
        except FileNotFoundError:
            print('File not Found')

        self.scroll = [0, 0]

        self.assets = {
            "base": load_images("/tiles/test")
        }

        self.tile_list = list(self.assets)

        self.tile_group = 0
        self.tile_variant = 0

        self.movment = [False, False, False, False]

        self.clicking = False
        self.right_clicking = False
        self.shift = False
        self.ongrid = True

    def run(self):
        self.running = True
        while self.running:
            self.clock.tick(60)
            self.update()
            self.draw()
        return self.running

    def update(self):
        self.mpos = pg.mouse.get_pos()
        self.mpos = (self.mpos[0] / RENDER_SCALE, self.mpos[1] / RENDER_SCALE)
        self.tile_pos = (int((self.mpos[0] + self.scroll[0]) // self.tilemap.tile_size), int((self.mpos[1] + self.scroll[1]) // self.tilemap.tile_size))

        self.scroll[0] += (self.movment[1] - self.movment[0]) * 2
        self.scroll[1] += (self.movment[3] - self.movment[2]) * 2

        if self.clicking and self.ongrid:
            self.tilemap.tilemap[str(self.tile_pos[0]) + ';' + str(self.tile_pos[1])] = {'type': self.tile_list[self.tile_group], 'variant': self.tile_variant, 'pos': self.tile_pos}
        if self.right_clicking:
            tile_loc = str(self.tile_pos[0]) + ';' + str(self.tile_pos[1])
            if tile_loc in self.tilemap.tilemap:
                del self.tilemap.tilemap[tile_loc]

        self.current_tile_img = self.assets[self.tile_list[self.tile_group]][self.tile_variant].copy()

        for event in pg.event.get():
            if event.type == pg.QUIT:
                    self.running = False
            if event.type == pg.MOUSEBUTTONDOWN:
                if event.button == 1:
                    self.clicking = True
                if event.button == 3:
                    self.right_clicking = True
                if self.shift:
                    if event.button == 4:
                        self.tile_variant = (self.tile_variant - 1) % len(self.assets[self.tile_list[self.tile_group]])
                    if event.button == 5:
                        self.tile_variant = (self.tile_variant + 1) % len(self.assets[self.tile_list[self.tile_group]])
                else:
                    if event.button == 4:
                        self.tile_group = (self.tile_group - 1) % len(self.tile_list)
                        self.tile_variant = 0
                    if event.button == 5:
                        self.tile_group = (self.tile_group + 1) % len(self.tile_list)
                        self.tile_variant = 0
            if event.type == pg.MOUSEBUTTONUP:
                if event.button == 1:
                    self.clicking = False
                if event.button == 3:
                    self.right_clicking = False

            if event.type == pg.KEYDOWN:
                if event.key == pg.K_a:
                    self.movment[0] = True
                if event.key == pg.K_d:
                    self.movment[1] = True
                if event.key == pg.K_w:
                    self.movment[2] = True
                if event.key == pg.K_s:
                    self.movment[3] = True
                if event.key == pg.K_LSHIFT:
                    self.shift = True
                if event.key == pg.K_o:
                    self.tilemap.save(self.path)
            if event.type == pg.KEYUP:
                if event.key == pg.K_a:
                    self.movment[0] = False
                if event.key == pg.K_d:
                    self.movment[1] = False
                if event.key == pg.K_w:
                    self.movment[2] = False
                if event.key == pg.K_s:
                    self.movment[3] = False
                if event.key == pg.K_LSHIFT:
                    self.shift = False
    
    def draw(self):
        self.display.fill((0, 0, 0))

        render_scroll = (int(self.scroll[0]), int(self.scroll[1]))

        self.display.blit(self.current_tile_img, (10, 10))

        self.tilemap.render(self.display, offset=render_scroll)
        if self.ongrid:
            self.display.blit(self.current_tile_img, (self.tile_pos[0] * self.tilemap.tile_size - self.scroll[0], self.tile_pos[1] * self.tilemap.tile_size - self.scroll[1]))
        else:
            self.display.blit(self.current_tile_img, self.mpos)

        self.screen.blit(pg.transform.scale(self.display, self.screen.get_size()))
        pg.display.update()