mirror of
https://codeberg.org/VnPower/PixivFE
synced 2024-12-06 19:16:23 +01:00
@@ -1,32 +0,0 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
|
||||
"codeberg.org/vnpower/pixivfe/v2/config"
|
||||
"codeberg.org/vnpower/pixivfe/v2/server/session"
|
||||
)
|
||||
|
||||
func SetDefaultCookies(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
// Check if it's a first visit
|
||||
_, err := r.Cookie("pixivfe-visited")
|
||||
if err == http.ErrNoCookie {
|
||||
// It's a first visit, set default cookies
|
||||
defaultCookies := config.GlobalConfig.DefaultCookies
|
||||
session.SetCookie(w, session.Cookie_ImageProxy, defaultCookies.ImageProxy)
|
||||
session.SetCookie(w, session.Cookie_NovelFontType, defaultCookies.NovelFontType)
|
||||
session.SetCookie(w, session.Cookie_NovelViewMode, defaultCookies.NovelViewMode)
|
||||
session.SetCookie(w, session.Cookie_ThumbnailToNewTab, defaultCookies.ThumbnailToNewTab)
|
||||
session.SetCookie(w, session.Cookie_ArtworkPreview, defaultCookies.ArtworkPreview)
|
||||
session.SetCookie(w, session.Cookie_HideArtR18, defaultCookies.HideArtR18)
|
||||
session.SetCookie(w, session.Cookie_HideArtR18G, defaultCookies.HideArtR18G)
|
||||
session.SetCookie(w, session.Cookie_HideArtAI, defaultCookies.HideArtAI)
|
||||
|
||||
// Set the visited cookie
|
||||
session.SetCookie(w, "pixivfe-visited", "true")
|
||||
}
|
||||
|
||||
next.ServeHTTP(w, r)
|
||||
})
|
||||
}
|
||||
@@ -30,9 +30,6 @@ func serveFile(filename string) http.HandlerFunc {
|
||||
func DefineRoutes() *mux.Router {
|
||||
router := mux.NewRouter()
|
||||
|
||||
// Apply the SetDefaultCookies middleware to all routes
|
||||
router.Use(SetDefaultCookies)
|
||||
|
||||
// Tutorial: Adding new routes
|
||||
// 1. Use router.HandleFunc to define the path and handler
|
||||
// 2. Wrap the handler function, defined in package routes, with CatchError for error handling
|
||||
|
||||
Reference in New Issue
Block a user