1.4.5.2

EdgeAPI/internal/db/models/user_dao_plus.go

@@ -0,0 +1,199 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build plus
+
+package models
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAPI/internal/errors"
+	"github.com/TeaOSLab/EdgeCommon/pkg/systemconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
+	"github.com/iwind/TeaGo/dbs"
+)
+
+// FindUserPriceInfo 查找用户计费方式和周期等账单信息
+func (this *UserDAO) FindUserPriceInfo(tx *dbs.Tx, userId int64) (priceType userconfigs.PriceType, pricePeriod userconfigs.PricePeriod, err error) {
+	if userId <= 0 {
+		return
+	}
+
+	one, err := this.Query(tx).
+		Pk(userId).
+		Result("priceType", "pricePeriod").
+		Find()
+	if err != nil || one == nil {
+		return "", "", err
+	}
+
+	var user = one.(*User)
+	priceType = user.PriceType
+	pricePeriod = user.PricePeriod
+
+	if len(priceType) > 0 && !userconfigs.IsValidPriceType(priceType) {
+		priceType = ""
+	}
+
+	if len(pricePeriod) > 0 && !userconfigs.IsValidPricePeriod(pricePeriod) {
+		pricePeriod = ""
+	}
+
+	return
+}
+
+// UpdateUserPriceType 修改用户计费方式
+func (this *UserDAO) UpdateUserPriceType(tx *dbs.Tx, userId int64, priceType userconfigs.PriceType) error {
+	if !userconfigs.IsValidPriceType(priceType) {
+		return errors.New("invalid price type '" + priceType + "'")
+	}
+	return this.Query(tx).
+		Pk(userId).
+		Set("priceType", priceType).
+		UpdateQuickly()
+}
+
+// UpdateUserPricePeriod 修改用户计费周期
+func (this *UserDAO) UpdateUserPricePeriod(tx *dbs.Tx, userId int64, pricePeriod userconfigs.PricePeriod) error {
+	if !userconfigs.IsValidPricePeriod(pricePeriod) {
+		return errors.New("invalid price period '" + pricePeriod + "'")
+	}
+	return this.Query(tx).
+		Pk(userId).
+		Set("pricePeriod", pricePeriod).
+		UpdateQuickly()
+}
+
+// FindUserBandwidthAlgoForFee 获取用户计费用的带宽算法
+func (this *UserDAO) FindUserBandwidthAlgoForFee(tx *dbs.Tx, userId int64, priceConfig *userconfigs.UserPriceConfig) (bandwidthAlgo string, err error) {
+	bandwidthAlgo, err = this.Query(tx).
+		Pk(userId).
+		Result("bandwidthAlgo").
+		FindStringCol("")
+	if len(bandwidthAlgo) > 0 {
+		return
+	}
+
+	if priceConfig == nil {
+		priceConfig, err = SharedSysSettingDAO.ReadUserPriceConfig(tx)
+		if err != nil {
+			return "", err
+		}
+		if priceConfig == nil {
+			return systemconfigs.BandwidthAlgoSecondly, nil
+		}
+	}
+
+	if priceConfig != nil &&
+		priceConfig.DefaultBandwidthPriceConfig != nil &&
+		len(priceConfig.DefaultBandwidthPriceConfig.BandwidthAlgo) > 0 {
+		return priceConfig.DefaultBandwidthPriceConfig.BandwidthAlgo, nil
+	}
+
+	return systemconfigs.BandwidthAlgoSecondly, nil
+}
+
+// CheckUserServersEnabled 判断用户是否可用服务功能
+func (this *UserDAO) CheckUserServersEnabled(tx *dbs.Tx, userId int64) (isEnabled bool, err error) {
+	if userId <= 0 {
+		return true, nil
+	}
+
+	// 是否已删除、未启用、已拒绝
+	one, err := this.Query(tx).
+		Result("id", "isRejected", "state", "isOn").
+		Pk(userId).
+		Find()
+	if err != nil {
+		return false, err
+	}
+	if one == nil {
+		return false, nil
+	}
+	var user = one.(*User)
+	if user.State != UserStateEnabled || !user.IsOn || user.IsRejected {
+		return false, nil
+	}
+
+	// 是否有逾期的账单
+	priceConfig, err := SharedSysSettingDAO.ReadUserPriceConfig(tx)
+	if err != nil {
+		return false, err
+	}
+	if priceConfig != nil &&
+		priceConfig.IsOn &&
+		priceConfig.UnpaidBillPolicy.IsOn &&
+		priceConfig.UnpaidBillPolicy.DisableServers &&
+		(priceConfig.UnpaidBillPolicy.MinDailyBillDays > 0 ||
+			priceConfig.UnpaidBillPolicy.MinMonthlyBillDays > 0) {
+		countUnpaidBills, err := SharedUserBillDAO.CountAllUserBills(tx, 0, userId, "", true, priceConfig.UnpaidBillPolicy.MinDailyBillDays, priceConfig.UnpaidBillPolicy.MinMonthlyBillDays)
+		if err != nil {
+			return false, err
+		}
+		if countUnpaidBills > 0 {
+			return false, nil
+		}
+	}
+
+	return true, nil
+}
+
+// FindUserNotificationConfig 查找通知设置
+func (this *UserDAO) FindUserNotificationConfig(tx *dbs.Tx, userId int64) (*userconfigs.UserNotificationConfig, error) {
+	notificationData, err := this.Query(tx).
+		Pk(userId).
+		Result("notification").
+		FindJSONCol()
+	if err != nil {
+		return nil, err
+	}
+
+	var config = userconfigs.NewUserNotificationConfig()
+	if IsNull(notificationData) {
+		return config, nil
+	}
+
+	err = json.Unmarshal(notificationData, config)
+	if err != nil {
+		return nil, err
+	}
+
+	return config, nil
+}
+
+// NotifyEmail 进行邮件通知
+func (this *UserDAO) NotifyEmail(tx *dbs.Tx, userId int64, notificationType userconfigs.UserNotificationType, subject string, body string) error {
+	// 当前用户是否有Email设置
+	one, err := this.Query(tx).
+		Pk(userId).
+		Result("email", "verifiedEmail", "notification").
+		Find()
+	if err != nil || one == nil {
+		return err
+	}
+	var user = one.(*User)
+	var email = user.Email
+	var verifiedEmail = user.VerifiedEmail
+	if len(email) == 0 {
+		email = verifiedEmail
+	}
+	if len(email) == 0 {
+		return nil
+	}
+
+	// 检查通知配置
+	var notificationData = user.Notification
+	var config = userconfigs.NewUserNotificationConfig()
+	if !IsNull(notificationData) {
+		err = json.Unmarshal(notificationData, config)
+		if err != nil {
+			return err
+		}
+	}
+
+	if !config.Support(notificationType, userconfigs.UserNotificationMediaEmail) {
+		return nil
+	}
+
+	// 创建邮件
+	_, err = SharedUserEmailNotificationDAO.CreateNotification(tx, email, subject, body)
+	return err
+}