haxeui/haxeui-core

package haxe.ui.util;

import haxe.ui.core.Component;
import haxe.ui.events.EventType;
import haxe.ui.events.UIEvent;

class EventMap  {
    private var _map:Map<String, FunctionArray<UIEvent->Void>>;

    public function new() {
        _map = new Map<String, FunctionArray<UIEvent->Void>>();
    }

    public function keys():Iterator<String> {
        return _map.keys();
    }

    public function removeAll<T:UIEvent>(type:EventType<T> = null) {
        if (type != null) {
            if (_map.exists(type)) {
                _map.get(type).removeAll();
                _map.remove(type);
            }
        } else {
            for (type in _map.keys()) {
                _map.get(type).removeAll();
            }
        }
    }

    public function add<T:UIEvent>(type:EventType<T>, listener:T->Void, priority:Int = 0):Bool { // returns true if a new FunctionArray was created
        if (listener == null) {
            return false;
        }
        var b:Bool = false;
        var arr:FunctionArray<UIEvent->Void> = _map.get(type);
        if (arr == null) {
            arr = new FunctionArray<UIEvent->Void>();
            arr.push(cast listener, priority);
            _map.set(type, arr);
            b = true;
        } else if (arr.contains(cast listener) == false) {
            arr.push(cast listener, priority);
        }
        return b;
    }

    public function remove<T:UIEvent>(type:EventType<T>, listener:T->Void):Bool { // returns true if a FunctionArray was removed
        if (listener == null) {
            return false;
        }
        var b:Bool = false;
        var arr:FunctionArray<UIEvent->Void> = _map.get(type);
        if (arr != null) {
            arr.remove(cast listener);
            if (arr.length == 0) {
                _map.remove(type);
                b = true;
            }
        }
        return b;
    }

    public function contains<T:UIEvent>(type:EventType<T>, listener:T->Void = null):Bool {
        var b:Bool = false;
        var arr:FunctionArray<UIEvent->Void> = _map.get(type);
        if (arr != null) {
            b = (listener != null) ? arr.contains(cast listener) : true;
        }
        return b;
    }

    public function invoke<T:UIEvent>(type:EventType<T>, event:T, target:Component = null) {
        if (event.bubble && event.target == null) {
            event.target = target;
        }

        var arr:FunctionArray<UIEvent->Void> = _map.get(type);
        if (arr != null && arr.length > 0) {
            arr = arr.copy();
            for (listener in arr) {
                if (event.canceled) {
                    break;
                }

                var c = event.clone();
                if (c.target == null) {
                    c.target = target;
                }
                listener.callback(c);
                event.copyFrom(c);
                event.canceled = c.canceled;
            }
        }
    }

    public function listenerCount<T:UIEvent>(type:EventType<T>):Int {
        var n:Int = 0;
        var arr:FunctionArray<UIEvent->Void> = _map.get(type);
        if (arr != null) {
            n = arr.length;
        }
        return n;
    }

    public function listeners<T:UIEvent>(type:EventType<T>):FunctionArray<UIEvent->Void> {
        var arr:FunctionArray<UIEvent->Void> = _map.get(type);
        if (arr == null) {
            return null;
        }
        return arr;
    }
}