package nats

import (
	"context"
	"github.com/gogf/gf/v2/errors/gerror"
)

// NatsMessageConfig nats Stream 消息配置
type NatsMessageConfig struct {
	CreateTaskStreamName   string
	CreateTaskSubjects     []string
	PublishSubject         string
	CreateTaskConsumerName string
	MsgCount               int
	HandleFunc             func(ctx context.Context, message map[string]interface{}) error
}

// MessageConfig 消息配置接口
type MessageConfig interface {
	createTaskStream(ctx context.Context) error
	publish(ctx context.Context, data interface{}) error
	createTaskConsumer(ctx context.Context) error
	//startConsumer(ctx context.Context, handleFunc func(ctx context.Context, msg *nats.Msg) error) error
}

func (n *NatsMessageConfig) createTaskStream(ctx context.Context) error {
	return createTaskStreamSimple(ctx, n.CreateTaskStreamName, n.CreateTaskSubjects)
}

// CreateTaskStreamBatch 批量创建任务消息队列流
func CreateTaskStreamBatch(ctx context.Context, configs ...MessageConfig) error {
	for _, cfg := range configs {
		if err := cfg.createTaskStream(ctx); err != nil {
			return gerror.Wrap(err, "创建任务消息队列流失败")
		}
	}
	return nil
}

func (n *NatsMessageConfig) publish(ctx context.Context, data interface{}) error {
	return publish(ctx, n.PublishSubject, data)
}

// PublishMessage 发布消息（统一入口）
func PublishMessage(ctx context.Context, cfg MessageConfig, data interface{}) (err error) {
	return cfg.publish(ctx, data)
}

func (n *NatsMessageConfig) createTaskConsumer(ctx context.Context) error {
	return CreateConsumerPushMode(ctx, n.CreateTaskStreamName, n.CreateTaskConsumerName, n.PublishSubject, n.MsgCount)
}

// CreateTaskConsumerBatch 批量创建任务消息队列消费者
func CreateTaskConsumerBatch(ctx context.Context, configs ...MessageConfig) error {
	for _, cfg := range configs {
		if err := cfg.createTaskConsumer(ctx); err != nil {
			return gerror.Wrap(err, "创建任务消息队列流失败")
		}
	}
	return nil
}

//func (n *NatsMessageConfig) startConsumer(ctx context.Context, handleFunc func(ctx context.Context, msg *nats.Msg) error) error {
//	return ConsumeMessages(ctx, n.CreateTaskStreamName, n.CreateTaskConsumerName, handleFunc)
//}