#include "snake.h"
#include <stdlib.h>

// Pridá novú časť hada na danú pozíciu a vráti novú hlavu hada
struct snake* add_snake(struct snake* snake, int x, int y) {
    struct snake* new_head = (struct snake*) malloc(sizeof(struct snake));
    new_head->x = x;
    new_head->y = y;
    new_head->next = snake;
    return new_head;
}

// Odstráni poslednú časť hada a vráti novú hlavu hada
struct snake* remove_snake(struct snake* snake) {
    if (snake == NULL || snake->next == NULL) {
        return NULL;
    }
    struct snake* temp = snake;
    while(temp->next->next != NULL) {
        temp = temp->next;
    }
    free(temp->next);
    temp->next = NULL;
    return snake;
}

// Zistí, či sú dané súradnice súčasťou hada
int is_snake(struct snake* snake, int x, int y) {
    while(snake != NULL) {
        if(snake->x == x && snake->y == y) {
            return 1;
        }
        snake = snake->next;
    }
    return 0;
}

// Uvoľní pamäť všetkých častí hada
void free_snake(struct snake* snake) {
    struct snake* temp;
    while(snake != NULL) {
        temp = snake;
        snake = snake->next;
        free(temp);
    }
}

// Zmení stav hry podľa pravidiel hry
int step_state(struct state* state) {
    int new_x = state->snake->x + state->sx;
    int new_y = state->snake->y + state->sy;

    // Ak je nová pozícia mimo hracej plochy, vráti END_WALL
    if(new_x < 0 || new_x >= state->width || new_y < 0 || new_y >= state->height) {
        return END_WALL;
    }

    // Ak je nová pozícia na hadovi, vráti END_SNAKE
    if(is_snake(state->snake, new_x, new_y)) {
        return END_SNAKE;
    }

    // Ak je nová pozícia na jedle, označí jedlo ako zjedené a pridá novú časť hada
    for(int i = 0; i < FOOD_COUNT; i++) {
        if(state->foodx[i] == new_x && state->foody[i] == new_y) {
            state->foodx[i] = -1;
            state->foody[i] = -1;
            state->snake = add_snake(state->snake, new_x, new_y);
            for(int j = 0; j < FOOD_COUNT; j++) {
                if(state->foodx[j] != -1) {
                    return END_CONTINUE;
                }
            }
            return END_FOOD;
        }
    }

    // Ak je nová pozícia voľná, pridá novú časť hada a odstráni poslednú časť
    state->snake = add_snake(state->snake, new_x, new_y);
    state->snake = remove_snake(state->snake);

    return END_CONTINUE;
}