|
| 1 | +import type { BlobType } from '../../shared' |
| 2 | + |
| 3 | +import { Err } from './err' |
| 4 | +import { Ok } from './ok' |
| 5 | + |
| 6 | +export class ResultAsync<T, N extends Std.ErrorValues, C extends Std.ErrorValues[] = []> |
| 7 | + implements PromiseLike<Std.Result<T, N, C>> |
| 8 | +{ |
| 9 | + private _promise: Promise<Std.Result<T, N, C>> |
| 10 | + |
| 11 | + constructor(res: Promise<Std.Result<T, N, C>>) { |
| 12 | + this._promise = res |
| 13 | + } |
| 14 | + |
| 15 | + // Makes ResultAsync implement PromiseLike<Result> |
| 16 | + // biome-ignore lint/suspicious/noThenProperty: Redundant |
| 17 | + then<A, B>( |
| 18 | + successCallback?: (res: Std.Result<T, N, C>) => A | PromiseLike<A>, |
| 19 | + failureCallback?: (reason: unknown) => B | PromiseLike<B> |
| 20 | + ): PromiseLike<A | B> { |
| 21 | + return this._promise.then(successCallback, failureCallback) |
| 22 | + } |
| 23 | + |
| 24 | + async *[Symbol.asyncIterator](): AsyncGenerator<Err<never, N, C>, T> { |
| 25 | + const result = await this._promise |
| 26 | + |
| 27 | + if (result.isErr()) { |
| 28 | + // @ts-expect-error -- This is structurally equivalent and safe |
| 29 | + yield errAsync(result.error) |
| 30 | + } |
| 31 | + |
| 32 | + // @ts-expect-error -- This is structurally equivalent and safe |
| 33 | + return result.value |
| 34 | + } |
| 35 | +} |
| 36 | + |
| 37 | +export const okAsync = <T, N extends Std.ErrorValues>(value: T): ResultAsync<T, N, never> => |
| 38 | + new ResultAsync(Promise.resolve(new Ok<T, never, never>(value))) |
| 39 | + |
| 40 | +export const errAsync = <N extends Std.ErrorValues>( |
| 41 | + err: N, |
| 42 | + message = '' |
| 43 | +): ResultAsync<never, N, never> => |
| 44 | + new ResultAsync(Promise.resolve(new Err<never, N, never>(err, message))) |
| 45 | + |
| 46 | +export const fromThrowable = <A extends readonly BlobType[], T, R>( |
| 47 | + fn: (...args: A) => Promise<T>, |
| 48 | + errorFn: (err: unknown) => R |
| 49 | +): (( |
| 50 | + ...args: A |
| 51 | +) => ResultAsync< |
| 52 | + T, |
| 53 | + Std.InferNameType<Std.UnionsToResult<R>>, |
| 54 | + Std.InferCauseType<Std.UnionsToResult<R>> |
| 55 | +>) => { |
| 56 | + return ((...args: BlobType) => { |
| 57 | + return new ResultAsync( |
| 58 | + (async () => { |
| 59 | + try { |
| 60 | + return new Ok(await fn(...args)) |
| 61 | + } catch (error) { |
| 62 | + return errorFn(error) as BlobType |
| 63 | + } |
| 64 | + })() |
| 65 | + ) |
| 66 | + }) as BlobType |
| 67 | +} |
0 commit comments