Fix withTaskGuard import path

Author: LauraBeatris

packages/clerk-js/src/ui/components/SessionTasks/tasks/TaskChooseOrganization/index.tsx

@@ -8,7 +8,7 @@ import { withCardStateProvider } from '@/ui/elements/contexts';
 import { useMultipleSessions } from '@/ui/hooks/useMultipleSessions';
 import { useOrganizationListInView } from '@/ui/hooks/useOrganizationListInView';
 
-import { withTaskGuard } from '../shared/withTaskGuard';
+import { withTaskGuard } from '../shared';
 import { ChooseOrganizationScreen } from './ChooseOrganizationScreen';
 import { CreateOrganizationScreen } from './CreateOrganizationScreen';
 

packages/clerk-js/src/ui/components/SessionTasks/tasks/TaskResetPassword/index.tsx

@@ -14,7 +14,7 @@ import { handleError } from '@/ui/utils/errorHandler';
 import { createPasswordError } from '@/ui/utils/passwordUtils';
 import { useFormControl } from '@/ui/utils/useFormControl';
 
-import { withTaskGuard } from '../shared/withTaskGuard';
+import { withTaskGuard } from '../shared';
 
 const TaskResetPasswordInternal = () => {
   const clerk = useClerk();

packages/clerk-js/src/ui/components/SessionTasks/tasks/shared/index.ts

@@ -0,0 +1 @@
+export { withTaskGuard } from './withTaskGuard';

packages/clerk-js/src/ui/components/SessionTasks/tasks/shared/withTaskGuard.ts

@@ -0,0 +1,30 @@
+import type { SessionTask } from '@clerk/shared/types';
+import type { ComponentType } from 'react';
+
+import { warnings } from '@/core/warnings';
+import { withRedirect } from '@/ui/common';
+import { useSessionTasksContext } from '@/ui/contexts/components/SessionTasks';
+import type { AvailableComponentProps } from '@/ui/types';
+
+export const withTaskGuard = <P extends AvailableComponentProps>(
+  Component: ComponentType<P>,
+  taskKey: SessionTask['key'],
+) => {
+  const displayName = Component.displayName || Component.name || 'Component';
+  Component.displayName = displayName;
+
+  const HOC = (props: P) => {
+    const ctx = useSessionTasksContext();
+    return withRedirect(
+      Component,
+      clerk => !clerk.session?.currentTask || clerk.session.currentTask.key !== taskKey,
+      ({ clerk }) =>
+        !clerk.session ? clerk.buildSignInUrl() : (ctx.redirectUrlComplete ?? clerk.buildAfterSignInUrl()),
+      warnings.cannotRenderComponentWhenTaskDoesNotExist,
+    )(props);
+  };
+
+  HOC.displayName = `withTaskGuard(${displayName})`;
+
+  return HOC;
+};