YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-06-05 22:09:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: update get request origin

Browse Source
This commit is contained in:
Steven
2024-03-03 14:10:48 +08:00
parent 3b089eeae3
commit c1da87a819
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
server/route/api/v2/auth_service.go
View File

@@ -248,11 +248,17 @@ func (s *APIV2Service) buildAccessTokenCookie(ctx context.Context, accessToken s
} else { } else {
attrs = append(attrs, "Expires="+expireTime.Format(time.RFC1123)) attrs = append(attrs, "Expires="+expireTime.Format(time.RFC1123))
} }
workspaceGeneralSetting, err := s.Store.GetWorkspaceGeneralSetting(ctx)
if err != nil { md, ok := metadata.FromIncomingContext(ctx)
return "", errors.Wrap(err, "failed to get workspace setting") if !ok {
return "", errors.New("failed to get metadata from context")
} }
if strings.HasPrefix(workspaceGeneralSetting.InstanceUrl, "https://") { var origin string
for _, v := range md.Get("origin") {
origin = v
}
isHTTPS := strings.HasPrefix(origin, "https://")
if isHTTPS {
attrs = append(attrs, "SameSite=None") attrs = append(attrs, "SameSite=None")
attrs = append(attrs, "Secure") attrs = append(attrs, "Secure")
} else { } else {