34 lines
765 B
Go
34 lines
765 B
Go
package auth
|
|
|
|
import (
|
|
"fmt"
|
|
"ghproxy/config"
|
|
|
|
"github.com/cloudwego/hertz/pkg/app"
|
|
)
|
|
|
|
func AuthHeaderHandler(c *app.RequestContext, cfg *config.Config) (isValid bool, err error) {
|
|
if !cfg.Auth.Enabled {
|
|
return true, nil
|
|
}
|
|
// 获取"GH-Auth"的值
|
|
var authToken string
|
|
if cfg.Auth.Key != "" {
|
|
authToken = string(c.GetHeader(cfg.Auth.Key))
|
|
|
|
} else {
|
|
authToken = string(c.GetHeader("GH-Auth"))
|
|
}
|
|
logDebug("%s %s %s %s %s AUTH_TOKEN: %s", c.Method(), string(c.Path()), c.Request.Header.UserAgent(), c.Request.Header.GetProtocol(), authToken)
|
|
if authToken == "" {
|
|
return false, fmt.Errorf("Auth token not found")
|
|
}
|
|
|
|
isValid = authToken == cfg.Auth.Token
|
|
if !isValid {
|
|
return false, fmt.Errorf("Auth token incorrect")
|
|
}
|
|
|
|
return isValid, nil
|
|
}
|