Initial commit (code only without large binaries)

EdgeAdmin/internal/web/actions/default/finance/fee/index.go

@@ -0,0 +1,188 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build plus
+
+package fee
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/configloaders"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/finance/financeutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/systemconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/userconfigs"
+	"github.com/iwind/TeaGo/actions"
+)
+
+type IndexAction struct {
+	actionutils.ParentAction
+}
+
+func (this *IndexAction) Init() {
+	this.Nav("", "", "basic")
+}
+
+func (this *IndexAction) RunGet(params struct{}) {
+	config, err := financeutils.ReadPriceConfig(this.AdminContext())
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["config"] = config
+
+	this.Show()
+}
+
+func (this *IndexAction) RunPost(params struct {
+	IsOn                     bool
+	PriceType                string
+	PricePeriod              string
+	TrafficPriceJSON         []byte
+	BandwidthPriceJSON       []byte
+	UserCanChangePriceType   bool
+	UserCanChangePricePeriod bool
+	EnablePlans              bool
+	ShowPlansInUserSystem    bool
+	EnableTrafficPackages    bool
+	ShowTrafficPackages      bool
+	UserUIShowPrices         bool
+
+	UnpaidBillPolicyIsOn               bool
+	UnpaidBillPolicyMinDailyBillDays   int32
+	UnpaidBillPolicyMinMonthlyBillDays int32
+	UnpaidBillPolicyDisableServers     bool
+
+	Must *actions.Must
+	CSRF *actionutils.CSRF
+}) {
+	defer this.CreateLogInfo(codes.FinanceFee_LogUpdateFeeSetting)
+
+	valueResp, err := this.RPC().SysSettingRPC().ReadSysSetting(this.AdminContext(), &pb.ReadSysSettingRequest{Code: systemconfigs.SettingCodeUserPriceConfig})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var valueJSON = valueResp.ValueJSON
+	var priceConfig = userconfigs.DefaultUserPriceConfig()
+	if len(valueJSON) > 0 {
+		err = json.Unmarshal(valueJSON, priceConfig)
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+	}
+
+	priceConfig.IsOn = params.IsOn
+
+	if !userconfigs.IsValidPricePeriod(params.PricePeriod) {
+		this.Fail("错误的结算周期 '" + params.PricePeriod + "'")
+		return
+	}
+	priceConfig.DefaultPricePeriod = params.PricePeriod
+	priceConfig.UserCanChangePriceType = params.UserCanChangePriceType
+	priceConfig.UserCanChangePricePeriod = params.UserCanChangePricePeriod
+
+	priceConfig.EnablePlans = params.EnablePlans
+	priceConfig.ShowPlansInUserSystem = params.ShowPlansInUserSystem
+
+	priceConfig.EnableTrafficPackages = params.EnableTrafficPackages
+	priceConfig.ShowTrafficPackages = params.ShowTrafficPackages
+	priceConfig.UserUI.ShowPrices = params.UserUIShowPrices
+
+	priceConfig.DefaultPriceType = params.PriceType
+
+	switch params.PriceType {
+	case serverconfigs.PlanPriceTypeTraffic:
+		if len(params.TrafficPriceJSON) == 0 {
+			this.Fail("请设置流量价格")
+		}
+	case serverconfigs.PlanPriceTypeBandwidth:
+		if len(params.BandwidthPriceJSON) == 0 {
+			this.Fail("请设置带宽价格")
+		}
+	default:
+		this.Fail("请选择默认计费方式")
+		return
+	}
+
+	// 流量价格
+	if len(params.TrafficPriceJSON) > 0 {
+		var config = &serverconfigs.PlanTrafficPriceConfig{}
+		err := json.Unmarshal(params.TrafficPriceJSON, config)
+		if err != nil {
+			this.Fail("流量价格设置错误：" + err.Error())
+		}
+
+		if params.PriceType == serverconfigs.PlanPriceTypeTraffic || params.UserCanChangePriceType {
+			if config.Base <= 0 && len(config.Ranges) == 0 {
+				this.Fail("流量基础价格和流量阶梯价格必须至少填写一个")
+			}
+		}
+		priceConfig.DefaultTrafficPriceConfig = config
+	}
+
+	// 带宽价格
+	if len(params.BandwidthPriceJSON) > 0 {
+		var config = &serverconfigs.PlanBandwidthPriceConfig{}
+		err := json.Unmarshal(params.BandwidthPriceJSON, config)
+		if err != nil {
+			this.Fail("带宽价格设置错误：" + err.Error())
+		}
+
+		if params.PriceType == serverconfigs.PlanPriceTypeBandwidth || params.UserCanChangePriceType {
+			if config.Percentile <= 0 {
+				this.Fail("带宽百分位必须大于0")
+			}
+			if config.Percentile > 100 {
+				this.Fail("带宽百分位必须不大于100")
+			}
+			if config.Base <= 0 && len(config.Ranges) == 0 {
+				this.Fail("带宽基础价格和带宽阶梯价格必须至少填写一个")
+			}
+		}
+		priceConfig.DefaultBandwidthPriceConfig = config
+
+		// 同步修改UserUI中的用量统计设置
+		uiConfig, err := configloaders.LoadUserUIConfig()
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		if uiConfig != nil && uiConfig.TrafficStats.BandwidthAlgo != config.BandwidthAlgo {
+			uiConfig.TrafficStats.BandwidthAlgo = config.BandwidthAlgo
+			err = configloaders.UpdateUserUIConfig(uiConfig)
+			if err != nil {
+				this.ErrorPage(err)
+				return
+			}
+		}
+	}
+
+	// 未支付账单策略
+	priceConfig.UnpaidBillPolicy.IsOn = params.UnpaidBillPolicyIsOn
+	priceConfig.UnpaidBillPolicy.MinDailyBillDays = params.UnpaidBillPolicyMinDailyBillDays
+	priceConfig.UnpaidBillPolicy.MinMonthlyBillDays = params.UnpaidBillPolicyMinMonthlyBillDays
+	priceConfig.UnpaidBillPolicy.DisableServers = params.UnpaidBillPolicyDisableServers
+
+	financeConfigJSON, err := json.Marshal(priceConfig)
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	_, err = this.RPC().SysSettingRPC().UpdateSysSetting(this.AdminContext(), &pb.UpdateSysSettingRequest{
+		Code:      systemconfigs.SettingCodeUserPriceConfig,
+		ValueJSON: financeConfigJSON,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	// 重置配置
+	configloaders.ResetUserPriceConfig(priceConfig)
+
+	this.Success()
+}