friends/nex/friends-wiiu/remove_friend.go
2023-11-12 06:48:13 -05:00

57 lines
1.8 KiB
Go

package nex_friends_wiiu
import (
"github.com/PretendoNetwork/friends/database"
database_wiiu "github.com/PretendoNetwork/friends/database/wiiu"
"github.com/PretendoNetwork/friends/globals"
notifications_wiiu "github.com/PretendoNetwork/friends/notifications/wiiu"
nex "github.com/PretendoNetwork/nex-go"
friends_wiiu "github.com/PretendoNetwork/nex-protocols-go/friends-wiiu"
)
func RemoveFriend(err error, packet nex.PacketInterface, callID uint32, pid uint32) uint32 {
if err != nil {
globals.Logger.Error(err.Error())
return nex.Errors.FPD.InvalidArgument
}
client := packet.Sender().(*nex.PRUDPClient)
err = database_wiiu.RemoveFriendship(client.PID(), pid)
if err != nil {
if err == database.ErrFriendshipNotFound {
return nex.Errors.FPD.NotInMyFriendList
} else {
globals.Logger.Critical(err.Error())
return nex.Errors.FPD.Unknown
}
}
connectedUser := globals.ConnectedUsers[pid]
if connectedUser != nil {
go notifications_wiiu.SendFriendshipRemoved(connectedUser.Client, pid)
}
rmcResponse := nex.NewRMCSuccess(nil)
rmcResponse.ProtocolID = friends_wiiu.ProtocolID
rmcResponse.MethodID = friends_wiiu.MethodRemoveFriend
rmcResponse.CallID = callID
rmcResponseBytes := rmcResponse.Bytes()
responsePacket, _ := nex.NewPRUDPPacketV0(client, nil)
responsePacket.SetType(nex.DataPacket)
responsePacket.AddFlag(nex.FlagNeedsAck)
responsePacket.AddFlag(nex.FlagReliable)
responsePacket.SetSourceStreamType(packet.(nex.PRUDPPacketInterface).DestinationStreamType())
responsePacket.SetSourcePort(packet.(nex.PRUDPPacketInterface).DestinationPort())
responsePacket.SetDestinationStreamType(packet.(nex.PRUDPPacketInterface).SourceStreamType())
responsePacket.SetDestinationPort(packet.(nex.PRUDPPacketInterface).SourcePort())
responsePacket.SetPayload(rmcResponseBytes)
globals.SecureServer.Send(responsePacket)
return 0
}