package logic

import (
	"context"
	"lone-services/pkg/modelbase"
	"lone-services/pkg/utils"
	"lone-services/services/equipment/internal/dao"
	"lone-services/services/equipment/internal/model"

	"lone-services/rpc/equipment/pb"
	"lone-services/services/equipment/internal/svc"

	"github.com/zeromicro/go-zero/core/logx"
)

type ItemsLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
	BaseLogic
}

func NewItemsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ItemsLogic {
	return &ItemsLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *ItemsLogic) Items(in *equipment.ItemsRequest) (*equipment.Response, error) {
	modelObj := model.VersionModel{}.Init()
	w := modelbase.Params{
		Page: int(in.Page),
		Size: int(in.Size),
	}
	if in.Type != utils.NumberZero {
		w.Eq = map[string]string{"type": string(in.Type)}
	}

	var data []dao.VersionInfo
	err := modelObj.Items(w, &data)
	if err != nil {
		return l.fail(utils.ErrorNotFund)
	}

	if len(data) < utils.NumberOne {
		return l.ok(data)
	}

	return l.ok(data)
}