import pygame
import random
import time
from Jablko import Jablko
from Kierunek import Kierunek
from Waz import Waz

SZEROKOSC_EKRANU = 800
WYSOKOSC_EKRANU = 608

tlo = pygame.Surface((SZEROKOSC_EKRANU, WYSOKOSC_EKRANU))

for i in range(25):
    for j in range(19):
        obraz = pygame.image.load("images/background.png")
        maska = (random.randrange(0,20), random.randrange(0,20), random.randrange(0,20))

        obraz.fill(maska, special_flags=pygame.BLEND_ADD)
        tlo.blit(obraz, (i*32, j*32))

pygame.init()
pygame.display.set_caption("Snakeee")
ekran = pygame.display.set_mode([SZEROKOSC_EKRANU, WYSOKOSC_EKRANU])
zegar = pygame.time.Clock()

# wąż
waz = Waz()
PORUSZ_WEZEM = pygame.USEREVENT+1
pygame.time.set_timer(PORUSZ_WEZEM, 200) # co 200 ms, wykonuje się przemieszczenie węża


jablko = Jablko()
jablka = pygame.sprite.Group()
jablka.add(jablko) 

gra_dziala = True
while gra_dziala:
    for zdarzenie in pygame.event.get():
        if zdarzenie.type == pygame.KEYDOWN:
            if zdarzenie == pygame.K_ESCAPE:
                gra_dziala = False
            if zdarzenie.key == pygame.K_w:
                waz.zmien_kierunek(Kierunek.GORA)
            if zdarzenie.key == pygame.K_s:
                waz.zmien_kierunek(Kierunek.DOL)
            if zdarzenie.key == pygame.K_a:
                waz.zmien_kierunek(Kierunek.LEWO)
            if zdarzenie.key == pygame.K_d:
                waz.zmien_kierunek(Kierunek.PRAWO)

        elif zdarzenie.type == PORUSZ_WEZEM:
            waz.aktualizuj()
        elif zdarzenie.type == pygame.QUIT:
            gra_dziala = False

    # czy snake dotknal jablko
    kolizja_z_jablkiem = pygame.sprite.spritecollideany(waz, jablka)
    if kolizja_z_jablkiem != None:
        kolizja_z_jablkiem.kill()
        waz.jedz_jablko()
        jablko = Jablko()
        jablka.add(jablko)

    ekran.blit(tlo, (0,0))

    waz.rysuj_segmenty(ekran)

    ekran.blit(waz.obraz, waz.rect)

    for jablko in jablka:
        ekran.blit(jablko.obraz, jablko.rect)

    if waz.sprawdz_kolizje():
        gra_dziala = False
        
    pygame.display.flip()
    zegar.tick(30)

time.sleep(1)
pygame.quit()