feat: update Discover flow to support update resource mechanism

web/packages/teleport/src/Discover/Discover.test.tsx

@@ -25,7 +25,8 @@ import { render, screen } from 'design/utils/testing';
 import { Resource } from 'gen-proto-ts/teleport/userpreferences/v1/onboard_pb';
 
 import TeleportContextProvider from 'teleport/TeleportContextProvider';
-import { Discover } from 'teleport/Discover/Discover';
+import { Discover, DiscoverComponent } from 'teleport/Discover/Discover';
+import { ResourceViewConfig } from 'teleport/Discover/flow';
 import { FeaturesContextProvider } from 'teleport/FeaturesContext';
 import { createTeleportContext, getAcl } from 'teleport/mocks/contexts';
 import { getOSSFeatures } from 'teleport/features';
@@ -47,6 +48,9 @@ import { makeTestUserContext } from 'teleport/User/testHelpers/makeTestUserConte
 import { makeDefaultUserPreferences } from 'teleport/services/userPreferences/userPreferences';
 
 import { ResourceKind } from './Shared';
+import { useDiscover } from './useDiscover';
+
+import type { ResourceSpec } from 'teleport/Discover/SelectResource/types';
 
 beforeEach(() => {
   jest.restoreAllMocks();
@@ -204,3 +208,101 @@ describe('location state', () => {
     ).not.toBeInTheDocument();
   });
 });
+
+type updateProps = {
+  resourceName?: string;
+  resourceSpecForUpdate?: ResourceSpec;
+};
+
+const renderUpdate = (props: updateProps) => {
+  const defaultPref = makeDefaultUserPreferences();
+  defaultPref.onboard.preferredResources = [Resource.WEB_APPLICATIONS];
+
+  mockUserContextProviderWith(
+    makeTestUserContext({ preferences: defaultPref })
+  );
+
+  const userAcl = getAcl();
+  const ctx = createTeleportContext({ customAcl: userAcl });
+
+  const MockComponent1 = () => {
+    const { agentMeta } = useDiscover();
+    return (
+      <>
+        {agentMeta.resourceName === 'saml2' ? agentMeta.resourceName : 'saml1'}
+      </>
+    );
+  };
+
+  const testViews: ResourceViewConfig[] = [
+    {
+      kind: ResourceKind.SamlApplication,
+      views() {
+        return [
+          {
+            title: 'MockComponent1',
+            component: MockComponent1,
+          },
+        ];
+      },
+    },
+  ];
+
+  return render(
+    <MemoryRouter
+      initialEntries={[
+        { pathname: cfg.routes.discover, state: { entity: '' } },
+      ]}
+    >
+      <TeleportContextProvider ctx={ctx}>
+        <DiscoverComponent
+          eViewConfigs={testViews}
+          isUpdateFlow={true}
+          resourceSpecForUpdate={props.resourceSpecForUpdate}
+          agentMetaForUpdate={{ resourceName: props.resourceName }}
+        />
+      </TeleportContextProvider>
+    </MemoryRouter>
+  );
+};
+
+test('update flow: renders single component based on resourceSpecForUpdatvalue', () => {
+  const resourceSpecForUpdate: ResourceSpec = {
+    name: 'Connect My Computer',
+    kind: ResourceKind.ConnectMyComputer,
+    event: null,
+    icon: 'Laptop',
+    keywords: '',
+    hasAccess: true,
+  };
+
+  renderUpdate({ resourceSpecForUpdate: resourceSpecForUpdate });
+
+  expect(screen.queryAllByTestId(ResourceKind.Server).length).toBeFalsy();
+
+  expect(screen.queryAllByTestId(ResourceKind.Database).length).toBeFalsy();
+
+  expect(screen.queryAllByTestId(ResourceKind.Application).length).toBeFalsy();
+
+  expect(screen.queryAllByTestId(ResourceKind.Kubernetes).length).toBeFalsy();
+
+  expect(screen.getByText('Sign In & Connect My Computer')).toBeInTheDocument();
+});
+
+test('update flow: agentMeta is prepopulated based on agentMetaForUpdate', () => {
+  const resourceSpecForUpdate: ResourceSpec = {
+    name: 'MockComponent1',
+    kind: ResourceKind.SamlApplication,
+    event: null,
+    icon: 'Application',
+    keywords: '',
+    hasAccess: true,
+  };
+
+  renderUpdate({
+    resourceName: 'saml2',
+    resourceSpecForUpdate: resourceSpecForUpdate,
+  });
+
+  expect(screen.getByText('saml2')).toBeInTheDocument();
+});

