@skyhook-io/k8s-ui 1.13.3 → 1.14.0

package/src/utils/workload-rollout.ts

@@ -0,0 +1,292 @@
+import type { WorkloadPodInfo } from '../types/core'
+
+export type WorkloadRolloutPhase =
+  | 'idle'
+  | 'applying'
+  | 'progressing'
+  | 'waiting'
+  | 'paused'
+  | 'partition-reached'
+  | 'stalled'
+
+export interface WorkloadRolloutActivity {
+  phase: WorkloadRolloutPhase
+  active: boolean
+  manual: boolean
+  label: string
+  detail?: string
+  desired: number
+  updated: number
+  ready: number
+  available: number
+}
+
+export function isArgoRolloutResource(resource: any): boolean {
+  return String(resource?.apiVersion || '').startsWith('argoproj.io/')
+}
+
+export function rolloutMayAdvanceAutomatically(activity: WorkloadRolloutActivity): boolean {
+  return activity.phase === 'applying' ||
+    activity.phase === 'progressing' ||
+    (activity.phase === 'waiting' && !activity.manual)
+}
+
+export function isRolloutActivityVisible(activity: WorkloadRolloutActivity): boolean {
+  return activity.active || activity.phase === 'stalled'
+}
+
+export function getObservedGeneration(resource: any): number {
+  const observed = resource?.status?.observedGeneration
+  if (typeof observed === 'number' && Number.isFinite(observed)) return observed
+  return typeof observed === 'string' && /^\d+$/.test(observed)
+    ? Number.parseInt(observed, 10)
+    : 0
+}
+
+export function getWorkloadRolloutActivity(
+  resource: any,
+  kind?: string,
+  workloadPods?: WorkloadPodInfo[],
+): WorkloadRolloutActivity {
+  const normalized = (kind || resource?.kind || '').toLowerCase().replace(/s$/, '')
+  if (normalized === 'rollout' && !isArgoRolloutResource(resource)) {
+    return activity('idle', false, false, 'Stable', '', 0, 0, 0, 0)
+  }
+  let result: WorkloadRolloutActivity
+  switch (normalized) {
+    case 'deployment':
+      result = deploymentActivity(resource)
+      break
+    case 'statefulset':
+      result = statefulSetActivity(resource)
+      break
+    case 'daemonset':
+      result = daemonSetActivity(resource)
+      break
+    case 'rollout':
+      result = argoRolloutActivity(resource)
+      break
+    default:
+      result = activity('idle', false, false, 'Stable', '', 0, 0, 0, 0)
+  }
+  return withUpdatedRevisionFailure(result, workloadPods)
+}
+
+function deploymentActivity(resource: any): WorkloadRolloutActivity {
+  const spec = resource?.spec || {}
+  const status = resource?.status || {}
+  const desired = spec.replicas ?? 1
+  const updated = status.updatedReplicas ?? 0
+  const ready = status.readyReplicas ?? 0
+  const available = status.availableReplicas ?? 0
+  const base = activity('idle', false, false, 'Stable', '', desired, updated, ready, available)
+  const progress = (status.conditions || []).find((condition: any) => condition.type === 'Progressing')
+
+  if ((status.observedGeneration ?? 0) < (resource?.metadata?.generation ?? 0)) {
+    return merge(base, 'applying', true, 'Applying change', 'Waiting for the Deployment controller to observe generation')
+  }
+  if (progress?.status === 'False' || progress?.reason === 'ProgressDeadlineExceeded') {
+    return merge(base, 'stalled', false, 'Rollout stalled', progress.message || 'Progress deadline exceeded')
+  }
+  const old = Math.max(0, (status.replicas ?? 0) - updated)
+  const pending = updated < desired || old > 0
+  if (spec.paused) {
+    return merge(base, 'paused', pending, 'Rollout paused', pending ? `Paused at ${updated}/${desired} updated` : 'No rollout is pending')
+  }
+  if (desired === 0) return merge(base, 'idle', false, 'Scaled to zero', 'No replicas requested')
+  if (!pending) return merge(base, 'idle', false, 'Stable', `${available}/${desired} available`)
+  if (spec.strategy?.type === 'Recreate') {
+    if (updated === 0 && old > 0) {
+      return merge(base, 'waiting', true, 'Replacing replicas', `Waiting for old replicas to stop · ${available} available`)
+    }
+    return updated === 0
+      ? merge(base, 'waiting', true, 'Waiting for new revision', replicaDetail(updated, desired, available))
+      : merge(base, 'progressing', true, 'Replacing replicas', replicaDetail(updated, desired, available))
+  }
+  if (updated === 0) return merge(base, 'waiting', true, 'Waiting for new revision', replicaDetail(updated, desired, available))
+  return old === 0 && updated === (status.replicas ?? 0)
+    ? merge(base, 'progressing', true, 'Scaling', replicaDetail(updated, desired, available))
+    : merge(base, 'progressing', true, 'Rolling out', replicaDetail(updated, desired, available))
+}
+
+function statefulSetActivity(resource: any): WorkloadRolloutActivity {
+  const spec = resource?.spec || {}
+  const status = resource?.status || {}
+  const desired = spec.replicas ?? 1
+  const updated = status.updatedReplicas ?? 0
+  const ready = status.readyReplicas ?? 0
+  const base = activity('idle', false, false, 'Stable', '', desired, updated, ready, ready)
+
+  if ((status.observedGeneration ?? 0) < (resource?.metadata?.generation ?? 0)) {
+    return merge(base, 'applying', true, 'Applying change', 'Waiting for the StatefulSet controller to observe generation')
+  }
+  if (desired === 0) return merge(base, 'idle', false, 'Scaled to zero', 'No replicas requested')
+  if (spec.updateStrategy?.type === 'OnDelete') {
+    return updated === desired && ready === desired
+      ? merge(base, 'idle', false, 'Stable', `${ready}/${desired} ready`)
+      : mergeManual(base, 'waiting', true, 'Waiting for Pod restart', `OnDelete strategy · ${updated}/${desired} updated`)
+  }
+  const partition = spec.updateStrategy?.rollingUpdate?.partition ?? 0
+  const target = Math.max(0, desired - partition)
+  if (updated >= target) {
+    const pendingRevision = Boolean(status.currentRevision && status.updateRevision && status.currentRevision !== status.updateRevision)
+    return partition > 0 && pendingRevision
+      ? merge(base, 'partition-reached', false, 'Partition reached', `${partition}/${desired} Pods intentionally retained`)
+      : merge(base, 'idle', false, 'Stable', `${ready}/${desired} ready`)
+  }
+  const detail = `${updated}/${target} target Pods updated · ${ready}/${desired} ready`
+  return updated === 0
+    ? merge(base, 'waiting', true, 'Waiting for new revision', detail)
+    : merge(base, 'progressing', true, 'Rolling out', detail)
+}
+
+function daemonSetActivity(resource: any): WorkloadRolloutActivity {
+  const spec = resource?.spec || {}
+  const status = resource?.status || {}
+  const desired = status.desiredNumberScheduled ?? 0
+  const updated = status.updatedNumberScheduled ?? 0
+  const ready = status.numberReady ?? 0
+  const available = status.numberAvailable ?? 0
+  const base = activity('idle', false, false, 'Stable', '', desired, updated, ready, available)
+
+  if ((status.observedGeneration ?? 0) < (resource?.metadata?.generation ?? 0)) {
+    return merge(base, 'applying', true, 'Applying change', 'Waiting for the DaemonSet controller to observe generation')
+  }
+  if (desired === 0) return merge(base, 'idle', false, 'No targets', 'Selector matches no nodes')
+  if (spec.updateStrategy?.type === 'OnDelete') {
+    return updated === desired && available === desired
+      ? merge(base, 'idle', false, 'Stable', `${available}/${desired} available`)
+      : mergeManual(base, 'waiting', true, 'Waiting for Pod restart', `OnDelete strategy · ${updated}/${desired} updated`)
+  }
+  if (updated === desired) return merge(base, 'idle', false, 'Stable', `${available}/${desired} available`)
+  return updated === 0
+    ? merge(base, 'waiting', true, 'Waiting for new revision', replicaDetail(updated, desired, available))
+    : merge(base, 'progressing', true, 'Rolling out', replicaDetail(updated, desired, available))
+}
+
+function argoRolloutActivity(resource: any): WorkloadRolloutActivity {
+  const spec = resource?.spec || {}
+  const status = resource?.status || {}
+  const desired = spec.replicas ?? ((status.replicas ?? 0) || 1)
+  const updated = status.updatedReplicas ?? 0
+  const ready = status.readyReplicas ?? 0
+  const available = status.availableReplicas ?? 0
+  const base = activity('idle', false, false, 'Stable', '', desired, updated, ready, available)
+  const detail = argoDetail(resource, updated, desired, available)
+  const observedGeneration = getObservedGeneration(resource)
+  if (observedGeneration > 0 && observedGeneration < (resource?.metadata?.generation ?? 0)) {
+    return merge(base, 'applying', true, 'Applying change', 'Waiting for the Rollout controller to observe generation')
+  }
+  const failedCondition = (status.conditions || []).find((condition: any) =>
+    (condition.type === 'InvalidSpec' && condition.status === 'True') ||
+    (condition.type === 'Progressing' && (condition.status === 'False' || condition.reason === 'ProgressDeadlineExceeded')),
+  )
+  if (failedCondition || status.abort) {
+    return merge(base, 'stalled', false, 'Rollout stalled', failedCondition?.message || failedCondition?.reason || status.message || (status.abort ? 'The Rollout was aborted' : '') || 'The Rollout controller reported a failed revision')
+  }
+  switch (String(status.phase || '').toLowerCase()) {
+    case 'degraded':
+    case 'error':
+    case 'failed':
+    case 'aborted':
+      return merge(base, 'stalled', false, 'Rollout stalled', status.message || 'The Rollout controller reported a failed revision')
+    case 'paused':
+      return merge(base, 'paused', true, 'Rollout paused', detail)
+    case 'progressing':
+      return updated === 0
+        ? merge(base, 'waiting', true, 'Waiting for new revision', detail)
+        : merge(base, 'progressing', true, 'Rolling out', detail)
+    case 'healthy':
+      return merge(base, 'idle', false, 'Stable', `${available}/${desired} available`)
+    default:
+      return desired > 0 && (updated < desired || available < desired)
+        ? merge(base, 'applying', true, 'Applying change', detail)
+        : merge(base, 'idle', false, 'Stable', status.message || '')
+  }
+}
+
+function withUpdatedRevisionFailure(
+  rollout: WorkloadRolloutActivity,
+  pods?: WorkloadPodInfo[],
+): WorkloadRolloutActivity {
+  if ((!rollout.active && rollout.phase !== 'stalled') || rollout.phase === 'applying' || !pods) return rollout
+  const failed = pods
+    .filter((pod) => pod.updatedRevision === true && pod.healthLevel === 'unhealthy')
+    .sort((a, b) => a.name.localeCompare(b.name))[0]
+  if (!failed) return rollout
+  const reason = failed.reason || failed.lastTerminationReason || failed.phase || 'Pod failed'
+  return merge(rollout, 'stalled', false, 'New revision cannot start', `${reason} · Pod ${failed.name}`)
+}
+
+export function getArgoRolloutStepNumber(resource: any): number | null {
+  const steps = resource?.spec?.strategy?.canary?.steps || []
+  const currentIndex = resource?.status?.currentStepIndex
+  if (steps.length === 0 || typeof currentIndex !== 'number') return null
+  return Math.min(Math.max(currentIndex + 1, 1), steps.length)
+}
+
+function argoDetail(resource: any, updated: number, desired: number, available: number): string {
+  const stepNumber = getArgoRolloutStepNumber(resource)
+  const step = stepNumber === null ? '' : `Step ${stepNumber} · `
+  return `${step}${updated}/${desired} updated · ${available} available`
+}
+
+function replicaDetail(updated: number, desired: number, available: number): string {
+  return `${updated}/${desired} updated · ${available} available`
+}
+
+function activity(
+  phase: WorkloadRolloutPhase,
+  active: boolean,
+  manual: boolean,
+  label: string,
+  detail: string,
+  desired: number,
+  updated: number,
+  ready: number,
+  available: number,
+): WorkloadRolloutActivity {
+  return { phase, active, manual, label, detail, desired, updated, ready, available }
+}
+
+function merge(
+  base: WorkloadRolloutActivity,
+  phase: WorkloadRolloutPhase,
+  active: boolean,
+  label: string,
+  detail: string,
+): WorkloadRolloutActivity {
+  return { ...base, phase, active, manual: false, label, detail }
+}
+
+function mergeManual(
+  base: WorkloadRolloutActivity,
+  phase: WorkloadRolloutPhase,
+  active: boolean,
+  label: string,
+  detail: string,
+): WorkloadRolloutActivity {
+  return { ...base, phase, active, manual: true, label, detail }
+}
+
+export function rolloutActivityLevel(activity: WorkloadRolloutActivity): 'healthy' | 'neutral' | 'degraded' | 'unhealthy' {
+  switch (activity.phase) {
+    case 'stalled':
+      return 'unhealthy'
+    case 'paused':
+      return 'degraded'
+    case 'waiting':
+      return activity.manual ? 'degraded' : 'neutral'
+    case 'applying':
+    case 'progressing':
+    case 'partition-reached':
+      return 'neutral'
+    default:
+      return activity.label === 'Stable' ? 'healthy' : 'neutral'
+  }
+}
+
+export function rolloutActivityBadge(activity: WorkloadRolloutActivity): { text: string; color: string; level: 'healthy' | 'neutral' | 'degraded' | 'unhealthy' } {
+  const level = rolloutActivityLevel(activity)
+  return { text: activity.label, color: `status-${level}`, level }
+}