go-go-try 7.4.0 → 8.0.0

package/src/goTryAll.ts

@@ -1,5 +1,43 @@
-import type { GoTryAllOptions } from './types.js'
+import type { GoTryAllOptions, GoTryAllRawOptions, ErrorConstructor } from './types.js'
 import { isError, getErrorMessage, type PromiseFactory } from './internals.js'
+import { UnknownError } from './unknown-error.js'
+
+/**
+ * Checks if a value is a tagged error (has a _tag property).
+ */
+function isTaggedError(err: unknown): err is { _tag: string } {
+  return isError(err) && '_tag' in err && typeof (err as { _tag?: unknown })._tag === 'string'
+}
+
+/**
+ * Wraps an error based on the provided options (errorClass/systemErrorClass).
+ * Consistent with goTryRaw behavior.
+ */
+function wrapError<E>(err: unknown, errorClass: ErrorConstructor<E> | undefined, systemErrorClass: ErrorConstructor<E> | undefined): E {
+  // If errorClass is specified, wrap all errors with it
+  if (errorClass) {
+    if (err === undefined) {
+      return new errorClass('undefined')
+    }
+    if (isError(err)) {
+      return new errorClass(err.message, { cause: err })
+    }
+    return new errorClass(String(err))
+  }
+
+  // If systemErrorClass is specified (or defaulted to UnknownError), only wrap non-tagged errors
+  const actualSystemErrorClass = systemErrorClass ?? UnknownError
+  if (isTaggedError(err)) {
+    return err as unknown as E
+  }
+  if (err === undefined) {
+    return new actualSystemErrorClass('undefined') as unknown as E
+  }
+  if (isError(err)) {
+    return new actualSystemErrorClass(err.message, { cause: err }) as unknown as E
+  }
+  return new actualSystemErrorClass(String(err)) as unknown as E
+}
 
 async function runWithConcurrency<T extends readonly unknown[]>(
   items: { [K in keyof T]: Promise<T[K]> | PromiseFactory<T[K]> },
@@ -106,33 +144,39 @@ export async function goTryAll<T extends readonly unknown[]>(
 
 /**
  * Like `goTryAll`, but returns raw Error objects instead of error messages.
+ * Non-tagged errors are wrapped in `UnknownError` by default (consistent with `goTryRaw`).
+ * Tagged errors pass through unchanged.
+ *
+ * Supports `errorClass` and `systemErrorClass` options (mutually exclusive):
+ * - `errorClass`: Wrap ALL errors in the specified class
+ * - `systemErrorClass`: Only wrap non-tagged errors (defaults to UnknownError)
  *
  * @template T The tuple type of all promise results
+ * @template E The type of the error
  * @param {readonly [...{ [K in keyof T]: Promise<T[K]> | (() => Promise<T[K]>) }]} items - Array of promises or factories
- * @param {GoTryAllOptions} options - Optional configuration
- * @returns {Promise<[{ [K in keyof T]: Error | undefined }, { [K in keyof T]: T[K] | undefined }]>}
+ * @param {GoTryAllRawOptions<E>} options - Optional configuration
+ * @returns {Promise<[{ [K in keyof T]: E | undefined }, { [K in keyof T]: T[K] | undefined }]>}
  *          A tuple where the first element is a tuple of Error objects (or undefined) and
  *          the second element is a tuple of results (or undefined), preserving input order
  */
-export async function goTryAllRaw<T extends readonly unknown[]>(
+export async function goTryAllRaw<T extends readonly unknown[], E = InstanceType<typeof UnknownError>>(
   items: { [K in keyof T]: Promise<T[K]> | (() => Promise<T[K]>) },
-  options?: GoTryAllOptions,
-): Promise<[{ [K in keyof T]: Error | undefined }, { [K in keyof T]: T[K] | undefined }]> {
+  options?: GoTryAllRawOptions<E>,
+): Promise<[{ [K in keyof T]: E | undefined }, { [K in keyof T]: T[K] | undefined }]> {
+  const { errorClass, systemErrorClass } = options || {}
   const settled = await runWithConcurrency(items, options?.concurrency ?? 0)
 
-  const errors = [] as { [K in keyof T]: Error | undefined }
+  const errors = [] as { [K in keyof T]: E | undefined }
   const results = [] as { [K in keyof T]: T[K] | undefined }
 
   for (let i = 0; i < settled.length; i++) {
     const item = settled[i]!
     if (item.status === 'fulfilled') {
-      ;(errors as (Error | undefined)[])[i] = undefined
+      ;(errors as (E | undefined)[])[i] = undefined
       ;(results as unknown[])[i] = (item as PromiseFulfilledResult<T[number]>).value
     } else {
       const reason = (item as PromiseRejectedResult).reason
-      ;(errors as (Error | undefined)[])[i] = isError(reason)
-        ? reason
-        : new Error(String(reason))
+      ;(errors as (E | undefined)[])[i] = wrapError(reason, errorClass, systemErrorClass)
       ;(results as unknown[])[i] = undefined
     }
   }

package/src/index.test.ts

@@ -1107,7 +1107,7 @@ describe('goTryAllRaw', () => {
     assert.deepEqual(results, ['a', 42, true])
   })
 
-  test('returns Error objects for failed promises', async () => {
+  test('wraps non-tagged errors in UnknownError', async () => {
     const [errors, results] = await goTryAllRaw([
       Promise.resolve('success'),
       Promise.reject(new Error('fail1')),
@@ -1115,6 +1115,8 @@ describe('goTryAllRaw', () => {
     ])
 
     assert.equal(errors[0], undefined)
+    assert.ok(errors[1] instanceof UnknownError)
+    assert.equal(errors[1]?._tag, 'UnknownError')
     assert.equal(errors[1]?.message, 'fail1')
     assert.equal(errors[2], undefined)
 
@@ -1123,15 +1125,44 @@ describe('goTryAllRaw', () => {
     assert.equal(results[2], 42)
   })
 
-  test('converts non-Error rejections to Error objects', async () => {
+  test('tagged errors pass through unchanged', async () => {
+    const DatabaseError = taggedError('DatabaseError')
+    const NetworkError = taggedError('NetworkError')
+
+    const [errors] = await goTryAllRaw([
+      Promise.reject(new DatabaseError('db failed')),
+      Promise.reject(new NetworkError('network timeout')),
+      Promise.reject(new Error('plain error')),
+    ])
+
+    // Tagged errors pass through
+    assert.ok(errors[0] instanceof DatabaseError)
+    assert.equal(errors[0]?._tag, 'DatabaseError')
+    assert.equal(errors[0]?.message, 'db failed')
+    assert.ok(errors[1] instanceof NetworkError)
+    assert.equal(errors[1]?._tag, 'NetworkError')
+    assert.equal(errors[1]?.message, 'network timeout')
+    // Non-tagged errors get wrapped in UnknownError
+    assert.ok(errors[2] instanceof UnknownError)
+    assert.equal(errors[2]?._tag, 'UnknownError')
+    assert.equal(errors[2]?.message, 'plain error')
+  })
+
+  test('converts non-Error rejections to UnknownError objects', async () => {
     const [errors] = await goTryAllRaw([
       Promise.reject('string error'),
       Promise.reject(42),
       Promise.reject(undefined),
     ])
 
+    assert.ok(errors[0] instanceof UnknownError)
+    assert.equal(errors[0]?._tag, 'UnknownError')
     assert.equal(errors[0]?.message, 'string error')
+    assert.ok(errors[1] instanceof UnknownError)
+    assert.equal(errors[1]?._tag, 'UnknownError')
     assert.equal(errors[1]?.message, '42')
+    assert.ok(errors[2] instanceof UnknownError)
+    assert.equal(errors[2]?._tag, 'UnknownError')
     assert.equal(errors[2]?.message, 'undefined')
   })
 
@@ -1141,6 +1172,60 @@ describe('goTryAllRaw', () => {
     assert.deepEqual(errors, [])
     assert.deepEqual(results, [])
   })
+
+  test('errorClass wraps all errors', async () => {
+    const DatabaseError = taggedError('DatabaseError')
+    const NetworkError = taggedError('NetworkError')
+
+    const [errors] = await goTryAllRaw([
+      Promise.reject(new DatabaseError('db error')),
+      Promise.reject(new NetworkError('network error')),
+      Promise.reject(new Error('plain error')),
+    ], { errorClass: DatabaseError })
+
+    // All errors should be wrapped in DatabaseError
+    assert.ok(errors[0] instanceof DatabaseError)
+    assert.equal(errors[0]?._tag, 'DatabaseError')
+    assert.ok(errors[1] instanceof DatabaseError)
+    assert.equal(errors[1]?._tag, 'DatabaseError')
+    assert.ok(errors[2] instanceof DatabaseError)
+    assert.equal(errors[2]?._tag, 'DatabaseError')
+  })
+
+  test('systemErrorClass only wraps non-tagged errors', async () => {
+    const DatabaseError = taggedError('DatabaseError')
+    const NetworkError = taggedError('NetworkError')
+    const SystemError = taggedError('SystemError')
+
+    const [errors] = await goTryAllRaw([
+      Promise.reject(new DatabaseError('db error')),
+      Promise.reject(new NetworkError('network error')),
+      Promise.reject(new Error('plain error')),
+    ], { systemErrorClass: SystemError })
+
+    // Tagged errors pass through
+    assert.ok(errors[0] instanceof DatabaseError)
+    assert.equal(errors[0]?._tag, 'DatabaseError')
+    assert.ok(errors[1] instanceof NetworkError)
+    assert.equal(errors[1]?._tag, 'NetworkError')
+    // Non-tagged error wrapped in SystemError
+    assert.ok(errors[2] instanceof SystemError)
+    assert.equal(errors[2]?._tag, 'SystemError')
+  })
+
+  test('concurrency option works with errorClass', async () => {
+    const DatabaseError = taggedError('DatabaseError')
+
+    const [errors] = await goTryAllRaw([
+      Promise.reject(new Error('error 1')),
+      Promise.reject(new Error('error 2')),
+      Promise.reject(new Error('error 3')),
+    ], { concurrency: 2, errorClass: DatabaseError })
+
+    assert.ok(errors[0] instanceof DatabaseError)
+    assert.ok(errors[1] instanceof DatabaseError)
+    assert.ok(errors[2] instanceof DatabaseError)
+  })
 })
 
 

package/src/index.ts

@@ -10,6 +10,7 @@ export type {
   ErrorConstructor,
   TaggedUnion,
   GoTryRawOptions,
+  GoTryAllRawOptions,
 } from './types.js'
 
 // Export core functions
@@ -21,7 +22,11 @@ export { goTryAll, goTryAllRaw } from './goTryAll.js'
 // Export helper functions
 export { taggedError } from './tagged-error.js'
 export { assert } from './assert.js'
+export { ensure } from './ensure.js'
 export { isSuccess, isFailure, success, failure, assertNever } from './result-helpers.js'
 
 // Export UnknownError tagged error
 export { UnknownError } from './unknown-error.js'
+
+// Export go aliases
+export { go, goAll, goElse } from './go.js'

package/src/types.ts

@@ -47,6 +47,16 @@ export type GoTryRawOptions<E = Error> =
   | { errorClass?: never; systemErrorClass: ErrorConstructor<E> }
   | { errorClass?: never; systemErrorClass?: never }
 
+/**
+ * Options for goTryAllRaw function.
+ * Includes concurrency control and error class options.
+ * errorClass and systemErrorClass are mutually exclusive.
+ */
+export type GoTryAllRawOptions<E = Error> =
+  | { concurrency?: number; errorClass: ErrorConstructor<E>; systemErrorClass?: never }
+  | { concurrency?: number; errorClass?: never; systemErrorClass: ErrorConstructor<E> }
+  | { concurrency?: number; errorClass?: never; systemErrorClass?: never }
+
 /**
  * Creates a union type from multiple tagged error classes.
  *