simstudioai--sim/apps/sim/lib/workflows/operations/socket-operations.ts

import { createLogger } from '@sim/logger'
import { generateId } from '@sim/utils/id'
import { client } from '@/lib/auth/auth-client'
import { useOperationQueueStore } from '@/stores/operation-queue/store'
import type { WorkflowState } from '@/stores/workflows/workflow/types'
import { normalizeWorkflowState } from '@/stores/workflows/workflow/validation'

const logger = createLogger('WorkflowSocketOperations')

async function resolveUserId(): Promise<string> {
  try {
    const sessionResult = await client.getSession()
    const userId = sessionResult.data?.user?.id
    if (userId) {
      return userId
    }
  } catch (error) {
    logger.warn('Failed to resolve session user id for workflow operation', { error })
  }

  return 'unknown'
}

interface EnqueueWorkflowOperationArgs {
  operation: string
  target: string
  payload: any
  workflowId: string
  operationId?: string
}

/**
 * Queues a workflow socket operation so it flows through the standard operation queue,
 * ensuring consistent retries, confirmations, and telemetry.
 */
async function enqueueWorkflowOperation({
  operation,
  target,
  payload,
  workflowId,
  operationId,
}: EnqueueWorkflowOperationArgs): Promise<string> {
  const userId = await resolveUserId()
  const opId = operationId ?? generateId()

  useOperationQueueStore.getState().addToQueue({
    id: opId,
    operation: {
      operation,
      target,
      payload,
    },
    workflowId,
    userId,
  })

  logger.debug('Queued workflow operation', {
    workflowId,
    operation,
    target,
    operationId: opId,
  })

  return opId
}

interface EnqueueReplaceStateArgs {
  workflowId: string
  state: WorkflowState
  operationId?: string
}

/**
 * Convenience wrapper for broadcasting a full workflow state replacement via the queue.
 */
export async function enqueueReplaceWorkflowState({
  workflowId,
  state,
  operationId,
}: EnqueueReplaceStateArgs): Promise<string> {
  const { state: validatedState, warnings } = normalizeWorkflowState(state)

  if (warnings.length > 0) {
    logger.warn('Normalized state before enqueuing replace-state', {
      workflowId,
      warningCount: warnings.length,
      warnings,
    })
  }

  return enqueueWorkflowOperation({
    workflowId,
    operation: 'replace-state',
    target: 'workflow',
    payload: { state: validatedState },
    operationId,
  })
}