From 821ae1e537cd3ae3ec47bc1cfa3d041e1ca1b3a5 Mon Sep 17 00:00:00 2001
From: mkwcat <cat@mkw.re>
Date: Tue, 12 Dec 2023 08:46:14 -0500
Subject: [PATCH] GPCM: Send pending friend requests on login

It will now show your friends' statuses immediately on login
---
 gpcm/friend.go | 30 +++++++++++++++++++++++++++++-
 gpcm/login.go  |  2 ++
 2 files changed, 31 insertions(+), 1 deletion(-)

diff --git a/gpcm/friend.go b/gpcm/friend.go
index 730d50b..c541e7a 100644
--- a/gpcm/friend.go
+++ b/gpcm/friend.go
@@ -29,6 +29,8 @@ func (g *GameSpySession) isFriendAuthorized(profileId uint32) bool {
 	return false
 }
 
+const addFriendMessage = "\r\n\r\n|signed|00000000000000000000000000000000"
+
 func (g *GameSpySession) addFriend(command common.GameSpyCommand) {
 	strNewProfileId := command.OtherValues["newprofileid"]
 	newProfileId, err := strconv.ParseUint(strNewProfileId, 10, 32)
@@ -83,8 +85,34 @@ func (g *GameSpySession) addFriend(command common.GameSpyCommand) {
 	// Friends are now mutual!
 	// TODO: Add a limit
 	g.AuthFriendList = append(g.AuthFriendList, uint32(newProfileId))
+	newSession.AuthFriendList = append(newSession.AuthFriendList, uint32(g.User.ProfileId))
 
-	sendMessageToProfileId("2", g.User.ProfileId, uint32(newProfileId), "\r\n\r\n|signed|"+common.RandomHexString(32))
+	sendMessageToProfileId("2", g.User.ProfileId, uint32(newProfileId), addFriendMessage)
+}
+
+func (g *GameSpySession) sendFriendRequests() {
+	mutex.Lock()
+	defer mutex.Unlock()
+
+	// Condense all requests into one packet
+	var message string
+
+	for _, newSession := range sessions {
+		if newSession.isFriendAdded(g.User.ProfileId) {
+			message += common.CreateGameSpyMessage(common.GameSpyCommand{
+				Command:      "bm",
+				CommandValue: "2",
+				OtherValues: map[string]string{
+					"f":   strconv.FormatUint(uint64(newSession.User.ProfileId), 10),
+					"msg": addFriendMessage,
+				},
+			})
+		}
+	}
+
+	if message != "" {
+		g.Conn.Write([]byte(message))
+	}
 }
 
 func (g *GameSpySession) removeFriend(command common.GameSpyCommand) {
diff --git a/gpcm/login.go b/gpcm/login.go
index 88d0859..83ab989 100644
--- a/gpcm/login.go
+++ b/gpcm/login.go
@@ -203,6 +203,8 @@ func (g *GameSpySession) login(command common.GameSpyCommand) {
 	})
 
 	g.Conn.Write([]byte(payload))
+
+	g.sendFriendRequests()
 }
 
 func IsLoggedIn(profileID uint32) bool {