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

import type { Schema } from '$lib/common'
import type { Flow, FlowModule } from '$lib/gen'
import { derived, writable } from 'svelte/store'
import { emptyFlowModuleSchema, isEmptyFlowModule, loadFlowModuleSchema } from './flowStateUtils'

export type FlowModuleSchema = {
	flowModule: FlowModule
	schema: Schema
	childFlowModules?: FlowModuleSchema[]
	previewResult?: any
}

export type FlowState = FlowModuleSchema[]

export const flowStateStore = writable<FlowState>(undefined)

export async function initFlowState(flow: Flow) {
	const flowState = await flowModulesToFlowState(flow.value.modules)
	flowStateStore.set(flowState)
}

export const isCopyFirstStepSchemaDisabled = derived(
	flowStateStore,
	(flowState: FlowState | undefined) => {
		if (flowState) {
			const firstModule = flowState[0]
			if (!firstModule) {
				return true
			}
			const fm = firstModule.flowModule
			return flowState.length === 0 || isEmptyFlowModule(fm)
		} else {
			return true
		}
	}
)

export async function flowModulesToFlowState(flowModules: FlowModule[]): Promise<FlowState> {
	return Promise.all(
		flowModules.map(async (flowModule: FlowModule) => {
			const value = flowModule.value
			if (value.type === 'forloopflow') {
				const childFlowModules = await Promise.all(
					value.modules.map(async (module) => loadFlowModuleSchema(module))
				)
				const loopFlowModule = await loadFlowModuleSchema(flowModule)

				return {
					...loopFlowModule,
					childFlowModules
				}
			}

			if (isEmptyFlowModule(flowModule)) {
				return emptyFlowModuleSchema()
			}

			return loadFlowModuleSchema(flowModule)
		})
	)
}

export function flowStateToFlow(flowState: FlowState, flow: Flow): Flow {
	if (!flowState || !flow) {
		return flow
	}

	const modules = flowState.map(({ flowModule, childFlowModules }) => {
		const fmv = flowModule.value

		if (fmv.type === 'forloopflow' && childFlowModules && Array.isArray(childFlowModules)) {
			fmv.modules = childFlowModules.map((cfm) => cfm.flowModule)
			flowModule.value = fmv
		}

		return flowModule
	})

	flow.value.modules = modules
	return flow
}