usememos/memos
1
1
Fork 0
mirror of https://github.com/usememos/memos.git synced 2024-12-19 17:12:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix current user store

Browse Source
This commit is contained in:
Steven 2023-11-25 22:58:17 +08:00
parent a12844f5db
commit 472afce98f
3 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
web/src/pages/AuthCallback.tsx
View File

@ -7,6 +7,7 @@ import * as api from "@/helpers/api";
import { absolutifyLink } from "@/helpers/utils";
import useNavigateTo from "@/hooks/useNavigateTo";
import { useUserStore } from "@/store/module";
import { useUserV1Store } from "@/store/v1";
import { useTranslate } from "@/utils/i18n";
interface State {
@ -19,6 +20,7 @@ const AuthCallback = () => {
const navigateTo = useNavigateTo();
const [searchParams] = useSearchParams();
const userStore = useUserStore();
const userV1Store = useUserV1Store();
const [state, setState] = useState<State>({
loading: true,
errorMessage: "",
@ -42,6 +44,7 @@ const AuthCallback = () => {
if (user) {
userStore.setCurrentUser(user);
await userStore.fetchCurrentUser();
await userV1Store.getOrFetchUserByUsername(user.username);
navigateTo("/");
} else {
toast.error(t("message.login-failed"));

3
web/src/pages/SignIn.tsx
View File

@ -9,6 +9,7 @@ import { absolutifyLink } from "@/helpers/utils";
import useLoading from "@/hooks/useLoading";
import useNavigateTo from "@/hooks/useNavigateTo";
import { useGlobalStore, useUserStore } from "@/store/module";
import { useUserV1Store } from "@/store/v1";
import { useTranslate } from "@/utils/i18n";
const SignIn = () => {
@ -16,6 +17,7 @@ const SignIn = () => {
const navigateTo = useNavigateTo();
const globalStore = useGlobalStore();
const userStore = useUserStore();
const userV1Store = useUserV1Store();
const actionBtnLoadingState = useLoading(false);
const { appearance, locale, systemStatus } = globalStore.state;
const mode = systemStatus.profile.mode;
@ -78,6 +80,7 @@ const SignIn = () => {
if (user) {
userStore.setCurrentUser(user);
await userStore.fetchCurrentUser();
await userV1Store.getOrFetchUserByUsername(user.username);
navigateTo("/");
} else {
toast.error(t("message.login-failed"));

3
web/src/pages/SignUp.tsx
View File

@ -8,6 +8,7 @@ import * as api from "@/helpers/api";
import useLoading from "@/hooks/useLoading";
import useNavigateTo from "@/hooks/useNavigateTo";
import { useGlobalStore, useUserStore } from "@/store/module";
import { useUserV1Store } from "@/store/v1";
import { useTranslate } from "@/utils/i18n";
const SignUp = () => {
@ -15,6 +16,7 @@ const SignUp = () => {
const navigateTo = useNavigateTo();
const globalStore = useGlobalStore();
const userStore = useUserStore();
const userV1Store = useUserV1Store();
const actionBtnLoadingState = useLoading(false);
const { appearance, locale, systemStatus } = globalStore.state;
const [username, setUsername] = useState("");
@ -58,6 +60,7 @@ const SignUp = () => {
if (user) {
userStore.setCurrentUser(user);
await userStore.fetchCurrentUser();
await userV1Store.getOrFetchUserByUsername(user.username);
navigateTo("/");
} else {
toast.error(t("message.signup-failed"));