waf-platform/EdgeAdmin/internal/web/actions/default/finance/fee/index.go

// 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()
}