#include "elephant_attack_system.h"
#include "../core/component.h"
#include "../core/world.h"
#include "../units/spawn_type.h"
#include <algorithm>
#include <array>
#include <cmath>
#include <cstdlib>
#include <vector>

namespace Game::Systems {

namespace {

struct FootOffset {
  float x;
  float z;
};

constexpr float k_pi = 3.14159265F;

auto pick_stomp_position(Engine::Core::TransformComponent *transform,
                         Engine::Core::ElephantComponent *elephant_comp)
    -> FootOffset {
  float scale =
      std::max(1.0F, (transform->scale.x + transform->scale.z) * 0.5F);
  float forward = 0.6F * scale;
  float side = 0.45F * scale;
  float const max_offset = elephant_comp->trample_radius * 0.95F;
  if (max_offset > 0.0F) {
    forward = std::min(forward, max_offset);
    side = std::min(side, max_offset);
  }

  std::array<FootOffset, 4> const offsets = {
      FootOffset{side, forward},
      FootOffset{-side, forward},
      FootOffset{side, -forward},
      FootOffset{-side, -forward},
  };

  int const index = std::rand() % static_cast<int>(offsets.size());
  FootOffset const local = offsets[index];

  float const yaw = transform->rotation.y * (k_pi / 180.0F);
  float const cos_y = std::cos(yaw);
  float const sin_y = std::sin(yaw);
  FootOffset world;
  world.x = transform->position.x + local.x * cos_y + local.z * sin_y;
  world.z = transform->position.z - local.x * sin_y + local.z * cos_y;
  return world;
}

} // namespace

void ElephantAttackSystem::update(Engine::Core::World *world,
                                  float delta_time) {
  process_elephant_behavior(world, delta_time);
}

void ElephantAttackSystem::process_elephant_behavior(Engine::Core::World *world,
                                                     float delta_time) {
  auto entities = world->get_entities_with<Engine::Core::UnitComponent>();

  for (auto *entity : entities) {
    auto *unit = entity->get_component<Engine::Core::UnitComponent>();
    if (unit == nullptr || unit->health <= 0) {
      continue;
    }

    if (unit->spawn_type != Game::Units::SpawnType::Elephant) {
      continue;
    }

    if (entity->has_component<Engine::Core::PendingRemovalComponent>()) {
      continue;
    }

    auto *elephant = entity->get_component<Engine::Core::ElephantComponent>();
    if (elephant == nullptr) {
      elephant = entity->add_component<Engine::Core::ElephantComponent>();
    }

    float const health_ratio =
        static_cast<float>(unit->health) / static_cast<float>(unit->max_health);
    if (health_ratio < 0.3F && !elephant->is_panicked) {

      if (std::rand() % 100 < 50) {
        elephant->is_panicked = true;
        elephant->panic_duration = 10.0F;
      }
    }

    if (elephant->is_panicked) {
      process_panic_mechanic(entity, world, delta_time);
    }

    if (elephant->charge_cooldown > 0.0F) {
      elephant->charge_cooldown -= delta_time;
    }

    process_charge_attack(entity, world, delta_time);

    process_trample_damage(entity, world, delta_time);

    process_melee_attack(entity, world, delta_time);
  }
}

void ElephantAttackSystem::process_charge_attack(Engine::Core::Entity *elephant,
                                                 Engine::Core::World *world,
                                                 float delta_time) {
  auto *elephant_comp =
      elephant->get_component<Engine::Core::ElephantComponent>();
  auto *unit = elephant->get_component<Engine::Core::UnitComponent>();
  auto *transform = elephant->get_component<Engine::Core::TransformComponent>();
  auto *movement = elephant->get_component<Engine::Core::MovementComponent>();
  auto *attack_target =
      elephant->get_component<Engine::Core::AttackTargetComponent>();

  if (elephant_comp == nullptr || unit == nullptr || transform == nullptr ||
      movement == nullptr) {
    return;
  }

  switch (elephant_comp->charge_state) {
  case Engine::Core::ElephantComponent::ChargeState::Idle: {

    if (attack_target != nullptr && attack_target->target_id != 0 &&
        elephant_comp->charge_cooldown <= 0.0F && !elephant_comp->is_panicked) {
      auto *target = world->get_entity(attack_target->target_id);
      if (target != nullptr) {
        auto *target_transform =
            target->get_component<Engine::Core::TransformComponent>();
        if (target_transform != nullptr) {
          float const dx = target_transform->position.x - transform->position.x;
          float const dz = target_transform->position.z - transform->position.z;
          float const dist = std::sqrt(dx * dx + dz * dz);

          if (dist >= 5.0F && dist <= 15.0F) {
            elephant_comp->charge_state =
                Engine::Core::ElephantComponent::ChargeState::Charging;
            elephant_comp->charge_duration = 3.0F;
          }
        }
      }
    }
    break;
  }

  case Engine::Core::ElephantComponent::ChargeState::Charging: {

    elephant_comp->charge_duration -= delta_time;

    if (elephant_comp->charge_duration <= 0.0F) {
      elephant_comp->charge_state =
          Engine::Core::ElephantComponent::ChargeState::Recovering;
      elephant_comp->charge_cooldown = 8.0F;
    }
    break;
  }

  case Engine::Core::ElephantComponent::ChargeState::Recovering: {
    elephant_comp->charge_state =
        Engine::Core::ElephantComponent::ChargeState::Idle;
    break;
  }

  default:
    break;
  }
}

void ElephantAttackSystem::process_trample_damage(
    Engine::Core::Entity *elephant, Engine::Core::World *world,
    float delta_time) {
  auto *elephant_comp =
      elephant->get_component<Engine::Core::ElephantComponent>();
  auto *unit = elephant->get_component<Engine::Core::UnitComponent>();
  auto *transform = elephant->get_component<Engine::Core::TransformComponent>();
  auto *movement = elephant->get_component<Engine::Core::MovementComponent>();
  auto *attack = elephant->get_component<Engine::Core::AttackComponent>();
  auto *attack_target =
      elephant->get_component<Engine::Core::AttackTargetComponent>();

  if (elephant_comp == nullptr || unit == nullptr || transform == nullptr ||
      movement == nullptr) {
    return;
  }

  constexpr float k_movement_threshold = 0.1F;
  bool const is_moving = (std::abs(movement->vx) > k_movement_threshold ||
                          std::abs(movement->vz) > k_movement_threshold);

  bool has_close_target = false;
  if (!is_moving && attack_target != nullptr && attack_target->target_id != 0) {
    auto *target = world->get_entity(attack_target->target_id);
    if (target != nullptr) {
      auto *target_transform =
          target->get_component<Engine::Core::TransformComponent>();
      if (target_transform != nullptr) {
        float const dx = target_transform->position.x - transform->position.x;
        float const dz = target_transform->position.z - transform->position.z;
        float const dist = std::sqrt(dx * dx + dz * dz);
        float const engage_range =
            (attack != nullptr)
                ? std::max(elephant_comp->trample_radius, attack->melee_range)
                : elephant_comp->trample_radius;
        has_close_target = (dist <= engage_range);
      }
    }
  }

  if (!is_moving && !has_close_target) {
    elephant_comp->trample_damage_accumulator = 0.0F;
    return;
  }

  elephant_comp->trample_damage_accumulator +=
      static_cast<float>(elephant_comp->trample_damage) * delta_time;
  int const damage =
      static_cast<int>(elephant_comp->trample_damage_accumulator);
  if (damage <= 0) {
    return;
  }

  auto *stomp_impact =
      elephant->get_component<Engine::Core::ElephantStompImpactComponent>();
  if (stomp_impact == nullptr) {
    stomp_impact =
        elephant->add_component<Engine::Core::ElephantStompImpactComponent>();
  }

  bool hit_any = false;
  auto entities = world->get_entities_with<Engine::Core::UnitComponent>();
  for (auto *other_entity : entities) {
    if (other_entity == elephant) {
      continue;
    }

    auto *other_unit =
        other_entity->get_component<Engine::Core::UnitComponent>();
    auto *other_transform =
        other_entity->get_component<Engine::Core::TransformComponent>();

    if (other_unit == nullptr || other_transform == nullptr ||
        other_unit->health <= 0) {
      continue;
    }

    bool const is_enemy = other_unit->owner_id != unit->owner_id;
    bool const should_damage = is_enemy || elephant_comp->is_panicked;

    if (!should_damage) {
      continue;
    }

    float const dx = other_transform->position.x - transform->position.x;
    float const dz = other_transform->position.z - transform->position.z;
    float const dist = std::sqrt(dx * dx + dz * dz);

    if (dist <= elephant_comp->trample_radius) {
      int const old_health = other_unit->health;
      other_unit->health -= damage;
      if (other_unit->health < 0) {
        other_unit->health = 0;
      }

      if (old_health > 0 && other_unit->health < old_health) {
        FootOffset const stomp_pos =
            pick_stomp_position(transform, elephant_comp);
        Engine::Core::ElephantStompImpactComponent::ImpactRecord impact;
        impact.x = stomp_pos.x;
        impact.z = stomp_pos.z;
        impact.time = 0.0F;
        stomp_impact->impacts.push_back(impact);
        hit_any = true;
      }
    }
  }

  if (hit_any) {
    elephant_comp->trample_damage_accumulator -= damage;
  } else {
    elephant_comp->trample_damage_accumulator = 0.0F;
  }
}

void ElephantAttackSystem::process_panic_mechanic(
    Engine::Core::Entity *elephant, Engine::Core::World *world,
    float delta_time) {
  auto *elephant_comp =
      elephant->get_component<Engine::Core::ElephantComponent>();
  auto *movement = elephant->get_component<Engine::Core::MovementComponent>();

  if (elephant_comp == nullptr || movement == nullptr) {
    return;
  }

  elephant_comp->panic_duration -= delta_time;

  if (elephant_comp->panic_duration <= 0.0F) {
    elephant_comp->is_panicked = false;
    elephant_comp->panic_duration = 0.0F;
    return;
  }

  static float random_target_timer = 0.0F;
  random_target_timer += delta_time;

  if (random_target_timer >= 2.0F) {
    random_target_timer = 0.0F;

    auto *transform =
        elephant->get_component<Engine::Core::TransformComponent>();
    if (transform != nullptr) {

      float const angle = static_cast<float>(std::rand()) /
                          static_cast<float>(RAND_MAX) * 3.14159F * 2.0F;
      float const distance = 10.0F;

      movement->target_x = transform->position.x + std::cos(angle) * distance;
      movement->target_y = transform->position.z + std::sin(angle) * distance;
      movement->has_target = true;
    }
  }
}

void ElephantAttackSystem::process_melee_attack(Engine::Core::Entity *elephant,
                                                Engine::Core::World *world,
                                                float delta_time) {}

} // namespace Game::Systems