@prefecthq/prefect-ui-library/src/automations/maps/workPoolStatusTrigger.ts

This library is the Vue and Typescript component library for [Prefect 2](https://github.com/PrefectHQ/prefect) and [Prefect Cloud 2](https://www.prefect.io/cloud/). _The components and utilities in this project are not meant to be used independently_.

import { toResourceId, fromResourceId } from '@/automations/maps/utilities'
import { AutomationTriggerEvent } from '@/automations/types'
import { WorkPoolStatusEvent, WorkPoolStatusTrigger, isWorkPoolStatusEvent } from '@/automations/types/workPoolStatusTrigger'
import { WorkPoolStatus, workPoolStatus } from '@/models/WorkPoolStatus'
import { MapFunction } from '@/services'

export const mapWorkPoolStatusTriggerToAutomationTrigger: MapFunction<WorkPoolStatusTrigger, AutomationTriggerEvent> = function(source) {
  if (source.posture === 'Reactive') {
    return mapReactiveWorkPoolStatusTriggerToAutomationTrigger(source)
  }

  return mapProactiveWorkPoolStatusTriggerToAutomationTrigger(source)
}

export const mapAutomationTriggerToWorkPoolStatusTrigger: MapFunction<AutomationTriggerEvent, WorkPoolStatusTrigger> = function(source) {
  if (source.posture === 'Reactive') {
    return mapReactiveAutomationTriggerToWorkPoolStatusTrigger(source)
  }

  return mapProactiveAutomationTriggerToWorkPoolStatusTrigger(source)
}

function mapReactiveWorkPoolStatusTriggerToAutomationTrigger(source: WorkPoolStatusTrigger): AutomationTriggerEvent {
  return new AutomationTriggerEvent({
    posture: 'Reactive',
    match: {
      'prefect.resource.id': toResourceId('prefect.work-pool', source.workPools),
    },
    forEach: ['prefect.resource.id'],
    expect: mapWorkPoolStatusToEvent(source.status),
  })
}

function mapProactiveWorkPoolStatusTriggerToAutomationTrigger(source: WorkPoolStatusTrigger): AutomationTriggerEvent {
  return new AutomationTriggerEvent({
    posture: 'Proactive',
    match: {
      'prefect.resource.id': toResourceId('prefect.work-pool', source.workPools),
    },
    forEach: ['prefect.resource.id'],
    expect: anyStatusExcept(source.status).flatMap(mapWorkPoolStatusToEvent),
    after: mapWorkPoolStatusToEvent(source.status),
    within: source.time,
  })
}

function mapReactiveAutomationTriggerToWorkPoolStatusTrigger(trigger: AutomationTriggerEvent): WorkPoolStatusTrigger {
  return {
    workPools: fromResourceId('prefect.work-pool', trigger.match['prefect.resource.id']),
    posture: 'Reactive',
    status: statusFromWorkPoolStatusEvents(trigger.expect),
    time: trigger.within,
  }
}

function mapProactiveAutomationTriggerToWorkPoolStatusTrigger(trigger: AutomationTriggerEvent): WorkPoolStatusTrigger {
  return {
    workPools: fromResourceId('prefect.work-pool', trigger.match['prefect.resource.id']),
    posture: 'Proactive',
    status: statusFromWorkPoolStatusEvents(trigger.after),
    time: trigger.within,
  }
}

function anyStatusExcept(status: WorkPoolStatus | null): WorkPoolStatus[] {
  return workPoolStatus.filter(_status => _status !== status)
}

function mapWorkPoolStatusToEvent(status: WorkPoolStatus | null): WorkPoolStatusEvent[] {
  switch (status) {
    case 'ready':
      return ['prefect.work-pool.ready']
    case 'not_ready':
      return [
        'prefect.work-pool.not-ready',
        // compatibility with old event name
        // remove after data migration to adjust existing automation triggers
        // to use new event name
        'prefect.work-pool.not_ready',
      ]
    case 'paused':
      return ['prefect.work-pool.paused']
    case null:
      return []
    default:
      const exhaustiveCheck: never = status
      return [`prefect.work-pool.${exhaustiveCheck}`]
  }
}

const statusEventToStatus: Record<WorkPoolStatusEvent, WorkPoolStatus> = {
  'prefect.work-pool.ready': 'ready',
  'prefect.work-pool.not-ready': 'not_ready',
  'prefect.work-pool.not_ready': 'not_ready',
  'prefect.work-pool.paused': 'paused',
}

function statusFromWorkPoolStatusEvents(events: string[]): WorkPoolStatus {
  for (const event of events) {
    if (isWorkPoolStatusEvent(event)) {
      return statusEventToStatus[event]
    }
  }

  throw new Error(`Unknown work pool status events: ${events}`)
}