Added arn to the main repository

arn/UserHelper.go

@@ -0,0 +1,166 @@
+package arn
+
+import (
+	"errors"
+	"sort"
+
+	"github.com/aerogo/nano"
+)
+
+// GetUser fetches the user with the given ID from the database.
+func GetUser(id UserID) (*User, error) {
+	obj, err := DB.Get("User", id)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return obj.(*User), nil
+}
+
+// GetUserByNick fetches the user with the given nick from the database.
+func GetUserByNick(nick string) (*User, error) {
+	obj, err := DB.Get("NickToUser", nick)
+
+	if err != nil {
+		return nil, err
+	}
+
+	userID := obj.(*NickToUser).UserID
+	user, err := GetUser(userID)
+
+	return user, err
+}
+
+// GetUserByEmail fetches the user with the given email from the database.
+func GetUserByEmail(email string) (*User, error) {
+	if email == "" {
+		return nil, errors.New("Email is empty")
+	}
+
+	obj, err := DB.Get("EmailToUser", email)
+
+	if err != nil {
+		return nil, err
+	}
+
+	userID := obj.(*EmailToUser).UserID
+	user, err := GetUser(userID)
+
+	return user, err
+}
+
+// GetUserByFacebookID fetches the user with the given Facebook ID from the database.
+func GetUserByFacebookID(facebookID string) (*User, error) {
+	obj, err := DB.Get("FacebookToUser", facebookID)
+
+	if err != nil {
+		return nil, err
+	}
+
+	userID := obj.(*FacebookToUser).UserID
+	user, err := GetUser(userID)
+
+	return user, err
+}
+
+// GetUserByTwitterID fetches the user with the given Twitter ID from the database.
+func GetUserByTwitterID(twitterID string) (*User, error) {
+	obj, err := DB.Get("TwitterToUser", twitterID)
+
+	if err != nil {
+		return nil, err
+	}
+
+	userID := obj.(*TwitterToUser).UserID
+	user, err := GetUser(userID)
+
+	return user, err
+}
+
+// GetUserByGoogleID fetches the user with the given Google ID from the database.
+func GetUserByGoogleID(googleID string) (*User, error) {
+	obj, err := DB.Get("GoogleToUser", googleID)
+
+	if err != nil {
+		return nil, err
+	}
+
+	userID := obj.(*GoogleToUser).UserID
+	user, err := GetUser(userID)
+
+	return user, err
+}
+
+// StreamUsers returns a stream of all users.
+func StreamUsers() <-chan *User {
+	channel := make(chan *User, nano.ChannelBufferSize)
+
+	go func() {
+		for obj := range DB.All("User") {
+			channel <- obj.(*User)
+		}
+
+		close(channel)
+	}()
+
+	return channel
+}
+
+// AllUsers returns a slice of all users.
+func AllUsers() ([]*User, error) {
+	all := make([]*User, 0, DB.Collection("User").Count())
+
+	for obj := range StreamUsers() {
+		all = append(all, obj)
+	}
+
+	return all, nil
+}
+
+// FilterUsers filters all users by a custom function.
+func FilterUsers(filter func(*User) bool) []*User {
+	var filtered []*User
+
+	for obj := range StreamUsers() {
+		if filter(obj) {
+			filtered = append(filtered, obj)
+		}
+	}
+
+	return filtered
+}
+
+// SortUsersLastSeenFirst sorts a list of users by their last seen date.
+func SortUsersLastSeenFirst(users []*User) {
+	sort.Slice(users, func(i, j int) bool {
+		return users[i].LastSeen > users[j].LastSeen
+	})
+}
+
+// SortUsersLastSeenLast sorts a list of users by their last seen date.
+func SortUsersLastSeenLast(users []*User) {
+	sort.Slice(users, func(i, j int) bool {
+		return users[i].LastSeen < users[j].LastSeen
+	})
+}
+
+// SortUsersFollowers sorts a list of users by their number of followers.
+func SortUsersFollowers(users []*User) {
+	followCount := UserFollowerCountMap()
+
+	sort.Slice(users, func(i, j int) bool {
+		if users[i].HasAvatar() != users[j].HasAvatar() {
+			return users[i].HasAvatar()
+		}
+
+		followersA := followCount[users[i].ID]
+		followersB := followCount[users[j].ID]
+
+		if followersA == followersB {
+			return users[i].Nick < users[j].Nick
+		}
+
+		return followersA > followersB
+	})
+}