#include "../world.h"
#include "../npc.h"
#include "../player.h"
#include "../aabb.h"
#include <SDL3/SDL_events.h>
#include <SDL3/SDL_keycode.h>
#include <SDL3/SDL_oldnames.h>
#include "../map.h"

void UpdateWorld(F32 delta, World *world)
{
  UpdateBandit(delta, &world->bandit, world);
  UpdateNPCs(delta, world);
  PlayerUpdate(delta, &world->player);
}

void UpdateNPCs(F32 delta, World *world)
{
  for (U32 i = 0; i < world->npcCount; i++)
  {
    NPC *npc = &world->npcs[i];
    UpdateNPC(delta, npc, world);
    if (
        world->player.controls.shot && AABB_Slab(world->player.pos, world->player.shotPos, npc->collision) && npc->currentArea == world->player.currentArea)
    {
      printf("You shot %*.s\n", Sv(world->npcs[i].name));
    }
  }
}

void ProcessEvents(SDL_Event *event, World *world)
{
  PlayerInput(event, &world->player);
  if(event->type == SDL_EVENT_KEY_DOWN && event->key.key == SDLK_F5){
    SaveWorld(world->arena, world);
  }
  if(event->type == SDL_EVENT_KEY_DOWN && event->key.key == SDLK_F6){
    LoadWorld(world->arena, world);
  }
}

void RenderWorld(World *world, D_Context *draw) {
  if(world->player.currentArea == WORLD_AREA_OUTSIDE) {
    for (int i = 0; i < 4800; i++) {
      D_Rect(
        draw,
        (F32) (i % 96), (F32) (i / 96),
        .texture = D_ImageHandle(draw,world->tileTypes[world->map[i]].tag),
        .angle = (F32) world->tileTypes[world->map[i]].rotation,
      );
    }
  }
  for (int i = 0; i < world->propCount; i++) {
    if(world->props[i].area == world->player.currentArea) {
      D_Rect(
        draw,
        world->props[i].pos.x,
        world->props[i].pos.y,
        .texture = D_ImageHandle(draw,world->propTypes[world->props[i].propType].tag),
        .scale = world->propTypes[world->props[i].propType].scale,
      );
    }
  }
  for(U32 i = 0; i < world->npcCount; i++) {
    NPC npc = world->npcs[i];
    if(npc.currentArea == world->player.currentArea) {
      V2f drawPos = AABB_Centre(npc.collision);
      D_Rect(draw, drawPos.x, drawPos.y, .texture = 1);
    }
  }
  if(world->bandit.currentArea == world->player.currentArea) {
    V2f drawPos = AABB_Centre(world->bandit.collision);
    D_Rect(draw, drawPos.x, drawPos.y, .texture = 9);
  }
  D_Rect(draw, world->player.pos.x, world->player.pos.y, .texture = 1);
}

void SaveWorld(M_Arena *arena, World *world) {
  printf("Saving world\n");
  OS_Handle file = FS_FileOpen(S("world.sgdat"), FS_ACCESS_WRITE);
  U32 offset = 0;
  for(int i = 0; i < WORLD_TILE_TYPE_MAX; i++) {
    FS_FileWrite(file, &world->tileTypes[i].tag.count, sizeof(S64), offset);
    offset += sizeof(S64);
    FS_FileWrite(file, world->tileTypes[i].tag.data, world->tileTypes[i].tag.count, offset);
    offset += sizeof(U8) * world->tileTypes[i].tag.count;
    FS_FileWrite(file, &world->tileTypes[i].rotation, sizeof(F32), offset);
    offset += sizeof(F32);
  }
  for(int i = 0; i < WORLD_PROP_TYPE_MAX; i++) {
    FS_FileWrite(file, &world->propTypes[i].tag.count, sizeof(S64), offset);
    offset += sizeof(S64);
    FS_FileWrite(file, world->propTypes[i].tag.data, world->propTypes[i].tag.count, offset);
    offset += sizeof(U8) * world->propTypes[i].tag.count;
    FS_FileWrite(file, &world->propTypes[i].scale, sizeof(F32), offset);
    offset += sizeof(F32);
  }

  FS_FileWrite(file, world->propTypes, sizeof(World_PropType)*WORLD_PROP_TYPE_MAX, offset);
  offset += sizeof(World_PropType)*WORLD_PROP_TYPE_MAX;

  FS_FileWrite(file, &world->propCount, sizeof(U32), offset);
  offset += sizeof(U32);

  FS_FileWrite(file, world->props, sizeof(World_Prop)*WORLD_PROP_MAX, offset);
  offset += sizeof(World_Prop)*WORLD_PROP_MAX;

  FS_FileWrite(file, &world->hitboxCount, sizeof(U32), offset);
  offset += sizeof(U32);

  FS_FileWrite(file, world->hitboxes, sizeof(AABB)*WORLD_HITBOX_MAX, offset);
  offset += sizeof(AABB)*WORLD_HITBOX_MAX;

  FS_FileWrite(file, world->map, sizeof(U32)*WORLD_MAP_MAX, offset);
  offset += sizeof(U32)*WORLD_MAP_MAX;

  FS_FileClose(file);
  printf("Saved world :)\n");
}

