/**
 *
 */
import { Disposable, IDisposable, IDisposableOrSubscription, ISubscription } from './Disposable';
export class CompositeDisposable extends Set<IDisposableOrSubscription> implements IDisposable, ISubscription {
    private _isDisposed = false;

    constructor(...disposables: IDisposableOrSubscription[]) {
        super(disposables);
    }

    public get isDisposed() {
        return this._isDisposed;
    }

    public dispose() {
        if (this._isDisposed) return;
        this._isDisposed = true;
        if (this.size) {
            this.forEach(disposable => Disposable.of(disposable).dispose());
            this.clear();
        }
    }

    public unsubscribe(): void {
        this.dispose();
    }

    public add(...disposables: IDisposableOrSubscription[]) {
        if (this.isDisposed) {
            disposables.forEach(item => Disposable.of(item).dispose());
        } else {
            disposables.forEach(item => super.add(item));
        }
        return this;
    }

    public remove(disposable: IDisposableOrSubscription) {
        this.delete(disposable);
        return this;
    }
}