diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx index a37489700..9c1f21660 100644 --- a/src/frontend/src/CustomNodes/GenericNode/index.tsx +++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx @@ -119,6 +119,7 @@ function GenericNode({ state.componentsToUpdate.find((component) => component.id === data.id), ), ); + const { outdated: isOutdated, breakingChange: hasBreakingChange, @@ -330,7 +331,7 @@ function GenericNode({ openAdvancedModal={false} onCloseAdvancedModal={() => {}} updateNode={() => handleUpdateCode()} - isOutdated={isOutdated && dismissAll} + isOutdated={isOutdated && (dismissAll || isUserEdited)} isUserEdited={isUserEdited} hasBreakingChange={hasBreakingChange} /> diff --git a/src/frontend/src/pages/FlowPage/components/UpdateAllComponents/index.tsx b/src/frontend/src/pages/FlowPage/components/UpdateAllComponents/index.tsx index 29c447ce0..d9fb532d6 100644 --- a/src/frontend/src/pages/FlowPage/components/UpdateAllComponents/index.tsx +++ b/src/frontend/src/pages/FlowPage/components/UpdateAllComponents/index.tsx @@ -48,7 +48,8 @@ export default function UpdateAllComponents({}: {}) { const componentsToUpdateFiltered = useMemo( () => componentsToUpdate.filter( - (component) => !dismissedNodes.includes(component.id), + (component) => + !dismissedNodes.includes(component.id) && !component.userEdited, ), [componentsToUpdate, dismissedNodes], ); diff --git a/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx index 125c3dd2b..866f086f5 100644 --- a/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx +++ b/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx @@ -606,7 +606,9 @@ const NodeToolbarComponent = memo( shortcuts.find((obj) => obj.name === "Update") ?.shortcut! } - style={hasBreakingChange ? "text-warning" : ""} + style={ + hasBreakingChange ? "text-accent-amber-foreground" : "" + } value={isUserEdited ? "Restore" : "Update"} icon={isUserEdited ? "RefreshCcwDot" : "CircleArrowUp"} dataTestId="update-button-modal"