web/notify.moe
Code Issues Pull Requests Activity

Implemented new user registrations

Browse Source
This commit is contained in:
Eduard Urbach
2017-06-22 22:26:52 +02:00
parent cf0d47bfb0
commit efe8a36b7a
6 changed files with 105 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
middleware/UserInfo.go
View File

@ -2,7 +2,6 @@ package middleware
import (
"encoding/json"
"fmt"
"io/ioutil"
"net/http"
"strings"
@ -46,8 +45,7 @@ func updateUserInfo(ctx *aero.Context) {
}
// Ignore non-HTML requests
println(ctx.GetRequestHeader("Accept-Type"))
if strings.Index(ctx.GetRequestHeader("Accept-Type"), "text/html") == -1 {
if strings.Index(ctx.GetRequestHeader("Accept"), "text/html") == -1 {
return
}
@ -63,37 +61,36 @@ func updateUserInfo(ctx *aero.Context) {
// Browser
user.Browser.Name, user.Browser.Version = parsed.Browser()
arn.PrettyPrint(user.Browser)
// OS
os := parsed.OSInfo()
user.OS.Name = os.Name
user.OS.Version = os.Version
arn.PrettyPrint(user.OS)
}
if user.IP != newIP {
user.IP = newIP
locationAPI := "https://api.ipinfodb.com/v3/ip-city/?key=" + apiKeys.IPInfoDB.ID + "&ip=" + "2a02:8108:8dc0:3000:6cf1:af03:ce6e:679a" + "&format=json"
locationAPI := "https://api.ipinfodb.com/v3/ip-city/?key=" + apiKeys.IPInfoDB.ID + "&ip=" + user.IP + "&format=json"
response, data, err := gorequest.New().Get(locationAPI).EndBytes()
if len(err) > 0 && err[0] != nil {
color.Red(err[0].Error())
color.Red("Couldn't fetch location data | Error: %s | IP: %s", err[0].Error(), user.IP)
return
}
if response.StatusCode != http.StatusOK {
body, _ := ioutil.ReadAll(response.Body)
fmt.Println(response.StatusCode, locationAPI)
fmt.Println(string(body))
color.Red("Couldn't fetch location data | Status: %d | IP: %s", response.StatusCode, user.IP)
return
}
json.Unmarshal(data, &user.Location)
arn.PrettyPrint(user.Location)
newLocation := arn.UserLocation{}
json.Unmarshal(data, &newLocation)
if newLocation.CountryName != "-" {
user.Location = newLocation
}
}
// user.Save()
user.LastSeen = arn.DateTimeUTC()
user.Save()
}