Victor_Jay
6229c81f6e
- 修复注册关闭时标题“创建账号”和副标题仍显示的问题，三态互斥（维护中 > 关闭注册 > 正常） - 新增 maintenance.enabled 站点设置，默认关闭，仅 Owner 可见可调节 - 维护中间件：全局拦截，维护期间仅站长可访问，白名单放行登录相关路径 - 登录服务：维护期间仅 Owner 角色可登录，Register 优先检查维护状态 - Nav 模板新增维护通知条（黄色横幅） - 管理后台站点设置新增“启用维护模式”开关 - BuildPageData 自动注入 IsMaintenance 到所有页面模板
85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
package common
|
||
|
||
import (
|
||
"regexp"
|
||
|
||
"metazone.cc/metalab/internal/model"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// mdImageRe 匹配 Markdown 图片语法 
|
||
var mdImageRe = regexp.MustCompile(`!\[.*?\]\((.*?)\)`)
|
||
|
||
// ExtractFirstImage 从 Markdown 正文提取第一张图片 URL
|
||
func ExtractFirstImage(md string) string {
|
||
match := mdImageRe.FindStringSubmatch(md)
|
||
if len(match) > 1 {
|
||
return match[1]
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// BuildPageData 构建页面模板数据，自动注入登录状态、站点信息与 CSRF token
|
||
func BuildPageData(c *gin.Context, extra gin.H) gin.H {
|
||
data := gin.H{}
|
||
for k, v := range extra {
|
||
data[k] = v
|
||
}
|
||
if username, exists := c.Get("username"); exists {
|
||
data["IsLoggedIn"] = true
|
||
data["Username"] = username
|
||
}
|
||
// 注入 CSRF token（由 CSRF 中间件设置到上下文中）
|
||
if token, exists := c.Get("csrf_token"); exists {
|
||
data["CSRFToken"] = token
|
||
}
|
||
// 注入站点信息（由 InjectSiteInfo 中间件设置）
|
||
injectSiteInfo(c, data)
|
||
// 注入管理入口权限（moderator 及以上可看到页脚管理面板链接）
|
||
if role, exists := c.Get("role"); exists {
|
||
data["CanAccessAdmin"] = model.HasMinRole(role.(string), model.RoleModerator)
|
||
}
|
||
// 注入维护状态
|
||
if isMaintenance, exists := c.Get("is_maintenance"); exists {
|
||
data["IsMaintenance"] = isMaintenance
|
||
}
|
||
return data
|
||
}
|
||
|
||
// BuildAdminPageData 构建管理后台模板数据
|
||
// 额外注入 UID、Role、CanManageUsers、CurrentPath 等权限信息
|
||
func BuildAdminPageData(c *gin.Context, extra gin.H) gin.H {
|
||
data := BuildPageData(c, extra)
|
||
data["IsAdmin"] = true
|
||
data["CurrentPath"] = c.Request.URL.Path
|
||
|
||
if uid, exists := c.Get("uid"); exists {
|
||
data["UID"] = uid
|
||
}
|
||
if role, exists := c.Get("role"); exists {
|
||
roleStr := role.(string)
|
||
data["Role"] = roleStr
|
||
if name, ok := model.RoleDisplayNames[roleStr]; ok {
|
||
data["RoleDisplayName"] = name
|
||
} else {
|
||
data["RoleDisplayName"] = roleStr
|
||
}
|
||
data["CanManageUsers"] = model.HasMinRole(roleStr, model.RoleAdmin)
|
||
data["CanManageAudits"] = model.HasMinRole(roleStr, model.RoleAdmin)
|
||
data["CanManageSettings"] = model.HasMinRole(roleStr, model.RoleOwner)
|
||
data["IsOwner"] = model.HasMinRole(roleStr, model.RoleOwner)
|
||
}
|
||
return data
|
||
}
|
||
|
||
// injectSiteInfo 从 context 注入站点展示信息
|
||
func injectSiteInfo(c *gin.Context, data gin.H) {
|
||
if framework, exists := c.Get("site_framework"); exists {
|
||
data["Framework"] = framework
|
||
}
|
||
if copyrightStart, exists := c.Get("site_copyright_start"); exists {
|
||
data["CopyrightStart"] = copyrightStart
|
||
}
|
||
}
|