Initial commit (code only without large binaries)

EdgeAdmin/internal/web/actions/default/plans/userPlans/createPopup.go

@@ -0,0 +1,143 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build plus
+
+package userPlans
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/actions"
+	"github.com/iwind/TeaGo/maps"
+	"github.com/iwind/TeaGo/types"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+	"time"
+)
+
+type CreatePopupAction struct {
+	actionutils.ParentAction
+}
+
+func (this *CreatePopupAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *CreatePopupAction) RunGet(params struct{}) {
+	// 所有套餐
+	planResp, err := this.RPC().PlanRPC().FindAllAvailablePlans(this.AdminContext(), &pb.FindAllAvailablePlansRequest{})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var planMaps = []maps.Map{}
+	for _, plan := range planResp.Plans {
+		planMaps = append(planMaps, maps.Map{
+			"id":              plan.Id,
+			"name":            plan.Name,
+			"priceType":       plan.PriceType,
+			"monthlyPrice":    plan.MonthlyPrice,
+			"seasonallyPrice": plan.SeasonallyPrice,
+			"yearlyPrice":     plan.YearlyPrice,
+		})
+	}
+	this.Data["plans"] = planMaps
+
+	// 默认结束日期
+	this.Data["defaultDayTo"] = timeutil.Format("Y-m-d", time.Now().AddDate(20, 0, 0))
+
+	this.Show()
+}
+
+func (this *CreatePopupAction) RunPost(params struct {
+	UserId int64
+	PlanId int64
+	Name   string
+
+	Period       string
+	CountMonths  int32
+	CountSeasons int32
+	CountYears   int32
+
+	DayTo string
+
+	Must *actions.Must
+	CSRF *actionutils.CSRF
+}) {
+	defer this.CreateLogInfo(codes.UserPlan_LogBuyUserPlan, params.UserId, params.PlanId)
+
+	params.Must.
+		Field("userId", params.UserId).
+		Gt(0, "请选择用户").
+		Field("name", params.Name).
+		Require("请输入备注名称").
+		Field("planId", params.PlanId).
+		Gt(0, "请选择套餐")
+
+	// 检查余额
+	planResp, err := this.RPC().PlanRPC().FindEnabledPlan(this.AdminContext(), &pb.FindEnabledPlanRequest{PlanId: params.PlanId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var plan = planResp.Plan
+	if plan == nil {
+		this.Fail("找不到要购买的套餐")
+	}
+
+	var countPeriod int32
+	if plan.PriceType == serverconfigs.PlanPriceTypePeriod {
+		var cost float64
+
+		switch params.Period {
+		case "monthly":
+			countPeriod = params.CountMonths
+			cost = plan.MonthlyPrice * float64(countPeriod)
+		case "seasonally":
+			countPeriod = params.CountSeasons
+			cost = plan.SeasonallyPrice * float64(countPeriod)
+		case "yearly":
+			countPeriod = params.CountYears
+			cost = plan.YearlyPrice * float64(countPeriod)
+		default:
+			this.Fail("invalid period '" + params.Period + "'")
+		}
+
+		// 账号
+		accountResp, err := this.RPC().UserAccountRPC().FindEnabledUserAccountWithUserId(this.AdminContext(), &pb.FindEnabledUserAccountWithUserIdRequest{UserId: params.UserId})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		var account = accountResp.UserAccount
+		if account == nil {
+			this.Fail("找不到用户'" + types.String(params.UserId) + "'的账户")
+		}
+		if account.Total < cost {
+			this.Fail("用户账户余额不足")
+		}
+	} else if plan.PriceType == serverconfigs.PlanPriceTypeTraffic {
+		params.Must.Field("dayTo", params.DayTo).
+			Match(`^\d+-\d+-\d+$`, "请输入正确的结束日期")
+	} else if plan.PriceType == serverconfigs.PlanPriceTypeBandwidth {
+		params.Must.Field("dayTo", params.DayTo).
+			Match(`^\d+-\d+-\d+$`, "请输入正确的结束日期")
+	} else {
+		this.Fail("不支持的付款方式：" + plan.PriceType)
+	}
+
+	_, err = this.RPC().UserPlanRPC().BuyUserPlan(this.AdminContext(), &pb.BuyUserPlanRequest{
+		UserId:      params.UserId,
+		PlanId:      params.PlanId,
+		Name:        params.Name,
+		DayTo:       params.DayTo,
+		Period:      params.Period,
+		CountPeriod: countPeriod,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}

EdgeAdmin/internal/web/actions/default/plans/userPlans/delete.go

@@ -0,0 +1,28 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build plus
+
+package userPlans
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+type DeleteAction struct {
+	actionutils.ParentAction
+}
+
+func (this *DeleteAction) RunPost(params struct {
+	UserPlanId int64
+}) {
+	defer this.CreateLogInfo(codes.UserPlan_LogDeleteUserPlan, params.UserPlanId)
+
+	_, err := this.RPC().UserPlanRPC().DeleteUserPlan(this.AdminContext(), &pb.DeleteUserPlanRequest{UserPlanId: params.UserPlanId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}

EdgeAdmin/internal/web/actions/default/plans/userPlans/index.go

@@ -0,0 +1,195 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build plus
+
+package userPlans
+
+import (
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/default/finance/financeutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/maps"
+)
+
+type IndexAction struct {
+	actionutils.ParentAction
+}
+
+func (this *IndexAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *IndexAction) RunGet(params struct {
+	Type   string
+	UserId int64
+}) {
+	if len(params.Type) == 0 {
+		params.Type = "available"
+	}
+	this.Data["type"] = params.Type
+
+	var total int64 = 0
+
+	// 有效
+	{
+		countResp, err := this.RPC().UserPlanRPC().CountAllEnabledUserPlans(this.AdminContext(), &pb.CountAllEnabledUserPlansRequest{
+			IsAvailable:  true,
+			IsExpired:    false,
+			ExpiringDays: 0,
+			UserId:       params.UserId,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		this.Data["countAvailable"] = countResp.Count
+
+		if params.Type == "available" {
+			total = countResp.Count
+		}
+	}
+
+	// 过期
+	{
+		countResp, err := this.RPC().UserPlanRPC().CountAllEnabledUserPlans(this.AdminContext(), &pb.CountAllEnabledUserPlansRequest{
+			IsAvailable:  false,
+			IsExpired:    true,
+			ExpiringDays: 0,
+			UserId:       params.UserId,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		this.Data["countExpired"] = countResp.Count
+
+		if params.Type == "expired" {
+			total = countResp.Count
+		}
+	}
+
+	// 7天过期
+	{
+		countResp, err := this.RPC().UserPlanRPC().CountAllEnabledUserPlans(this.AdminContext(), &pb.CountAllEnabledUserPlansRequest{
+			IsAvailable:  false,
+			IsExpired:    false,
+			ExpiringDays: 7,
+			UserId:       params.UserId,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		this.Data["countExpiring7"] = countResp.Count
+
+		if params.Type == "expiring7" {
+			total = countResp.Count
+		}
+	}
+
+	// 30天过期
+	{
+		countResp, err := this.RPC().UserPlanRPC().CountAllEnabledUserPlans(this.AdminContext(), &pb.CountAllEnabledUserPlansRequest{
+			IsAvailable:  false,
+			IsExpired:    false,
+			ExpiringDays: 30,
+			UserId:       params.UserId,
+		})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		this.Data["countExpiring30"] = countResp.Count
+
+		if params.Type == "expiring30" {
+			total = countResp.Count
+		}
+	}
+
+	// 列表
+	var page = this.NewPage(total)
+	this.Data["page"] = page.AsHTML()
+
+	var expiringDays = int32(0)
+	if params.Type == "expiring7" {
+		expiringDays = 7
+	} else if params.Type == "expiring30" {
+		expiringDays = 30
+	}
+	userPlansResp, err := this.RPC().UserPlanRPC().ListEnabledUserPlans(this.AdminContext(), &pb.ListEnabledUserPlansRequest{
+		IsAvailable:  params.Type == "available",
+		IsExpired:    params.Type == "expired",
+		ExpiringDays: expiringDays,
+		UserId:       params.UserId,
+		Offset:       page.Offset,
+		Size:         page.Size,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var userPlanMaps = []maps.Map{}
+	for _, userPlan := range userPlansResp.UserPlans {
+		// user
+		var userMap = maps.Map{"id": 0}
+		if userPlan.User != nil {
+			userMap = maps.Map{
+				"id":       userPlan.User.Id,
+				"fullname": userPlan.User.Fullname,
+				"username": userPlan.User.Username,
+			}
+		}
+
+		// plan
+		var planMap = maps.Map{"id": 0}
+		if userPlan.Plan != nil {
+			planMap = maps.Map{
+				"id":   userPlan.Plan.Id,
+				"name": userPlan.Plan.Name,
+			}
+		}
+
+		// server
+		var serverMaps = []maps.Map{}
+		if len(userPlan.Servers) > 0 {
+			for _, server := range userPlan.Servers {
+				// 补足网站名称
+				if len(server.Name) == 0 && len(server.ServerNamesJSON) > 0 {
+					var serverNames = []*serverconfigs.ServerNameConfig{}
+					err = json.Unmarshal(server.ServerNamesJSON, &serverNames)
+					if err == nil && len(serverNames) > 0 {
+						server.Name = serverNames[0].FirstName()
+					}
+				}
+
+				serverMaps = append(serverMaps, maps.Map{
+					"id":   server.Id,
+					"name": server.Name,
+					"type": server.Type,
+				})
+			}
+		}
+
+		userPlanMaps = append(userPlanMaps, maps.Map{
+			"id":      userPlan.Id,
+			"isOn":    userPlan.IsOn,
+			"name":    userPlan.Name,
+			"dayTo":   userPlan.DayTo,
+			"user":    userMap,
+			"plan":    planMap,
+			"servers": serverMaps,
+		})
+	}
+	this.Data["userPlans"] = userPlanMaps
+
+	// 价格设置
+	priceConfig, err := financeutils.ReadPriceConfig(this.AdminContext())
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	this.Data["canUsePlans"] = priceConfig != nil && priceConfig.EnablePlans
+
+	this.Show()
+}

EdgeAdmin/internal/web/actions/default/plans/userPlans/renewPopup.go

@@ -0,0 +1,206 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build plus
+
+package userPlans
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs/codes"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/actions"
+	"github.com/iwind/TeaGo/maps"
+	"github.com/iwind/TeaGo/types"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+	"regexp"
+	"time"
+)
+
+type RenewPopupAction struct {
+	actionutils.ParentAction
+}
+
+func (this *RenewPopupAction) Init() {
+	this.Nav("", "", "")
+}
+
+func (this *RenewPopupAction) RunGet(params struct {
+	UserPlanId int64
+}) {
+	userPlanResp, err := this.RPC().UserPlanRPC().FindEnabledUserPlan(this.AdminContext(), &pb.FindEnabledUserPlanRequest{UserPlanId: params.UserPlanId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var userPlan = userPlanResp.UserPlan
+	if userPlan == nil {
+		this.Fail("找不到要修改的套餐")
+	}
+	var planId = userPlan.PlanId
+	var userId = userPlan.UserId
+
+	this.Data["userPlanId"] = params.UserPlanId
+
+	// 套餐
+	planResp, err := this.RPC().PlanRPC().FindEnabledPlan(this.AdminContext(), &pb.FindEnabledPlanRequest{PlanId: planId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var plan = planResp.Plan
+	if plan == nil {
+		this.Fail("找不到要购买的套餐")
+	}
+	this.Data["plan"] = maps.Map{
+		"id":              plan.Id,
+		"name":            plan.Name,
+		"priceType":       plan.PriceType,
+		"monthlyPrice":    plan.MonthlyPrice,
+		"seasonallyPrice": plan.SeasonallyPrice,
+		"yearlyPrice":     plan.YearlyPrice,
+	}
+
+	// 用户
+	userResp, err := this.RPC().UserRPC().FindEnabledUser(this.AdminContext(), &pb.FindEnabledUserRequest{UserId: userId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var user = userResp.User
+	if user == nil {
+		this.Fail("找不到套餐所属用户")
+	}
+	this.Data["user"] = maps.Map{
+		"id":       user.Id,
+		"username": user.Username,
+		"fullname": user.Fullname,
+	}
+
+	// 账户
+	accountResp, err := this.RPC().UserAccountRPC().FindEnabledUserAccountWithUserId(this.AdminContext(), &pb.FindEnabledUserAccountWithUserIdRequest{UserId: userId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var account = accountResp.UserAccount
+	if account == nil {
+		this.Fail("找不到用户'" + types.String(userId) + "'的账户")
+	}
+	this.Data["userAccount"] = maps.Map{"total": account.Total}
+
+	// 默认结束日期
+	var hasDefaultDayTo = false
+	if len(userPlan.DayTo) > 0 {
+		// 如果已经有日期，则自动增加1年
+		var reg = regexp.MustCompile(`^(\d+)-(\d+)-(\d+)$`)
+		if reg.MatchString(userPlan.DayTo) {
+			var matches = reg.FindStringSubmatch(userPlan.DayTo)
+			var year = types.Int(matches[1])
+			var month = types.Int(matches[2])
+			var day = types.Int(matches[3])
+			if year > 1970 {
+				this.Data["defaultDayTo"] = timeutil.Format("Y-m-d", time.Date(year+1, time.Month(month), day, 0, 0, 0, 0, time.Local))
+				hasDefaultDayTo = true
+			}
+		}
+	}
+	if !hasDefaultDayTo {
+		this.Data["defaultDayTo"] = timeutil.Format("Y-m-d", time.Now().AddDate(20, 0, 0))
+	}
+
+	this.Show()
+}
+
+func (this *RenewPopupAction) RunPost(params struct {
+	UserPlanId int64
+
+	Period       string
+	CountMonths  int32
+	CountSeasons int32
+	CountYears   int32
+
+	DayTo string
+
+	Must *actions.Must
+	CSRF *actionutils.CSRF
+}) {
+	defer this.CreateLogInfo(codes.UserPlan_LogRenewUserPlan, params.UserPlanId)
+
+	userPlanResp, err := this.RPC().UserPlanRPC().FindEnabledUserPlan(this.AdminContext(), &pb.FindEnabledUserPlanRequest{UserPlanId: params.UserPlanId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var userPlan = userPlanResp.UserPlan
+	if userPlan == nil {
+		this.Fail("找不到要修改的套餐")
+	}
+
+	var planId = userPlan.PlanId
+	var userId = userPlan.UserId
+
+	// 检查余额
+	planResp, err := this.RPC().PlanRPC().FindEnabledPlan(this.AdminContext(), &pb.FindEnabledPlanRequest{PlanId: planId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+	var plan = planResp.Plan
+	if plan == nil {
+		this.Fail("找不到要购买的套餐")
+	}
+
+	var countPeriod int32
+	if plan.PriceType == serverconfigs.PlanPriceTypePeriod {
+		var cost float64
+
+		switch params.Period {
+		case "monthly":
+			countPeriod = params.CountMonths
+			cost = plan.MonthlyPrice * float64(countPeriod)
+		case "seasonally":
+			countPeriod = params.CountSeasons
+			cost = plan.SeasonallyPrice * float64(countPeriod)
+		case "yearly":
+			countPeriod = params.CountYears
+			cost = plan.YearlyPrice * float64(countPeriod)
+		default:
+			this.Fail("invalid period '" + params.Period + "'")
+		}
+
+		// 账号
+		accountResp, err := this.RPC().UserAccountRPC().FindEnabledUserAccountWithUserId(this.AdminContext(), &pb.FindEnabledUserAccountWithUserIdRequest{UserId: userId})
+		if err != nil {
+			this.ErrorPage(err)
+			return
+		}
+		var account = accountResp.UserAccount
+		if account == nil {
+			this.Fail("找不到用户'" + types.String(userId) + "'的账户")
+		}
+		if account.Total < cost {
+			this.Fail("用户账户余额不足")
+		}
+	} else if plan.PriceType == serverconfigs.PlanPriceTypeTraffic {
+		params.Must.Field("dayTo", params.DayTo).
+			Match(`^\d+-\d+-\d+$`, "请输入正确的结束日期")
+	} else if plan.PriceType == serverconfigs.PlanPriceTypeBandwidth {
+		params.Must.Field("dayTo", params.DayTo).
+			Match(`^\d+-\d+-\d+$`, "请输入正确的结束日期")
+	} else {
+		this.Fail("不支持的付款方式：" + plan.PriceType)
+	}
+
+	_, err = this.RPC().UserPlanRPC().RenewUserPlan(this.AdminContext(), &pb.RenewUserPlanRequest{
+		UserPlanId:  params.UserPlanId,
+		DayTo:       params.DayTo,
+		Period:      params.Period,
+		CountPeriod: countPeriod,
+	})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	this.Success()
+}

EdgeAdmin/internal/web/actions/default/plans/userPlans/userAccount.go

@@ -0,0 +1,35 @@
+// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build plus
+
+package userPlans
+
+import (
+	"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/iwind/TeaGo/maps"
+)
+
+type UserAccountAction struct {
+	actionutils.ParentAction
+}
+
+func (this *UserAccountAction) RunPost(params struct {
+	UserId int64
+}) {
+	resp, err := this.RPC().UserAccountRPC().FindEnabledUserAccountWithUserId(this.AdminContext(), &pb.FindEnabledUserAccountWithUserIdRequest{UserId: params.UserId})
+	if err != nil {
+		this.ErrorPage(err)
+		return
+	}
+
+	var account = resp.UserAccount
+	if account == nil {
+		account = &pb.UserAccount{}
+	}
+	this.Data["account"] = maps.Map{
+		"total":       account.Total,
+		"totalFrozen": account.TotalFrozen,
+	}
+
+	this.Success()
+}