mirror of
https://github.com/PretendoNetwork/splatoon.git
synced 2026-03-21 17:45:08 -05:00
ef521aea15
I would have liked to save the gathering info too but I couldn't work out how to extract that from MatchmakingManager you can kinda reverse-engineer it from the tracking logs anyway
114 lines
5.2 KiB
Go
114 lines
5.2 KiB
Go
package nex
|
|
|
|
import (
|
|
"github.com/PretendoNetwork/nex-go/v2"
|
|
"github.com/PretendoNetwork/nex-go/v2/types"
|
|
commonmatchmaking "github.com/PretendoNetwork/nex-protocols-common-go/v2/match-making"
|
|
commonmatchmakingext "github.com/PretendoNetwork/nex-protocols-common-go/v2/match-making-ext"
|
|
commonmatchmakeextension "github.com/PretendoNetwork/nex-protocols-common-go/v2/matchmake-extension"
|
|
commonnattraversal "github.com/PretendoNetwork/nex-protocols-common-go/v2/nat-traversal"
|
|
commonranking "github.com/PretendoNetwork/nex-protocols-common-go/v2/ranking"
|
|
commonsecure "github.com/PretendoNetwork/nex-protocols-common-go/v2/secure-connection"
|
|
matchmaking "github.com/PretendoNetwork/nex-protocols-go/v2/match-making"
|
|
matchmakingext "github.com/PretendoNetwork/nex-protocols-go/v2/match-making-ext"
|
|
match_making_types "github.com/PretendoNetwork/nex-protocols-go/v2/match-making/types"
|
|
matchmakeextension "github.com/PretendoNetwork/nex-protocols-go/v2/matchmake-extension"
|
|
nattraversal "github.com/PretendoNetwork/nex-protocols-go/v2/nat-traversal"
|
|
ranking "github.com/PretendoNetwork/nex-protocols-go/v2/ranking/splatoon"
|
|
secure "github.com/PretendoNetwork/nex-protocols-go/v2/secure-connection"
|
|
"github.com/PretendoNetwork/splatoon/globals"
|
|
)
|
|
|
|
// that's right boys it's a FACTORY METHOD IN GO LET'S GOOOO
|
|
func fCreateReportDBRecord() func(*types.PID, *types.PrimitiveU32, *types.QBuffer) error {
|
|
createReportStmt, err := globals.Postgres.Prepare(`INSERT INTO reports (ts, report_id, report) VALUES (NOW(), $1, $2)`)
|
|
if err != nil {
|
|
globals.Logger.Error(err.Error())
|
|
}
|
|
|
|
return func(pid *types.PID, id *types.PrimitiveU32, report *types.QBuffer) error {
|
|
if createReportStmt == nil {
|
|
return nil // Just drop the report in case of DB error - no need to freak out the client with a reported err
|
|
}
|
|
|
|
_, err := createReportStmt.Exec(id.Value, report.Value)
|
|
return err // consider not reporting this error to the client
|
|
}
|
|
}
|
|
|
|
func stubGetPlayingSession(err error, packet nex.PacketInterface, callID uint32, _ *types.List[*types.PID]) (*nex.RMCMessage, *nex.Error) {
|
|
if err != nil {
|
|
globals.Logger.Error(err.Error())
|
|
return nil, nex.NewError(nex.ResultCodes.Core.InvalidArgument, "change_error")
|
|
}
|
|
|
|
connection := packet.Sender().(*nex.PRUDPConnection)
|
|
endpoint := connection.Endpoint().(*nex.PRUDPEndPoint)
|
|
|
|
lstPlayingSession := types.NewList[*match_making_types.PlayingSession]()
|
|
|
|
rmcResponseStream := nex.NewByteStreamOut(endpoint.LibraryVersions(), endpoint.ByteStreamSettings())
|
|
|
|
lstPlayingSession.WriteTo(rmcResponseStream)
|
|
|
|
rmcResponseBody := rmcResponseStream.Bytes()
|
|
|
|
rmcResponse := nex.NewRMCSuccess(endpoint, rmcResponseBody)
|
|
rmcResponse.ProtocolID = matchmakeextension.ProtocolID
|
|
rmcResponse.MethodID = matchmakeextension.MethodGetSimplePlayingSession
|
|
rmcResponse.CallID = callID
|
|
|
|
return rmcResponse, nil
|
|
}
|
|
|
|
func cleanupMatchmakeSessionSearchCriteriasHandler(searchCriterias *types.List[*match_making_types.MatchmakeSessionSearchCriteria]) {
|
|
for _, searchCriteria := range searchCriterias.Slice() {
|
|
_ = searchCriteria.Attribs.SetIndex(4, types.NewString(""))
|
|
}
|
|
}
|
|
|
|
func onAfterAutoMatchmakeWithParamPostpone(_ nex.PacketInterface, _ *match_making_types.AutoMatchmakeParam) {
|
|
globals.MatchmakingManager.Mutex.Lock()
|
|
|
|
_, err := globals.MatchmakingManager.Database.Exec(`UPDATE matchmaking.matchmake_sessions SET open_participation=true WHERE game_mode=12`)
|
|
if err != nil {
|
|
globals.Logger.Error(err.Error())
|
|
}
|
|
|
|
globals.MatchmakingManager.Mutex.Unlock()
|
|
}
|
|
|
|
func registerCommonSecureServerProtocols() {
|
|
secureProtocol := secure.NewProtocol()
|
|
globals.SecureEndpoint.RegisterServiceProtocol(secureProtocol)
|
|
commonSecureProtocol := commonsecure.NewCommonProtocol(secureProtocol)
|
|
|
|
commonSecureProtocol.CreateReportDBRecord = fCreateReportDBRecord()
|
|
|
|
natTraversalProtocol := nattraversal.NewProtocol()
|
|
globals.SecureEndpoint.RegisterServiceProtocol(natTraversalProtocol)
|
|
commonnattraversal.NewCommonProtocol(natTraversalProtocol)
|
|
|
|
matchMakingProtocol := matchmaking.NewProtocol()
|
|
globals.SecureEndpoint.RegisterServiceProtocol(matchMakingProtocol)
|
|
commonMatchMakingProtocol := commonmatchmaking.NewCommonProtocol(matchMakingProtocol)
|
|
commonMatchMakingProtocol.SetManager(globals.MatchmakingManager)
|
|
|
|
matchMakingExtProtocol := matchmakingext.NewProtocol()
|
|
globals.SecureEndpoint.RegisterServiceProtocol(matchMakingExtProtocol)
|
|
commonMatchMakingExtProtocol := commonmatchmakingext.NewCommonProtocol(matchMakingExtProtocol)
|
|
commonMatchMakingExtProtocol.SetManager(globals.MatchmakingManager)
|
|
|
|
matchmakeExtensionProtocol := matchmakeextension.NewProtocol()
|
|
globals.SecureEndpoint.RegisterServiceProtocol(matchmakeExtensionProtocol)
|
|
commonMatchmakeExtensionProtocol := commonmatchmakeextension.NewCommonProtocol(matchmakeExtensionProtocol)
|
|
matchmakeExtensionProtocol.SetHandlerGetPlayingSession(stubGetPlayingSession)
|
|
commonMatchmakeExtensionProtocol.CleanupMatchmakeSessionSearchCriterias = cleanupMatchmakeSessionSearchCriteriasHandler
|
|
commonMatchmakeExtensionProtocol.OnAfterAutoMatchmakeWithParamPostpone = onAfterAutoMatchmakeWithParamPostpone
|
|
commonMatchmakeExtensionProtocol.SetManager(globals.MatchmakingManager)
|
|
|
|
rankingProtocol := ranking.NewProtocol(globals.SecureEndpoint)
|
|
globals.SecureEndpoint.RegisterServiceProtocol(rankingProtocol)
|
|
commonranking.NewCommonProtocol(rankingProtocol)
|
|
}
|