forked from github/plane
116 lines
3.2 KiB
TypeScript
116 lines
3.2 KiB
TypeScript
import { useEffect, useState } from "react";
|
|
import { useRouter } from "next/router";
|
|
// services
|
|
import { WorkspaceService } from "@/services/workspace.service";
|
|
// types
|
|
import { IUser } from "@plane/types";
|
|
|
|
const workspaceService = new WorkspaceService();
|
|
type Props = {
|
|
routeAuth?: "sign-in" | "onboarding" | "admin" | null;
|
|
user: IUser | null;
|
|
isLoading: boolean;
|
|
};
|
|
const useUserAuth = (props: Props) => {
|
|
const { routeAuth, user, isLoading } = props;
|
|
// states
|
|
const [isRouteAccess, setIsRouteAccess] = useState(true);
|
|
// router
|
|
const router = useRouter();
|
|
const { next_path } = router.query;
|
|
|
|
const isValidURL = (url: string): boolean => {
|
|
const disallowedSchemes = /^(https?|ftp):\/\//i;
|
|
return !disallowedSchemes.test(url);
|
|
};
|
|
|
|
useEffect(() => {
|
|
const handleWorkSpaceRedirection = async () => {
|
|
workspaceService.userWorkspaces().then(async (userWorkspaces) => {
|
|
if (!user) return;
|
|
|
|
const firstWorkspace = Object.values(userWorkspaces ?? {})?.[0];
|
|
const lastActiveWorkspace = userWorkspaces.find((workspace) => workspace.id === user?.last_workspace_id);
|
|
|
|
if (lastActiveWorkspace) {
|
|
router.push(`/${lastActiveWorkspace.slug}`);
|
|
return;
|
|
} else if (firstWorkspace) {
|
|
router.push(`/${firstWorkspace.slug}`);
|
|
return;
|
|
} else {
|
|
router.push(`/profile`);
|
|
return;
|
|
}
|
|
});
|
|
};
|
|
|
|
const handleUserRouteAuthentication = async () => {
|
|
if (user && user.is_active) {
|
|
if (routeAuth === "sign-in") {
|
|
if (user.is_onboarded) handleWorkSpaceRedirection();
|
|
else {
|
|
router.push("/onboarding");
|
|
return;
|
|
}
|
|
} else if (routeAuth === "onboarding") {
|
|
if (user.is_onboarded) handleWorkSpaceRedirection();
|
|
else {
|
|
setIsRouteAccess(() => false);
|
|
return;
|
|
}
|
|
} else {
|
|
if (!user.is_onboarded) {
|
|
router.push("/onboarding");
|
|
return;
|
|
} else {
|
|
setIsRouteAccess(() => false);
|
|
return;
|
|
}
|
|
}
|
|
} else {
|
|
// user is not active and we can redirect to no access page
|
|
// router.push("/no-access");
|
|
// remove token
|
|
return;
|
|
}
|
|
};
|
|
|
|
if (routeAuth === null) {
|
|
setIsRouteAccess(() => false);
|
|
return;
|
|
} else {
|
|
if (!isLoading) {
|
|
setIsRouteAccess(() => true);
|
|
if (user) {
|
|
if (next_path) {
|
|
if (isValidURL(next_path.toString())) {
|
|
router.push(next_path.toString());
|
|
return;
|
|
} else {
|
|
router.push("/");
|
|
return;
|
|
}
|
|
} else handleUserRouteAuthentication();
|
|
} else {
|
|
if (routeAuth === "sign-in") {
|
|
setIsRouteAccess(() => false);
|
|
return;
|
|
} else {
|
|
router.push("/");
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}, [user, isLoading, routeAuth, router, next_path]);
|
|
|
|
return {
|
|
isLoading: isRouteAccess,
|
|
// assignedIssuesLength: user?.assigned_issues ?? 0,
|
|
// workspaceInvitesLength: user?.workspace_invites ?? 0,
|
|
};
|
|
};
|
|
|
|
export default useUserAuth;
|