usememos/memos
1
1
Fork 0
mirror of https://github.com/usememos/memos.git synced 2024-11-23 22:07:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

chore: migrate get current user

Browse Source
This commit is contained in:
Steven 2023-11-23 08:50:33 +08:00
parent c3db4ee236
commit 45d7d0d5f6
7 changed files with 53 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/v1/jwt.go
View File

@ -62,7 +62,7 @@ func JWTMiddleware(server *APIV1Service, next echo.HandlerFunc, secret string) e
}
// Skip validation for server status endpoints.
if util.HasPrefixes(path, "/api/v1/ping", "/api/v1/idp", "/api/v1/status", "/api/v1/user") && path != "/api/v1/user/me" && path != "/api/v1/user" && method == http.MethodGet {
if util.HasPrefixes(path, "/api/v1/ping", "/api/v1/idp", "/api/v1/status") && method == http.MethodGet {
return next(c)
}
@ -73,7 +73,7 @@ func JWTMiddleware(server *APIV1Service, next echo.HandlerFunc, secret string) e
return next(c)
}
// When the request is not authenticated, we allow the user to access the memo endpoints for those public memos.
if util.HasPrefixes(path, "/api/v1/memo") && method == http.MethodGet {
if util.HasPrefixes(path, "/api/v1/memo", "/api/v1/user") && path != "/api/v1/user" && method == http.MethodGet {
return next(c)
}
return echo.NewHTTPError(http.StatusUnauthorized, "Missing access token")

8
api/v1/user.go
View File

@ -281,8 +281,12 @@ func (s *APIV1Service) GetUserByID(c echo.Context) error {
}
userMessage := convertUserFromStore(user)
// data desensitize
userMessage.Email = ""
userID, ok := c.Get(userIDContextKey).(int32)
if !ok || userID != user.ID {
// Data desensitize.
userMessage.Email = ""
}
return c.JSON(http.StatusOK, userMessage)
}

14
web/src/helpers/api.ts
View File

