diff options
Diffstat (limited to 'internal/api/security')
-rw-r--r-- | internal/api/security/extraheaders.go | 8 | ||||
-rw-r--r-- | internal/api/security/security.go | 1 |
2 files changed, 9 insertions, 0 deletions
diff --git a/internal/api/security/extraheaders.go b/internal/api/security/extraheaders.go new file mode 100644 index 000000000..dfcddfbe1 --- /dev/null +++ b/internal/api/security/extraheaders.go @@ -0,0 +1,8 @@ +package security + +import "github.com/gin-gonic/gin" + +// ExtraHeaders adds any additional required headers to the response +func (m *Module) ExtraHeaders(c *gin.Context) { + c.Header("Server", "Mastodon") +} diff --git a/internal/api/security/security.go b/internal/api/security/security.go index c80b568b3..eaae8471e 100644 --- a/internal/api/security/security.go +++ b/internal/api/security/security.go @@ -42,5 +42,6 @@ func New(config *config.Config, log *logrus.Logger) api.ClientModule { // Route attaches security middleware to the given router func (m *Module) Route(s router.Router) error { s.AttachMiddleware(m.FlocBlock) + s.AttachMiddleware(m.ExtraHeaders) return nil } |