useragent.go (1324B)
1 // GoToSocial 2 // Copyright (C) GoToSocial Authors admin@gotosocial.org 3 // SPDX-License-Identifier: AGPL-3.0-or-later 4 // 5 // This program is free software: you can redistribute it and/or modify 6 // it under the terms of the GNU Affero General Public License as published by 7 // the Free Software Foundation, either version 3 of the License, or 8 // (at your option) any later version. 9 // 10 // This program is distributed in the hope that it will be useful, 11 // but WITHOUT ANY WARRANTY; without even the implied warranty of 12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13 // GNU Affero General Public License for more details. 14 // 15 // You should have received a copy of the GNU Affero General Public License 16 // along with this program. If not, see <http://www.gnu.org/licenses/>. 17 18 package middleware 19 20 import ( 21 "errors" 22 "net/http" 23 24 "github.com/gin-gonic/gin" 25 ) 26 27 // UserAgent returns a gin middleware which aborts requests with 28 // empty user agent strings, returning code 418 - I'm a teapot. 29 func UserAgent() gin.HandlerFunc { 30 // todo: make this configurable 31 return func(c *gin.Context) { 32 if ua := c.Request.UserAgent(); ua == "" { 33 code := http.StatusTeapot 34 err := errors.New(http.StatusText(code) + ": no user-agent sent with request") 35 c.AbortWithStatusJSON(code, gin.H{"error": err.Error()}) 36 } 37 } 38 }