Passport/pkg/services/friendships.go

package services

import (
	"errors"
	"fmt"
	"git.solsynth.dev/hydrogen/identity/pkg/database"
	"git.solsynth.dev/hydrogen/identity/pkg/models"
	"gorm.io/gorm"
)

func ListFriend(anyside models.Account, status models.FriendshipStatus) ([]models.AccountFriendship, error) {
	var relationships []models.AccountFriendship
	if err := database.C.
		Where(&models.AccountFriendship{Status: status}).
		Where(&models.AccountFriendship{AccountID: anyside.ID}).
		Or(&models.AccountFriendship{RelatedID: anyside.ID}).
		Preload("Account").
		Preload("Related").
		Find(&relationships).Error; err != nil {
		return relationships, err
	}

	return relationships, nil
}

func GetFriend(anysideId uint) (models.AccountFriendship, error) {
	var relationship models.AccountFriendship
	if err := database.C.
		Where(&models.AccountFriendship{AccountID: anysideId}).
		Or(&models.AccountFriendship{RelatedID: anysideId}).
		Preload("Account").
		Preload("Related").
		First(&relationship).Error; err != nil {
		return relationship, err
	}

	return relationship, nil
}

func GetFriendWithTwoSides(userId, relatedId uint, noPreload ...bool) (models.AccountFriendship, error) {
	var tx *gorm.DB
	if len(noPreload) > 0 && noPreload[0] {
		tx = database.C
	} else {
		tx = database.C.Preload("Account").Preload("Related")
	}

	var relationship models.AccountFriendship
	if err := tx.
		Where(&models.AccountFriendship{AccountID: userId, RelatedID: relatedId}).
		Or(&models.AccountFriendship{RelatedID: userId, AccountID: relatedId}).
		First(&relationship).Error; err != nil {
		return relationship, err
	}

	return relationship, nil
}

func NewFriend(user models.Account, related models.Account, status models.FriendshipStatus) (models.AccountFriendship, error) {
	relationship := models.AccountFriendship{
		AccountID: user.ID,
		RelatedID: related.ID,
		Status:    status,
	}

	if user.ID == related.ID {
		return relationship, fmt.Errorf("you cannot make friendship with yourself")
	} else if _, err := GetFriendWithTwoSides(user.ID, related.ID, true); err == nil || !errors.Is(err, gorm.ErrRecordNotFound) {
		return relationship, fmt.Errorf("you already have a friendship with him or her")
	}

	if err := database.C.Save(&relationship).Error; err != nil {
		return relationship, err
	} else {
		_ = NewNotification(models.Notification{
			Subject:     fmt.Sprintf("New friend request from %s", user.Name),
			Content:     fmt.Sprintf("You got a new friend request from %s. Go to your settings and decide how to deal it.", user.Nick),
			RecipientID: related.ID,
		})
	}

	return relationship, nil
}

func EditFriend(relationship models.AccountFriendship) (models.AccountFriendship, error) {
	if err := database.C.Save(&relationship).Error; err != nil {
		return relationship, err
	}
	return relationship, nil
}

func DeleteFriend(relationship models.AccountFriendship) error {
	if err := database.C.Delete(&relationship).Error; err != nil {
		return err
	}
	return nil
}