115 lines
2.7 KiB
Go
Raw Normal View History

package middleware
import (
"encoding/json"
"io/ioutil"
"net/http"
2017-06-22 20:50:50 +00:00
"strconv"
"strings"
"github.com/aerogo/aero"
"github.com/animenotifier/arn"
"github.com/animenotifier/notify.moe/utils"
"github.com/fatih/color"
"github.com/mssola/user_agent"
"github.com/parnurzeal/gorequest"
)
var apiKeys arn.APIKeys
func init() {
data, _ := ioutil.ReadFile("security/api-keys.json")
err := json.Unmarshal(data, &apiKeys)
if err != nil {
panic(err)
}
}
// UserInfo updates user related information after each request.
func UserInfo() aero.Middleware {
return func(ctx *aero.Context, next func()) {
next()
2017-06-23 14:58:12 +00:00
// Ignore non-HTML requests
2017-06-23 17:22:39 +00:00
if ctx.IsMediaResponse() {
return
}
// Ignore API requests
if strings.HasPrefix(ctx.URI(), "/api/") {
2017-06-23 14:58:12 +00:00
return
}
2017-06-23 14:58:12 +00:00
user := utils.GetUser(ctx)
2017-06-23 14:58:12 +00:00
// When there's no user logged in, nothing to update
if user == nil {
return
}
2017-06-23 14:58:12 +00:00
// This works asynchronously so it doesn't block the response
go updateUserInfo(ctx, user)
}
2017-06-23 14:58:12 +00:00
}
2017-06-23 15:00:41 +00:00
// Update browser and OS data
2017-06-23 14:58:12 +00:00
func updateUserInfo(ctx *aero.Context, user *arn.User) {
newIP := ctx.RealIP()
newUserAgent := ctx.UserAgent()
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
}
if user.IP != newIP {
2017-06-23 14:58:12 +00:00
updateUserLocation(user, newIP)
}
2017-06-23 14:58:12 +00:00
user.LastSeen = arn.DateTimeUTC()
user.Save()
}
2017-06-23 15:00:41 +00:00
// Updates the location of the user.
2017-06-23 14:58:12 +00:00
func updateUserLocation(user *arn.User, newIP string) {
user.IP = newIP
locationAPI := "https://api.ipinfodb.com/v3/ip-city/?key=" + apiKeys.IPInfoDB.ID + "&ip=" + user.IP + "&format=json"
2017-06-23 14:58:12 +00:00
response, data, err := gorequest.New().Get(locationAPI).EndBytes()
2017-06-23 14:58:12 +00:00
if len(err) > 0 && err[0] != nil {
color.Red("Couldn't fetch location data | Error: %s | IP: %s", err[0].Error(), user.IP)
return
}
2017-06-23 14:58:12 +00:00
if response.StatusCode != http.StatusOK {
color.Red("Couldn't fetch location data | Status: %d | IP: %s", response.StatusCode, user.IP)
return
}
newLocation := arn.IPInfoDBLocation{}
json.Unmarshal(data, &newLocation)
if newLocation.CountryName != "-" {
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
}
}