library IsUnitInvulnerableCheck initializer Init

globals
    unit DummyAttacker
    trigger gg_trg_Damage_Blocker
    group gg_grp_CheckedGroup
    hashtable GameData = InitHashtable( )
    boolean IsInvul = false
endglobals

function Trig_Damage_Blocker_Conditions takes nothing returns boolean
    return GetEventDamage() > 0.00 and GetEventDamageSource( ) == DummyAttacker
endfunction

function Timer_Heal_Unit_Expires takes nothing returns nothing
    local timer t = GetExpiredTimer( )
    local integer id = GetHandleId(t)
    local unit Damaged = LoadUnitHandle( GameData, id, 0 )
    
    call UnitRemoveAbility( Damaged, 'AMHP' )
    call SetUnitState( Damaged, UNIT_STATE_LIFE, LoadReal( GameData, id, 1 ) )
    call SetUnitState( Damaged, UNIT_STATE_MANA, LoadReal( GameData, id, 2 ) )
   
    call DestroyTimer(t)
    call FlushChildHashtable( GameData, id )
    
    set Damaged = null
    set t = null
endfunction 

function Trig_Damage_Blocker_Actions takes nothing returns nothing
    local unit Damaged = GetTriggerUnit( )
    local timer t = CreateTimer()
    local integer id = GetHandleId( t )

    call SaveUnitHandle( GameData, id, 0, Damaged )
    call SaveReal( GameData, id, 1, GetUnitState( Damaged, UNIT_STATE_LIFE ) )
    call SaveReal( GameData, id, 2, GetUnitState( Damaged, UNIT_STATE_MANA ) )
    
    call TimerStart( t, 0.00, false, function Timer_Heal_Unit_Expires )
    
    call UnitAddAbility( Damaged, 'AMHP' )
    call SetUnitState( Damaged, UNIT_STATE_LIFE, GetUnitState( Damaged, UNIT_STATE_MAX_LIFE ) )
    
    set IsInvul = true
    
    set Damaged = null
    set t = null
endfunction


function IsUnitInvul takes unit enemy returns boolean
    local real hp = GetUnitState( enemy, UNIT_STATE_LIFE )
    local real mp = GetUnitState( enemy, UNIT_STATE_MANA )
    
	set IsInvul = false

    if hp < 0.405 or GetUnitTypeId(enemy) < 1 then
        return true
    elseif GetUnitAbilityLevel( enemy, 'Avul' ) > 0 then
        return true
    endif
        
    if not IsUnitInGroup( enemy, gg_grp_CheckedGroup ) then
        call TriggerRegisterUnitEvent( gg_trg_Damage_Blocker, enemy, EVENT_UNIT_DAMAGED )
        call GroupAddUnit( gg_grp_CheckedGroup, enemy )
    endif

    call EnableTrigger( gg_trg_Damage_Blocker )
    call UnitDamageTarget( DummyAttacker, enemy, 100.00, false, false, ATTACK_TYPE_CHAOS, DAMAGE_TYPE_UNKNOWN, WEAPON_TYPE_WHOKNOWS )
    call DisableTrigger( gg_trg_Damage_Blocker )
    
    return IsInvul
endfunction 

private function Init takes nothing returns nothing
    set DummyAttacker = CreateUnit( Player(12), 'hpea', 0.00,0.00, 270.00 )
    
    call SetUnitPathing( DummyAttacker, true )
    call PauseUnit( DummyAttacker, true )
    call SetUnitInvulnerable( DummyAttacker, true )
    call ShowUnit( DummyAttacker, false )
    
    
    set gg_grp_CheckedGroup = CreateGroup( )
    set gg_trg_Damage_Blocker = CreateTrigger( )

    call TriggerAddCondition( gg_trg_Damage_Blocker, Condition( function Trig_Damage_Blocker_Conditions ) )
    call TriggerAddAction( gg_trg_Damage_Blocker, function Trig_Damage_Blocker_Actions )
    call DisableTrigger( gg_trg_Damage_Blocker )
endfunction

endlibrary

function GetAbilityDisabledCount takes integer pAbility returns integer
        return Memory[pAbility / 4 + 15]
    endfunction