From 116a19631ea3d8f6ced2ec495bae17c0de166d45 Mon Sep 17 00:00:00 2001
From: light <57009359+hauntii@users.noreply.github.com>
Date: Mon, 27 Mar 2023 01:25:51 -0400
Subject: [PATCH] Stop duplicate friend requests

---
 database/wiiu/save_friend_request.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/database/wiiu/save_friend_request.go b/database/wiiu/save_friend_request.go
index b322a8d..debae41 100644
--- a/database/wiiu/save_friend_request.go
+++ b/database/wiiu/save_friend_request.go
@@ -1,13 +1,25 @@
 package database_wiiu
 
 import (
+	"database/sql"
+
 	"github.com/PretendoNetwork/friends-secure/database"
 	"github.com/PretendoNetwork/friends-secure/globals"
 )
 
 func SaveFriendRequest(senderPID uint32, recipientPID uint32, sentTime uint64, expireTime uint64, message string) uint64 {
 	var id uint64
-	err := database.Postgres.QueryRow(`
+
+	// Make sure we don't already have that friend request! If we do, give them the one we already have.
+	err := database.Postgres.QueryRow(`SELECT id FROM wiiu.friend_requests WHERE sender_pid=$1 AND recipient_pid=$2`, senderPID, recipientPID).Scan(&id)
+	if err != nil && err != sql.ErrNoRows {
+		globals.Logger.Critical(err.Error())
+		return 0
+	} else if id != 0 {
+		return id
+	}
+
+	err = database.Postgres.QueryRow(`
 		INSERT INTO wiiu.friend_requests (sender_pid, recipient_pid, sent_on, expires_on, message, received, accepted, denied)
 		VALUES ($1, $2, $3, $4, $5, false, false, false) RETURNING id`, senderPID, recipientPID, sentTime, expireTime, message).Scan(&id)