library SpellLib
globals
    constant hashtable H = InitHashtable( )
    
    private constant real AuraRange        = 400.00
    private constant integer AuraID        = 'A000'
    private constant integer AbilityID     = 'A001'
    private constant integer AbilityMaxLvl = 10
    private constant integer UnitGroup     = StringHash( SCOPE_PRIVATE + "UnitGroup" )
    
    private constant group TempG   = CreateGroup( )
    private constant group TempG_1 = CreateGroup( )
    private group TempG_2          = null
    
    private boolexpr CondAura = null
    private boolexpr CondAdd  = null
    
    private unit TempUnit            = null
    private unit bj_lastFilterUnit   = null
    private unit bj_lastFilterUnit_1 = null
    
endglobals

native UnitAlive takes unit id returns boolean

private function CheckRange takes nothing returns nothing
    local integer i
    set TempUnit = GetEnumUnit( )
    
    if not IsUnitInGroup( TempUnit, TempG_1 ) then
        call GroupRemoveUnit( TempG_2, TempUnit )
        set i = GetUnitAbilityLevel( TempUnit, AbilityID ) - 1
        
        if i > 0 then
            call SetUnitAbilityLevel( TempUnit, AbilityID, i )
        else
            call UnitRemoveAbility( TempUnit, AbilityID )
        endif
    endif
endfunction

private function EnumAddBuff takes nothing returns boolean
    local integer i
    set bj_lastFilterUnit_1 = GetFilterUnit( )
    
    if IsUnitAlly( bj_lastFilterUnit_1, bj_groupEnumOwningPlayer ) and not IsUnitInGroup( bj_lastFilterUnit_1, TempG_2 ) then
        call GroupAddUnit( TempG_2, bj_lastFilterUnit_1 )
        set i = GetUnitAbilityLevel( bj_lastFilterUnit_1, AbilityID )
        if i == 0 then
            call UnitAddAbility( bj_lastFilterUnit_1, AbilityID )
        else
            call SetUnitAbilityLevel( bj_lastFilterUnit_1, AbilityID, i + 1 )
        endif
    endif
    
    return true
endfunction

private function EnumWithAura takes nothing returns boolean
    set bj_lastFilterUnit = GetFilterUnit( )
    
    if GetUnitAbilityLevel( bj_lastFilterUnit, AuraID ) > 0 then
        set TempG_2 = LoadGroupHandle( H, GetHandleId( bj_lastFilterUnit ), UnitGroup )
        if TempG_2 == null and UnitAlive( bj_lastFilterUnit ) then
            set TempG_2 = CreateGroup( )
            call SaveGroupHandle( H, GetHandleId( bj_lastFilterUnit ), UnitGroup, TempG_2 )
        endif
        
        if UnitAlive( bj_lastFilterUnit ) then
            set bj_groupEnumOwningPlayer = GetOwningPlayer( bj_lastFilterUnit )
            call GroupEnumUnitsInRange( TempG_1, GetUnitX( bj_lastFilterUnit ), GetUnitY( bj_lastFilterUnit ), AuraRange, CondAdd )
            call ForGroup( TempG_2, function CheckRange )
            call GroupClear( TempG_1 )
        else
            loop
                set TempUnit = FirstOfGroup( TempG_2 )
                exitwhen TempUnit == null
                call UnitRemoveAbility( TempUnit, AbilityID )
            endloop
            
            call DestroyGroup( TempG_2 )
            call RemoveSavedHandle( H, GetHandleId( bj_lastFilterUnit ), UnitGroup )
        endif
    endif
    
    return false
endfunction

function Spell_Actions takes nothing returns nothing
    call GroupEnumUnitsInRect( TempG, bj_mapInitialPlayableArea, CondAura )
endfunction

//===========================================================================
function InitTrig_Spell takes nothing returns nothing
    set gg_trg_Spell = CreateTrigger( )
    call TriggerRegisterTimerEvent( gg_trg_Spell, 1.00, true )
    call TriggerAddAction( gg_trg_Spell, function Spell_Actions )
    
    set CondAura = Condition( function EnumWithAura )
    set CondAdd  = Condition( function EnumAddBuff )
    
    set TempUnit = CreateUnit( Player( 0x0F ), 'hfoo', 0.00, 0.00, 0.00 )
    call UnitAddAbility( TempUnit, AbilityID )
    call SetUnitAbilityLevel( TempUnit, AbilityID, AbilityMaxLvl )
    call KillUnit( TempUnit )
    call RemoveUnit( TempUnit )
endfunction
endlibrary