windmill/frontend/src/lib/components/otherUserDraftLoad.svelte.ts

import { SvelteMap, SvelteSet } from 'svelte/reactivity'
import { goto } from '$lib/navigation'
import { UserDraft, draftValuesEqual, type UserDraftItemKind } from '$lib/userDraft.svelte'
import { UserDraftDbSyncer } from '$lib/userDraftDbSyncer.svelte'

/**
 * Coordinates "Load another user's draft into the editor as if it were ours".
 *
 * Two cases, decided by the route once it knows whether WE already have a draft
 * at this path (`is_draft` for us):
 *  - No own draft  → the loaded value just becomes our draft (normal autosave).
 *  - Own draft     → "overlay" mode: the loaded value is shown but NEVER saved
 *    (so our own draft on the server is untouched). The first edit prompts
 *    "overwrite your current draft?". Confirm persists the edited value as our
 *    draft; Reset restores our own draft.
 *
 * The save block is a hard per-key lock in the syncer (see `lockSync`) — it
 * covers the reactive mirror AND the navigation / tab-death flush paths, so the
 * foreign value can't leak onto the server through any route.
 */

function keyOf(workspace: string, itemKind: UserDraftItemKind, path: string): string {
	return `${workspace}/${itemKind}/${path}`
}

export type PendingOtherUserDraftLoad = {
	workspace: string
	itemKind: UserDraftItemKind
	path: string
	value: unknown
	ownerLabel: string
}

type ActiveSession = {
	workspace: string
	itemKind: UserDraftItemKind
	path: string
	ownerLabel: string
	/** The loaded value the editor shows. A blocked save whose value still
	 *  equals this is a programmatic load-cascade write, not an edit — so the
	 *  overwrite prompt fires only once the value actually diverges. */
	loadedValue: unknown
	/** Reload our own draft into the editor (AutosaveIndicator's "Reset to draft"). */
	onResetToOwnDraft: () => void | Promise<void>
}

// One-shot handoff: set by the Load action, consumed by the editor's loader.
const pending = new SvelteMap<string, PendingOtherUserDraftLoad>()
// Live overlay sessions, keyed by (workspace, itemKind, path).
const active = new SvelteMap<string, ActiveSession>()
// Keys whose overwrite-confirmation modal is currently open.
const overwriteOpen = new SvelteSet<string>()

export function editRouteFor(itemKind: UserDraftItemKind, path: string): string {
	switch (itemKind) {
		case 'script':
			return `/scripts/edit/${path}`
		case 'flow':
			return `/flows/edit/${path}`
		case 'app':
			return `/apps/edit/${path}`
		case 'raw_app':
			return `/apps_raw/edit/${path}`
		default:
			throw new Error(`Cannot load drafts of kind ${itemKind}`)
	}
}

export const OtherUserDraftLoad = {
	/**
	 * Stage a load. The editor's loader picks it up via `takePending`. When
	 * `navigate`, route to the item's edit page (home-page entry point); the
	 * in-editor entry point reloads in place instead.
	 */
	stage(
		workspace: string,
		itemKind: UserDraftItemKind,
		value: unknown,
		path: string,
		ownerLabel: string,
		opts: { navigate: boolean }
	): void {
		pending.set(keyOf(workspace, itemKind, path), { workspace, itemKind, path, value, ownerLabel })
		if (opts.navigate) goto(editRouteFor(itemKind, path))
	},

	/** Consume the staged load for this key (returns + removes it). */
	takePending(
		workspace: string,
		itemKind: UserDraftItemKind,
		path: string
	): PendingOtherUserDraftLoad | undefined {
		const k = keyOf(workspace, itemKind, path)
		const v = pending.get(k)
		if (v) pending.delete(k)
		return v
	},

	/**
	 * Enter overlay mode: lock all server saves for this key and remember how
	 * to restore our own draft. A blocked save opens the overwrite modal ONLY
	 * once the value diverges from `loadedValue` — so the programmatic load
	 * cascade (which writes back the same value) never trips the prompt, while
	 * the user's very first real edit does, with no timing window.
	 */
	beginOverlay(session: ActiveSession): void {
		const k = keyOf(session.workspace, session.itemKind, session.path)
		active.set(k, session)
		UserDraftDbSyncer.lockSync(
			{ workspace: session.workspace, itemKind: session.itemKind, path: session.path },
			() => {
				const current = UserDraft.get(session.itemKind, session.path, {
					workspace: session.workspace
				})
				if (current !== undefined && !draftValuesEqual(current, session.loadedValue)) {
					this.requestOverwriteModal(session.workspace, session.itemKind, session.path)
				}
			}
		)
	},

	isActive(workspace: string, itemKind: UserDraftItemKind, path: string): boolean {
		return active.has(keyOf(workspace, itemKind, path))
	},

	getSession(
		workspace: string,
		itemKind: UserDraftItemKind,
		path: string
	): ActiveSession | undefined {
		return active.get(keyOf(workspace, itemKind, path))
	},

	requestOverwriteModal(workspace: string, itemKind: UserDraftItemKind, path: string): void {
		const k = keyOf(workspace, itemKind, path)
		if (active.has(k)) overwriteOpen.add(k)
	},

	isOverwriteModalOpen(workspace: string, itemKind: UserDraftItemKind, path: string): boolean {
		return overwriteOpen.has(keyOf(workspace, itemKind, path))
	},

	/** Cancel: keep editing the loaded value, stay paused; re-prompt on the next edit. */
	dismissOverwriteModal(workspace: string, itemKind: UserDraftItemKind, path: string): void {
		overwriteOpen.delete(keyOf(workspace, itemKind, path))
	},

	/** Confirm: adopt the current (edited) value as our own draft and resume saving. */
	confirmOverwrite(workspace: string, itemKind: UserDraftItemKind, path: string): void {
		const current = UserDraft.get(itemKind, path, { workspace })
		this.clear(workspace, itemKind, path)
		if (current !== undefined) {
			void UserDraftDbSyncer.save({ workspace, itemKind, path, value: current, immediate: true })
		}
	},

	/** Discard the loaded view and restore our own draft. */
	async resetToOwnDraft(
		workspace: string,
		itemKind: UserDraftItemKind,
		path: string
	): Promise<void> {
		const session = active.get(keyOf(workspace, itemKind, path))
		this.clear(workspace, itemKind, path)
		await session?.onResetToOwnDraft()
	},

	/** Exit overlay mode: unlock saves, drop the session, close the modal. */
	clear(workspace: string, itemKind: UserDraftItemKind, path: string): void {
		const k = keyOf(workspace, itemKind, path)
		active.delete(k)
		overwriteOpen.delete(k)
		UserDraftDbSyncer.unlockSync({ workspace, itemKind, path })
	}
}