mirror of
https://github.com/PretendoNetwork/friends.git
synced 2025-04-02 11:01:46 -04:00
94 lines
3.2 KiB
Go
94 lines
3.2 KiB
Go
package nex_friends_3ds
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/PretendoNetwork/friends/database"
|
|
database_3ds "github.com/PretendoNetwork/friends/database/3ds"
|
|
"github.com/PretendoNetwork/friends/globals"
|
|
notifications_3ds "github.com/PretendoNetwork/friends/notifications/3ds"
|
|
nex "github.com/PretendoNetwork/nex-go/v2"
|
|
"github.com/PretendoNetwork/nex-go/v2/types"
|
|
friends_3ds "github.com/PretendoNetwork/nex-protocols-go/v2/friends-3ds"
|
|
friends_3ds_types "github.com/PretendoNetwork/nex-protocols-go/v2/friends-3ds/types"
|
|
)
|
|
|
|
func SyncFriend(err error, packet nex.PacketInterface, callID uint32, lfc types.UInt64, pids types.List[types.PID], lfcList types.List[types.UInt64]) (*nex.RMCMessage, *nex.Error) {
|
|
if err != nil {
|
|
globals.Logger.Error(err.Error())
|
|
return nil, nex.NewError(nex.ResultCodes.FPD.InvalidArgument, "") // TODO - Add error message
|
|
}
|
|
|
|
connection := packet.Sender().(*nex.PRUDPConnection)
|
|
|
|
friendRelationships, err := database_3ds.GetUserFriends(uint32(connection.PID()))
|
|
if err != nil && err != sql.ErrNoRows {
|
|
globals.Logger.Critical(err.Error())
|
|
return nil, nex.NewError(nex.ResultCodes.FPD.Unknown, "") // TODO - Add error message
|
|
}
|
|
|
|
for _, relationship := range friendRelationships {
|
|
var hasPID bool
|
|
for _, pid := range pids {
|
|
if pid == relationship.PID {
|
|
hasPID = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !hasPID {
|
|
err := database_3ds.RemoveFriendship(uint32(connection.PID()), uint32(relationship.PID))
|
|
if err != nil && err != database.ErrFriendshipNotFound {
|
|
globals.Logger.Critical(err.Error())
|
|
return nil, nex.NewError(nex.ResultCodes.FPD.Unknown, "") // TODO - Add error message
|
|
}
|
|
}
|
|
}
|
|
|
|
// TODO - Not needed?
|
|
relationships := friendRelationships.Copy().(types.List[friends_3ds_types.FriendRelationship])
|
|
|
|
for _, pid := range pids {
|
|
if !isPIDInRelationships(relationships, uint32(pid)) {
|
|
relationship, err := database_3ds.SaveFriendship(uint32(connection.PID()), uint32(pid))
|
|
if err != nil {
|
|
globals.Logger.Critical(err.Error())
|
|
return nil, nex.NewError(nex.ResultCodes.FPD.Unknown, "") // TODO - Add error message
|
|
}
|
|
|
|
relationships = append(relationships, relationship)
|
|
|
|
// * Alert the other side, in case they weren't able to get our presence data
|
|
connectedUser, ok := globals.ConnectedUsers.Get(uint32(pid))
|
|
if ok && connectedUser != nil {
|
|
go notifications_3ds.SendFriendshipCompleted(connectedUser.Connection, connection.PID())
|
|
}
|
|
}
|
|
}
|
|
|
|
// TODO - Not needed?
|
|
syncedRelationships := relationships.Copy().(types.List[friends_3ds_types.FriendRelationship])
|
|
|
|
rmcResponseStream := nex.NewByteStreamOut(globals.SecureEndpoint.LibraryVersions(), globals.SecureEndpoint.ByteStreamSettings())
|
|
|
|
syncedRelationships.WriteTo(rmcResponseStream)
|
|
|
|
rmcResponseBody := rmcResponseStream.Bytes()
|
|
|
|
rmcResponse := nex.NewRMCSuccess(globals.SecureEndpoint, rmcResponseBody)
|
|
rmcResponse.ProtocolID = friends_3ds.ProtocolID
|
|
rmcResponse.MethodID = friends_3ds.MethodSyncFriend
|
|
rmcResponse.CallID = callID
|
|
|
|
return rmcResponse, nil
|
|
}
|
|
|
|
func isPIDInRelationships(relationships []friends_3ds_types.FriendRelationship, pid uint32) bool {
|
|
for i := range relationships {
|
|
if uint32(relationships[i].PID) == pid {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|