#include "Runtime/World/CScriptGenerator.hpp"

#include "Runtime/CStateManager.hpp"
#include "Runtime/World/CWallCrawlerSwarm.hpp"

#include "TCastTo.hpp" // Generated file, do not modify include path

namespace metaforce {

CScriptGenerator::CScriptGenerator(TUniqueId uid, std::string_view name, const CEntityInfo& info, u32 spawnCount,
                                   bool noReuseFollowers, const zeus::CVector3f& vec1, bool noInheritXf, bool active,
                                   float minScale, float maxScale)
: CEntity(uid, info, active, name)
, x34_spawnCount(spawnCount)
, x38_24_noReuseFollowers(noReuseFollowers)
, x38_25_noInheritTransform(noInheritXf)
, x3c_offset(vec1)
, x48_minScale(minScale)
, x4c_maxScale(maxScale) {}

void CScriptGenerator::Accept(IVisitor& visitor) { visitor.Visit(this); }

void CScriptGenerator::AcceptScriptMsg(EScriptObjectMessage msg, TUniqueId sender, CStateManager& stateMgr) {
  switch (msg) {
  case EScriptObjectMessage::SetToZero: {
    if (!GetActive()) {
      break;
    }

    std::vector<TUniqueId> follows;
    follows.reserve(x20_conns.size());
    for (const SConnection& conn : x20_conns) {
      if (conn.x0_state != EScriptObjectState::Zero || conn.x4_msg != EScriptObjectMessage::Follow) {
        continue;
      }

      const TUniqueId uid = stateMgr.GetIdForScript(conn.x8_objId);
      if (uid != kInvalidUniqueId) {
        const CEntity* ent = stateMgr.GetObjectById(uid);
        if (ent && ent->GetActive()) {
          follows.push_back(uid);
        }
      }
    }

    if (follows.empty()) {
      follows.push_back(sender);
    }

    std::vector<std::pair<TUniqueId, TEditorId>> activates;
    activates.reserve(x20_conns.size());

    for (const SConnection& conn : x20_conns) {
      if (conn.x0_state != EScriptObjectState::Zero) {
        continue;
      }

      TUniqueId uid = stateMgr.GetIdForScript(conn.x8_objId);
      if (uid == kInvalidUniqueId) {
        continue;
      }

      if (conn.x4_msg == EScriptObjectMessage::Activate) {
        if (!stateMgr.GetObjectById(uid)) {
          continue;
        }
        activates.emplace_back(uid, conn.x8_objId);
      } else {
        stateMgr.SendScriptMsgAlways(uid, GetUniqueId(), conn.x4_msg);
      }
    }

    if (activates.empty()) {
      break;
    }

    for (u32 i = 0; i < x34_spawnCount; ++i) {
      if (activates.size() == 0 || follows.size() == 0) {
        break;
      }

      u32 activatesRand = 0.99f * (stateMgr.GetActiveRandom()->Float() * activates.size());
      const u32 followsRand = 0.99f * (stateMgr.GetActiveRandom()->Float() * follows.size());

      for (u32 j = 0; j < activates.size(); ++j) {
        if (TCastToConstPtr<CScriptSound>(stateMgr.GetObjectById(activates[j].first))) {
          activatesRand = j;
          break;
        }
      }

      const std::pair<TUniqueId, TEditorId> idPair = activates[activatesRand];
      CEntity* activate = stateMgr.ObjectById(idPair.first);
      CEntity* follow = stateMgr.ObjectById(follows[followsRand]);

      if (!activate || !follow) {
        break;
      }

      const bool oldGeneratingObject = stateMgr.GetIsGeneratingObject();
      stateMgr.SetIsGeneratingObject(true);
      const std::pair<TEditorId, TUniqueId> objId = stateMgr.GenerateObject(idPair.second);
      stateMgr.SetIsGeneratingObject(oldGeneratingObject);

      if (objId.second != kInvalidUniqueId) {
        if (CEntity* genObj = stateMgr.ObjectById(objId.second)) {
          const TCastToPtr<CActor> activateActor(genObj);
          const TCastToConstPtr<CActor> followActor(follow);
          const TCastToConstPtr<CWallCrawlerSwarm> wallCrawlerSwarm(follow);

          if (activateActor && wallCrawlerSwarm) {
            if (!x38_25_noInheritTransform) {
              activateActor->SetTransform(wallCrawlerSwarm->GetTransform());
            }
            activateActor->SetTranslation(wallCrawlerSwarm->GetLastKilledOffset() + x3c_offset);
          } else if (activateActor && followActor) {
            if (!x38_25_noInheritTransform) {
              activateActor->SetTransform(followActor->GetTransform());
            }
            activateActor->SetTranslation(followActor->GetTranslation() + x3c_offset);
          }

          const float rnd = stateMgr.GetActiveRandom()->Range(x48_minScale, x4c_maxScale);
          CModelData* mData = activateActor->GetModelData();
          if (mData && !mData->IsNull()) {
            mData->SetScale(rnd * mData->GetScale());
          }

          stateMgr.SendScriptMsg(genObj, GetUniqueId(), EScriptObjectMessage::Activate);
        }
      }

      activates.erase(std::find(activates.begin(), activates.end(), idPair));
      if (x38_24_noReuseFollowers) {
        follows.erase(std::find(follows.begin(), follows.end(), follows[followsRand]));
      }
    }
    break;
  }
  default:
    break;
  }

  CEntity::AcceptScriptMsg(msg, sender, stateMgr);
}
} // namespace metaforce