local time = GetDateTimeLocal()
local seed = 3600*tonumber(string.sub(time,-8,-7))+60*tonumber(string.sub(time,-5,-4))+tonumber(string.sub(time,-2,-1))
local RNG = CreateRandomNumberGenerator(seed + mthTruncF(mthRndF() * 1000))
local rndIntRange = function(a,b)
  return (mthFloorF(RNG:RndF()*(b-a+1)) % (b-a+1) + a)
end

worldGlobals.CreateRPC("server","reliable","MysteryBoxPickupSound",function(player)
  player:PlaySound(LoadResource("Content/SeriousSam3/Sounds/Items/WeaponPickup/WeaponPickup0" ..rndIntRange(1,4).. ".wav"),1,1,0)
end)

if worldGlobals.netIsHost then

  if not scrFileExists("Content/SeriousSamFusion/Scripts/Startup/00_NSKuberWeaponDatabase.lua") then
    print("Weapon database not found! Please subscribe to NSKuber's 'Weapon Resources Database' mod!")
    WaitForever()
  end

  if (worldGlobals.CWMDoNotReplace == nil) then
    worldGlobals.CWMDoNotReplace = {}
  end

  local worldInfo = worldGlobals.worldInfo
  local availableWeaponsTable
  local boxCreateds = NewGroupVar()
  local boxPlacement
  local BoxesPlacement = {}
  local boxSpawners = NewGroupVar()
  local crates = worldInfo:GetAllEntitiesOfClass("CGenericAmmoRefillItemEntity")
  local createdsPerBox = NewGroupVar()
  local createdWeapon
  local cyclePlacement
  local farPlacement = mthQuatVect(mthHPBToQuaternion(0,0,0),mthVector3f(0,-100000,0))
  local maxAmmo
  local mysteryBoxes = NewGroupVar()
  local mysteryBoxesCrates = NewGroupVar()
  local mysteryBoxPicker
  local mysteryBoxPuppet
  local mysteryBoxSpawner
  local mysteryBoxTemplate
  local rechargeDelay
  local randomWeapon
  local spawnersPerBox = NewGroupVar()
  local templatePointer
  local thisBox
  local thisWeaponTable
  local thisWeapon
  local weaponTemplate
  worldGlobals.MysBoxWeaponTables = {}

  local getTableByBox = function(box,printMessage)
    thisWeaponTable = {}
    availableWeaponsTable = {}
    for i=1,#worldGlobals.MysBoxWeaponTables,1 do
      if box == worldGlobals.MysBoxWeaponTables[i][1] then
        thisWeaponTable = worldGlobals.MysBoxWeaponTables[i][2]
        maxAmmo = worldGlobals.MysBoxWeaponTables[i][3]
        rechargeDelay = worldGlobals.MysBoxWeaponTables[i][4]
        break
      end
    end

    if #thisWeaponTable == 0 then
      thisWeaponTable = worldGlobals.MysBoxDefaultWeaponTable
      maxAmmo = false
      rechargeDelay = 30
    end

    for j=1,#thisWeaponTable,1 do
      templatePointer = "Content/SeriousSamFusion/Scripts/Templates/WeaponDatabase/" ..thisWeaponTable[j].. "_BFE.rsc"
      if scrFileExists(templatePointer) then
        table.insert(availableWeaponsTable, thisWeaponTable[j])
      else
        if printMessage == true then
          print("Weapon " ..thisWeaponTable[j].. " does not exist in Weapon Resource Database or is missing a BFE variant!\nIt cannot be included in Mystery Box (" ..box:GetName().. ") item pool.")
        end
      end
    end

    return availableWeaponsTable, maxAmmo, rechargeDelay
  end

  for i=1,#crates,1 do
    if crates[i]:GetItemParams():GetFileName() == "Content/SeriousSam3/Databases/Items/MysteryBox/MysteryBox.ep" then
      table.insert(mysteryBoxesCrates, crates[i])
    end
  end

  Wait(CustomEvent("OnStep"))

  local HandleMysteryBox = function(mysteryBoxCrate,index)
    RunAsync(function()
      BoxesPlacement[index] = mysteryBoxCrate:GetPlacement()
      mysteryBoxCrate:SetPlacement(farPlacement)
      local isBoxDepleted = false
      local isBoxOpened = false
      local mysteryBox
      while not isBoxDepleted do
        if IsDeleted(mysteryBoxPuppet) then
          mysteryBoxPuppet = mysteryBoxTemplate:SpawnEntityFromTemplate(0,worldInfo,BoxesPlacement[index])
          mysteryBoxSpawner = mysteryBoxPuppet:GetEffectiveEntity()
        end
        mysteryBox = mysteryBoxSpawner:SpawnOne()
        mysteryBox:BeInvulnerable(true)
        Wait(CustomEvent("OnStep"))
        if not isBoxOpened then
          mysteryBox:EnableUsage()
        else
          mysteryBox:PlayAnim("Open")
          while isBoxOpened do
            Wait(CustomEvent("OnStep"))
          end
        end
        if IsDeleted(mysteryBox) then
          if IsDeleted(mysteryBoxPuppet) then
            mysteryBoxPuppet = mysteryBoxTemplate:SpawnEntityFromTemplate(0,worldInfo,BoxesPlacement[index])
            mysteryBoxSpawner = mysteryBoxPuppet:GetEffectiveEntity()
          end
          mysteryBox = mysteryBoxSpawner:SpawnOne()
          mysteryBox:BeInvulnerable(true)
          mysteryBox:EnableUsage()
        end
        RunHandled(function()
          while not IsDeleted(mysteryBox) and not isBoxDepleted do
            Wait(CustomEvent("OnStep"))
          end
        end,

        On(Event(mysteryBox.Used)),
        function(payload)
          isBoxOpened = true
          mysteryBoxPicker = payload:GetUser()
          local cyclePlacement = mysteryBox:GetPlacement()
          cyclePlacement.vy = cyclePlacement.vy+0.6

          mysteryBox:PlayAnim("Open")
          Wait(Delay(0.75))
          local spawnersInBox = boxSpawners[index]
          local baseDelay = 4/(3*#spawnersInBox)
          local spawnedWeapon
          for j=1,#spawnersInBox*3,1 do
            local k=j%#spawnersInBox
            if k == 0 then k = #spawnersInBox end
            boxCreateds[index][k]:SetPlacement(cyclePlacement)
            spawnedWeapon = spawnersInBox[k]:SpawnOne()
            worldGlobals.CWMDoNotReplace[spawnedWeapon] = {false}
            spawnedWeapon:DisablePicking()
            cyclePlacement.vy = cyclePlacement.vy+(0.3/(3*#spawnersInBox))
            local delayFormula = 0.8*(j/(#spawnersInBox*3))+0.6
            Wait(Delay(baseDelay*delayFormula))
            spawnedWeapon:Delete()
          end

          randomWeapon = rndIntRange(1,#spawnersInBox)
          spawnedWeapon = spawnersInBox[randomWeapon]:SpawnOne()
          spawnedWeapon:DisablePicking()
          local thisWeaponTable
          local maxAmmo
          local rechargeDelay
          thisWeaponTable,maxAmmo,rechargeDelay = getTableByBox(mysteryBoxesCrates[index],false)
          thisWeapon = thisWeaponTable[randomWeapon]

          Wait(Delay(1))
          local weaponParams
          local weaponAmmo
          if worldGlobals.MysBoxIsBFE == true then
            weaponParams = LoadResource(worldGlobals.DBWeaponParamsPathsB[thisWeapon])
            if maxAmmo == true then
              weaponAmmo = worldGlobals.DBWeaponAmmoB[thisWeapon][3]
            else
              weaponAmmo = worldGlobals.DBWeaponAmmoB[thisWeapon][1]
            end
          else
            weaponParams = LoadResource(worldGlobals.DBWeaponParamsPaths[thisWeapon])
            if maxAmmo == true then
              weaponAmmo = worldGlobals.DBWeaponAmmo[thisWeapon][3]
            else
              weaponAmmo = worldGlobals.DBWeaponAmmo[thisWeapon][1]
            end
          end

          mysteryBoxPicker:AwardWeapon(weaponParams)
          mysteryBoxPicker:AwardAmmoForWeapon(weaponParams,weaponAmmo)
          if worldInfo:IsSinglePlayer() then
            mysteryBoxPicker:PlaySound(LoadResource("Content/SeriousSam3/Sounds/Items/WeaponPickup/WeaponPickup0" ..rndIntRange(1,4).. ".wav"),1,1,0)
          else
            worldGlobals.MysteryBoxPickupSound(mysteryBoxPicker)
          end
          spawnedWeapon:Delete()

          if rechargeDelay > 0 then
            Wait(Delay(rechargeDelay))
            isBoxOpened = false
            if not IsDeleted(mysteryBox) then
              mysteryBox:PlayAnim("Close")
              mysteryBox:EnableUsage()
            end
          elseif rechargeDelay <= 0 then
            isBoxDepleted = true
          end
        end)
      end
      if not IsDeleted(mysteryBox) then
        mysteryBox:Delete()
      end

    end)
  end

  if #mysteryBoxesCrates > 0 then
    mysteryBoxTemplate = LoadResource("Content/SeriousSamFusion/Scripts/Templates/MysteryBox.rsc")
    for i=1,#mysteryBoxesCrates,1 do
      HandleMysteryBox(mysteryBoxesCrates[i],i)
    end
  else
    WaitForever()
  end

  RunHandled(function()
    WaitForever()
  end,

  On(Delay(0.1)),
  function()
    for i=1,#mysteryBoxesCrates,1 do
      local boxPlacement = mthCloneQuatVect(BoxesPlacement[i])
      boxPlacement.vy = boxPlacement.vy+0.6
      thisWeaponTable = getTableByBox(mysteryBoxesCrates[i],true)

      for k=1,#thisWeaponTable,1 do
        templatePointer = "Content/SeriousSamFusion/Scripts/Templates/WeaponDatabase/" ..thisWeaponTable[k].. "_BFE.rsc"
        weaponTemplate = LoadResource(templatePointer)
        createdWeapon = weaponTemplate:SpawnEntityFromTemplate(0,worldInfo,boxPlacement)
        worldGlobals.CWMDoNotReplace[createdWeapon] = {false}
        table.insert(createdsPerBox,createdWeapon)
        table.insert(spawnersPerBox,createdWeapon:GetEffectiveEntity())
        createdWeapon:SetPlacement(farPlacement)
      end

      print("inserting "..#spawnersPerBox)
      table.insert(boxSpawners,spawnersPerBox)
      table.insert(boxCreateds,createdsPerBox)
      createdsPerBox = NewGroupVar()
      spawnersPerBox = NewGroupVar()
      thisWeaponTable = nil

      for l=1,#createdsPerBox,1 do
        createdsPerBox[l]:SetPlacement(boxPlacement)
      end
    end
  end)

end