import { PromptCancelledError } from 'lib/errors.js'
import { getOutput } from 'lib/output/get-output.js'
import {
  promptNumber,
  promptSelect,
  promptText,
  withBackHint,
} from 'lib/prompt.js'

export const interactForArray = async (
  array: string[],
  message: string,
): Promise<string[]> => {
  const updatedArray = [...array]
  const output = getOutput()

  const displayList = () => {
    output.info(`${message} Current list:`)
    if (updatedArray.length > 0) {
      updatedArray.forEach((item, index) => {
        output.info(`${index + 1}: ${item}`)
      })
    } else {
      output.info('The list is currently empty.')
    }
  }

  let action: string
  do {
    displayList()

    try {
      action = await promptSelect({
        message: withBackHint('Choose an action:'),
        choices: [
          { label: 'Add an item', value: 'add' },
          { label: 'Remove an item', value: 'remove' },
          { label: 'Finish editing', value: 'done' },
        ],
      })
    } catch (error) {
      if (!(error instanceof PromptCancelledError)) throw error
      // Dismissing the action menu finishes editing, keeping the changes.
      break
    }

    try {
      if (action === 'add') {
        const newItem = await promptText({
          message: withBackHint('Enter the new item:'),
        })
        if (newItem) {
          updatedArray.push(newItem)
        }
      } else if (action === 'remove') {
        const index = await promptNumber({
          message: withBackHint('Enter the index of the item to remove:'),
          validate: (value) =>
            value > 0 && value <= updatedArray.length
              ? undefined
              : 'Invalid index',
        })
        updatedArray.splice(index - 1, 1)
      }
    } catch (error) {
      if (!(error instanceof PromptCancelledError)) throw error
      // Dismissing an inner prompt returns to the action menu.
    }
  } while (action !== 'done')

  return updatedArray
}