@@ -29,6 +29,7 @@ import {
2929 toStyleModeForTabLanguage ,
3030} from './modules/app-core/workspace-local-helpers.js'
3131import { createWorkspaceEditorHelpers } from './modules/app-core/workspace-editor-helpers.js'
32+ import { createEditedIndicatorVisibilityController } from './modules/app-core/edited-indicator-visibility-controller.js'
3233import { createLayoutDiagnosticsSetup } from './modules/app-core/layout-diagnostics-setup.js'
3334import { createWorkspaceControllersSetup } from './modules/app-core/workspace-controllers-setup.js'
3435import { createGitHubWorkflowsSetup } from './modules/app-core/github-workflows-setup.js'
@@ -436,6 +437,11 @@ const prContextUi = createGitHubPrContextUiController({
436437 closeWorkspacesDrawer : ( ) => workspacesDrawerController ?. setOpen ( false ) ,
437438} )
438439
440+ const editedIndicatorVisibilityController = createEditedIndicatorVisibilityController ( {
441+ getToken : ( ) => githubAiContextState . token ,
442+ getActivePrContext : ( ) => githubAiContextState . activePrContext ,
443+ } )
444+
439445const byotControls = createGitHubByotControls ( {
440446 controlsRoot : githubAiControls ,
441447 tokenInput : githubTokenInput ,
@@ -472,6 +478,7 @@ const byotControls = createGitHubByotControls({
472478 prContextUi . syncAiChatTokenVisibility ( token )
473479 chatDrawerController . setToken ( token )
474480 prDrawerController . setToken ( token )
481+ editedIndicatorVisibilityController . refreshIndicators ( )
475482 } ,
476483 setStatus,
477484} )
@@ -510,6 +517,8 @@ const {
510517 editorPanelsByKind,
511518 editorHeaderLabelByKind,
512519 editorHeaderDirtyStatusByKind,
520+ getShouldShowEditedDesign :
521+ editedIndicatorVisibilityController . getShouldShowEditedDesign ,
513522 defaultTabNameByKind,
514523 toNonEmptyWorkspaceText,
515524 getLoadedStylesTabId : ( ) => loadedStylesTabId ,
@@ -660,6 +669,8 @@ const {
660669 setHasPendingWorkspaceTabsRender : value => ( hasPendingWorkspaceTabsRender = value ) ,
661670 persistActiveTabEditorContent,
662671 getWorkspaceTabDisplay,
672+ getShouldShowEditedDesign :
673+ editedIndicatorVisibilityController . getShouldShowEditedDesign ,
663674 workspaceTabsShell,
664675 workspaceTabAddWrap,
665676 setWorkspaceTabRenameState : value => ( workspaceTabRenameState = value ) ,
@@ -684,6 +695,11 @@ const {
684695 createWorkspaceTabId,
685696} )
686697
698+ editedIndicatorVisibilityController . setRefreshHandlers ( {
699+ syncHeaderLabels,
700+ renderWorkspaceTabs,
701+ } )
702+
687703const githubWorkflows = createGitHubWorkflowsSetup ( {
688704 factories : {
689705 createGitHubPrEditorSyncController,
@@ -758,6 +774,9 @@ const githubWorkflows = createGitHubWorkflowsSetup({
758774 getStyleMode : ( ) => styleMode . value ,
759775 getActivePrContextSyncKey,
760776 prContextUi,
777+ onPrContextStateChange : ( ) => {
778+ editedIndicatorVisibilityController . refreshIndicators ( )
779+ } ,
761780 getTokenForVisibility : ( ) => githubAiContextState . token ,
762781 closeWorkspacesDrawer : ( ) => {
763782 void workspacesDrawerController ?. setOpen ( false )
0 commit comments