From ea33857afbd8ee565abd28b4ee5734fc970c181d Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Wed, 26 Jun 2024 15:52:58 +0800
Subject: [PATCH] :sparkles: Recycle accounts

---
 .idea/workspace.xml                 | 13 +++++---
 pkg/internal/models/tokens.go       |  2 +-
 pkg/internal/server/api/auth_api.go |  2 ++
 pkg/internal/services/accounts.go   | 49 ++++++++++++++++++++++++++++-
 pkg/internal/services/tokens.go     |  6 ++--
 pkg/main.go                         |  1 +
 6 files changed, 64 insertions(+), 9 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 52b7edb..9dde2a4 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -4,8 +4,13 @@
     <option name="autoReloadType" value="ALL" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":bug: Bug fixes">
-      <change beforePath="$PROJECT_DIR$/web/src/stores/notifications.ts" beforeDir="false" afterPath="$PROJECT_DIR$/web/src/stores/notifications.ts" afterDir="false" />
+    <list default="true" id="3fefb2c4-b6f9-466b-a523-53352e8d6f95" name="更改" comment=":arrow_up: Fix notification listen">
+      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/models/tokens.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/models/tokens.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/server/api/auth_api.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/server/api/auth_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/tokens.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/tokens.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/main.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/main.go" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -143,7 +148,6 @@
     </option>
   </component>
   <component name="VcsManagerConfiguration">
-    <MESSAGE value=":sparkles: Firebase is back" />
     <MESSAGE value=":sparkles: Apple push notification services" />
     <MESSAGE value=":bug: Bug fix and fix" />
     <MESSAGE value=":bug: Fix APNs non-production" />
@@ -168,7 +172,8 @@
     <MESSAGE value=":recycle: OAuth authenticate" />
     <MESSAGE value=":sparkles: Recommend app component" />
     <MESSAGE value=":bug: Bug fixes" />
-    <option name="LAST_COMMIT_MESSAGE" value=":bug: Bug fixes" />
+    <MESSAGE value=":arrow_up: Fix notification listen" />
+    <option name="LAST_COMMIT_MESSAGE" value=":arrow_up: Fix notification listen" />
   </component>
   <component name="VgoProject">
     <settings-migrated>true</settings-migrated>
diff --git a/pkg/internal/models/tokens.go b/pkg/internal/models/tokens.go
index 9c432a1..760d7b4 100644
--- a/pkg/internal/models/tokens.go
+++ b/pkg/internal/models/tokens.go
@@ -14,6 +14,6 @@ type MagicToken struct {
 
 	Code      string     `json:"code"`
 	Type      int8       `json:"type"`
-	AssignTo  *uint      `json:"assign_to"`
+	AccountID *uint      `json:"account_id"`
 	ExpiredAt *time.Time `json:"expired_at"`
 }
