Correct handling of crashing player

when assigned to a task

Moose Development/Moose/Tasking/CommandCenter.lua

@@ -189,9 +189,12 @@ function COMMANDCENTER:New( CommandCenterPositionable, CommandCenterName )
     function( self, EventData )
       local PlayerUnit = EventData.IniUnit
       for MissionID, Mission in pairs( self:GetMissions() ) do
+        local Mission = Mission -- Tasking.Mission#MISSION
+        if Mission:IsENGAGED() then
           Mission:CrashUnit( PlayerUnit )
         end
       end
+    end
   )
   
   self:SetMenu()

Moose Development/Moose/Tasking/Mission.lua

@@ -308,19 +308,17 @@ end
 -- If the Unit is part of a Task in the Mission, true is returned.
 -- @param #MISSION self
 -- @param Wrapper.Unit#UNIT PlayerUnit The CLIENT or UNIT of the Player crashing.
--- @return #boolean true if Unit is part of a Task in the Mission.
+-- @return #MISSION
 function MISSION:CrashUnit( PlayerUnit )
   self:F( { PlayerUnit = PlayerUnit } )
   
-  local PlayerUnitRemoved = false
-  
   for TaskID, Task in pairs( self:GetTasks() ) do
-    if Task:CrashUnit( PlayerUnit ) then
+    local Task = Task -- Tasking.Task#TASK
-      PlayerUnitRemoved = true
+    local PlayerGroup = PlayerUnit:GetGroup()
-    end
+    Task:CrashGroup( PlayerGroup )
   end
   
-  return PlayerUnitRemoved
+  return self
 end
 
 --- Add a scoring to the mission.

Moose Development/Moose/Tasking/Task.lua

@@ -260,7 +260,7 @@ end
 -- If the Unit is part of the Task, true is returned.
 -- @param #TASK self
 -- @param Wrapper.Unit#UNIT PlayerUnit The CLIENT or UNIT of the Player aborting the Task.
--- @return #boolean true if Unit is part of the Task.
+-- @return #TASK
 function TASK:AbortGroup( PlayerGroup )
   self:F( { PlayerGroup = PlayerGroup } )
   
@@ -312,14 +312,11 @@ end
 -- If the Unit is part of the Task, true is returned.
 -- @param #TASK self
 -- @param Wrapper.Unit#UNIT PlayerUnit The CLIENT or UNIT of the Player aborting the Task.
--- @return #boolean true if Unit is part of the Task.
+-- @return #TASK
-function TASK:CrashUnit( PlayerUnit )
+function TASK:CrashGroup( PlayerGroup )
-  self:F( { PlayerUnit = PlayerUnit } )
+  self:F( { PlayerGroup = PlayerGroup } )
-  
-  local PlayerUnitCrashed = false
   
   local PlayerGroups = self:GetGroups()
-  local PlayerGroup = PlayerUnit:GetGroup()
 
   -- Is the PlayerGroup part of the PlayerGroups?  
   if PlayerGroups:IsIncludeObject( PlayerGroup ) then
@@ -330,18 +327,35 @@ function TASK:CrashUnit( PlayerUnit )
       local IsGroupAssigned = self:IsGroupAssigned( PlayerGroup )
       self:E( { IsGroupAssigned = IsGroupAssigned } )
       if IsGroupAssigned then
-        self:UnAssignFromUnit( PlayerUnit )
+        local PlayerName = PlayerGroup:GetUnit(1):GetPlayerName()
-        self:MessageToGroups( PlayerUnit:GetPlayerName() .. " crashed in Task " .. self:GetName() )
+        self:MessageToGroups( PlayerName .. " crashed! " )
-        self:E( { TaskGroup = PlayerGroup:GetName(), GetUnits = PlayerGroup:GetUnits() } )
+        self:UnAssignFromGroup( PlayerGroup )
-        if #PlayerGroup:GetUnits() == 1 then
+
-          self:ClearGroupAssignment( PlayerGroup )
+        -- Now check if the task needs to go to hold...
-        end
+        -- It will go to hold, if there are no players in the mission...
-        self:PlayerCrashed( PlayerUnit )
+        
-      end
+        PlayerGroups:Flush()
+        local IsRemaining = false
+        for GroupName, AssignedGroup in pairs( PlayerGroups:GetSet() or {} ) do
+          if self:IsGroupAssigned( AssignedGroup ) == true then
+            IsRemaining = true
+            self:F( { Task = self:GetName(), IsRemaining = IsRemaining } )
+           break
           end
         end
 
-  return PlayerUnitCrashed
+        self:F( { Task = self:GetName(), IsRemaining = IsRemaining } )
+        if IsRemaining == false then
+          self:Abort()
+        end
+        
+        self:PlayerCrashed( PlayerGroup:GetUnit(1) )
+      end
+      
+    end
+  end
+  
+  return self
 end