package packagecustomer

import (
	"context"
	"strings"

	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/rabbitmq"
	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"

	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
)

func PublishSync(ctx context.Context, message model.PackageCustomerSyncMessage) error {
	message.TenantId = strings.TrimSpace(message.TenantId)
	message.Trigger = strings.TrimSpace(message.Trigger)
	message.SourceBatchNo = strings.TrimSpace(message.SourceBatchNo)
	if message.TenantId == "" {
		return gerror.New("package customer sync tenantId is empty")
	}
	if message.DeptId <= 0 {
		return gerror.New("package customer sync deptId is empty")
	}
	if message.Trigger == "" {
		message.Trigger = "manual"
	}
	if err := rabbitmq.Publish(ctx, consts.QueuePackageCustomerSync, message); err != nil {
		return gerror.Wrapf(err, "publish package customer sync failed, queue:%s", consts.QueuePackageCustomerSync)
	}
	g.Log().Infof(ctx, "package customer sync published, tenantId:%s, deptId:%d, trigger:%s, sourceBatchNo:%s",
		message.TenantId, message.DeptId, message.Trigger, message.SourceBatchNo)
	return nil
}

func PublishSyncSafe(ctx context.Context, message model.PackageCustomerSyncMessage) {
	if err := PublishSync(ctx, message); err != nil {
		g.Log().Warningf(ctx, "package customer sync publish skipped, tenantId:%s, deptId:%d, trigger:%s, sourceBatchNo:%s, err:%+v",
			message.TenantId, message.DeptId, message.Trigger, message.SourceBatchNo, err)
	}
}

func PublishSyncForDeptsSafe(ctx context.Context, tenantId string, userId int64, deptIds []int64, trigger, sourceBatchNo string) {
	seen := make(map[int64]struct{}, len(deptIds))
	for _, deptId := range deptIds {
		if deptId <= 0 {
			continue
		}
		if _, ok := seen[deptId]; ok {
			continue
		}
		seen[deptId] = struct{}{}
		PublishSyncSafe(ctx, model.PackageCustomerSyncMessage{
			TenantId:      tenantId,
			DeptId:        deptId,
			UserId:        userId,
			Trigger:       trigger,
			SourceBatchNo: sourceBatchNo,
		})
	}
}