Passport/pkg/internal/services/relationships.go

package services

import (
	"errors"
	"fmt"

	"git.solsynth.dev/hydrogen/passport/pkg/internal/database"
	"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
	"github.com/samber/lo"
	"gorm.io/gorm"
)

func ListAllRelationship(user models.Account) ([]models.AccountRelationship, error) {
	var relationships []models.AccountRelationship
	if err := database.C.
		Where("account_id = ?", user.ID).
		Preload("Account").
		Preload("Related").
		Find(&relationships).Error; err != nil {
		return relationships, err
	}

	return relationships, nil
}

func ListRelationshipWithFilter(user models.Account, status models.RelationshipStatus) ([]models.AccountRelationship, error) {
	var relationships []models.AccountRelationship
	if err := database.C.
		Where("account_id = ? AND status = ?", user.ID, status).
		Preload("Account").
		Preload("Related").
		Find(&relationships).Error; err != nil {
		return relationships, err
	}

	return relationships, nil
}

func GetRelationship(otherId uint) (models.AccountRelationship, error) {
	var relationship models.AccountRelationship
	if err := database.C.
		Where(&models.AccountRelationship{AccountID: otherId}).
		Preload("Account").
		Preload("Related").
		First(&relationship).Error; err != nil {
		return relationship, err
	}

	return relationship, nil
}

func GetRelationWithTwoNode(userId, relatedId uint, noPreload ...bool) (models.AccountRelationship, 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.AccountRelationship
	if err := tx.
		Where(&models.AccountRelationship{AccountID: userId, RelatedID: relatedId}).
		First(&relationship).Error; err != nil {
		return relationship, err
	}

	return relationship, nil
}

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

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

func NewBlockship(userA models.Account, userB models.Account) (models.AccountRelationship, error) {
	var err error
	var rel models.AccountRelationship
	if rel, err = GetRelationWithTwoNode(userA.ID, userB.ID, true); err == nil {
		rel.Status = models.RelationshipBlocked
	} else {
		rel = models.AccountRelationship{
			AccountID: userA.ID,
			RelatedID: userB.ID,
			Status:    models.RelationshipBlocked,
		}
	}

	if err := database.C.Save(&rel).Error; err != nil {
		return rel, err
	}

	return rel, nil
}

func NewFriend(userA models.Account, userB models.Account, skipPending ...bool) (models.AccountRelationship, error) {
	relA := models.AccountRelationship{
		AccountID: userA.ID,
		RelatedID: userB.ID,
		Status:    models.RelationshipWaiting,
	}
	relB := models.AccountRelationship{
		AccountID: userB.ID,
		RelatedID: userA.ID,
		Status:    models.RelationshipPending,
	}

	if len(skipPending) > 0 && skipPending[0] {
		relA.Status = models.RelationshipFriend
		relB.Status = models.RelationshipFriend
	}

	if userA.ID == userB.ID {
		return relA, fmt.Errorf("unable to make relationship with yourself")
	} else if _, err := GetRelationWithTwoNode(userA.ID, userB.ID, true); err == nil || !errors.Is(err, gorm.ErrRecordNotFound) {
		return relA, fmt.Errorf("unable to recreate a relationship with that user")
	}

	if err := database.C.Save(&relA).Error; err != nil {
		return relA, err
	} else if err = database.C.Save(&relB).Error; err != nil {
		return relA, err
	} else {
		_ = NewNotification(models.Notification{
			Title:     "New Friend Request",
			Subtitle:  lo.ToPtr(fmt.Sprintf("New friend request from %s", userA.Name)),
			Body:      fmt.Sprintf("You got a new friend request from %s. Go to your account page and decide how to deal it.", userA.Nick),
			Account:   userB,
			AccountID: userB.ID,
		})
	}

	return relA, nil
}

func HandleFriend(userA models.Account, userB models.Account, isAccept bool) error {
	relA, err := GetRelationWithTwoNode(userA.ID, userB.ID, true)
	if err != nil {
		return fmt.Errorf("relationship was not found: %v", err)
	} else if relA.Status != models.RelationshipPending {
		return fmt.Errorf("relationship already handled")
	}

	if isAccept {
		relA.Status = models.RelationshipFriend
	} else {
		relA.Status = models.RelationshipBlocked
	}

	if err := database.C.Save(&relA).Error; err != nil {
		return err
	}

	relB, err := GetRelationWithTwoNode(userB.ID, userA.ID, true)
	if err == nil && relB.Status == models.RelationshipWaiting {
		relB.Status = models.RelationshipFriend
		if err := database.C.Save(&relB).Error; err != nil {
			return err
		}

		_ = NewNotification(models.Notification{
			Title:     "Friend Request Processed",
			Subtitle:  lo.ToPtr(fmt.Sprintf("Your friend request to %s has been processsed.", userA.Name)),
			Body:      fmt.Sprintf("Your relationship status with %s has been updated, go check it out!", userA.Nick),
			Account:   userB,
			AccountID: userB.ID,
		})
	}

	return nil
}