diff --git a/datahub-web-react/src/app/search/SearchablePage.tsx b/datahub-web-react/src/app/search/SearchablePage.tsx index 2bbb6f2451..472a8aa7d7 100644 --- a/datahub-web-react/src/app/search/SearchablePage.tsx +++ b/datahub-web-react/src/app/search/SearchablePage.tsx @@ -8,6 +8,7 @@ import { useEntityRegistry } from '../useEntityRegistry'; import { useGetAutoCompleteResultsLazyQuery } from '../../graphql/search.generated'; import { navigateToSearchUrl } from './utils/navigateToSearchUrl'; import { EntityType } from '../../types.generated'; +import { useGetAuthenticatedUser } from '../useGetAuthenticatedUser'; const ALL_ENTITIES_SEARCH_TYPE_NAME = 'All Entities'; @@ -29,6 +30,9 @@ export const SearchablePage = ({ selectedType, initialQuery, children }: Props) const entityRegistry = useEntityRegistry(); const searchTypes = entityRegistry.getSearchEntityTypes(); + + const { data: userData } = useGetAuthenticatedUser(); + const searchTypeNames = [ ALL_ENTITIES_SEARCH_TYPE_NAME, ...searchTypes.map((entityType) => entityRegistry.getCollectionName(entityType)), @@ -78,7 +82,8 @@ export const SearchablePage = ({ selectedType, initialQuery, children }: Props) } onSearch={search} onQueryChange={autoComplete} - authenticatedUserUrn="urn:li:corpuser:0" + authenticatedUserUrn={userData?.corpUser?.urn || ''} + authenticatedUserPictureLink={userData?.corpUser?.editableInfo?.pictureLink || ''} />
{children}
diff --git a/datahub-web-react/src/app/useGetAuthenticatedUser.tsx b/datahub-web-react/src/app/useGetAuthenticatedUser.tsx new file mode 100644 index 0000000000..c8eb86ebc8 --- /dev/null +++ b/datahub-web-react/src/app/useGetAuthenticatedUser.tsx @@ -0,0 +1,8 @@ +import { useGetUserQuery } from '../graphql/user.generated'; + +/** + * Fetch a CorpUser object corresponding to the currently authenticated user. + */ +export function useGetAuthenticatedUser() { + return useGetUserQuery({ variables: { urn: localStorage.getItem('userUrn') as string } }); +}