From b1f6cf8f6e988e7d3edd9f9010be1510646040b9 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Wed, 26 Jun 2024 20:28:12 +0800
Subject: [PATCH] :sparkles: Last seen at

---
 .idea/workspace.xml                     | 14 +++++++++-----
 pkg/internal/models/profiles.go         |  1 +
 pkg/internal/server/api/statuses_api.go | 10 +++++++---
 pkg/internal/services/accounts.go       | 11 +++++++++++
 pkg/internal/services/connections.go    |  6 ++++++
 5 files changed, 34 insertions(+), 8 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 6eb7413..63da3dc 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -4,8 +4,12 @@
     <option name="autoReloadType" value="ALL" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":sparkles: Status system">
-      <change beforePath="$PROJECT_DIR$/pkg/internal/services/statuses.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/statuses.go" afterDir="false" />
+    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":bug: Fix online condition">
+      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/models/profiles.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/models/profiles.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/server/api/statuses_api.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/server/api/statuses_api.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/services/accounts.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/accounts.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/services/connections.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/connections.go" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -151,8 +155,6 @@
     </option>
   </component>
   <component name="VcsManagerConfiguration">
-    <MESSAGE value=":sparkles: Consul registration" />
-    <MESSAGE value=":wastebasket: Remove HTTP provision to consul" />
     <MESSAGE value=":sparkles: Drop direct connection and uses consul" />
     <MESSAGE value=":technologist: Add the server side Hyper SDK" />
     <MESSAGE value=":recycle: Improve code structure and much easier to read&#10;:bug: Fix auth middleware" />
@@ -176,7 +178,9 @@
     <MESSAGE value=":bug: Authenticate wrong payload hotfix" />
     <MESSAGE value=":sparkles: Can pick up mfa request" />
     <MESSAGE value=":sparkles: Status system" />
-    <option name="LAST_COMMIT_MESSAGE" value=":sparkles: Status system" />
+    <MESSAGE value=":bug: Fix status expired in cache" />
+    <MESSAGE value=":bug: Fix online condition" />
+    <option name="LAST_COMMIT_MESSAGE" value=":bug: Fix online condition" />
   </component>
   <component name="VgoProject">
     <settings-migrated>true</settings-migrated>
diff --git a/pkg/internal/models/profiles.go b/pkg/internal/models/profiles.go
index 1aa5452..8c873a8 100644
--- a/pkg/internal/models/profiles.go
+++ b/pkg/internal/models/profiles.go
@@ -10,6 +10,7 @@ type AccountProfile struct {
 	FirstName  string     `json:"first_name"`
 	LastName   string     `json:"last_name"`
 	Experience uint64     `json:"experience"`
+	LastSeenAt *time.Time `json:"last_seen_at"`
 	Birthday   *time.Time `json:"birthday"`
 	AccountID  uint       `json:"account_id"`
 }
diff --git a/pkg/internal/server/api/statuses_api.go b/pkg/internal/server/api/statuses_api.go
index b4293e4..5649d89 100644
--- a/pkg/internal/server/api/statuses_api.go
+++ b/pkg/internal/server/api/statuses_api.go
@@ -2,6 +2,7 @@ package api
 
 import (
 	"fmt"
+	"git.solsynth.dev/hydrogen/passport/pkg/internal/database"
 	"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
 	"git.solsynth.dev/hydrogen/passport/pkg/internal/server/exts"
 	"git.solsynth.dev/hydrogen/passport/pkg/internal/services"
@@ -13,9 +14,11 @@ import (
 func getStatus(c *fiber.Ctx) error {
 	alias := c.Params("alias")
 
-	user, err := services.GetAccountWithName(alias)
-	if err != nil {
-		return fiber.NewError(fiber.StatusNotFound, fmt.Sprintf("account not found: %s", alias))
+	var user models.Account
+	if err := database.C.Where(models.Account{
+		Name: alias,
+	}).Preload("Profile").First(&user).Error; err != nil {
+		return fiber.NewError(fiber.StatusNotFound, fmt.Sprintf("account not found: %s", err))
 	}
 
 	status, err := services.GetStatus(user.ID)
@@ -24,6 +27,7 @@ func getStatus(c *fiber.Ctx) error {
 
 	return c.JSON(fiber.Map{
 		"status":         lo.Ternary(err == nil, &status, nil),
+		"last_seen_at":   user.Profile.LastSeenAt,
 		"is_disturbable": disturbable,
 		"is_online":      online,
 	})
diff --git a/pkg/internal/services/accounts.go b/pkg/internal/services/accounts.go
index e49f341..ab764ca 100644
--- a/pkg/internal/services/accounts.go
+++ b/pkg/internal/services/accounts.go
@@ -179,3 +179,14 @@ func RecycleUnConfirmAccount() {
 		}
 	}
 }
+
+func SetAccountLastSeen(uid uint) error {
+	var profile models.AccountProfile
+	if err := database.C.Where("account_id = ?", uid).First(&profile).Error; err != nil {
+		return err
+	}
+
+	profile.LastSeenAt = lo.ToPtr(time.Now())
+
+	return database.C.Save(&profile).Error
+}
diff --git a/pkg/internal/services/connections.go b/pkg/internal/services/connections.go
index f38ad3a..87484e9 100644
--- a/pkg/internal/services/connections.go
+++ b/pkg/internal/services/connections.go
@@ -28,4 +28,10 @@ func ClientUnregister(user models.Account, conn *websocket.Conn) {
 	}
 	delete(wsConn[user.ID], conn)
 	wsMutex.Unlock()
+
+	if status, err := GetStatus(user.ID); err != nil || !status.IsInvisible {
+		if len(wsConn[user.ID]) == 0 {
+			_ = SetAccountLastSeen(user.ID)
+		}
+	}
 }