diff --git a/pkg/internal/server/api/auth_api.go b/pkg/internal/server/api/auth_api.go
index 9d3763c..dd6f546 100644
--- a/pkg/internal/server/api/auth_api.go
+++ b/pkg/internal/server/api/auth_api.go
@@ -23,6 +23,8 @@ func doAuthenticate(c *fiber.Ctx) error {
 	user, err := services.LookupAccount(data.Username)
 	if err != nil {
 		return fiber.NewError(fiber.StatusBadRequest, fmt.Sprintf("account was not found: %v", err.Error()))
+	} else if user.ConfirmedAt == nil {
+		return fiber.NewError(fiber.StatusForbidden, "account was not confirmed")
 	}
 
 	ticket, err := services.NewTicket(user, c.IP(), c.Get(fiber.HeaderUserAgent))
diff --git a/pkg/internal/services/accounts.go b/pkg/internal/services/accounts.go
index a84d9f9..a4b23a0 100644
--- a/pkg/internal/services/accounts.go
+++ b/pkg/internal/services/accounts.go
@@ -2,6 +2,7 @@ package services
 
 import (
 	"fmt"
+	"github.com/rs/zerolog/log"
 	"github.com/spf13/viper"
 	"gorm.io/datatypes"
 	"time"
@@ -93,7 +94,7 @@ func ConfirmAccount(code string) error {
 
 	var user models.Account
 	if err := database.C.Where(&models.Account{
-		BaseModel: models.BaseModel{ID: *token.AssignTo},
+		BaseModel: models.BaseModel{ID: *token.AccountID},
 	}).First(&user).Error; err != nil {
 		return err
 	}
@@ -121,3 +122,49 @@ func ConfirmAccount(code string) error {
 		return nil
 	})
 }
+
+func DeleteAccount(id uint) error {
+	tx := database.C.Begin()
+
+	for _, model := range []any{
+		&models.Badge{},
+		&models.RealmMember{},
+		&models.AccountContact{},
+		&models.AuthFactor{},
+		&models.AuthTicket{},
+		&models.MagicToken{},
+		&models.ThirdClient{},
+		&models.Notification{},
+		&models.NotificationSubscriber{},
+		&models.AccountFriendship{},
+	} {
+		if err := tx.Delete(model, "account_id = ?", id).Error; err != nil {
+			tx.Rollback()
+			return err
+		}
+	}
+
+	if err := tx.Delete(&models.Account{}, "id = ?", id).Error; err != nil {
+		tx.Rollback()
+		return err
+	}
+
+	return tx.Commit().Error
+}
+
+func RecycleUnConfirmAccount() {
+	var hitList []models.Account
+	if err := database.C.Where("confirmed_at IS NULL").Find(&hitList).Error; err != nil {
+		log.Error().Err(err).Msg("An error occurred while recycling accounts...")
+		return
+	}
+
+	if len(hitList) > 0 {
+		log.Info().Int("count", len(hitList)).Msg("Going to recycle those un-confirmed accounts...")
+		for _, entry := range hitList {
+			if err := DeleteAccount(entry.ID); err != nil {
+				log.Error().Err(err).Msg("An error occurred while recycling accounts...")
+			}
+		}
+	}
+}
diff --git a/pkg/internal/services/tokens.go b/pkg/internal/services/tokens.go
index f297444..9dd9419 100644
--- a/pkg/internal/services/tokens.go
+++ b/pkg/internal/services/tokens.go
@@ -47,7 +47,7 @@ func NewMagicToken(mode models.MagicTokenType, assignTo *models.Account, expired
 	token := models.MagicToken{
 		Code:      strings.Replace(uuid.NewString(), "-", "", -1),
 		Type:      mode,
-		AssignTo:  &uid,
+		AccountID: &uid,
 		ExpiredAt: expiredAt,
 	}
 
@@ -59,13 +59,13 @@ func NewMagicToken(mode models.MagicTokenType, assignTo *models.Account, expired
 }
 
 func NotifyMagicToken(token models.MagicToken) error {
-	if token.AssignTo == nil {
+	if token.AccountID == nil {
 		return fmt.Errorf("could notify a non-assign magic token")
 	}
 
 	var user models.Account
 	if err := database.C.Where(&models.Account{
-		BaseModel: models.BaseModel{ID: *token.AssignTo},
+		BaseModel: models.BaseModel{ID: *token.AccountID},
 	}).Preload("Contacts").First(&user).Error; err != nil {
 		return err
 	}
diff --git a/pkg/main.go b/pkg/main.go
index 636fc1f..dae1fd4 100644
--- a/pkg/main.go
+++ b/pkg/main.go
@@ -68,6 +68,7 @@ func main() {
 	quartz.AddFunc("@every 60m", services.DoAutoSignoff)
 	quartz.AddFunc("@every 60m", services.DoAutoDatabaseCleanup)
 	quartz.AddFunc("@every 60s", services.RecycleAuthContext)
+	quartz.AddFunc("@every 60m", services.RecycleUnConfirmAccount)
 	quartz.AddFunc("@every 5m", services.KexCleanup)
 	quartz.Start()