web/packages/teleport/src/Discover/Discover.tsx

@@ -29,9 +29,11 @@ import { findViewAtIndex } from 'teleport/components/Wizard/flow';
 
 import { EViewConfigs } from './types';
 
-import { DiscoverProvider, useDiscover } from './useDiscover';
+import { DiscoverProvider, useDiscover, AgentMeta } from './useDiscover';
 import { DiscoverIcon } from './SelectResource/icons';
 
+import type { ResourceSpec } from './SelectResource';
+
 function DiscoverContent() {
   const {
     currentStep,
@@ -94,10 +96,21 @@ function DiscoverContent() {
   );
 }
 
-export function DiscoverComponent({ eViewConfigs = [] }: Props) {
+export function DiscoverComponent({
+  eViewConfigs = [],
+  isUpdateFlow,
+  resourceSpecForUpdate,
+  agentMetaForUpdate,
+}: DiscoverComponentProps) {
   const location = useLocation();
   return (
-    <DiscoverProvider eViewConfigs={eViewConfigs} key={location.key}>
+    <DiscoverProvider
+      eViewConfigs={eViewConfigs}
+      key={location.key}
+      isUpdateFlow={isUpdateFlow}
+      resourceSpecForUpdate={resourceSpecForUpdate}
+      agentMetaForUpdate={agentMetaForUpdate}
+    >
       <DiscoverContent />
     </DiscoverProvider>
   );
@@ -107,6 +120,9 @@ export function Discover() {
   return <DiscoverComponent />;
 }
 
-type Props = {
+export type DiscoverComponentProps = {
   eViewConfigs?: EViewConfigs;
+  isUpdateFlow?: boolean;
+  resourceSpecForUpdate?: ResourceSpec;
+  agentMetaForUpdate?: AgentMeta;
 };

web/packages/teleport/src/Discover/useDiscover.tsx

@@ -70,6 +70,7 @@ export interface DiscoverContextState<T = any> {
   emitErrorEvent(errorStr: string): void;
   emitEvent(status: DiscoverEventStepStatus, custom?: CustomEventInput): void;
   eventState: EventState;
+  isUpdateFlow?: boolean;
 }
 
 type EventState = {
@@ -93,6 +94,14 @@ type DiscoverProviderProps = {
   mockCtx?: DiscoverContextState;
   // Extra view configs that are passed in. This is used to add view configs from Enterprise.
   eViewConfigs?: EViewConfigs;
+  // isUpdateFlow indicates if the Discover flow is an update resource flow.
+  isUpdateFlow?: boolean;
+  // resourceSpecForUpdate specifies ResourceSpec which should be used to
+  // start a Discover flow.
+  resourceSpecForUpdate?: ResourceSpec;
+  // agentMetaForUpdate includes data that will be used to prepopulate input fields
+  // in the respective Discover compnents.
+  agentMetaForUpdate?: AgentMeta;
 };
 
 // DiscoverUrlLocationState define fields to preserve state between
@@ -117,6 +126,9 @@ export function DiscoverProvider({
   mockCtx,
   children,
   eViewConfigs = [],
+  isUpdateFlow,
+  resourceSpecForUpdate,
+  agentMetaForUpdate,
 }: React.PropsWithChildren<DiscoverProviderProps>) {
   const history = useHistory();
   const location = useLocation<DiscoverUrlLocationState>();
@@ -229,6 +241,14 @@ export function DiscoverProvider({
     });
   }
 
+  // trigger update Discover flow.
+  useEffect(() => {
+    if (isUpdateFlow) {
+      onSelectResource(resourceSpecForUpdate);
+      updateAgentMeta(agentMetaForUpdate);
+    }
+  }, [agentMetaForUpdate]);
+
   // If a location.state.discover was provided, that means the user is
   // coming back from another location to resume the flow.
   // Users will resume at the step that is +1 from the step they left from.
@@ -453,6 +473,7 @@ export function DiscoverProvider({
     emitErrorEvent,
     emitEvent,
     eventState,
+    isUpdateFlow,
   };
 
   return (