package portal

import (
	"context"
	"encoding/json"

	"github.com/perfect-panel/server/internal/svc"
	"github.com/perfect-panel/server/internal/types"
	"github.com/perfect-panel/server/pkg/logger"
	"github.com/perfect-panel/server/pkg/tool"
	"github.com/perfect-panel/server/pkg/xerr"
	"github.com/pkg/errors"
)

type GetSubscriptionLogic struct {
	logger.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

// NewGetSubscriptionLogic Get Subscription
func NewGetSubscriptionLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetSubscriptionLogic {
	return &GetSubscriptionLogic{
		Logger: logger.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *GetSubscriptionLogic) GetSubscription() (resp *types.GetSubscriptionResponse, err error) {
	resp = &types.GetSubscriptionResponse{
		List: make([]types.Subscribe, 0),
	}
	// Get the subscription list
	data, err := l.svcCtx.SubscribeModel.QuerySubscribeListByShow(l.ctx)
	if err != nil {
		l.Errorw("[Site GetSubscription]", logger.Field("err", err.Error()))
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "get subscription list error: %v", err.Error())
	}
	list := make([]types.Subscribe, len(data))
	for i, item := range data {
		var sub types.Subscribe
		tool.DeepCopy(&sub, item)
		if item.Discount != "" {
			var discount []types.SubscribeDiscount

			_ = json.Unmarshal([]byte(item.Discount), &discount)
			sub.Discount = discount
		}

		// 计算节点数量（通过服务组查询关联的实际节点数量）
		if item.ServerGroup != "" {
			// 获取服务组ID列表
			groupIds := tool.StringToInt64Slice(item.ServerGroup)

			// 通过服务组查询关联的节点数量
			servers, err := l.svcCtx.ServerModel.FindServerListByGroupIds(l.ctx, groupIds)
			if err != nil {
				l.Errorw("[Site GetSubscription] FindServerListByGroupIds error", logger.Field("error", err.Error()))
				sub.ServerCount = 0
			} else {
				sub.ServerCount = int64(len(servers))
			}
		}

		list[i] = sub
	}
	resp.List = list
	return
}