Added arn to the main repository

2019-06-03 18:32:43 +09:00
parent cf258573a8
commit 29a48d94a5
465 changed files with 15968 additions and 288 deletions
--- a/arn/User.go
+++ b/arn/User.go
@ -0,0 +1,575 @@
+package arn
+
+import (
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"strconv"
+	"strings"
+	"sync"
+	"time"
+
+	"github.com/aerogo/aero"
+
+	"github.com/aerogo/http/client"
+	"github.com/animenotifier/notify.moe/arn/autocorrect"
+	"github.com/animenotifier/notify.moe/arn/validate"
+	"github.com/animenotifier/ffxiv"
+	"github.com/animenotifier/osu"
+	"github.com/animenotifier/overwatch"
+	gravatar "github.com/ungerik/go-gravatar"
+)
+
+var setNickMutex sync.Mutex
+var setEmailMutex sync.Mutex
+
+// Register data lists.
+func init() {
+	DataLists["genders"] = []*Option{
+		// &Option{"", "Prefer not to say"},
+		{"male", "Male"},
+		{"female", "Female"},
+	}
+}
+
+// UserID represents a user ID.
+type UserID = string
+
+// User is a registered person.
+type User struct {
+	ID           UserID       `json:"id"`
+	Nick         string       `json:"nick" editable:"true"`
+	FirstName    string       `json:"firstName" private:"true"`
+	LastName     string       `json:"lastName" private:"true"`
+	Email        string       `json:"email" editable:"true" private:"true"`
+	Role         string       `json:"role"`
+	Registered   string       `json:"registered"`
+	LastLogin    string       `json:"lastLogin" private:"true"`
+	LastSeen     string       `json:"lastSeen" private:"true"`
+	ProExpires   string       `json:"proExpires" editable:"true"`
+	Gender       string       `json:"gender" editable:"true" private:"true" datalist:"genders"`
+	Language     string       `json:"language"`
+	Introduction string       `json:"introduction" editable:"true" type:"textarea"`
+	Website      string       `json:"website" editable:"true"`
+	BirthDay     string       `json:"birthDay" editable:"true" private:"true"`
+	IP           string       `json:"ip" private:"true"`
+	UserAgent    string       `json:"agent" private:"true"`
+	Balance      int          `json:"balance" private:"true"`
+	Avatar       UserAvatar   `json:"avatar"`
+	Cover        UserCover    `json:"cover"`
+	Accounts     UserAccounts `json:"accounts" private:"true"`
+	Browser      UserBrowser  `json:"browser" private:"true"`
+	OS           UserOS       `json:"os" private:"true"`
+	Location     *Location    `json:"location" private:"true"`
+
+	hasPosts
+
+	eventStreams struct {
+		sync.Mutex
+		value []*aero.EventStream
+	}
+}
+
+// NewUser creates an empty user object with a unique ID.
+func NewUser() *User {
+	user := &User{
+		ID: GenerateID("User"),
+
+		// Avoid nil value fields
+		Location: &Location{},
+	}
+
+	return user
+}
+
+// RegisterUser registers a new user in the database and sets up all the required references.
+func RegisterUser(user *User) {
+	user.Registered = DateTimeUTC()
+	user.LastLogin = user.Registered
+	user.LastSeen = user.Registered
+
+	// Save nick in NickToUser collection
+	DB.Set("NickToUser", user.Nick, &NickToUser{
+		Nick:   user.Nick,
+		UserID: user.ID,
+	})
+
+	// Save email in EmailToUser collection
+	if user.Email != "" {
+		DB.Set("EmailToUser", user.Email, &EmailToUser{
+			Email:  user.Email,
+			UserID: user.ID,
+		})
+	}
+
+	// Create default settings
+	NewSettings(user).Save()
+
+	// Add empty anime list
+	DB.Set("AnimeList", user.ID, &AnimeList{
+		UserID: user.ID,
+		Items:  []*AnimeListItem{},
+	})
+
+	// Add empty inventory
+	NewInventory(user.ID).Save()
+
+	// Add draft index
+	NewDraftIndex(user.ID).Save()
+
+	// Add empty push subscriptions
+	DB.Set("PushSubscriptions", user.ID, &PushSubscriptions{
+		UserID: user.ID,
+		Items:  []*PushSubscription{},
+	})
+
+	// Add empty follow list
+	NewUserFollows(user.ID).Save()
+
+	// Add empty notifications list
+	NewUserNotifications(user.ID).Save()
+
+	// Fetch gravatar
+	if user.Email != "" && !IsDevelopment() {
+		gravatarURL := gravatar.Url(user.Email) + "?s=" + fmt.Sprint(AvatarMaxSize) + "&d=404&r=pg"
+		gravatarURL = strings.Replace(gravatarURL, "http://", "https://", 1)
+
+		response, err := client.Get(gravatarURL).End()
+
+		if err == nil && response.StatusCode() == http.StatusOK {
+			data := response.Bytes()
+			err = user.SetAvatarBytes(data)
+
+			if err != nil {
+				fmt.Println(err)
+			}
+		}
+	}
+}
+
+// SendNotification accepts a PushNotification and generates a new Notification object.
+// The notification is then sent to all registered push devices.
+func (user *User) SendNotification(pushNotification *PushNotification) {
+	// Don't ever send notifications in development mode
+	if IsDevelopment() && user.ID != "4J6qpK1ve" {
+		return
+	}
+
+	// Save notification in database
+	notification := NewNotification(user.ID, pushNotification)
+	notification.Save()
+
+	userNotifications := user.Notifications()
+	err := userNotifications.Add(notification.ID)
+
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+
+	userNotifications.Save()
+
+	// Send push notification
+	subs := user.PushSubscriptions()
+	expired := []*PushSubscription{}
+
+	for _, sub := range subs.Items {
+		resp, err := sub.SendNotification(pushNotification)
+
+		if resp != nil && resp.StatusCode == http.StatusGone {
+			expired = append(expired, sub)
+			continue
+		}
+
+		// Print errors
+		if err != nil {
+			fmt.Println(err)
+			continue
+		}
+
+		// Print bad status codes
+		if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusCreated {
+			body, _ := ioutil.ReadAll(resp.Body)
+			fmt.Println(resp.StatusCode, string(body))
+			continue
+		}
+
+		sub.LastSuccess = DateTimeUTC()
+	}
+
+	// Remove expired items
+	if len(expired) > 0 {
+		for _, sub := range expired {
+			subs.Remove(sub.ID())
+		}
+	}
+
+	// Save changes
+	subs.Save()
+
+	// Send an event to the user's open tabs
+	user.BroadcastEvent(&aero.Event{
+		Name: "notificationCount",
+		Data: userNotifications.CountUnseen(),
+	})
+}
+
+// RealName returns the real name of the user.
+func (user *User) RealName() string {
+	if user.LastName == "" {
+		return user.FirstName
+	}
+
+	if user.FirstName == "" {
+		return user.LastName
+	}
+
+	return user.FirstName + " " + user.LastName
+}
+
+// RegisteredTime returns the time the user registered his account.
+func (user *User) RegisteredTime() time.Time {
+	reg, _ := time.Parse(time.RFC3339, user.Registered)
+	return reg
+}
+
+// LastSeenTime returns the time the user was last seen on the site.
+func (user *User) LastSeenTime() time.Time {
+	lastSeen, _ := time.Parse(time.RFC3339, user.LastSeen)
+	return lastSeen
+}
+
+// AgeInYears returns the user's age in years.
+func (user *User) AgeInYears() int {
+	return AgeInYears(user.BirthDay)
+}
+
+// IsActive tells you whether the user is active.
+func (user *User) IsActive() bool {
+	lastSeen, _ := time.Parse(time.RFC3339, user.LastSeen)
+	twoWeeksAgo := time.Now().Add(-14 * 24 * time.Hour)
+
+	if lastSeen.Unix() < twoWeeksAgo.Unix() {
+		return false
+	}
+
+	if len(user.AnimeList().Items) == 0 {
+		return false
+	}
+
+	return true
+}
+
+// IsPro returns whether the user is a PRO user or not.
+func (user *User) IsPro() bool {
+	if user.ProExpires == "" {
+		return false
+	}
+
+	return DateTimeUTC() < user.ProExpires
+}
+
+// ExtendProDuration extends the PRO account duration by the given duration.
+func (user *User) ExtendProDuration(duration time.Duration) {
+	now := time.Now().UTC()
+	expires, _ := time.Parse(time.RFC3339, user.ProExpires)
+
+	// If the user never had a PRO account yet,
+	// or if it already expired,
+	// use the current time as the start time.
+	if user.ProExpires == "" || now.Unix() > expires.Unix() {
+		expires = now
+	}
+
+	user.ProExpires = expires.Add(duration).Format(time.RFC3339)
+}
+
+// TimeSinceRegistered returns the duration since the user registered his account.
+func (user *User) TimeSinceRegistered() time.Duration {
+	registered, _ := time.Parse(time.RFC3339, user.Registered)
+	return time.Since(registered)
+}
+
+// HasNick returns whether the user has a custom nickname.
+func (user *User) HasNick() bool {
+	return !strings.HasPrefix(user.Nick, "g") && !strings.HasPrefix(user.Nick, "fb") && !strings.HasPrefix(user.Nick, "t") && user.Nick != ""
+}
+
+// WebsiteURL adds https:// to the URL.
+func (user *User) WebsiteURL() string {
+	return "https://" + user.WebsiteShortURL()
+}
+
+// WebsiteShortURL returns the user website without the protocol.
+func (user *User) WebsiteShortURL() string {
+	return strings.Replace(strings.Replace(user.Website, "https://", "", 1), "http://", "", 1)
+}
+
+// Link returns the URI to the user page.
+func (user *User) Link() string {
+	return "/+" + user.Nick
+}
+
+// GetID returns the ID.
+func (user *User) GetID() string {
+	return user.ID
+}
+
+// HasAvatar tells you whether the user has an avatar or not.
+func (user *User) HasAvatar() bool {
+	return user.Avatar.Extension != ""
+}
+
+// AvatarLink returns the URL to the user avatar.
+// Expects "small" (50 x 50) or "large" (560 x 560) for the size parameter.
+func (user *User) AvatarLink(size string) string {
+	if user.HasAvatar() {
+		return fmt.Sprintf("//%s/images/avatars/%s/%s%s?%v", MediaHost, size, user.ID, user.Avatar.Extension, user.Avatar.LastModified)
+	}
+
+	return fmt.Sprintf("//%s/images/elements/no-avatar.svg", MediaHost)
+}
+
+// CoverLink ...
+func (user *User) CoverLink(size string) string {
+	if user.Cover.Extension != "" && user.IsPro() {
+		return fmt.Sprintf("//%s/images/covers/%s/%s%s?%v", MediaHost, size, user.ID, user.Cover.Extension, user.Cover.LastModified)
+	}
+
+	return "/images/elements/default-cover.jpg"
+}
+
+// Gravatar returns the URL to the gravatar if an email has been registered.
+func (user *User) Gravatar() string {
+	if user.Email == "" {
+		return ""
+	}
+
+	return gravatar.SecureUrl(user.Email) + "?s=" + fmt.Sprint(AvatarMaxSize)
+}
+
+// EditorScore returns the editor score.
+func (user *User) EditorScore() int {
+	ignoreDifferences := FilterIgnoreAnimeDifferences(func(entry *IgnoreAnimeDifference) bool {
+		return entry.CreatedBy == user.ID
+	})
+
+	score := len(ignoreDifferences) * IgnoreAnimeDifferenceEditorScore
+
+	logEntries := FilterEditLogEntries(func(entry *EditLogEntry) bool {
+		return entry.UserID == user.ID
+	})
+
+	for _, entry := range logEntries {
+		score += entry.EditorScore()
+	}
+
+	return score
+}
+
+// ActivateItemEffect activates an item in the user inventory by the given item ID.
+func (user *User) ActivateItemEffect(itemID string) error {
+	month := 30 * 24 * time.Hour
+
+	switch itemID {
+	case "pro-account-1":
+		user.ExtendProDuration(1 * month)
+		user.Save()
+		return nil
+
+	case "pro-account-3":
+		user.ExtendProDuration(3 * month)
+		user.Save()
+		return nil
+
+	case "pro-account-6":
+		user.ExtendProDuration(6 * month)
+		user.Save()
+		return nil
+
+	case "pro-account-12":
+		user.ExtendProDuration(12 * month)
+		user.Save()
+		return nil
+
+	case "pro-account-24":
+		user.ExtendProDuration(24 * month)
+		user.Save()
+		return nil
+
+	default:
+		return errors.New("Can't activate unknown item: " + itemID)
+	}
+}
+
+// SetNick changes the user's nickname safely.
+func (user *User) SetNick(newName string) error {
+	setNickMutex.Lock()
+	defer setNickMutex.Unlock()
+
+	newName = autocorrect.UserNick(newName)
+
+	if !validate.Nick(newName) {
+		return errors.New("Invalid nickname")
+	}
+
+	if newName == user.Nick {
+		return nil
+	}
+
+	// Make sure the nickname doesn't exist already
+	_, err := GetUserByNick(newName)
+
+	// If there was no error: the username exists.
+	// If "not found" is not included in the error message it's a different error type.
+	if err == nil || !strings.Contains(err.Error(), "not found") {
+		return errors.New("Username '" + newName + "' is taken already")
+	}
+
+	user.ForceSetNick(newName)
+	return nil
+}
+
+// ForceSetNick forces a nickname overwrite.
+func (user *User) ForceSetNick(newName string) {
+	// Delete old nick reference
+	DB.Delete("NickToUser", user.Nick)
+
+	// Set new nick
+	user.Nick = newName
+
+	DB.Set("NickToUser", user.Nick, &NickToUser{
+		Nick:   user.Nick,
+		UserID: user.ID,
+	})
+}
+
+// CleanNick only returns the nickname if it was set by user, otherwise empty string.
+func (user *User) CleanNick() string {
+	if user.HasNick() {
+		return user.Nick
+	}
+
+	return ""
+}
+
+// Creator needs to be implemented for the PostParent interface.
+func (user *User) Creator() *User {
+	return user
+}
+
+// CreatorID needs to be implemented for the PostParent interface.
+func (user *User) CreatorID() UserID {
+	return user.ID
+}
+
+// TitleByUser returns the username.
+func (user *User) TitleByUser(viewUser *User) string {
+	return user.Nick
+}
+
+// SetEmail changes the user's email safely.
+func (user *User) SetEmail(newEmail string) error {
+	setEmailMutex.Lock()
+	defer setEmailMutex.Unlock()
+
+	if !validate.Email(newEmail) {
+		return errors.New("Invalid email address")
+	}
+
+	// Delete old email reference
+	DB.Delete("EmailToUser", user.Email)
+
+	// Set new email
+	user.Email = newEmail
+
+	DB.Set("EmailToUser", user.Email, &EmailToUser{
+		Email:  user.Email,
+		UserID: user.ID,
+	})
+
+	return nil
+}
+
+// HasBasicInfo returns true if the user has a username, an avatar and an introduction.
+func (user *User) HasBasicInfo() bool {
+	return user.HasAvatar() && user.HasNick() && user.Introduction != ""
+}
+
+// TypeName returns the type name.
+func (user *User) TypeName() string {
+	return "User"
+}
+
+// Self returns the object itself.
+func (user *User) Self() Loggable {
+	return user
+}
+
+// RefreshOsuInfo refreshes a user's Osu information.
+func (user *User) RefreshOsuInfo() error {
+	if user.Accounts.Osu.Nick == "" {
+		return nil
+	}
+
+	osu, err := osu.GetUser(user.Accounts.Osu.Nick)
+
+	if err != nil {
+		return err
+	}
+
+	user.Accounts.Osu.PP, _ = strconv.ParseFloat(osu.PPRaw, 64)
+	user.Accounts.Osu.Level, _ = strconv.ParseFloat(osu.Level, 64)
+	user.Accounts.Osu.Accuracy, _ = strconv.ParseFloat(osu.Accuracy, 64)
+
+	return nil
+}
+
+// RefreshFFXIVInfo refreshes a user's FFXIV information.
+func (user *User) RefreshFFXIVInfo() error {
+	if user.Accounts.FinalFantasyXIV.Nick == "" || user.Accounts.FinalFantasyXIV.Server == "" {
+		return nil
+	}
+
+	characterID, err := ffxiv.GetCharacterID(user.Accounts.FinalFantasyXIV.Nick, user.Accounts.FinalFantasyXIV.Server)
+
+	if err != nil {
+		return err
+	}
+
+	character, err := ffxiv.GetCharacter(characterID)
+
+	if err != nil {
+		return err
+	}
+
+	user.Accounts.FinalFantasyXIV.Class = character.Class
+	user.Accounts.FinalFantasyXIV.Level = character.Level
+	user.Accounts.FinalFantasyXIV.ItemLevel = character.ItemLevel
+
+	return nil
+}
+
+// RefreshOverwatchInfo refreshes a user's Overwatch information.
+func (user *User) RefreshOverwatchInfo() error {
+	if user.Accounts.Overwatch.BattleTag == "" {
+		return nil
+	}
+
+	stats, err := overwatch.GetPlayerStats(user.Accounts.Overwatch.BattleTag)
+
+	if err != nil {
+		return err
+	}
+
+	skillRating, tier := stats.HighestSkillRating()
+
+	// Only show career highest skill rating
+	if skillRating > user.Accounts.Overwatch.SkillRating {
+		user.Accounts.Overwatch.SkillRating = skillRating
+		user.Accounts.Overwatch.Tier = tier
+	}
+
+	return nil
+}