2023-08-13 00:06:03 +08:00
|
|
|
package v2
|
|
|
|
|
|
|
|
import "strings"
|
|
|
|
|
|
|
|
var authenticationAllowlistMethods = map[string]bool{
|
|
|
|
"/memos.api.v2.SystemService/GetSystemInfo": true,
|
2023-11-30 20:58:36 +08:00
|
|
|
"/memos.api.v2.AuthService/GetAuthStatus": true,
|
2023-08-13 00:06:03 +08:00
|
|
|
"/memos.api.v2.UserService/GetUser": true,
|
|
|
|
"/memos.api.v2.MemoService/ListMemos": true,
|
|
|
|
}
|
|
|
|
|
|
|
|
// 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]
|
|
|
|
}
|