import os
import pygame
from kb import tractor_kb
import pytholog as pl
import random
from config import TILE_SIZE

class Tile(pygame.sprite.Sprite):

    def __init__(self, id, field):
        super().__init__()
        self.id = id
        x = id%16
        y = id//16

        self.field = field

        # temporary solution to have vegetables act as obstacles
        if random.randint(1, 10) % 3 == 0:
            vegetables = tractor_kb.query(pl.Expr("warzywo(Nazwa_warzywa)"))
            random_vegetable = vegetables[random.randint(0, len(vegetables)-1)]['Nazwa_warzywa']
            self.set_type(random_vegetable)
        else:
            self.set_type('grass')

        self.faza = 'posadzono'

        self.rect = self.image.get_rect()
        self.rect.topleft = (x * TILE_SIZE, y * TILE_SIZE)
    

    def draw(self, surface):
        self.tiles.draw(surface)

    def set_type(self, type):
        self.type = type
        if self.type == 'grass':
            image_path = "images/grass.png"
        else:
            image_path = f"images/vegetables/{self.type}.png"
            if not os.path.exists(image_path):
                image_path = "images/question.jpg"
        
        self.image = pygame.image.load(image_path).convert()
        self.image = pygame.transform.scale(self.image, (TILE_SIZE, TILE_SIZE))