package middleware import ( "net/http" "strconv" "strings" "github.com/aerogo/aero" "github.com/aerogo/http/client" "github.com/akyoto/color" "github.com/animenotifier/notify.moe/arn" "github.com/animenotifier/notify.moe/utils" "github.com/mssola/user_agent" ) // UserInfo updates user related information after each request. func UserInfo(next aero.Handler) aero.Handler { return func(ctx aero.Context) error { err := next(ctx) // Ignore non-HTML requests contentType := ctx.Response().Header("Content-Type") if !strings.HasPrefix(contentType, "text/html") { return nil } user := utils.GetUser(ctx) // When there's no user logged in, nothing to update if user == nil { return nil } updateUserInfo(ctx, user) return err } } // Update browser and OS data func updateUserInfo(ctx aero.Context, user *arn.User) { newIP := ctx.IP() newUserAgent := ctx.Request().Header("User-Agent") if user.UserAgent != newUserAgent { user.UserAgent = newUserAgent // Parse user agent parsed := user_agent.New(user.UserAgent) // Browser user.Browser.Name, user.Browser.Version = parsed.Browser() // OS os := parsed.OSInfo() user.OS.Name = os.Name user.OS.Version = os.Version } user.LastSeen = arn.DateTimeUTC() user.Save() if user.IP != newIP { go updateUserLocation(user, newIP) } } // Updates the location of the user. func updateUserLocation(user *arn.User, newIP string) { user.IP = newIP if arn.APIKeys.IPInfoDB.ID == "" { if arn.IsProduction() { color.Red("IPInfoDB key not defined") } return } locationAPI := "https://api.ipinfodb.com/v3/ip-city/?key=" + arn.APIKeys.IPInfoDB.ID + "&ip=" + user.IP + "&format=json" response, err := client.Get(locationAPI).End() if err != nil { color.Red("Couldn't fetch location data | Error: %s | IP: %s", err.Error(), user.IP) return } if response.StatusCode() != http.StatusOK { color.Red("Couldn't fetch location data | Status: %d | IP: %s", response.StatusCode, user.IP) return } newLocation := arn.IPInfoDBLocation{} err = response.Unmarshal(&newLocation) if err != nil { color.Red("Couldn't deserialize location data | Status: %d | IP: %s", response.StatusCode, user.IP) return } if newLocation.CountryName == "" || newLocation.CountryName == "-" { return } user.Location.CountryName = newLocation.CountryName user.Location.CountryCode = newLocation.CountryCode user.Location.Latitude, _ = strconv.ParseFloat(newLocation.Latitude, 64) user.Location.Longitude, _ = strconv.ParseFloat(newLocation.Longitude, 64) user.Location.CityName = newLocation.CityName user.Location.RegionName = newLocation.RegionName user.Location.TimeZone = newLocation.TimeZone user.Location.ZipCode = newLocation.ZipCode // Make South Korea easier to read if user.Location.CountryName == "Korea, Republic of" { user.Location.CountryName = "South Korea" } user.Save() }