FIX: Users failed to log in when the username changes

Author: small-lovely-cat

internal/apps/oauth/utils.go

@@ -109,20 +109,32 @@ func doOAuth(ctx context.Context, code string) (*User, error) {
 	// 处理用户信息同步逻辑
 	var user User
 
-	txByUsername := db.DB(ctx).Where("username = ?", userInfo.Username).First(&user)
-	if txByUsername.Error != nil {
-		txByID := db.DB(ctx).Where("id = ?", userInfo.Id).First(&user)
-		if txByID.Error == nil {
-			// ID 存在但 username 不匹配(用户改名)
-			if err = user.CheckActive(); err != nil {
-				span.SetStatus(codes.Error, err.Error())
-				return nil, err
-			}
-			user.UpdateFromOAuthInfo(&userInfo)
-			if err = db.DB(ctx).Save(&user).Error; err != nil {
+	// 优先通过 ID 查询用户
+	txByID := db.DB(ctx).Where("id = ?", userInfo.Id).First(&user)
+	if txByID.Error == nil {
+		// ID 存在，更新用户信息（包括可能的用户名变更）
+		if err = user.CheckActive(); err != nil {
+			span.SetStatus(codes.Error, err.Error())
+			return nil, err
+		}
+		user.UpdateFromOAuthInfo(&userInfo)
+		if err = db.DB(ctx).Save(&user).Error; err != nil {
+			span.SetStatus(codes.Error, err.Error())
+			return nil, err
+		}
+	} else if errors.Is(txByID.Error, gorm.ErrRecordNotFound) {
+		// ID 不存在，检查 username 是否被其他用户占用
+		var existingUser User
+		txByUsername := db.DB(ctx).Where("username = ?", userInfo.Username).First(&existingUser)
+		if txByUsername.Error == nil {
+			// username 被其他用户占用（账户注销后被新用户占用）
+			newUser, err := existingUser.MarkAsDeactivatedAndCreateNew(ctx, &userInfo)
+			if err != nil {
 				span.SetStatus(codes.Error, err.Error())
 				return nil, err
 			}
+			user = *newUser
+			user.EnqueueBadgeScoreTask(ctx)
 		} else if errors.Is(txByUsername.Error, gorm.ErrRecordNotFound) {
 			user = User{
 				ID:          userInfo.Id,
@@ -139,29 +151,14 @@ func doOAuth(ctx context.Context, code string) (*User, error) {
 			}
 			user.EnqueueBadgeScoreTask(ctx)
 		} else {
-			// query failed
+			// username 查询失败
 			span.SetStatus(codes.Error, txByUsername.Error.Error())
 			return nil, txByUsername.Error
 		}
 	} else {
-		if user.ID != userInfo.Id {
-			// username 相同但 ID 不同(账户注销后被新用户占用)
-			if _, err = user.MarkAsDeactivatedAndCreateNew(ctx, &userInfo); err != nil {
-				span.SetStatus(codes.Error, err.Error())
-				return nil, err
-			}
-			user.EnqueueBadgeScoreTask(ctx)
-		} else {
-			if err = user.CheckActive(); err != nil {
-				span.SetStatus(codes.Error, err.Error())
-				return nil, err
-			}
-			user.UpdateFromOAuthInfo(&userInfo)
-			if err = db.DB(ctx).Save(&user).Error; err != nil {
-				span.SetStatus(codes.Error, err.Error())
-				return nil, err
-			}
-		}
+		// ID 查询失败
+		span.SetStatus(codes.Error, txByID.Error.Error())
+		return nil, txByID.Error
 	}
 	return &user, nil
 }