From 14b6aab9024f456203f8dc08ca8e6fa53ba28601 Mon Sep 17 00:00:00 2001
From: wendrul <53628737+wendrul@users.noreply.github.com>
Date: Thu, 26 Feb 2026 17:43:30 +0100
Subject: [PATCH]  add button to edit in workspace fork (#8119)

* Add button to edit in workspace fork

* Remove old fork from script row

* Remove fork button from approw

* fix frontend check
---
 .../src/lib/components/FlowBuilder.svelte     |  9 +++
 .../src/lib/components/ScriptBuilder.svelte   | 12 ++++
 .../apps/editor/AppEditorHeader.svelte        | 14 ++++-
 .../lib/components/common/table/AppRow.svelte | 25 +++++---
 .../components/common/table/FlowRow.svelte    | 25 +++++---
 .../components/common/table/ScriptRow.svelte  | 63 +++++++++++--------
 .../raw_apps/RawAppEditorHeader.svelte        | 14 ++++-
 frontend/src/lib/utils/editInFork.ts          | 22 +++++++
 .../(logged)/flows/get/[...path]/+page.svelte | 14 +++++
 .../(root)/(logged)/run/[...run]/+page.svelte | 11 ++++
 .../scripts/get/[...hash]/+page.svelte        | 14 +++++
 11 files changed, 176 insertions(+), 47 deletions(-)
 create mode 100644 frontend/src/lib/utils/editInFork.ts

diff --git a/frontend/src/lib/components/FlowBuilder.svelte b/frontend/src/lib/components/FlowBuilder.svelte
index 95e659bdf4..5266f4f169 100644
--- a/frontend/src/lib/components/FlowBuilder.svelte
+++ b/frontend/src/lib/components/FlowBuilder.svelte
@@ -96,6 +96,8 @@
 	import type { FlowBuilderProps } from './flow_builder'
 	import { ModulesTestStates } from './modulesTest.svelte'
 	import FlowAssetsHandler, { initFlowGraphAssetsCtx } from './flows/FlowAssetsHandler.svelte'
+	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 
 	let {
 		initialPath = $bindable(''),
@@ -810,6 +812,13 @@
 				onClick: () => window.open(`/flows/add?template=${initialPath}`)
 			})
 		}
