54 lines
1.6 KiB
TypeScript
54 lines
1.6 KiB
TypeScript
import { useEffect } from "react";
|
|
import { usePathname, useRouter } from "next/navigation";
|
|
import { useCurrentUser, useToken } from "./auth";
|
|
import { useSetAtom } from "jotai";
|
|
import { isCreateAiLimitReachedDialogOpenAtom } from "@/atoms/global";
|
|
import { VipType } from "@/services/wallet";
|
|
import { isVipDrawerOpenAtom } from "@/atoms/im";
|
|
|
|
const CREATE_CHARACTER_ROUTE = "/create/type";
|
|
|
|
const useCreatorNavigation = () => {
|
|
const router = useRouter();
|
|
const pathname = usePathname();
|
|
const { data: currentUser } = useCurrentUser();
|
|
const { isLogin } = useToken();
|
|
const { isMember, canCreateAiCount, createdAiCount } = currentUser || {}
|
|
const setIsCreateAiLimitReachedDialogOpen = useSetAtom(isCreateAiLimitReachedDialogOpenAtom)
|
|
const setIsVipDrawerOpen = useSetAtom(isVipDrawerOpenAtom)
|
|
|
|
const loginRedirectHref = `/login?redirect=${encodeURIComponent(CREATE_CHARACTER_ROUTE)}`;
|
|
|
|
useEffect(() => {
|
|
router.prefetch(CREATE_CHARACTER_ROUTE);
|
|
router.prefetch(loginRedirectHref);
|
|
}, [router, loginRedirectHref]);
|
|
|
|
const routerToCreate = () => {
|
|
// 检查是否登录，如果未登录则跳转到登录页面
|
|
if (!isLogin) {
|
|
router.push(loginRedirectHref);
|
|
return;
|
|
}
|
|
|
|
if ((createdAiCount ?? 0) >= (canCreateAiCount ?? 0)) {
|
|
if (isMember) {
|
|
setIsCreateAiLimitReachedDialogOpen(true)
|
|
return;
|
|
} else {
|
|
setIsVipDrawerOpen({ open: true, vipType: VipType.ADD_CREATE_AI })
|
|
return;
|
|
}
|
|
}
|
|
|
|
localStorage.setItem('before_creator_navigation', pathname);
|
|
router.push(CREATE_CHARACTER_ROUTE);
|
|
}
|
|
|
|
return {
|
|
routerToCreate,
|
|
};
|
|
}
|
|
|
|
export default useCreatorNavigation;
|