package agent

import (
	"context"
	v1 "git.5iot.com/iotlcm-public/iotlcm-admin-svr/api/agent/v1"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/service"
)

// List 查看企业客户单位信息列表
func (c *ControllerV1) TAgentEnterpriseList(ctx context.Context, req *v1.TAgentEnterpriseListReq) (res *v1.TAgentEnterpriseListRes, err error) {
	list, totalCount, err := service.AgentEnterprise().List(ctx, &req.TAgentEnterpriseListParam)
	if err != nil {
		return
	}

	if list == nil {
		list = []*model.TAgentEnterpriseListModel{}
	}

	res = new(v1.TAgentEnterpriseListRes)
	res.Items = list
	res.PageResult.Page = req.Page
	res.PageResult.PageSize = req.PageSize
	res.PageResult.Total = totalCount
	return
}

// Export 导出企业客户单位信息列表
func (c *ControllerV1) TAgentEnterpriseExport(ctx context.Context, req *v1.TAgentEnterpriseExportReq) (res *v1.TAgentEnterpriseExportRes, err error) {
	err = service.AgentEnterprise().Export(ctx, &req.TAgentEnterpriseListParam)
	return
}

// Edit 更新企业客户单位信息
func (c *ControllerV1) TAgentEnterpriseEdit(ctx context.Context, req *v1.TAgentEnterpriseEditReq) (res *v1.TAgentEnterpriseEditRes, err error) {
	err = service.AgentEnterprise().Edit(ctx, &req.TAgentEnterpriseEditParam)
	return
}

// View 获取指定企业客户单位信息信息
func (c *ControllerV1) TAgentEnterpriseView(ctx context.Context, req *v1.TAgentEnterpriseViewReq) (res *v1.TAgentEnterpriseViewRes, err error) {
	data, err := service.AgentEnterprise().View(ctx, &req.TAgentEnterpriseViewParam)
	if err != nil {
		return
	}

	res = new(v1.TAgentEnterpriseViewRes)
	res.TAgentEnterpriseViewModel = data
	return
}

// Delete 删除企业客户单位信息
func (c *ControllerV1) TAgentEnterpriseDelete(ctx context.Context, req *v1.TAgentEnterpriseDeleteReq) (res *v1.TAgentEnterpriseDeleteRes, err error) {
	err = service.AgentEnterprise().Delete(ctx, &req.TAgentEnterpriseDeleteParam)
	return
}