Function handleEdgesOperationTx

apps/realtime/src/database/operations.ts:1408–1625

(
  tx: any,
  workflowId: string,
  operation: string,
  payload: any
)

}

async function handleEdgesOperationTx(
  tx: any,
  workflowId: string,
  operation: string,
  payload: any
) {
  switch (operation) {
    case EDGES_OPERATIONS.BATCH_REMOVE_EDGES: {
      const { ids } = payload
      if (!Array.isArray(ids) || ids.length === 0) {
        logger.debug('No edge IDs provided for batch remove')
        return
      }

      logger.info(`Batch removing ${ids.length} edges from workflow ${workflowId}`)

      // Get edges to check connected blocks
      const edgesToRemove = await tx
        .select({
          id: workflowEdges.id,
          sourceBlockId: workflowEdges.sourceBlockId,
          targetBlockId: workflowEdges.targetBlockId,
        })
        .from(workflowEdges)
        .where(and(eq(workflowEdges.workflowId, workflowId), inArray(workflowEdges.id, ids)))

      if (edgesToRemove.length === 0) {
        logger.debug('No edges found to remove')
        return
      }

      type EdgeToRemove = (typeof edgesToRemove)[number]

      // Get all connected block IDs
      const connectedBlockIds = new Set<string>()
      edgesToRemove.forEach((e: EdgeToRemove) => {
        connectedBlockIds.add(e.sourceBlockId)
        connectedBlockIds.add(e.targetBlockId)
      })

      // Fetch blocks to check lock status
      const connectedBlocks = await tx
        .select({
          id: workflowBlocks.id,
          locked: workflowBlocks.locked,
          data: workflowBlocks.data,
        })
        .from(workflowBlocks)
        .where(
          and(
            eq(workflowBlocks.workflowId, workflowId),
            inArray(workflowBlocks.id, Array.from(connectedBlockIds))
          )
        )

      type EdgeBlockRecord = (typeof connectedBlocks)[number]
      const blocksById: Record<string, EdgeBlockRecord> = Object.fromEntries(
        connectedBlocks.map((b: EdgeBlockRecord) => [b.id, b])