moved folders

42/media/lua/client/TOC/Admin.lua

@@ -0,0 +1,115 @@
+local CommandsData = require("TOC/CommandsData")
+local ClientRelayCommands = require("TOC/ClientRelayCommands")
+local StaticData = require("TOC/StaticData")
+local DataController = require("TOC/Controllers/DataController")
+-------------------
+
+---@param playerNum number
+---@param context ISContextMenu
+---@param worldobjects table
+local function AddAdminTocOptions(playerNum, context, worldobjects)
+    if not(isClient() and isAdmin() or isDebugEnabled()) then return end
+
+    local players = {}
+    for _, v in ipairs(worldobjects) do
+        for x = v:getSquare():getX() - 1, v:getSquare():getX() + 1 do
+            for y = v:getSquare():getY() - 1, v:getSquare():getY() + 1 do
+                local sq = getCell():getGridSquare(x, y, v:getSquare():getZ());
+                if sq then
+                    for z = 0, sq:getMovingObjects():size() - 1 do
+                        local o = sq:getMovingObjects():get(z)
+                        if instanceof(o, "IsoPlayer") then
+                            ---@cast o IsoPlayer
+
+                            local oId = o:getOnlineID()
+                            players[oId] = o
+                        end
+                    end
+                end
+            end
+        end
+    end
+
+
+    for _, pl in pairs(players) do
+        ---@cast pl IsoPlayer
+
+        local clickedPlayerNum = pl:getOnlineID()
+
+        local option = context:addOption(getText("ContextMenu_Admin_TOC") .. " - " .. pl:getUsername(), nil, nil)
+        local subMenu = ISContextMenu:getNew(context)
+        context:addSubMenu(option, subMenu)
+
+        subMenu:addOption(getText("ContextMenu_Admin_ResetTOC"), nil, function()
+            if isClient() then
+                sendClientCommand(CommandsData.modules.TOC_RELAY, CommandsData.server.Relay.RelayExecuteInitialization,
+                    { patientNum = clickedPlayerNum })
+            else
+                -- TODO ugly
+                ClientRelayCommands.ReceiveExecuteInitialization()
+            end
+        end)
+
+        -- Force amputation
+        local forceAmpOption = subMenu:addOption(getText("ContextMenu_Admin_ForceAmputation"), nil, nil)
+        local forceAmpSubMenu = ISContextMenu:getNew(subMenu)
+        context:addSubMenu(forceAmpOption, forceAmpSubMenu)
+
+        for i = 1, #StaticData.LIMBS_STR do
+            local limbName = StaticData.LIMBS_STR[i]
+            local limbTranslatedName = getText("ContextMenu_Limb_" .. limbName)
+
+            forceAmpSubMenu:addOption(limbTranslatedName, nil, function()
+                if isClient() then
+                    sendClientCommand(CommandsData.modules.TOC_RELAY, CommandsData.server.Relay.RelayForcedAmputation,
+                    { patientNum = clickedPlayerNum, limbName = limbName })
+                else
+                    ClientRelayCommands.ReceiveExecuteAmputationAction({surgeonNum=clickedPlayerNum, limbName=limbName, damagePlayer=false})
+                    -- todo ugly
+                end
+
+            end)
+        end
+    end
+
+    
+end
+Events.OnFillWorldObjectContextMenu.Add(AddAdminTocOptions)
+
+
+--* Override to cheats to fix stuff
+
+local og_ISHealthPanel_onCheatCurrentPlayer = ISHealthPanel.onCheatCurrentPlayer
+
+---Override to onCheatCurrentPlayer to fix behaviour with TOC
+---@param bodyPart BodyPart
+---@param action any
+---@param player IsoPlayer
+function ISHealthPanel.onCheatCurrentPlayer(bodyPart, action, player)
+    og_ISHealthPanel_onCheatCurrentPlayer(bodyPart, action, player)
+    local bptString = BodyPartType.ToString(bodyPart:getType())
+
+    if action == "healthFullBody" then
+        -- loop all limbs and reset them if infected
+        local dcInst = DataController.GetInstance()
+
+        for i = 1, #StaticData.LIMBS_STR do
+            local limbName = StaticData.LIMBS_STR[i]
+
+            dcInst:setIsInfected(limbName, false)
+        end
+
+        dcInst:setIsIgnoredPartInfected(false)
+
+        dcInst:apply()
+    end
+
+    if action == "healthFull" then
+        -- Get the limbName for that BodyPart and fix the values in TOC Data
+        local limbName = StaticData.BODYLOCS_TO_LIMBS_IND_STR[bptString]
+        local dcInst = DataController.GetInstance()
+
+        dcInst:setIsInfected(limbName, false)
+        dcInst:apply()
+    end
+end