new useragent lib

go.mod

@@ -2,4 +2,4 @@ module github.com/davewhit3/traefik-cf-device-detector
 
 go 1.19
 
-require github.com/mssola/useragent v1.0.0
+require github.com/mileusna/useragent v1.2.1

go.sum

@@ -1,2 +1,2 @@
-github.com/mssola/useragent v1.0.0 h1:WRlDpXyxHDNfvZaPEut5Biveq86Ze4o4EMffyMxmH5o=
-github.com/mssola/useragent v1.0.0/go.mod h1:hz9Cqz4RXusgg1EdI4Al0INR62kP7aPSRNHnpU+b85Y=
+github.com/mileusna/useragent v1.2.1 h1:p3RJWhi3LfuI6BHdddojREyK3p6qX67vIfOVMnUIVr0=
+github.com/mileusna/useragent v1.2.1/go.mod h1:3d8TOmwL/5I8pJjyVDteHtgDGcefrFUX4ccGOMKNYYc=

middleware.go

@@ -8,7 +8,7 @@ import (
 	"os"
 	"strconv"
 
-	"github.com/mssola/useragent"
+	"github.com/mileusna/useragent"
 )
 
 const (
@@ -57,12 +57,12 @@ func New(ctx context.Context, next http.Handler, config *Config, name string) (h
 }
 
 func (mw *CfDeviceDetector) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
-	ua := useragent.New(req.Header.Get(UserAgentHeader))
+	ua := useragent.Parse(req.Header.Get(UserAgentHeader))
 
-	req.Header.Set(DeviceIsMobileHeader, strconv.FormatBool(ua.Mobile()))
-	req.Header.Set(DeviceIsDesktopHeader, strconv.FormatBool(!ua.Mobile()))
-	req.Header.Set(DeviceIsTabletHeader, strconv.FormatBool(false))
-	req.Header.Set(DeviceIsSmartTVHeader, strconv.FormatBool(false))
+	req.Header.Set(DeviceIsMobileHeader, strconv.FormatBool(ua.Mobile))
+	req.Header.Set(DeviceIsDesktopHeader, strconv.FormatBool(ua.Desktop))
+	req.Header.Set(DeviceIsTabletHeader, strconv.FormatBool(ua.Tablet))
+	req.Header.Set(DeviceIsSmartTVHeader, strconv.FormatBool(!ua.Tablet && !ua.Desktop && !ua.Mobile))
 
 	mw.next.ServeHTTP(rw, req)
 }