iterative/vscode-dvc

import { Args, Flag } from './constants'
import { Prompt } from './output'
import { MaybeConsoleError } from '../error'
import { Modal } from '../../vscode/modal'
import { CommandId, InternalCommands } from '../../commands/internal'
import { Response } from '../../vscode/response'

const offerToForce = async (
  stderr: string,
  internalCommands: InternalCommands,
  commandId: CommandId,
  ...args: Args
): Promise<string | undefined> => {
  const text = stderr.replace(
    Prompt.TRY_FORCE,
    '\n\nWould you like to force this action?'
  )
  const response = await Modal.warnOfConsequences(text, Response.FORCE)
  if (response !== Response.FORCE) {
    return
  }
  return internalCommands.executeCommand(commandId, ...args, Flag.FORCE)
}

export const tryThenMaybeForce = async (
  internalCommands: InternalCommands,
  commandId: CommandId,
  ...args: Args
): Promise<string | undefined> => {
  try {
    return await internalCommands.executeCommand(commandId, ...args)
  } catch (error: unknown) {
    const stderr = (error as MaybeConsoleError).stderr

    if (stderr && Prompt.TRY_FORCE.test(stderr)) {
      return offerToForce(stderr, internalCommands, commandId, ...args)
    }

    throw error
  }
}