mirror of
https://codeberg.org/VnPower/PixivFE
synced 2024-12-06 19:16:23 +01:00
perennial
109 lines
2.7 KiB
Go
109 lines
2.7 KiB
Go
// User Settings (Using Browser Cookies)
|
|
|
|
package session
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
_ "time/tzdata"
|
|
)
|
|
|
|
type CookieName string
|
|
|
|
const ( // the __Host thing force it to be secure and same-origin (no subdomain) >> https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Set-Cookie
|
|
Cookie_Token CookieName = "pixivfe-Token"
|
|
Cookie_CSRF CookieName = "pixivfe-CSRF"
|
|
Cookie_ImageProxy CookieName = "pixivfe-ImageProxy"
|
|
Cookie_TZ CookieName = "pixivfe-TZ"
|
|
Cookie_NovelFontType CookieName = "pixivfe-NovelFontType"
|
|
Cookie_NovelViewMode CookieName = "pixivfe-NovelViewMode"
|
|
Cookie_ThumbnailToNewTab CookieName = "pixivfe-ThumbnailToNewTab"
|
|
Cookie_ArtworkPreview CookieName = "pixivfe-ArtworkPreview"
|
|
Cookie_HideArtR18 CookieName = "pixivfe-HideArtR18"
|
|
Cookie_HideArtR18G CookieName = "pixivfe-HideArtR18G"
|
|
Cookie_HideArtAI CookieName = "pixivfe-HideArtAI"
|
|
Cookie_Locale CookieName = "pixivfe-Locale"
|
|
Cookie_LogoStyle CookieName = "pixivfe-LogoStyle"
|
|
)
|
|
|
|
// Go can't make this a const...
|
|
var AllCookieNames []CookieName = []CookieName{
|
|
Cookie_Token,
|
|
Cookie_CSRF,
|
|
Cookie_ImageProxy,
|
|
Cookie_TZ,
|
|
Cookie_NovelFontType,
|
|
Cookie_NovelViewMode,
|
|
Cookie_ThumbnailToNewTab,
|
|
Cookie_ArtworkPreview,
|
|
Cookie_HideArtR18,
|
|
Cookie_HideArtR18G,
|
|
Cookie_HideArtAI,
|
|
Cookie_Locale,
|
|
Cookie_LogoStyle,
|
|
}
|
|
|
|
func GetCookie(r *http.Request, name CookieName) string {
|
|
cookie, err := r.Cookie(string(name))
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return cookie.Value
|
|
}
|
|
|
|
func SetCookie(w http.ResponseWriter, name CookieName, value string) {
|
|
http.SetCookie(w, &http.Cookie{
|
|
Name: string(name),
|
|
Value: value,
|
|
Path: "/",
|
|
// expires in 30 days from now
|
|
Expires: time.Now().Add(30 * (24 * time.Hour)),
|
|
HttpOnly: true,
|
|
Secure: true,
|
|
SameSite: http.SameSiteStrictMode, // bye-bye cross site forgery
|
|
})
|
|
}
|
|
|
|
var CookieExpireDelete = time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC)
|
|
|
|
func ClearCookie(w http.ResponseWriter, name CookieName) {
|
|
http.SetCookie(w, &http.Cookie{
|
|
Name: string(name),
|
|
Value: "",
|
|
Path: "/",
|
|
// expires in 30 days from now
|
|
Expires: CookieExpireDelete,
|
|
HttpOnly: true,
|
|
Secure: true,
|
|
SameSite: http.SameSiteStrictMode,
|
|
})
|
|
}
|
|
|
|
func ClearAllCookies(w http.ResponseWriter) {
|
|
for _, name := range AllCookieNames {
|
|
ClearCookie(w, name)
|
|
}
|
|
}
|
|
|
|
func ValidateTimezone(tz string) (*time.Location, error) {
|
|
if tz == "" {
|
|
return time.UTC, nil
|
|
}
|
|
|
|
location, err := time.LoadLocation(tz)
|
|
if err != nil {
|
|
return time.UTC, err
|
|
}
|
|
|
|
return location, nil
|
|
}
|
|
|
|
func GetUserTimezone(r *http.Request) *time.Location {
|
|
tz := GetCookie(r, Cookie_TZ)
|
|
location, err := ValidateTimezone(tz)
|
|
if err != nil {
|
|
return time.UTC
|
|
}
|
|
return location
|
|
}
|