#include "vulkan_engine/vulkan/engine.h"

#include "precomp.h"
#include "socket/packet.h"

namespace veng {

void Engine::init() {
  glm::ivec2 window_size_ = vulkan_graphics->window->GetFramebufferSize();
  view = glm::lookAt(glm::vec3(0.f, 0.f, -5.f), glm::vec3(0.f, 0.f, 0.f),
                     glm::vec3(0.f, -1.f, 0.f));
  projection = glm::perspective(
      glm::radians(103.f),
      (std::float_t)window_size_.x / (std::float_t)window_size_.y, 0.1f,
      1000.f);
  vulkan_graphics->SetViewProjection(view, projection);

  if (BeginPlay != nullptr) BeginPlay(*this);
}

void Engine::LoadModelAsset(std::string path, std::string name) {
  veng::Model model(vulkan_graphics);
  asset_loader_.setPath(path);
  asset_loader_.loadModel(model);

  model.vertex_buffer = vulkan_graphics->CreateVertexBuffer(model.vertices);
  model.index_buffer = vulkan_graphics->CreateIndexBuffer(model.indices);
  model.material.texture_handle =
      vulkan_graphics->CreateTexture(asset_loader_.readTexture());
  model_assets_[name] = std::move(model);
}

const Model* Engine::GetStaticModel(std::string name) {
  if (model_assets_.find(name) != model_assets_.end())
    return &model_assets_[name];
  return nullptr;
}

Model* Engine::SpawnModel(std::string asset_name, std::string name) {
  if (asset_name == "") {
    Model model_to_spawn(nullptr);
    model_to_spawn.visible = false;
    models_[name] = std::move(model_to_spawn);
    return &models_[name];
  }

  if (models_.find(name) == models_.end()) {
    Model model_to_spawn(*GetStaticModel(asset_name));
    models_[name] = std::move(model_to_spawn);
    return &models_[name];
  }

  std::uint32_t i = 0;
  for (i = 0; i < std::numeric_limits<std::uint32_t>::max();) {
    if (models_.find(name + std::to_string(i)) == models_.end()) {
      Model model_to_spawn(*GetStaticModel(asset_name));
      models_[name + std::to_string(i)] = std::move(model_to_spawn);
      break;
    }
    i++;
  }

  if (i == std::numeric_limits<std::uint32_t>::max() - 1)
    return nullptr;
  else
    return &models_[name + std::to_string(i)];
}

Model* Engine::SpawnLifedModel(std::string asset_name, std::string name,
                               std::float_t lifespan) {
  if (asset_name == "") {
    Model model_to_spawn(nullptr);
    model_to_spawn.visible = false;
    model_to_spawn.lifespan = lifespan;
    models_[name] = std::move(model_to_spawn);
    return &models_[name];
  }

  if (models_.find(name) == models_.end()) {
    Model model_to_spawn(*GetStaticModel(asset_name));
    model_to_spawn.lifespan = lifespan;
    models_[name] = std::move(model_to_spawn);
    return &models_[name];
  }

  std::uint32_t i = 0;
  for (i = 0; i < std::numeric_limits<std::uint32_t>::max();) {
    if (models_.find(name + std::to_string(i)) == models_.end()) {
      Model model_to_spawn(*GetStaticModel(asset_name));
      model_to_spawn.lifespan = lifespan;
      models_[name + std::to_string(i)] = std::move(model_to_spawn);
      break;
    }
    i++;
  }

  if (i == std::numeric_limits<std::uint32_t>::max() - 1)
    return nullptr;
  else
    return &models_[name + std::to_string(i)];
}

Model* Engine::GetSpawnedObject(std::string name) {
  for (auto it = models_.begin(); it != models_.end();) {
    if (it->first == name) return &it->second;
    ++it;
  }

  return nullptr;
}

void Engine::Update() {
  glm::ivec2 framebuffer_size = vulkan_graphics->window->GetFramebufferSize();
  if (framebuffer_size != window_size_ && framebuffer_size.x != 0 &&
      framebuffer_size.y != 0) {
    window_size_ = framebuffer_size;
    auto grater = (framebuffer_size.x > framebuffer_size.y)
                      ? framebuffer_size.x
                      : framebuffer_size.y;
    view = glm::lookAt(glm::vec3(0.f, 0.f, -5.f), glm::vec3(0.f, 0.f, 0.f),
                       glm::vec3(0.f, -1.f, 0.f));
    projection = glm::perspective(
        glm::radians(103.f),
        (std::float_t)framebuffer_size.x / (std::float_t)framebuffer_size.y,
        0.1f, 1000.f);
  }
  vulkan_graphics->SetViewProjection(view, projection);

  if (vulkan_graphics->BeginFrame()) {
    std::double_t current_time = glfwGetTime();
    std::float_t delta_time =
        static_cast<std::float_t>(current_time - last_frame_time_);
    last_frame_time_ = current_time;

    if (Tick != nullptr) Tick(*this, delta_time);

    std::vector<Model*> models;
    models.reserve(models_.size());

    for (auto it = models_.begin(); it != models_.end();) {
      auto& model = it->second;

      if (std::abs(model.lifespan + 1.f) <
          std::numeric_limits<float>::epsilon()) {
        models.push_back(&model);
        ++it;
        continue;
      }

      if (model.lifespan <= 0.f) {
        it = models_.erase(it);
        continue;
      }

      model.lifespan -= delta_time;

      models.push_back(&model);
      ++it;
    }

    for (auto& it : models_) {
      it.second.Update(delta_time);
      vulkan_graphics->RenderModel(&it.second);
    }

    physics_controller_.invokeOnColisionEvent(thread_pool_,
                                              {models.data(), models.size()});

    vulkan_graphics->EndFrame();
  }
}
void Engine::NetUpdate(std::shared_ptr<Network::Socket> sock) {
  NetworkUpload(sock);
  ResponseToServerAndRefresh(sock);
}

void Engine::NetworkUpload(std::shared_ptr<Network::Socket> sock) {
  std::vector<char> data;
  for (auto& it : models_) {
    if (!it.second.networkReplicated) continue;

    Packet::Header header;
    header.opcode = Packet::Opcode::UPDATEMODEL;
    std::vector<char> model = it.second.Serialize();
    header.body_length = model.size();
    std::vector<char> header_serialized = header.Serialize();

    data.insert(data.end(), header_serialized.begin(), header_serialized.end());
    data.insert(data.end(), model.begin(), model.end());
  }
  iocp_->send(*sock, data);
}

void Engine::ResponseToServerAndRefresh(std::shared_ptr<Network::Socket> sock) {
  while (iocp_->GetRecvedBytes(sock->sock) < 6);
  std::vector<char> recv_data(6);
  iocp_->recv(*sock, recv_data);
  Packet::Header header;
  header.Deserialize(recv_data);
  recv_data.resize(header.body_length);
  while (iocp_->GetRecvedBytes(sock->sock) < header.body_length);
  iocp_->recv(*sock, recv_data);
}

}  // namespace veng