1.4.5.2

EdgeAPI/internal/rpc/services/nameservers/service_ns_plan.go

@@ -0,0 +1,191 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+//go:build plus
+
+package nameservers
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models/nameservers"
+	"github.com/TeaOSLab/EdgeAPI/internal/rpc/services"
+	"github.com/TeaOSLab/EdgeCommon/pkg/dnsconfigs"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+// NSPlanService DNS套餐服务
+type NSPlanService struct {
+	services.BaseService
+}
+
+// CreateNSPlan 创建DNS套餐
+func (this *NSPlanService) CreateNSPlan(ctx context.Context, req *pb.CreateNSPlanRequest) (*pb.CreateNSPlanResponse, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+
+	if len(req.ConfigJSON) == 0 {
+		return nil, errors.New("invalid 'configJSON'")
+	}
+
+	var config = dnsconfigs.DefaultNSPlanConfig()
+	err = json.Unmarshal(req.ConfigJSON, config)
+	if err != nil {
+		return nil, errors.New("decode 'configJSON' failed: " + err.Error())
+	}
+
+	planId, err := nameservers.SharedNSPlanDAO.CreatePlan(tx, req.Name, req.MonthlyPrice, req.YearlyPrice, config)
+	if err != nil {
+		return nil, err
+	}
+
+	return &pb.CreateNSPlanResponse{NsPlanId: planId}, nil
+}
+
+// UpdateNSPlan 修改DNS套餐
+func (this *NSPlanService) UpdateNSPlan(ctx context.Context, req *pb.UpdateNSPlanRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+
+	if len(req.ConfigJSON) == 0 {
+		return nil, errors.New("invalid 'configJSON'")
+	}
+
+	var config = dnsconfigs.DefaultNSPlanConfig()
+	err = json.Unmarshal(req.ConfigJSON, config)
+	if err != nil {
+		return nil, errors.New("decode 'configJSON' failed: " + err.Error())
+	}
+
+	err = nameservers.SharedNSPlanDAO.UpdatePlan(tx, req.NsPlanId, req.Name, req.IsOn, req.MonthlyPrice, req.YearlyPrice, config)
+	if err != nil {
+		return nil, err
+	}
+
+	return this.Success()
+}
+
+// SortNSPlanOrders 修改DNS套餐顺序
+func (this *NSPlanService) SortNSPlanOrders(ctx context.Context, req *pb.SortNSPlansRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	err = nameservers.SharedNSPlanDAO.UpdatePlanOrders(tx, req.NsPlanIds)
+	if err != nil {
+		return nil, err
+	}
+
+	return this.Success()
+}
+
+// FindAllNSPlans 查找所有DNS套餐
+func (this *NSPlanService) FindAllNSPlans(ctx context.Context, req *pb.FindAllNSPlansRequest) (*pb.FindAllNSPlansResponse, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+
+	var pbPlans = []*pb.NSPlan{}
+	plans, err := nameservers.SharedNSPlanDAO.FindAllPlans(tx)
+	if err != nil {
+		return nil, err
+	}
+	for _, plan := range plans {
+		pbPlans = append(pbPlans, &pb.NSPlan{
+			Id:           int64(plan.Id),
+			Name:         plan.Name,
+			IsOn:         plan.IsOn,
+			MonthlyPrice: float32(plan.MonthlyPrice),
+			YearlyPrice:  float32(plan.YearlyPrice),
+			ConfigJSON:   plan.Config,
+		})
+	}
+
+	return &pb.FindAllNSPlansResponse{NsPlans: pbPlans}, nil
+}
+
+// FindAllEnabledNSPlans 查找所有可用DNS套餐
+func (this *NSPlanService) FindAllEnabledNSPlans(ctx context.Context, req *pb.FindAllEnabledNSPlansRequest) (*pb.FindAllEnabledNSPlansResponse, error) {
+	_, _, err := this.ValidateAdminAndUser(ctx, true)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+
+	var pbPlans = []*pb.NSPlan{}
+	plans, err := nameservers.SharedNSPlanDAO.FindAllEnabledPlans(tx)
+	if err != nil {
+		return nil, err
+	}
+	for _, plan := range plans {
+		pbPlans = append(pbPlans, &pb.NSPlan{
+			Id:           int64(plan.Id),
+			Name:         plan.Name,
+			IsOn:         plan.IsOn,
+			MonthlyPrice: float32(plan.MonthlyPrice),
+			YearlyPrice:  float32(plan.YearlyPrice),
+			ConfigJSON:   plan.Config,
+		})
+	}
+
+	return &pb.FindAllEnabledNSPlansResponse{NsPlans: pbPlans}, nil
+}
+
+// FindNSPlan 查找DNS套餐
+func (this *NSPlanService) FindNSPlan(ctx context.Context, req *pb.FindNSPlanRequest) (*pb.FindNSPlanResponse, error) {
+	_, _, err := this.ValidateAdminAndUser(ctx, true)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	plan, err := nameservers.SharedNSPlanDAO.FindEnabledNSPlan(tx, req.NsPlanId)
+	if err != nil {
+		return nil, err
+	}
+
+	if plan == nil {
+		return &pb.FindNSPlanResponse{}, nil
+	}
+
+	return &pb.FindNSPlanResponse{
+		NsPlan: &pb.NSPlan{
+			Id:           int64(plan.Id),
+			Name:         plan.Name,
+			IsOn:         plan.IsOn,
+			MonthlyPrice: float32(plan.MonthlyPrice),
+			YearlyPrice:  float32(plan.YearlyPrice),
+			ConfigJSON:   plan.Config,
+		},
+	}, nil
+}
+
+// DeleteNSPlan 删除DNS套餐
+func (this *NSPlanService) DeleteNSPlan(ctx context.Context, req *pb.DeleteNSPlanRequest) (*pb.RPCSuccess, error) {
+	_, err := this.ValidateAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+
+	err = nameservers.SharedNSPlanDAO.DisableNSPlan(tx, req.NsPlanId)
+	if err != nil {
+		return nil, err
+	}
+
+	return this.Success()
+}