diff --git a/server/config/config.go b/server/config/config.go index acfbace..f96e55c 100644 --- a/server/config/config.go +++ b/server/config/config.go @@ -29,8 +29,10 @@ func LoadConfig() *Config { Ip: os.Getenv("BIND_IP"), AuthUsername: os.Getenv("AUTH_USERNAME"), AuthPassword: os.Getenv("AUTH_PASSWORD"), - AuthKey: os.Getenv("SENSOR_AUTH_KEY"), + AuthKey: os.Getenv("AUTH_KEY"), } + // TODO: Crash when any auth* param is empty + return &config } diff --git a/server/middleware/auth.go b/server/middleware/auth.go index 89dd8bf..3e164e0 100644 --- a/server/middleware/auth.go +++ b/server/middleware/auth.go @@ -22,8 +22,10 @@ func LoginAuthMiddleware(server *app.Server) gin.HandlerFunc { } func KeyAuthMiddleware(server *app.Server) gin.HandlerFunc { + keyWithBearer := "Bearer " + server.Config.AuthKey + return func(c *gin.Context) { - if c.GetHeader("authorization") != server.Config.AuthKey { + if c.GetHeader("authorization") != keyWithBearer { c.AbortWithStatus(http.StatusUnauthorized) return