import {
    Err,
    type UnwrapOption,
    Ok,
    type Result,
    FlattenResult,
} from "@shared/utils/result.ts";
import { none, None, Option, some, Some } from "@shared/utils/option.ts";

export class ResultAsync<T, E> implements PromiseLike<Result<T, E>> {
    constructor(private readonly _promise: Promise<Result<T, E>>) {
        this._promise = _promise;
    }

    static fromPromise<T, E>(
        promise: Promise<T>,
        errorMapper: (error: unknown) => E,
    ) {
        const promiseOfResult: Promise<Result<T, E>> = promise
            .then((value: T): Result<T, E> => {
                return new Ok<T, E>(value);
            })
            .catch((error: unknown): Result<T, E> => {
                return new Err<T, E>(errorMapper(error));
            });

        return new ResultAsync<T, E>(promiseOfResult);
    }

    static fromSafePromise<T>(promise: Promise<T>) {
        const promiseOfResult: Promise<Result<T, never>> = promise.then(
            (value: T): Result<T, never> => {
                return new Ok<T, never>(value);
            },
        );

        return new ResultAsync<T, never>(promiseOfResult);
    }

    static fromThrowable<Fn extends (...args: readonly any[]) => any, E>(
        fn: Fn,
        errorMapper?: (e: unknown) => E,
    ): (...args: Parameters<Fn>) => ResultAsync<ReturnType<Fn>, E> {
        return (...args: Parameters<Fn>): ResultAsync<ReturnType<Fn>, E> => {
            try {
                return okAsync(fn(args));
            } catch (e) {
                return errAsync(errorMapper ? errorMapper(e) : e);
            }
        };
    }

    async unwrap(): Promise<T> {
        const result = await this._promise;

        if (result.isErr()) {
            throw result.error;
        }

        return result.value;
    }

    async match<A, B = A>(
        ok: (value: T) => A,
        err: (err: E) => B,
    ): Promise<A | B> {
        const result = await this._promise;

        if (result.isErr()) {
            return err(result.error);
        }
        return ok(result.value);
    }

    map<U>(fn: (value: T) => U): ResultAsync<U, E> {
        return new ResultAsync(
            this._promise.then((result: Result<T, E>): Result<U, E> => {
                if (result.isErr()) {
                    return new Err<U, E>(result.error);
                }

                return new Ok<U, E>(fn(result.value));
            }),
        );
    }

    mapAsync<U>(fn: (value: T) => Promise<U>): ResultAsync<U, E> {
        return new ResultAsync(
            this._promise.then(
                async (result: Result<T, E>): Promise<Result<U, E>> => {
                    if (result.isErr()) {
                        return errAsync<U, E>(result.error);
                    }

                    return new Ok<U, E>(await fn(result.value));
                },
            ),
        );
    }

    mapErr<U>(fn: (err: E) => U): ResultAsync<T, U> {
        return new ResultAsync(
            this._promise.then((result: Result<T, E>): Result<T, U> => {
                if (result.isErr()) {
                    return new Err<T, U>(fn(result.error));
                }

                return new Ok<T, U>(result.value);
            }),
        );
    }

    mapErrAsync<U>(fn: (value: T) => Promise<U>): ResultAsync<T, U> {
        return new ResultAsync(
            this._promise.then(
                async (result: Result<T, E>): Promise<Result<T, U>> => {
                    if (result.isErr()) {
                        return errAsync<T, U>(await fn(result.error));
                    }

                    return new Ok<T, U>(result.value);
                },
            ),
        );
    }

    andThen<U, F>(fn: (value: T) => ResultAsync<U, F>): ResultAsync<U, E | F> {
        return new ResultAsync(
            this._promise.then(
                (result: Result<T, E>): ResultAsync<U, E | F> => {
                    if (result.isErr()) {
                        return errAsync(result.error);
                    }

                    return fn(result.value) as ResultAsync<U, E | F>;
                },
            ),
        );
    }

    nullableToOption(): ResultAsync<Option<NonNullable<T>>, E> {
        return this.map((v) => (v ? some(v) : none));
    }

    flatten(): FlattenResultAsync<ResultAsync<T, E>> {
        return new ResultAsync(
            this._promise.then(
                (result: Result<T, E>): FlattenResult<Result<T, E>> => {
                    return result.flatten();
                },
            ),
        ) as FlattenResultAsync<ResultAsync<T, E>>;
    }

    flattenOption<U = never>(
        errFn: () => U,
    ): ResultAsync<UnwrapOption<T>, E | U> {
        return new ResultAsync(
            this._promise.then(
                (result: Result<T, E>): Result<UnwrapOption<T>, E | U> => {
                    return result.flattenOption(errFn);
                },
            ),
        );
    }

    flattenOptionOrDefault<D = UnwrapOption<T>>(
        defaultValue: D,
    ): ResultAsync<UnwrapOption<T> | D, E> {
        return new ResultAsync(
            this._promise.then(
                (result: Result<T, E>): Result<UnwrapOption<T> | D, E> => {
                    return result.flattenOptionOrDefault(defaultValue);
                },
            ),
        );
    }

    matchOption<A, B = A>(
        some: (value: UnwrapOption<T>) => A,
        none: () => B,
    ): ResultAsync<A | B, E> {
        return new ResultAsync(
            this._promise.then((result: Result<T, E>): Result<A | B, E> => {
                return result.matchOption(some, none);
            }),
        );
    }

    matchOptionAndFlatten<A, B, U, F>(
        some: (value: UnwrapOption<T>) => Result<A, U>,
        none: () => Result<B, F>,
    ): ResultAsync<A | B, E | U | F> {
        return new ResultAsync(
            this._promise.then(
                (result: Result<T, E>): Result<A | B, E | U | F> => {
                    return result.matchOptionAndFlatten(some, none);
                },
            ),
        );
    }

    then<A, B>(
        onFulfilled?: (res: Result<T, E>) => A | PromiseLike<A>,
        onRejected?: (reason: unknown) => B | PromiseLike<B>,
    ): PromiseLike<A | B> {
        return this._promise.then(onFulfilled, onRejected);
    }
}

export function okAsync<T, E = never>(value: T): ResultAsync<T, E>;
export function okAsync<T extends void = void, E = never>(
    value: void,
): ResultAsync<void, E>;
export function okAsync<T, E = never>(value: T): ResultAsync<T, E> {
    return new ResultAsync(Promise.resolve(new Ok<T, E>(value)));
}

export function errAsync<T = never, E extends string = string>(
    err: E,
): ResultAsync<T, E>;
export function errAsync<T = never, E = unknown>(err: E): ResultAsync<T, E>;
export function errAsync<T = never, E extends void = void>(
    err: void,
): ResultAsync<T, void>;
export function errAsync<E, T = never>(err: E): ResultAsync<T, E> {
    return new ResultAsync(Promise.resolve(new Err<T, E>(err)));
}

export type FlattenResultAsync<R> =
    R extends ResultAsync<infer T, infer E>
        ? T extends ResultAsync<any, any>
            ? FlattenResultAsync<T> extends ResultAsync<infer V, infer innerE>
                ? ResultAsync<V, E | innerE>
                : never
            : R
        : never;