@ -18,7 +18,7 @@ export function vacuumDatabase() {
}
export function signin(username: string, password: string, remember: boolean) {
return axios.post("/api/v1/auth/signin", {
return axios.post<User>("/api/v1/auth/signin", {
username,
password,
remember,
@ -26,7 +26,7 @@ export function signin(username: string, password: string, remember: boolean) {
}
export function signinWithSSO(identityProviderId: IdentityProviderId, code: string, redirectUri: string) {
return axios.post("/api/v1/auth/signin/sso", {
return axios.post<User>("/api/v1/auth/signin/sso", {
identityProviderId,
code,
redirectUri,
@ -34,7 +34,7 @@ export function signinWithSSO(identityProviderId: IdentityProviderId, code: stri
}
export function signup(username: string, password: string) {
return axios.post("/api/v1/auth/signup", {
return axios.post<User>("/api/v1/auth/signup", {
username,
password,
});
@ -44,14 +44,14 @@ export function signout() {
return axios.post("/api/v1/auth/signout");
}
export function getMyselfUser() {
return axios.get<User>("/api/v1/user/me");
}
export function getUserList() {
return axios.get<User[]>("/api/v1/user");
}
export function getUserById(id: number) {
return axios.get<User>(`/api/v1/user/${id}`);
}
export function upsertUserSetting(upsert: UserSettingUpsert) {
return axios.post<UserSetting>(`/api/v1/user/setting`, upsert);
}

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

@ -5,6 +5,7 @@ import { useSearchParams } from "react-router-dom";
import Icon from "@/components/Icon";
import * as api from "@/helpers/api";
import { absolutifyLink } from "@/helpers/utils";
import useNavigateTo from "@/hooks/useNavigateTo";
import { useUserStore } from "@/store/module";
import { useTranslate } from "@/utils/i18n";
@ -15,6 +16,7 @@ interface State {
const AuthCallback = () => {
const t = useTranslate();
const navigateTo = useNavigateTo();
const [searchParams] = useSearchParams();
const userStore = useUserStore();
const [state, setState] = useState<State>({
@ -32,14 +34,15 @@ const AuthCallback = () => {
if (identityProviderId) {
api
.signinWithSSO(identityProviderId, code, redirectUri)
.then(async () => {
.then(async ({ data: user }) => {
setState({
loading: false,
errorMessage: "",
});
const user = await userStore.doSignIn();
if (user) {
window.location.href = "/";
userStore.setCurrentUser(user);
await userStore.fetchCurrentUser();
navigateTo("/");
} else {
toast.error(t("message.login-failed"));
}

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

@ -7,11 +7,13 @@ import LocaleSelect from "@/components/LocaleSelect";
import * as api from "@/helpers/api";
import { absolutifyLink } from "@/helpers/utils";
import useLoading from "@/hooks/useLoading";
import useNavigateTo from "@/hooks/useNavigateTo";
import { useGlobalStore, useUserStore } from "@/store/module";
import { useTranslate } from "@/utils/i18n";
const SignIn = () => {
const t = useTranslate();
const navigateTo = useNavigateTo();
const globalStore = useGlobalStore();
const userStore = useUserStore();
const actionBtnLoadingState = useLoading(false);
@ -72,10 +74,11 @@ const SignIn = () => {
try {
actionBtnLoadingState.setLoading();
await api.signin(username, password, remember);
const user = await userStore.doSignIn();
const { data: user } = await api.signin(username, password, remember);
if (user) {
window.location.href = "/";
userStore.setCurrentUser(user);
await userStore.fetchCurrentUser();
navigateTo("/");
} else {
toast.error(t("message.login-failed"));
}

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

@ -6,11 +6,13 @@ import AppearanceSelect from "@/components/AppearanceSelect";
import LocaleSelect from "@/components/LocaleSelect";
import * as api from "@/helpers/api";
import useLoading from "@/hooks/useLoading";
import useNavigateTo from "@/hooks/useNavigateTo";
import { useGlobalStore, useUserStore } from "@/store/module";
import { useTranslate } from "@/utils/i18n";
const SignUp = () => {
const t = useTranslate();
const navigateTo = useNavigateTo();
const globalStore = useGlobalStore();
const userStore = useUserStore();
const actionBtnLoadingState = useLoading(false);
@ -52,10 +54,11 @@ const SignUp = () => {
try {
actionBtnLoadingState.setLoading();
await api.signup(username, password);
const user = await userStore.doSignIn();
const { data: user } = await api.signup(username, password);
if (user) {
window.location.href = "/";
userStore.setCurrentUser(user);
await userStore.fetchCurrentUser();
navigateTo("/");
} else {
toast.error(t("message.signup-failed"));
}

36
web/src/store/module/user.ts
View File

@ -58,10 +58,8 @@ export const initialUserState = async () => {
store.dispatch(setHost(convertResponseModelUser(systemStatus.host)));
}
const { data } = await api.getMyselfUser();
if (data) {
const user = convertResponseModelUser(data);
store.dispatch(setUser(user));
const user = await fetchCurrentUser();
if (user) {
if (user.setting.locale) {
store.dispatch(setLocale(user.setting.locale));
}
@ -72,18 +70,21 @@ export const initialUserState = async () => {
}
};
const doSignIn = async () => {
const { data: user } = await api.getMyselfUser();
if (user) {
store.dispatch(setUser(convertResponseModelUser(user)));
} else {
doSignOut();
}
return user;
};
const doSignOut = async () => {
await api.signout();
localStorage.removeItem("userId");
};
const fetchCurrentUser = async () => {
const userId = localStorage.getItem("userId");
if (userId) {
const { data } = await api.getUserById(Number(userId));
const user = convertResponseModelUser(data);
if (user) {
store.dispatch(setUser(user));
return user;
}
}
};
export const useUserStore = () => {
@ -94,14 +95,17 @@ export const useUserStore = () => {
getState: () => {
return store.getState().user;
},
doSignIn,
doSignOut,
fetchCurrentUser,
setCurrentUser: async (user: User) => {
localStorage.setItem("userId", String(user.id));
},
upsertUserSetting: async (key: string, value: any) => {
await api.upsertUserSetting({
key: key as any,
value: JSON.stringify(value),
});
await doSignIn();
await fetchCurrentUser();
},
upsertLocalSetting: async (localSetting: LocalSetting) => {
storage.set({ localSetting });