import nodesnim
import random
randomize()


Window("Snake game")


type
  Snake = ref object
    dir: Vector2Obj
    size: Vector2Obj
    food: Vector2Obj
    body: seq[Vector2Obj]


build:
  - Scene main:
    - Canvas canvas

  - Scene game_over:
    - Label label_go:
      call:
        setText("Game Over")
        setTextAlign(0.5, 0.5, 0.5, 0.5)
        setSizeAnchor(1, 1)

var
  snake = Snake(
    dir: Vector2(),
    body: @[Vector2(0, 0)],
    size: Vector2(40, 40),
    food: Vector2(160, 160))
  time = 0


addKeyAction("forward", "w")
addKeyAction("backward", "s")
addKeyAction("left", "a")
addKeyAction("right", "d")


canvas@on_process(self):
  if isActionJustPressed("forward"):
    snake.dir = Vector2(0, -40)
  elif isActionJustPressed("backward"):
    snake.dir = Vector2(0, 40)
  elif isActionJustPressed("left"):
    snake.dir = Vector2(-40, 0)
  elif isActionJustPressed("right"):
    snake.dir = Vector2(40, 0)
  if time < 10:
    inc time
    return
  canvas.resize(main.rect_size.x, main.rect_size.y)
  let
    grid_size = Vector2(
      (main.rect_size.x.int div snake.size.x.int).float,
      (main.rect_size.y.int div snake.size.y.int).float
    )
  canvas.fill(Color(0, 0, 0, 0))


  if snake.dir.y < 0:
    for i in countdown(snake.body.len()-1, 0):
      if i > 0:
        snake.body[i] = Vector2(snake.body[i-1].x, snake.body[i-1].y)
      else:
        if snake.body[i].y > 0:
          snake.body[i].y += snake.dir.y
        else:
          snake.body[i].y = grid_size.y*snake.size.y - snake.size.y
  elif snake.dir.y > 0:
    for i in countdown(snake.body.len()-1, 0):
      if i > 0:
        snake.body[i] = Vector2(snake.body[i-1].x, snake.body[i-1].y)
      else:
        if snake.body[i].y < grid_size.y*snake.size.y - snake.size.y:
          snake.body[i].y += snake.dir.y
        else:
          snake.body[i].y = 0

  elif snake.dir.x < 0:
    for i in countdown(snake.body.len()-1, 0):
      if i > 0:
        snake.body[i] = Vector2(snake.body[i-1].x, snake.body[i-1].y)
      else:
        if snake.body[i].x > 0:
          snake.body[i].x += snake.dir.x
        else:
          snake.body[i].x = grid_size.x*snake.size.x - snake.size.x
  elif snake.dir.x > 0:
    for i in countdown(snake.body.len()-1, 0):
      if i > 0:
        snake.body[i] = Vector2(snake.body[i-1].x, snake.body[i-1].y)
      else:
        if snake.body[i].x < grid_size.x*snake.size.x - snake.size.x:
          snake.body[i].x += snake.dir.x
        else:
          snake.body[i].x = 0

  if snake.body[0] == snake.food:
    snake.body.add(snake.body[^1])
    snake.food.x = (rand(grid_size.x.int-1) * snake.size.x.int).float
    snake.food.y = (rand(grid_size.y.int-1) * snake.size.y.int).float
  elif snake.body[0] in snake.body[1..^1]:
    changeScene("game_over")
  # draw
  for i in snake.body:
    canvas.fillRect(i.x, i.y, snake.size.x, snake.size.y, Color(1f, 1f, 1f))
  # head
  canvas.fillRect(
    snake.body[0].x, snake.body[0].y,
    snake.size.x, snake.size.y,
    Color(0xaaccffff'u32))
  # food
  canvas.fillRect(
    snake.food.x, snake.food.y,
    snake.size.x, snake.size.y,
    Color(0xffccaaff'u32))
  time = 0


addScene(game_over)
addMainScene(main)
windowLaunch()