add forget password

internal/repository/auth/init.go

@@ -26,7 +26,8 @@ func (r *AuthRepository) CheckExistsUserAccount(ctx context.Context, email strin
 
 	err := r.db.
 		Table("users").
-		Select("users.*, user_roles.role_id, user_roles.partner_id, user_roles.site_id, sites.name, roles.role_name, partners.name as partner_name, partners.status as partner_status").
+		Select("users.*, user_roles.role_id, user_roles.partner_id, user_roles.site_id,"+
+			" sites.name, roles.role_name, partners.name as partner_name, partners.status as partner_status, users.reset_password").
 		Where("users.email = ?", email).
 		Joins("left join user_roles on users.id = user_roles.user_id").
 		Joins("left join roles on user_roles.role_id = roles.role_id").
@@ -49,3 +50,49 @@ func (r *AuthRepository) CheckExistsUserAccount(ctx context.Context, email strin
 
 	return &user, nil
 }
+
+func (r *AuthRepository) CheckExistsUserAccountByID(ctx context.Context, userID int64) (*entity.UserDB, error) {
+	var user entity.UserDB
+
+	err := r.db.
+		Table("users").
+		Select("users.*, user_roles.role_id, user_roles.partner_id, user_roles.site_id, sites.name, roles.role_name, partners.name as partner_name, partners.status as partner_status").
+		Where("users.id = ?", userID).
+		Joins("left join user_roles on users.id = user_roles.user_id").
+		Joins("left join roles on user_roles.role_id = roles.role_id").
+		Joins("left join partners on user_roles.partner_id = partners.id").
+		Joins("left join sites on user_roles.site_id = sites.id").
+		First(&user).Error
+
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, nil
+	}
+
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, fmt.Errorf("user with not exist") // or use a custom error type
+		}
+
+		logger.ContextLogger(ctx).Error(fmt.Sprintf("Failed to get user"), zap.Error(err))
+		return nil, err
+	}
+
+	return &user, nil
+}
+
+func (r *AuthRepository) UpdatePassword(ctx context.Context, trx *gorm.DB, newHashedPassword string, userID int64, resetPassword bool) error {
+	// Perform the update using a single Updates call
+	err := trx.Model(&entity.UserDB{}).
+		Where("id = ?", userID).
+		Updates(map[string]interface{}{
+			"password":       newHashedPassword,
+			"reset_password": resetPassword,
+		}).Error
+
+	if err != nil {
+		logger.ContextLogger(ctx).Error(fmt.Sprintf("Failed to update password for user with id: %d", userID), zap.Error(err))
+		return err
+	}
+
+	return nil
+}