169 lines
3.5 KiB
Go
Raw Normal View History

2019-06-03 09:32:43 +00:00
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.
2019-08-27 00:39:35 +00:00
func SortUsersFollowers(users []*User) map[string]int {
2019-06-03 09:32:43 +00:00
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
})
2019-08-27 00:39:35 +00:00
return followCount
2019-06-03 09:32:43 +00:00
}