1
0
This repository has been archived on 2024-02-27. You can view files and clone it, but cannot push or open issues or pull requests.
chatgpt-to-api/middleware.go
Antonio Cheong f54f15f791 refactor
2023-04-05 18:07:07 +08:00

34 lines
595 B
Go

package main
import (
"os"
gin "github.com/gin-gonic/gin"
)
var ADMIN_PASSWORD string
func init() {
ADMIN_PASSWORD = os.Getenv("ADMIN_PASSWORD")
if ADMIN_PASSWORD == "" {
ADMIN_PASSWORD = "TotallySecurePassword"
}
}
func adminCheck(c *gin.Context) {
password := c.Request.Header.Get("Authorization")
if password != ADMIN_PASSWORD {
c.String(401, "Unauthorized")
c.Abort()
return
}
c.Next()
}
func cors(c *gin.Context) {
c.Header("Access-Control-Allow-Origin", "*")
c.Header("Access-Control-Allow-Methods", "*")
c.Header("Access-Control-Allow-Headers", "*")
c.Next()
}