mirror of
https://github.com/nezhahq/nezha.git
synced 2025-01-23 13:18:13 -05:00
26 lines
605 B
Go
26 lines
605 B
Go
|
package mygin
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/naiba/nezha/model"
|
||
|
|
"github.com/naiba/nezha/service/singleton"
|
||
|
|
)
|
||
|
|
|
||
|
|
func PreferredTheme(c *gin.Context) {
|
||
|
|
// 采用前端传入的主题
|
||
|
|
if theme, err := c.Cookie("preferred_theme"); err == nil {
|
||
|
|
if _, has := model.Themes[theme]; has {
|
||
|
|
c.Set(model.CtxKeyPreferredTheme, theme)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetPreferredTheme(c *gin.Context, path string) string {
|
||
|
|
if theme, has := c.Get(model.CtxKeyPreferredTheme); has {
|
||
|
|
return fmt.Sprintf("theme-%s%s", theme, path)
|
||
|
|
}
|
||
|
|
return fmt.Sprintf("theme-%s%s", singleton.Conf.Site.Theme, path)
|
||
|
|
}
|