Hypernet/Nexus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

🐛 Bug fixes on auth

Browse Source
This commit is contained in:
LittleSheep 2024-10-30 23:26:39 +08:00
parent ae3583ead1
commit a785adfafa
2 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/internal/auth/http.go
View File

@ -18,7 +18,7 @@ func ContextMiddleware(c *fiber.Ctx) error {
return err
}
} else if err != nil {
c.Locals("nex_auth_error", err)
return fiber.NewError(fiber.StatusUnauthorized, err.Error())
}
return c.Next()
@ -26,12 +26,8 @@ func ContextMiddleware(c *fiber.Ctx) error {
func ValidatorMiddleware(c *fiber.Ctx) error {
if c.Locals("nex_principal") == nil {
if err, ok := c.Locals("nex_auth_error").(error); ok {
return fiber.NewError(fiber.StatusUnauthorized, err.Error())
} else {
return fiber.NewError(fiber.StatusUnauthorized, "unauthorized")
}
}
return c.Next()
}

4
pkg/internal/http/api/forward.go
View File

@ -29,9 +29,9 @@ func forwardService(c *fiber.Ctx) error {
url = *service.HttpAddr + url
if tk, ok := c.Locals("nex_token").(string); ok {
c.Set(fiber.HeaderAuthorization, fmt.Sprintf("Bearer %s", tk))
c.Request().Header.Set(fiber.HeaderAuthorization, fmt.Sprintf("Bearer %s", tk))
} else {
c.Set(fiber.HeaderAuthorization, "")
c.Request().Header.Del(fiber.HeaderAuthorization)
}
return proxy.Do(c, url)