2

internal/erpserver/model/form/user.go

@@ -1,5 +1,11 @@
 package form
 
+import (
+	"net/http"
+
+	"github.com/zhang2092/browser"
+)
+
 type Login struct {
 	Email     string `form:"email" binding:"required,email"`
 	Password  string `form:"password" binding:"required,min=6"`
@@ -14,6 +20,20 @@ type Login struct {
 	Url      string
 }
 
+func (login Login) SetAttributes(r *http.Request) Login {
+	login.Ip = r.RemoteAddr
+	login.Referrer = r.Header.Get("Referer")
+	login.Url = r.URL.RequestURI()
+
+	br, err := browser.NewBrowser(r.Header.Get("User-Agent"))
+	if err == nil {
+		login.Os = br.Platform().Name()
+		login.Browser = br.Name()
+	}
+
+	return login
+}
+
 type User struct {
 	ID             *int32 `form:"id" binding:"required"`
 	Email          string `form:"email" binding:"required,email"`