mirror of https://github.com/usememos/memos
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
898 B
Go
37 lines
898 B
Go
package api
|
|
|
|
import (
|
|
"memos/api/e"
|
|
"net/http"
|
|
)
|
|
|
|
func AuthCheckerMiddleWare(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
session, _ := SessionStore.Get(r, "session")
|
|
|
|
if userId, ok := session.Values["user_id"].(string); !ok || userId == "" {
|
|
e.ErrorHandler(w, "NOT_AUTH", "Need authorize")
|
|
return
|
|
}
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
func JSONResponseMiddleWare(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
|
|
func CorsMiddleWare(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Access-Control-Allow-Origin", "*")
|
|
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
|
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|