let snake;
let food;
let w;
let h;
let size = 20;

function setup() {
    createCanvas(600, 600);
    w = floor(width / size);
    h = floor(width / size);
    frameRate(8);
    snake = new Snake();
    generateFood();
}

function draw() {
    scale(size)
    background(220);

    if (snake.eat(food)) {
        generateFood();
    }

    if (snake.dead()) {
        print("END");
        background(200, 0, 0);
        noLoop();
    }

    snake.update();
    snake.show();

    // draw food
    noStroke();
    fill(255, 0, 0);
    rect(food.x, food.y, 1, 1);
}

function generateFood() {
    let x = floor(random(w));
    let y = floor(random(h));
    food = createVector(x, y);
}

function keyPressed() {
    if (keyCode === LEFT_ARROW) {
        snake.setDir(-1, 0);
    } else if (keyCode === RIGHT_ARROW) {
        snake.setDir(1, 0);
    } else if (keyCode === UP_ARROW) {
        snake.setDir(0, -1);
    } else if (keyCode === DOWN_ARROW) {
        snake.setDir(0, 1);
    }
}