import type { ReactNode } from 'react'
import React from 'react'
import { Text } from 'src/ink.js'
import type { TaskStatus } from 'src/Task.js'
import type { LocalShellTaskState } from 'src/tasks/LocalShellTask/guards.js'
import type { DeepImmutable } from 'src/types/utils.js'

type TaskStatusTextProps = {
  status: TaskStatus
  label?: string
  suffix?: string
}

export function TaskStatusText({
  status,
  label,
  suffix,
}: TaskStatusTextProps): ReactNode {
  const displayLabel = label ?? status
  const color =
    status === 'completed'
      ? 'success'
      : status === 'failed'
        ? 'error'
        : status === 'killed'
          ? 'warning'
          : undefined
  return (
    <Text color={color} dimColor>
      ({displayLabel}
      {suffix})
    </Text>
  )
}

export function ShellProgress({
  shell,
}: {
  shell: DeepImmutable<LocalShellTaskState>
}): ReactNode {
  switch (shell.status) {
    case 'completed':
      return <TaskStatusText status="completed" label="done" />
    case 'failed':
      return <TaskStatusText status="failed" label="error" />
    case 'killed':
      return <TaskStatusText status="killed" label="stopped" />
    case 'running':
    case 'pending':
      return <TaskStatusText status="running" />
  }
}