ComfyUI_frontend/src/storybook/mocks/useJobActions.ts

import { computed, ref, toValue } from 'vue'
import type { MaybeRefOrGetter } from 'vue'

import type { JobAction } from '../../composables/queue/useJobActions'
import type { JobListItem } from '../../composables/queue/useJobList'
import type { JobState } from '../../types/queue'

const actionsByJobId = ref<Record<string, JobAction[]>>({})
const cancellableStates: JobState[] = ['pending', 'initialization', 'running']
const cancelAction: JobAction = {
  icon: 'icon-[lucide--x]',
  label: 'Cancel',
  variant: 'destructive'
}

export function setMockJobActions(actions: Record<string, JobAction[]>) {
  actionsByJobId.value = actions
}

export function useJobActions(
  job?: MaybeRefOrGetter<JobListItem | null | undefined>
) {
  const jobRef = computed(() => (job ? (toValue(job) ?? null) : null))

  const canCancelJob = computed(() => {
    const currentJob = jobRef.value
    if (!currentJob) {
      return false
    }

    const configuredActions = actionsByJobId.value[currentJob.id]
    if (configuredActions) {
      return configuredActions.length > 0
    }

    return (
      currentJob.showClear !== false &&
      cancellableStates.includes(currentJob.state)
    )
  })

  async function runCancelJob() {
    return undefined
  }

  return {
    cancelAction,
    canCancelJob,
    runCancelJob
  }
}