2023-08-13 00:06:03 +08:00
|
|
|
package v2
|
|
|
|
|
|
|
|
import "strings"
|
|
|
|
|
|
|
|
var authenticationAllowlistMethods = map[string]bool{
|
2024-02-23 09:08:49 +08:00
|
|
|
"/memos.api.v2.WorkspaceService/GetWorkspaceProfile": true,
|
|
|
|
"/memos.api.v2.WorkspaceSettingService/GetWorkspaceSetting": true,
|
|
|
|
"/memos.api.v2.AuthService/GetAuthStatus": true,
|
|
|
|
"/memos.api.v2.AuthService/SignIn": true,
|
|
|
|
"/memos.api.v2.AuthService/SignInWithSSO": true,
|
|
|
|
"/memos.api.v2.AuthService/SignOut": true,
|
|
|
|
"/memos.api.v2.AuthService/SignUp": true,
|
|
|
|
"/memos.api.v2.UserService/GetUser": true,
|
|
|
|
"/memos.api.v2.MemoService/ListMemos": true,
|
|
|
|
"/memos.api.v2.MemoService/GetMemo": true,
|
|
|
|
"/memos.api.v2.MemoService/GetMemoByName": true,
|
|
|
|
"/memos.api.v2.MemoService/ListMemoResources": true,
|
|
|
|
"/memos.api.v2.MemoService/ListMemoRelations": true,
|
|
|
|
"/memos.api.v2.MemoService/ListMemoComments": true,
|
2023-08-13 00:06:03 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// isUnauthorizeAllowedMethod returns whether the method is exempted from authentication.
|
|
|
|
func isUnauthorizeAllowedMethod(fullMethodName string) bool {
|
|
|
|
if strings.HasPrefix(fullMethodName, "/grpc.reflection") {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
return authenticationAllowlistMethods[fullMethodName]
|
|
|
|
}
|
|
|
|
|
|
|
|
var allowedMethodsOnlyForAdmin = map[string]bool{
|
|
|
|
"/memos.api.v2.UserService/CreateUser": true,
|
|
|
|
}
|
|
|
|
|
|
|
|
// isOnlyForAdminAllowedMethod returns true if the method is allowed to be called only by admin.
|
|
|
|
func isOnlyForAdminAllowedMethod(methodName string) bool {
|
|
|
|
return allowedMethodsOnlyForAdmin[methodName]
|
|
|
|
}
|