libs5-go/protocol/message.go

package protocol

import (
	"git.lumeweb.com/LumeWeb/libs5-go/protocol/base"
	"git.lumeweb.com/LumeWeb/libs5-go/protocol/signed"
	"git.lumeweb.com/LumeWeb/libs5-go/types"
	"sync"
)

var (
	messageTypes sync.Map
)

var (
	_ base.IncomingMessage = (*base.IncomingMessageImpl)(nil)
)

func init() {
	messageTypes = sync.Map{}

	// Register factory functions instead of instances
	RegisterMessageType(types.ProtocolMethodHandshakeOpen, func() base.IncomingMessage {
		return NewHandshakeOpen([]byte{}, "")
	})
	RegisterMessageType(types.ProtocolMethodSignedMessage, func() base.IncomingMessage {
		return signed.NewSignedMessage()
	})
}

func RegisterMessageType(messageType types.ProtocolMethod, factoryFunc func() base.IncomingMessage) {
	if factoryFunc == nil {
		panic("factoryFunc cannot be nil")
	}
	messageTypes.Store(messageType, factoryFunc)
}

func GetMessageType(kind types.ProtocolMethod) (base.IncomingMessage, bool) {
	value, ok := messageTypes.Load(kind)
	if !ok {
		return nil, false
	}

	factoryFunc, ok := value.(func() base.IncomingMessage)
	if !ok {
		return nil, false
	}

	return factoryFunc(), true
}