add header filter
This commit is contained in:
17
mw/filter_headers.go
Normal file
17
mw/filter_headers.go
Normal file
@@ -0,0 +1,17 @@
|
||||
package mw
|
||||
|
||||
import (
|
||||
"github.com/labstack/echo/v4"
|
||||
)
|
||||
|
||||
var NOT_ALLOWED_HEADERS = [2]string{"UserId", "Authorization"}
|
||||
|
||||
func FilterHeaders(next echo.HandlerFunc) echo.HandlerFunc {
|
||||
return func(c echo.Context) error {
|
||||
for _, header := range NOT_ALLOWED_HEADERS {
|
||||
c.Request().Header.Set(header, "")
|
||||
}
|
||||
|
||||
return next(c)
|
||||
}
|
||||
}
|
||||
@@ -2,22 +2,16 @@ package mw
|
||||
|
||||
import (
|
||||
"github.com/labstack/echo/v4"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
func TokenFromCookies(next echo.HandlerFunc) echo.HandlerFunc {
|
||||
return func(c echo.Context) error {
|
||||
var token, err = c.Request().Cookie("accessToken")
|
||||
var authToken = c.Request().Header.Get("Authorization")
|
||||
|
||||
if err != nil {
|
||||
return next(c)
|
||||
}
|
||||
|
||||
if authToken != "" {
|
||||
return c.NoContent(http.StatusBadRequest)
|
||||
}
|
||||
|
||||
c.Request().Header.Set("Authorization", "Bearer "+token.Value)
|
||||
|
||||
return next(c)
|
||||
|
||||
Reference in New Issue
Block a user