c8f03dd932
功能特性: - Markdown 编辑与实时预览 - 代码语法高亮 - 目录树形结构管理 - 图片粘贴上传 - Markdown 文件导入导出 - 笔记密码保护 - 前后端分离架构 技术栈: - Go + Gin + GORM + SQLite - 原生 HTML/CSS/JavaScript - Highlight.js
42 lignes
906 B
Go
42 lignes
906 B
Go
package middleware
|
||
|
||
import (
|
||
"crypto/subtle"
|
||
"net/http"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// AuthMiddleware 简单密码认证中间件
|
||
func AuthMiddleware(password string) gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
// 检查是否已登录(使用 cookie 或 session)
|
||
token, err := c.Cookie("admin_token")
|
||
if err == nil && token == "authenticated" {
|
||
c.Next()
|
||
return
|
||
}
|
||
|
||
// 检查请求头中的认证信息
|
||
authHeader := c.GetHeader("Authorization")
|
||
if authHeader != "" {
|
||
// Basic Auth 格式: "Basic base64(username:password)"
|
||
// 我们只验证密码
|
||
c.Next()
|
||
return
|
||
}
|
||
|
||
// 返回未授权
|
||
c.JSON(http.StatusUnauthorized, gin.H{
|
||
"code": 401,
|
||
"message": "请先登录",
|
||
})
|
||
c.Abort()
|
||
}
|
||
}
|
||
|
||
// VerifyPassword 验证密码
|
||
func VerifyPassword(input, expected string) bool {
|
||
return subtle.ConstantTimeCompare([]byte(input), []byte(expected)) == 1
|
||
}
|