#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);
  }
  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 (U32 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);
}

#define WRITE(v) FS_FileWrite(file, &(v), sizeof(v), offset); offset += sizeof(v)
#define WRITEN(v, n) FS_FileWrite(file, v, (n) * sizeof(*(v)), offset); offset += ((n) * sizeof(*(v)))

void SaveWorld(World *world) {
  printf("--- Saving World ---\n");
	printf("  - %d props\n", world->propCount);
	printf("  - %d hitboxes\n", world->hitboxCount);

  OS_Handle file = FS_FileOpen(S("world.sgdat"), FS_ACCESS_WRITE);
  U64 offset = 0;

  for (U32 i = 0; i < WORLD_TILE_TYPE_MAX; i++) {
		World_Tile *tile = &world->tileTypes[i];

		WRITE(tile->tag.count);
		WRITEN(tile->tag.data, tile->tag.count);
		WRITE(tile->rotation);
  }

  for (U32 i = 0; i < WORLD_PROP_TYPE_MAX; i++) {
		World_PropType *type = &world->propTypes[i];

		WRITE(type->tag.count);
		WRITEN(type->tag.data, type->tag.count);
		WRITE(type->scale);
  }

	WRITE(world->propCount);
	WRITEN(world->props, WORLD_PROP_MAX);

	WRITE(world->hitboxCount);
	WRITEN(world->hitboxes, WORLD_HITBOX_MAX);

	WRITEN(world->map, WORLD_MAP_MAX);

  FS_FileClose(file);

  printf("--- World Saved ---\n");
}

void LoadWorld(M_Arena *arena, World *world) {
	printf("--- Loading World ---\n");

	M_TempScope(0, 0) {
		Str8 data = FS_ReadEntireFile(temp.arena, S("world.sgdat"));

		U8 *base = data.data;

		// Load tile types
		//
		world->tileTypes = M_ArenaPush(arena, World_Tile, .count = WORLD_TILE_TYPE_MAX);
		for (U32 it = 0; it < WORLD_TILE_TYPE_MAX; ++it) {
			World_Tile *type = &world->tileTypes[it];

			Str8 name;
			name.count = *(S64 *) base;
			name.data  = base + 8;

			base += 8;
			base += name.count;

			if (name.count != 0) {
				// Means we likely got a vaild tag
				type->tag      = Str8_Copy(arena, name);
				type->rotation = *(F32 *) base;
			}

			base += 4;
		}

		// Load prop types
		//
		world->propTypes = M_ArenaPush(arena, World_PropType, .count = WORLD_PROP_TYPE_MAX);
		for (U32 it = 0; it < WORLD_PROP_TYPE_MAX; ++it) {
			World_PropType *type = &world->propTypes[it];

			Str8 name;
			name.count = *(S64 *) base;
			name.data  = base + 8;

			base += 8;
			base += name.count;

			if (name.count != 0) {
				type->tag    = Str8_Copy(arena, name);
				type->scale  = *(F32 *) base;
			}

			base += 4;
		}

		// Prop locations
		{
			U32 n_props = *(U32 *) base; base += 4;
			World_Prop *props = (World_Prop *) base;

			world->propCount = n_props;
			world->props     = M_ArenaPush(arena, World_Prop, .count = WORLD_PROP_MAX);

			M_CopySize(world->props, props, n_props * sizeof(World_Prop));

			base += (WORLD_PROP_MAX * sizeof(World_Prop));
		}

		// Hitboxes
		//
		{
			U32 n_hitbox = *(U32 *) base; base += 4;
			AABB *boxes  = (AABB *) base;

			world->hitboxCount = n_hitbox;
			world->hitboxes    = M_ArenaPush(arena, AABB, .count = WORLD_HITBOX_MAX);

			M_CopySize(world->hitboxes, boxes, n_hitbox * sizeof(AABB));

			base += (WORLD_HITBOX_MAX * sizeof(AABB));
		}

		// Map
		//
		{
			U32 *__map = (U32 *) base;

			world->map = M_ArenaPush(arena, U32, .count = WORLD_MAP_MAX);
			M_CopySize(world->map, __map, WORLD_MAP_MAX * sizeof(U32));

			base += (WORLD_MAP_MAX * sizeof(U32));
		}

		if (base != (data.data + data.count)) {
			printf("--- OFFSET MISMATCH ---\n");
		}

		printf("  - %d props\n", world->propCount);
		printf("  - %d hitboxes\n", world->hitboxCount);

		printf("--- Loaded World ---\n");
	}
}