package proto

import (
	"go_dreamfactory/lego/utils/mapstructure"
)

type ProtoType uint8

const (
	Proto_Json ProtoType = 1
	Proto_Buff ProtoType = 2
)

type Option func(*Options)
type Options struct {
	MsgProtoType   ProtoType
	IsUseBigEndian bool
	MessageFactory IMessageFactory
}

func SetMsgProtoType(v ProtoType) Option {
	return func(o *Options) {
		o.MsgProtoType = v
	}
}

func SetIsUseBigEndian(v bool) Option {
	return func(o *Options) {
		o.IsUseBigEndian = v
	}
}

func SetMessageFactory(v IMessageFactory) Option {
	return func(o *Options) {
		o.MessageFactory = v
	}
}

func newOptions(config map[string]interface{}, opts ...Option) Options {
	options := Options{
		MsgProtoType:   Proto_Buff,
		IsUseBigEndian: false,
	}
	if config != nil {
		mapstructure.Decode(config, &options)
	}
	for _, o := range opts {
		o(&options)
	}
	if options.MessageFactory == nil {
		options.MessageFactory = new(DefMessageFactory)
	}
	return options
}

func newOptionsByOption(opts ...Option) Options {
	options := Options{
		MsgProtoType:   Proto_Buff,
		IsUseBigEndian: false,
	}
	for _, o := range opts {
		o(&options)
	}
	if options.MessageFactory == nil {
		options.MessageFactory = new(DefMessageFactory)
	}
	return options
}