windmill/frontend/src/lib/components/flows/flowDeleteController.ts

import type { GroupDisplayState } from '$lib/components/graph/groupEditor.svelte'
import { refreshStateStore } from '$lib/svelte5Utils.svelte'
import type { GroupedModulesProxy } from '$lib/components/graph/groupedModulesProxy.svelte'
import type { SelectionManager } from '$lib/components/graph/selectionUtils.svelte'
import type { FlowStructureNode } from '$lib/components/graph/flowStructure'
import type { OpenFlow } from '$lib/gen'
import { push, type History } from '$lib/history.svelte'
import type { StateStore } from '$lib/utils'
import { createDeletePlan, removeDeletePlanTools, type DeletePlan } from './flowDeleteUtils'
import type { FlowState } from './flowState'
import { deleteFlowStateById } from './flowStateUtils.svelte'

export type PreparedDeleteRequest = {
	plan: DeletePlan
	needsDependencyConfirmation: boolean
}

export type DeletePlanExecutionContext = {
	history?: History<OpenFlow>
	flowStore: StateStore<OpenFlow>
	flowStateStore: StateStore<FlowState>
	selectionManager: Pick<SelectionManager, 'clearSelection' | 'selectId'>
	onDelete?: (id: string) => void
}

export type DeletePlanExecutionResult = {
	removedStateIds: string[]
	removedToolStateIds: string[]
}

export function prepareDeleteRequest(args: {
	ids: string[]
	flow: OpenFlow
	tree: FlowStructureNode[]
	proxy: GroupedModulesProxy
	displayState: GroupDisplayState
}): PreparedDeleteRequest | undefined {
	const plan = createDeletePlan(args)
	if (!plan) {
		return undefined
	}

	return {
		plan,
		needsDependencyConfirmation: Object.keys(plan.dependents).length > 0
	}
}

export function executeDeletePlan(
	plan: DeletePlan,
	args: DeletePlanExecutionContext
): DeletePlanExecutionResult {
	push(args.history, args.flowStore.val)
	const removedStructureStateIds = resolvePlannedStructureStateIds(plan.targets)

	if (plan.selection.kind === 'clear') {
		args.selectionManager.clearSelection()
	} else {
		// Whatever remains selected after a delete was not asked for, so it must not
		// pop the modal panel open.
		args.selectionManager.selectId(plan.selection.id, { openPanel: false })
	}

	if (plan.targets.some((target) => target.kind === 'preprocessor')) {
		args.flowStore.val.value.preprocessor_module = undefined
	}

	plan.structureDelete?.commit({ removeDuplicates: plan.removeDuplicates })

	const removedToolStateIds = removeDeletePlanTools(plan.targets, args.flowStore.val.value.modules)
	const removedStateIds = uniqueIds([...removedStructureStateIds, ...removedToolStateIds])

	for (const id of removedStateIds) {
		deleteFlowStateById(id, args.flowStateStore)
	}

	refreshStateStore(args.flowStore)

	if (plan.inputIds.length === 1) {
		args.onDelete?.(plan.targets[0].id)
	}

	return {
		removedStateIds,
		removedToolStateIds
	}
}

function resolvePlannedStructureStateIds(targets: DeletePlan['targets']): string[] {
	const removedIds: string[] = []

	for (const target of targets) {
		if (target.kind !== 'preprocessor' && target.kind !== 'structure_node') {
			continue
		}
		removedIds.push(...target.stateIds)
	}

	return uniqueIds(removedIds)
}

function uniqueIds(ids: string[]): string[] {
	return [...new Set(ids)]
}