From b93ef24411478715270ed5a9675ad21548bf675d Mon Sep 17 00:00:00 2001
From: gamray <thomas.web13@gmail.com>
Date: Sat, 18 May 2024 13:01:34 +0200
Subject: [PATCH] added /account/friendsonline

---
 api/common.go    |  1 +
 api/endpoints.go | 28 ++++++++++++++++++++++++++++
 db/account.go    | 36 ++++++++++++++++++++++++++++++++++++
 defs/game.go     |  5 +++++
 4 files changed, 70 insertions(+)

diff --git a/api/common.go b/api/common.go
index d530b62..26953f8 100644
--- a/api/common.go
+++ b/api/common.go
@@ -44,6 +44,7 @@ func Init(mux *http.ServeMux) error {
 	mux.HandleFunc("GET /account/logout", handleAccountLogout)
 	mux.HandleFunc("POST /account/addfriend", handleAddFriend)
 	mux.HandleFunc("POST /account/removefriend", handleRemoveFriend)
+	mux.HandleFunc("GET /account/friendsonline", handleFriendsOnlineStat)
 
 	// game
 	mux.HandleFunc("GET /game/titlestats", handleGameTitleStats)
diff --git a/api/endpoints.go b/api/endpoints.go
index 4966897..de12aa0 100644
--- a/api/endpoints.go
+++ b/api/endpoints.go
@@ -186,6 +186,34 @@ func handleRemoveFriend(w http.ResponseWriter, r *http.Request) {
 	jsonResponse(w, r, response)
 }
 
+func handleFriendsOnlineStat(w http.ResponseWriter, r *http.Request) {
+	uuid, err := uuidFromRequest(r)
+	if err != nil {
+		httpError(w, r, err, http.StatusBadRequest)
+		return
+	}
+
+	friendsAmount, countErr := db.FriendCount(uuid)
+	if countErr != nil {
+		httpError(w, r, countErr, http.StatusInternalServerError)
+		return
+	}
+
+	friendsOnline, onlineErr := db.FriendOnlineCount(uuid)
+	if onlineErr != nil {
+		httpError(w, r, onlineErr, http.StatusInternalServerError)
+		return
+	}
+
+	stats := defs.FriendsOnlineStats{
+		FriendsOnline: friendsOnline,
+		FriendsAmount: friendsAmount,
+	}
+
+	jsonResponse(w, r, stats)
+}
+
+
 // game
 func handleGameTitleStats(w http.ResponseWriter, r *http.Request) {
 	stats := defs.TitleStats{
diff --git a/db/account.go b/db/account.go
index 0d1a32b..276a4c7 100644
--- a/db/account.go
+++ b/db/account.go
@@ -323,4 +323,40 @@ func RemoveFriend(uuid []byte, friendUsername string) (bool, error) {
 	}
 
 	return true, nil
+}
+
+func FriendCount(uuid []byte) (int, error) {
+	username, err := FetchUsernameFromUUID(uuid);
+	if err != nil {
+		return -1, err
+	}
+
+	var count int
+	err = handle.QueryRow("SELECT COUNT(*) FROM friends WHERE user = ?", username).Scan(&count)
+	if err != nil {
+		return -1, err
+	}
+
+	return count, nil
+}
+
+func FriendOnlineCount(uuid []byte) (int, error) {
+	username, err := FetchUsernameFromUUID(uuid);
+	if err != nil {
+		return -1, err
+	}
+	    
+	query := `SELECT COUNT(*) FROM accounts a
+  			  JOIN friends f
+  			  ON a.username = f.friend
+			  WHERE a.lastActivity > DATE_SUB(UTC_TIMESTAMP(), INTERVAL 5 MINUTE)
+			  AND f.user = ?;`
+
+	var count int
+	err = handle.QueryRow(query, username).Scan(&count)
+	if err != nil {
+		return -1, err
+	}
+
+	return count, nil
 }
\ No newline at end of file
diff --git a/defs/game.go b/defs/game.go
index 2134b3f..95e26aa 100644
--- a/defs/game.go
+++ b/defs/game.go
@@ -21,3 +21,8 @@ type TitleStats struct {
 	PlayerCount int `json:"playerCount"`
 	BattleCount int `json:"battleCount"`
 }
+
+type FriendsOnlineStats struct {
+	FriendsOnline int `json:"friendsOnline"`
+	FriendsAmount int `json:"friendsAmount"`
+}
\ No newline at end of file