import { IObservable, ICancellation, IDestroyable } from "@implab/core/interfaces";
import { Cancellation } from "@implab/core/Cancellation";
import { TraceEvent, LogLevel, WarnLevel, DebugLevel, TraceSource } from "@implab/core/log/TraceSource";
import * as tape from "tape";
import { argumentNotNull, destroy } from "@implab/core/safe";

export class TapeWriter implements IDestroyable {
    readonly _tape: tape.Test;

    _subscriptions = new Array<IDestroyable>();

    constructor(t: tape.Test) {
        argumentNotNull(t, "tape");
        this._tape = t;
    }

    writeEvents(source: IObservable<TraceEvent>, ct: ICancellation = Cancellation.none) {
        const subscription = source.on(this.writeEvent.bind(this));
        if (ct.isSupported()) {
            ct.register(subscription.destroy.bind(subscription));
        }
        this._subscriptions.push(subscription);
    }

    writeEvent(next: TraceEvent) {
        if (next.level >= DebugLevel) {
            this._tape.comment(`DEBUG ${next.source.id} ${next.arg}`);
        } else if (next.level >= LogLevel) {
            this._tape.comment(`LOG   ${next.source.id} ${next.arg}`);
        } else if (next.level >= WarnLevel) {
            this._tape.comment(`WARN  ${next.source.id} ${next.arg}`);
        } else {
            this._tape.comment(`ERROR ${next.source.id} ${next.arg}`);
        }
    }

    destroy() {
        this._subscriptions.forEach(destroy);
    }
}

export async function delay(timeout: number, ct: ICancellation = Cancellation.none) {
    let un: IDestroyable;

    try {
        await new Promise((resolve, reject) => {
            if (ct.isRequested()) {
                un = ct.register(reject);
            } else {
                const ht = setTimeout(() => {
                    resolve();
                }, timeout);

                un = ct.register(e => {
                    clearTimeout(ht);
                    reject(e);
                });
            }
        });
    } finally {
        destroy(un);
    }
}

export function test(name: string, cb: (t: tape.Test) => any) {
    tape(name, async t => {
        const writer = new TapeWriter(t);

        TraceSource.on(ts => {
            ts.level = DebugLevel;
            writer.writeEvents(ts.events);
        });

        try {
            await cb(t);
        } catch (e) {

            // verbose error information
            // tslint:disable-next-line
            console.error(e);
            t.fail(e);

        } finally {
            t.end();
            destroy(writer);
        }
    });
}