mirror of
https://github.com/go-gitea/gitea.git
synced 2026-08-10 05:24:18 +09:00
76a81b24f9
Enable [`forcetypeassert`](https://github.com/gostaticanalysis/forcetypeassert) linter to prevent unchecked type assertions. ~650 issues fixed, most fixes were clean, some use `setting.PanicInDevOrTesting`. The only behaviour changes are where code would previously send a 500 error or panic, a 4xx error is now emitted. Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
// Copyright 2020 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package session
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"gitea.dev/modules/setting"
|
|
|
|
"gitea.com/go-chi/session"
|
|
)
|
|
|
|
type RawStore = session.RawStore
|
|
|
|
type Store interface {
|
|
RawStore
|
|
Destroy(http.ResponseWriter, *http.Request) error
|
|
}
|
|
|
|
type mockStoreContextKeyStruct struct{}
|
|
|
|
var MockStoreContextKey = mockStoreContextKeyStruct{}
|
|
|
|
// RegenerateSession regenerates the underlying session and returns the new store
|
|
func RegenerateSession(resp http.ResponseWriter, req *http.Request) (Store, error) {
|
|
for _, f := range BeforeRegenerateSession {
|
|
f(resp, req)
|
|
}
|
|
if setting.IsInTesting {
|
|
if store, ok := req.Context().Value(MockStoreContextKey).(Store); ok {
|
|
return store, nil
|
|
}
|
|
}
|
|
return session.RegenerateSession(resp, req)
|
|
}
|
|
|
|
func GetContextSession(req *http.Request) Store {
|
|
if setting.IsInTesting {
|
|
if store, ok := req.Context().Value(MockStoreContextKey).(Store); ok {
|
|
return store
|
|
}
|
|
}
|
|
return session.GetSession(req)
|
|
}
|
|
|
|
// BeforeRegenerateSession is a list of functions that are called before a session is regenerated.
|
|
var BeforeRegenerateSession []func(http.ResponseWriter, *http.Request)
|