fix: allow option parsing if only one is present (#60)

This commit is contained in:
httpjamesm 2023-09-24 00:59:48 -04:00 committed by GitHub
parent 7cf3839f41
commit 991b5b46f4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 18 deletions

View File

@ -1,6 +1,8 @@
package middleware
import "github.com/gin-gonic/gin"
import (
"github.com/gin-gonic/gin"
)
func OptionsMiddleware() gin.HandlerFunc {
return func(c *gin.Context) {
@ -8,24 +10,18 @@ func OptionsMiddleware() gin.HandlerFunc {
c.Set("theme", "dark")
imagesCookie, err := c.Cookie("disable_images")
if err != nil {
c.Next()
return
}
if err == nil {
if imagesCookie == "true" {
c.Set("disable_images", true)
}
themeCookie, err := c.Cookie("theme")
if err != nil {
c.Next()
return
}
themeCookie, err := c.Cookie("theme")
if err == nil {
if themeCookie == "light" {
c.Set("theme", "light")
}
}
c.Next()
}

View File

@ -30,12 +30,10 @@ func ChangeOptions(c *gin.Context) {
if c.MustGet("theme").(string) == "dark" {
text = "light"
}
c.SetCookie("theme", text, 60*60*24*365*10, "/", "", false, true)
// get redirect url from query
redirectUrl := c.Query("redirect_url")
if redirectUrl == "" {
redirectUrl = os.Getenv("APP_URL")
}
if !strings.HasPrefix(redirectUrl, os.Getenv("APP_URL")) {
redirectUrl = os.Getenv("APP_URL")