import { Button, Node, } from "cc";

export function addBtnClick(btn: Button | Node, onClick: Function, target?) {
    if (btn instanceof Node) {
        btn.on(Node.EventType.TOUCH_END, onClick, target);
        return;
    }

    if (btn instanceof Button) {
        btn.node.on(Button.EventType.CLICK, onClick, target);
        return;
    }
}

export function removeBtnClick(btn: Button | Node, onClick, target) {
    if (btn instanceof Node) {
        btn.off(Node.EventType.TOUCH_END, onClick, target);
        return;
    }

    if (btn instanceof Button) {
        btn.node.off(Button.EventType.CLICK, onClick, target);
        return;
    }
}