+
+		if (!newFlow && !isRuleActive('DisableWorkspaceForking')) {
+			dropdownItems.push({
+				label: 'Edit in workspace fork',
+				onClick: () => window.open(buildForkEditUrl('flow', initialPath))
+			})
+		}
 	}
 
 	let flowCopilotContext: FlowCopilotContext = $state({
diff --git a/frontend/src/lib/components/ScriptBuilder.svelte b/frontend/src/lib/components/ScriptBuilder.svelte
index 9b634fd921..40c4c2b896 100644
--- a/frontend/src/lib/components/ScriptBuilder.svelte
+++ b/frontend/src/lib/components/ScriptBuilder.svelte
@@ -103,6 +103,8 @@
 	import { inputSizeClasses } from './text_input/TextInput.svelte'
 	import type { ButtonType } from './common/button/model'
 	import DebounceLimit from './flows/DebounceLimit.svelte'
+	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 	import OnBehalfOfSelector, { type OnBehalfOfChoice } from './OnBehalfOfSelector.svelte'
 
 	let {
@@ -769,6 +771,16 @@
 								window.open(`/scripts/add?template=${initialPath}`)
 							}
 						},
+						...(!isRuleActive('DisableWorkspaceForking')
+							? [
+									{
+										label: 'Edit in workspace fork',
+										onClick: () => {
+											window.open(buildForkEditUrl('script', initialPath))
+										}
+									}
+								]
+							: []),
 						...(customUi?.topBar?.diff !== false && savedScript && diffDrawer
 							? [
 									{
diff --git a/frontend/src/lib/components/apps/editor/AppEditorHeader.svelte b/frontend/src/lib/components/apps/editor/AppEditorHeader.svelte
index a85602ddfa..f194bc1df7 100644
--- a/frontend/src/lib/components/apps/editor/AppEditorHeader.svelte
+++ b/frontend/src/lib/components/apps/editor/AppEditorHeader.svelte
@@ -63,6 +63,8 @@
 	import AppEditorHeaderDeployInitialDraft from './AppEditorHeaderDeployInitialDraft.svelte'
 	import { computeSecretUrl } from './appDeploy.svelte'
 	import { updatePolicy } from './appPolicy'
+	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 
 	interface Props {
 		policy: Policy
@@ -1118,7 +1120,17 @@
 							onClick: () => {
 								window.open(`/apps/add?template=${appPath}`)
 							}
-						}
+						},
+						...(!isRuleActive('DisableWorkspaceForking')
+							? [
+									{
+										label: 'Edit in workspace fork',
+										onClick: () => {
+											window.open(buildForkEditUrl('app', $appPath))
+										}
+									}
+								]
+							: [])
 					]
 				: undefined}
 		>
diff --git a/frontend/src/lib/components/common/table/AppRow.svelte b/frontend/src/lib/components/common/table/AppRow.svelte
index 529b50ea0b..3998b15af5 100644
--- a/frontend/src/lib/components/common/table/AppRow.svelte
+++ b/frontend/src/lib/components/common/table/AppRow.svelte
@@ -32,6 +32,8 @@
 	import AppDeploymentHistory from '$lib/components/apps/editor/AppDeploymentHistory.svelte'
 	import { isDeployable } from '$lib/utils_deployable'
 	import { getDeployUiSettings } from '$lib/components/home/deploy_ui'
+	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 
 	interface Props {
 		app: ListableApp & { has_draft?: boolean; draft_only?: boolean; canWrite: boolean }
@@ -97,8 +99,8 @@
 	{/snippet}
 	{#snippet actions()}
 		<span class="hidden md:inline-flex gap-x-1">
-			{#if !$userStore?.operator && showEditButton}
-				{#if app.canWrite}
+			{#if !$userStore?.operator}
+				{#if showEditButton && app.canWrite}
 					<div>
 						<Button
 							aiId={`edit-app-button-${app.summary?.length > 0 ? app.summary : app.path}`}
@@ -111,18 +113,17 @@
 							Edit
 						</Button>
 					</div>
-				{:else}
+				{/if}
+				{#if !isRuleActive('DisableWorkspaceForking') && (!showEditButton || !app.canWrite)}
 					<div>
 						<Button
-							aiId={`fork-app-button-${app.summary?.length > 0 ? app.summary : app.path}`}
-							aiDescription={`Fork the app ${app.summary?.length > 0 ? app.summary : app.path}`}
-							variant="subtle"
-							wrapperClasses="w-20"
+							variant={!showEditButton ? 'default' : 'subtle'}
+							wrapperClasses="w-32"
 							unifiedSize="md"
 							startIcon={{ icon: GitFork }}
-							href="{base}/apps{app.raw_app ? '_raw' : ''}/add?template={app.path}"
+							href={buildForkEditUrl(app.raw_app ? 'raw_app' : 'app', app.path)}
 						>
-							Fork
+							Edit in fork
 						</Button>
 					</div>
 				{/if}
@@ -174,6 +175,12 @@
 						disabled: !showEditButton,
 						hide: $userStore?.operator
 					},
+					{
+						displayName: 'Edit in workspace fork',
+						icon: GitFork,
+						href: buildForkEditUrl(app.raw_app ? 'raw_app' : 'app', path),
+						hide: $userStore?.operator || isRuleActive('DisableWorkspaceForking')
+					},
 					{
 						displayName: 'Move/Rename',
 						icon: FolderOpen,
diff --git a/frontend/src/lib/components/common/table/FlowRow.svelte b/frontend/src/lib/components/common/table/FlowRow.svelte
index 9cfa55f948..3cf66b5238 100644
--- a/frontend/src/lib/components/common/table/FlowRow.svelte
+++ b/frontend/src/lib/components/common/table/FlowRow.svelte
@@ -34,6 +34,8 @@
 	} from 'lucide-svelte'
 	import FlowHistory from '$lib/components/flows/FlowHistory.svelte'
 	import { getDeployUiSettings } from '$lib/components/home/deploy_ui'
+	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 
 	interface Props {
 		flow: Flow & { has_draft?: boolean; draft_only?: boolean; canWrite: boolean }
@@ -120,8 +122,8 @@
 	{/snippet}
 	{#snippet actions()}
 		<span class="hidden md:inline-flex gap-x-1">
-			{#if !$userStore?.operator && showEditButton}
-				{#if flow.canWrite && !flow.archived}
+			{#if !$userStore?.operator}
+				{#if showEditButton && flow.canWrite && !flow.archived}
 					<div>
 						<Button
 							variant="subtle"
@@ -135,18 +137,17 @@
 							Edit
 						</Button>
 					</div>
-				{:else}
+				{/if}
+				{#if !isRuleActive('DisableWorkspaceForking') && (!showEditButton || !flow.canWrite)}
 					<div>
 						<Button
-							variant="subtle"
-							wrapperClasses="w-20"
+							variant={!showEditButton ? 'default' : 'subtle'}
+							wrapperClasses="w-32"
 							unifiedSize="md"
 							startIcon={{ icon: GitFork }}
-							href="{base}/flows/add?template={flow.path}"
-							aiId={`fork-flow-button-${flow.summary?.length > 0 ? flow.summary : flow.path}`}
-							aiDescription={`Fork the flow ${flow.summary?.length > 0 ? flow.summary : flow.path}`}
+							href={buildForkEditUrl('flow', flow.path)}
 						>
-							Fork
+							Edit in fork
 						</Button>
 					</div>
 				{/if}
@@ -194,6 +195,12 @@
 						disabled: !showEditButton,
 						hide: $userStore?.operator
 					},
+					{
+						displayName: 'Edit in workspace fork',
+						icon: GitFork,
+						href: buildForkEditUrl('flow', path),
+						hide: $userStore?.operator || isRuleActive('DisableWorkspaceForking')
+					},
 					{
 						displayName: 'Audit logs',
 						icon: Eye,
diff --git a/frontend/src/lib/components/common/table/ScriptRow.svelte b/frontend/src/lib/components/common/table/ScriptRow.svelte
index 42587a5a4d..6ad3918aff 100644
--- a/frontend/src/lib/components/common/table/ScriptRow.svelte
+++ b/frontend/src/lib/components/common/table/ScriptRow.svelte
@@ -43,6 +43,8 @@
 	import Tooltip from '$lib/components/Tooltip.svelte'
 	import { getDeployUiSettings } from '$lib/components/home/deploy_ui'
 	import { scriptToHubUrl } from '$lib/hub'
+	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 
 	interface Props {
 		script: Script & { canWrite: boolean; use_codebase: boolean }
@@ -150,39 +152,40 @@
 
 	{#snippet actions()}
 		<span class="hidden md:inline-flex gap-x-1">
-			{#if !$userStore?.operator && showEditButton}
-				{#if script.use_codebase}
-					<Badge
-						>bundle<Tooltip
-							>This script is deployed as a bundle and can only be deployed from the CLI for now</Tooltip
-						></Badge
-					>
-				{:else if script.canWrite && !script.archived}
-					<div>
-						<Button
-							aiId={`edit-script-button-${script.summary?.length > 0 ? script.summary : script.path}`}
-							aiDescription={`Edits the script ${script.summary?.length > 0 ? script.summary : script.path}`}
-							variant="subtle"
-							wrapperClasses="w-20"
-							unifiedSize="md"
-							startIcon={{ icon: Pen }}
-							href="{base}/scripts/edit/{script.path}"
+			{#if !$userStore?.operator}
+				{#if showEditButton}
+					{#if script.use_codebase}
+						<Badge
+							>bundle<Tooltip
+								>This script is deployed as a bundle and can only be deployed from the CLI for now</Tooltip
+							></Badge
 						>
-							Edit
-						</Button>
-					</div>
-				{:else if !script.draft_only}
+					{:else if script.canWrite && !script.archived}
+						<div>
+							<Button
+								aiId={`edit-script-button-${script.summary?.length > 0 ? script.summary : script.path}`}
+								aiDescription={`Edits the script ${script.summary?.length > 0 ? script.summary : script.path}`}
+								variant="subtle"
+								wrapperClasses="w-20"
+								unifiedSize="md"
+								startIcon={{ icon: Pen }}
+								href="{base}/scripts/edit/{script.path}"
+							>
+								Edit
+							</Button>
+						</div>
+					{/if}
+				{/if}
+				{#if !isRuleActive('DisableWorkspaceForking') && (!showEditButton || !script.canWrite)}
 					<div>
 						<Button
-							aiId={`fork-script-button-${script.summary ?? script.path}`}
-							aiDescription={`Fork the script ${script.summary ?? script.path}`}
-							variant="subtle"
-							wrapperClasses="w-20"
+							variant={!showEditButton ? 'default' : 'subtle'}
+							wrapperClasses="w-32"
 							unifiedSize="md"
 							startIcon={{ icon: GitFork }}
-							href="{base}/scripts/add?template={script.path}"
+							href={buildForkEditUrl('script', script.path)}
 						>
-							Fork
+							Edit in fork
 						</Button>
 					</div>
 				{/if}
@@ -237,6 +240,12 @@
 						disabled: !showEditButton,
 						hide: $userStore?.operator
 					},
+					{
+						displayName: 'Edit in workspace fork',
+						icon: GitFork,
+						href: buildForkEditUrl('script', script.path),
+						hide: $userStore?.operator || isRuleActive('DisableWorkspaceForking')
+					},
 					{
 						displayName: 'Move/Rename',
 						icon: FolderOpen,
diff --git a/frontend/src/lib/components/raw_apps/RawAppEditorHeader.svelte b/frontend/src/lib/components/raw_apps/RawAppEditorHeader.svelte
index e6646292da..7594567440 100644
--- a/frontend/src/lib/components/raw_apps/RawAppEditorHeader.svelte
+++ b/frontend/src/lib/components/raw_apps/RawAppEditorHeader.svelte
@@ -51,6 +51,8 @@
 	import { aiChatManager } from '../copilot/chat/AIChatManager.svelte'
 	import { AIBtnClasses } from '../copilot/chat/AIButtonStyle'
 	import type { RawAppData } from './dataTableRefUtils'
+	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 
 	// async function hash(message) {
 	// 	try {
@@ -931,7 +933,17 @@
 							onClick: () => {
 								window.open(`/apps/add?template=${appPath}`)
 							}
-						}
+						},
+						...(!isRuleActive('DisableWorkspaceForking')
+							? [
+									{
+										label: 'Edit in workspace fork',
+										onClick: () => {
+											window.open(buildForkEditUrl('raw_app', appPath))
+										}
+									}
+								]
+							: [])
 					]
 				: undefined}
 		>
diff --git a/frontend/src/lib/utils/editInFork.ts b/frontend/src/lib/utils/editInFork.ts
new file mode 100644
index 0000000000..2a0768694d
--- /dev/null
+++ b/frontend/src/lib/utils/editInFork.ts
@@ -0,0 +1,22 @@
+import { base } from '$lib/base'
+
+type ItemType = 'script' | 'flow' | 'app' | 'raw_app'
+
+export function buildForkEditUrl(itemType: ItemType, itemPath: string): string {
+	let editPath: string
+	switch (itemType) {
+		case 'script':
+			editPath = `${base}/scripts/edit/${itemPath}`
+			break
+		case 'flow':
+			editPath = `${base}/flows/edit/${itemPath}?nodraft=true`
+			break
+		case 'app':
+			editPath = `${base}/apps/edit/${itemPath}?nodraft=true`
+			break
+		case 'raw_app':
+			editPath = `${base}/apps_raw/edit/${itemPath}?nodraft=true`
+			break
+	}
+	return `${base}/user/fork_workspace?rd=${encodeURIComponent(editPath)}`
+}
diff --git a/frontend/src/routes/(root)/(logged)/flows/get/[...path]/+page.svelte b/frontend/src/routes/(root)/(logged)/flows/get/[...path]/+page.svelte
index 9c095fed58..847f708704 100644
--- a/frontend/src/routes/(root)/(logged)/flows/get/[...path]/+page.svelte
+++ b/frontend/src/routes/(root)/(logged)/flows/get/[...path]/+page.svelte
@@ -67,6 +67,8 @@
 	import { slide } from 'svelte/transition'
 	import { twMerge } from 'tailwind-merge'
 	import NoDirectDeployAlert from '$lib/components/NoDirectDeployAlert.svelte'
+	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 
 	let flow: Flow | undefined = $state()
 	let can_write = $state(false)
@@ -253,6 +255,18 @@
 			})
 		}
 
+		if (flow && !$userStore?.operator && !isRuleActive('DisableWorkspaceForking')) {
+			buttons.push({
+				label: 'Edit in fork',
+				buttonProps: {
+					href: buildForkEditUrl('flow', flow.path),
+					unifiedSize: 'md',
+					variant: !showEditButtons ? 'default' : 'subtle',
+					startIcon: GitFork
+				}
+			})
+		}
+
 		if (!flow) {
 			return buttons
 		}
diff --git a/frontend/src/routes/(root)/(logged)/run/[...run]/+page.svelte b/frontend/src/routes/(root)/(logged)/run/[...run]/+page.svelte
index 22d5474987..0a2754e3dd 100644
--- a/frontend/src/routes/(root)/(logged)/run/[...run]/+page.svelte
+++ b/frontend/src/routes/(root)/(logged)/run/[...run]/+page.svelte
@@ -34,6 +34,7 @@
 		Code2,
 		ClipboardCopy,
 		GitBranch,
+		GitFork,
 		EllipsisVertical
 	} from 'lucide-svelte'
 
@@ -86,6 +87,7 @@
 	import FlowRestartButton from '$lib/components/FlowRestartButton.svelte'
 	import JobOtelTraces from '$lib/components/JobOtelTraces.svelte'
 	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 	let job: (Job & { result?: any; result_stream?: string }) | undefined = $state()
 	let jobUpdateLastFetch: Date | undefined = $state()
 
@@ -654,6 +656,15 @@
 							startIcon={{ icon: Pen }}>Edit</Button
 						>
 					{/if}
+					{#if !showEditButton && !isRuleActive('DisableWorkspaceForking')}
+						<Button
+							href={buildForkEditUrl(isScript ? 'script' : 'flow', job?.script_path ?? '')}
+							unifiedSize="md"
+							variant="default"
+							size="sm"
+							startIcon={{ icon: GitFork }}>Edit in fork</Button
+						>
+					{/if}
 				{/if}
 			{/if}
 			{#if job?.job_kind === 'script' || job?.job_kind === 'script_hub' || job?.job_kind === 'flow'}
diff --git a/frontend/src/routes/(root)/(logged)/scripts/get/[...hash]/+page.svelte b/frontend/src/routes/(root)/(logged)/scripts/get/[...hash]/+page.svelte
index 0344736a67..f322f2bfaa 100644
--- a/frontend/src/routes/(root)/(logged)/scripts/get/[...hash]/+page.svelte
+++ b/frontend/src/routes/(root)/(logged)/scripts/get/[...hash]/+page.svelte
@@ -86,6 +86,8 @@
 	import TriggersEditor from '$lib/components/triggers/TriggersEditor.svelte'
 	import { Triggers } from '$lib/components/triggers/triggers.svelte'
 	import { page } from '$app/state'
+	import { isRuleActive } from '$lib/workspaceProtectionRules.svelte'
+	import { buildForkEditUrl } from '$lib/utils/editInFork'
 
 	let script: Script | undefined = $state()
 	let topHash: string | undefined = $state()
@@ -349,6 +351,18 @@
 			})
 		}
 
+		if (script && !$userStore?.operator && !isRuleActive('DisableWorkspaceForking')) {
+			buttons.push({
+				label: 'Edit in fork',
+				buttonProps: {
+					href: buildForkEditUrl('script', script.path),
+					unifiedSize: 'md',
+					variant: !showEditButtons ? 'default' : 'subtle',
+					startIcon: GitFork
+				}
+			})
+		}
+
 		if (!script) {
 			return buttons
 		}