mirror of
https://github.com/PretendoNetwork/friends.git
synced 2026-04-26 07:46:54 -05:00
8d8a05a7e2
Now database getters do error handling aswell, and in case of any error, the NEX or GRPC method throws a proper error about it. There are still some doubts on how to handle errors when a list of data is processed, so for now skip that element on the list and continue. Also add some constant errors to do better error handling.
64 lines
1.8 KiB
Go
64 lines
1.8 KiB
Go
package nex_friends_wiiu
|
|
|
|
import (
|
|
database_wiiu "github.com/PretendoNetwork/friends/database/wiiu"
|
|
"github.com/PretendoNetwork/friends/globals"
|
|
nex "github.com/PretendoNetwork/nex-go"
|
|
friends_wiiu "github.com/PretendoNetwork/nex-protocols-go/friends-wiiu"
|
|
friends_wiiu_types "github.com/PretendoNetwork/nex-protocols-go/friends-wiiu/types"
|
|
)
|
|
|
|
func GetRequestBlockSettings(err error, client *nex.Client, callID uint32, pids []uint32) uint32 {
|
|
if err != nil {
|
|
globals.Logger.Error(err.Error())
|
|
return nex.Errors.FPD.InvalidArgument
|
|
}
|
|
|
|
settings := make([]*friends_wiiu_types.PrincipalRequestBlockSetting, 0)
|
|
|
|
// TODO:
|
|
// Improve this. Use less database_wiiu.reads
|
|
for i := 0; i < len(pids); i++ {
|
|
requestedPID := pids[i]
|
|
|
|
setting := friends_wiiu_types.NewPrincipalRequestBlockSetting()
|
|
setting.PID = requestedPID
|
|
isBlocked, err := database_wiiu.IsFriendRequestBlocked(client.PID(), requestedPID)
|
|
if err != nil {
|
|
globals.Logger.Critical(err.Error())
|
|
continue
|
|
}
|
|
|
|
setting.IsBlocked = isBlocked
|
|
|
|
settings = append(settings, setting)
|
|
}
|
|
|
|
rmcResponseStream := nex.NewStreamOut(globals.SecureServer)
|
|
|
|
rmcResponseStream.WriteListStructure(settings)
|
|
|
|
rmcResponseBody := rmcResponseStream.Bytes()
|
|
|
|
// Build response packet
|
|
rmcResponse := nex.NewRMCResponse(friends_wiiu.ProtocolID, callID)
|
|
rmcResponse.SetSuccess(friends_wiiu.MethodGetRequestBlockSettings, rmcResponseBody)
|
|
|
|
rmcResponseBytes := rmcResponse.Bytes()
|
|
|
|
responsePacket, _ := nex.NewPacketV0(client, nil)
|
|
|
|
responsePacket.SetVersion(0)
|
|
responsePacket.SetSource(0xA1)
|
|
responsePacket.SetDestination(0xAF)
|
|
responsePacket.SetType(nex.DataPacket)
|
|
responsePacket.SetPayload(rmcResponseBytes)
|
|
|
|
responsePacket.AddFlag(nex.FlagNeedsAck)
|
|
responsePacket.AddFlag(nex.FlagReliable)
|
|
|
|
globals.SecureServer.Send(responsePacket)
|
|
|
|
return 0
|
|
}
|