void LoadWorld(M_Arena *arena, World *world) {
  printf("loading world\n");
  OS_Handle file = FS_FileOpen(S("world.sgdat"), FS_ACCESS_READ);
  U32 offset = 0;

  world->tileTypes = M_ArenaPush(arena, World_Tile, .count=WORLD_TILE_TYPE_MAX);
  for(int i = 0; i < WORLD_TILE_TYPE_MAX; i++) {
    FS_FileRead(file, &world->tileTypes[i].tag.count, sizeof(S64), offset);
    offset += sizeof(S64);
    world->tileTypes[i].tag.data = M_ArenaPush(arena, U8, .count=world->tileTypes[i].tag.count);
    FS_FileRead(file, world->tileTypes[i].tag.data, world->tileTypes[i].tag.count, offset);
    offset += sizeof(U8) * world->tileTypes[i].tag.count;
    FS_FileRead(file, &world->tileTypes[i].rotation, sizeof(F32), offset);
    offset += sizeof(F32);
  }
  world->propTypes = M_ArenaPush(arena, World_PropType, .count=WORLD_PROP_TYPE_MAX);
  for(int i = 0; i < WORLD_PROP_TYPE_MAX; i++) {
    FS_FileRead(file, &world->propTypes[i].tag.count, sizeof(S64), offset);
    offset += sizeof(S64);
    world->propTypes[i].tag.data = M_ArenaPush(arena, U8, .count=world->propTypes[i].tag.count);
    FS_FileRead(file, world->propTypes[i].tag.data, world->propTypes[i].tag.count, offset);
    offset += sizeof(U8) * world->propTypes[i].tag.count;
    FS_FileRead(file, &world->propTypes[i].scale, sizeof(F32), offset);
    offset += sizeof(F32);
  }

  FS_FileRead(file, &world->propCount, sizeof(U32), offset);
  offset += sizeof(U32);

  world->props = M_ArenaPush(arena, World_Prop, .count=WORLD_PROP_MAX);
  FS_FileRead(file, world->props, sizeof(World_Prop)*WORLD_PROP_MAX, offset);
  offset += sizeof(World_Prop)*world->propCount;

  FS_FileRead(file, &world->hitboxCount, sizeof(U32), offset);
  offset += sizeof(U32);

  world->hitboxes = M_ArenaPush(arena, AABB, .count=WORLD_HITBOX_MAX);
  FS_FileRead(file, world->hitboxes, sizeof(AABB)*WORLD_HITBOX_MAX, offset);
  offset += sizeof(AABB)*world->hitboxCount;

  world->map = M_ArenaPush(arena, U32, .count=WORLD_MAP_MAX);
  FS_FileRead(file, world->map, sizeof(U32)*WORLD_MAP_MAX, offset);
  offset += sizeof(U32)*WORLD_MAP_MAX;

  FS_FileClose(file);
  printf("loaded world\n");
}