notify.moe/arn/UserFollows.go

package arn

import (
	"errors"

	"github.com/aerogo/aero"
	"github.com/aerogo/api"
)

// Add adds an user to the user if it hasn't been added yet.
func (user *User) Follow(followUserID UserID) error {
	if followUserID == user.ID {
		return errors.New("You can't follow yourself")
	}

	if user.IsFollowing(followUserID) {
		return errors.New("User " + followUserID + " has already been added")
	}

	followedUser, err := GetUser(followUserID)

	if err != nil {
		return err
	}

	user.FollowIDs = append(user.FollowIDs, followUserID)

	// Send notification
	if !followedUser.Settings().Notification.NewFollowers {
		return nil
	}

	followedUser.SendNotification(&PushNotification{
		Title:   "You have a new follower!",
		Message: user.Nick + " started following you.",
		Icon:    "https:" + user.AvatarLink("large"),
		Link:    "https://notify.moe" + user.Link(),
		Type:    NotificationTypeFollow,
	})

	return nil
}

// Unfollow removes the user ID from the follow list.
func (user *User) Unfollow(userID UserID) bool {
	for index, item := range user.FollowIDs {
		if item == userID {
			user.FollowIDs = append(user.FollowIDs[:index], user.FollowIDs[index+1:]...)
			return true
		}
	}

	return false
}

// IsFollowing checks if the object follows the user ID.
func (user *User) IsFollowing(userID UserID) bool {
	if userID == user.ID {
		return true
	}

	for _, item := range user.FollowIDs {
		if item == userID {
			return true
		}
	}

	return false
}

// Follows returns a slice of all the users you are following.
func (user *User) Follows() []*User {
	followsObj := DB.GetMany("User", user.FollowIDs)
	follows := make([]*User, len(followsObj))

	for i, user := range followsObj {
		follows[i] = user.(*User)
	}

	return follows
}

// Friends returns a slice of all the users you are following that also follow you.
func (user *User) Friends() []*User {
	followsObj := DB.GetMany("User", user.FollowIDs)
	friends := make([]*User, 0, len(followsObj))

	for _, friendObj := range followsObj {
		friend := friendObj.(*User)

		if friend.IsFollowing(user.ID) {
			friends = append(friends, friend)
		}
	}

	return friends
}

// Followers returns the users who follow the user.
func (user *User) Followers() []*User {
	var followerIDs []string

	for follower := range StreamUsers() {
		if follower.IsFollowing(user.ID) {
			followerIDs = append(followerIDs, follower.ID)
		}
	}

	usersObj := DB.GetMany("User", followerIDs)
	users := make([]*User, len(usersObj))

	for i, obj := range usersObj {
		users[i] = obj.(*User)
	}

	return users
}

// FollowersCount returns how many followers the user has.
func (user *User) FollowersCount() int {
	count := 0

	for follower := range StreamUsers() {
		if follower.IsFollowing(user.ID) {
			count++
		}
	}

	return count
}

// UserFollowerCountMap returns a map of user ID keys and their corresping number of followers as the value.
func UserFollowerCountMap() map[string]int {
	followCount := map[string]int{}

	for user := range StreamUsers() {
		for _, followUserID := range user.FollowIDs {
			followCount[followUserID]++
		}
	}

	return followCount
}

// FollowAction returns an API action that adds a user ID to the follow list.
func FollowAction() *api.Action {
	return &api.Action{
		Name:  "follow",
		Route: "/follow/:follow-id",
		Run: func(obj interface{}, ctx aero.Context) error {
			user := obj.(*User)
			followID := ctx.Get("follow-id")
			err := user.Follow(followID)

			if err != nil {
				return err
			}

			user.Save()
			return nil
		},
	}
}

// UnfollowAction returns an API action that removes a user ID from the follow list.
func UnfollowAction() *api.Action {
	return &api.Action{
		Name:  "unfollow",
		Route: "/unfollow/:unfollow-id",
		Run: func(obj interface{}, ctx aero.Context) error {
			user := obj.(*User)
			unfollowID := ctx.Get("unfollow-id")

			if !user.Unfollow(unfollowID) {
				return errors.New("This item does not exist in the list")
			}

			user.Save()
			return nil
		},
	}
}