error.ts

const errorRe =
  /(Code|Error): (?<code>\d+).*Exception: (?<message>.+)\((?<type>(?=.+[A-Z]{3})[A-Z0-9_]+?)\)/s

interface ParsedClickHouseError {
  message: string
  code: string
  type?: string
}

/** An error that is thrown by the ClickHouse server. */
export class ClickHouseError extends Error {
  readonly code: string
  readonly type: string | undefined
  constructor({ message, code, type }: ParsedClickHouseError) {
    super(message)
    this.code = code
    this.type = type

    // Set the prototype explicitly, see:
    // https://github.com/Microsoft/TypeScript/wiki/Breaking-Changes#extending-built-ins-like-error-array-and-map-may-no-longer-work
    Object.setPrototypeOf(this, ClickHouseError.prototype)
  }
}

export function parseError(input: string | Error): ClickHouseError | Error {
  const inputIsError = input instanceof Error
  const message = inputIsError ? input.message : input
  const match = message.match(errorRe)
  const groups = match?.groups as ParsedClickHouseError | undefined
  if (groups) {
    return new ClickHouseError(groups)
  } else {
    return inputIsError ? input : new Error(input)
  }
}

export function getCurrentStackTrace(): string {
  const originalStackTraceLimit = Error.stackTraceLimit
  Error.stackTraceLimit = 100
  const stack = new Error().stack
  Error.stackTraceLimit = originalStackTraceLimit

  if (!stack) return ''

  let foundNewLines = 0
  let idx = 0

  // Skip the first three lines of the stack trace, containing useless information
  // - Text `Error`
  // - Info about this function call
  // - Info about the originator of this function call, e.g., `request`
  while (foundNewLines < 3) {
    idx = stack.indexOf('\n', idx + 1)
    if (idx === -1) {
      return ''
    } else {
      foundNewLines++
    }
  }

  return stack.substring(idx + 1)
}

export function addStackTrace<E extends Error>(err: E, stackTrace: string): E {
  if (err.stack) {
    const firstNewlineIndex = err.stack.indexOf('\n')
    const firstLine = err.stack.substring(0, firstNewlineIndex)
    const errStack = err.stack.substring(firstNewlineIndex + 1)
    err.stack = `${firstLine}\n${stackTrace}\n${errStack}`
  }
  return err
}