supermq/bootstrap/events/consumer/streams.go

// Copyright (c) Abstract Machines
// SPDX-License-Identifier: Apache-2.0

package consumer

import (
	"context"
	"time"

	"github.com/absmach/magistrala/bootstrap"
	svcerr "github.com/absmach/supermq/pkg/errors/service"
	"github.com/absmach/supermq/pkg/events"
)

const (
	clientRemove     = "client.remove"
	clientConnect    = "group.assign"
	clientDisconnect = "group.unassign"

	channelPrefix = "channels."
	channelUpdate = channelPrefix + "update"
	channelRemove = channelPrefix + "remove"

	memberKind = "client"
	relation   = "group"
)

type eventHandler struct {
	svc bootstrap.Service
}

// NewEventHandler returns new event store handler.
func NewEventHandler(svc bootstrap.Service) events.EventHandler {
	return &eventHandler{
		svc: svc,
	}
}

func (es *eventHandler) Handle(ctx context.Context, event events.Event) error {
	msg, err := event.Encode()
	if err != nil {
		return err
	}

	switch msg["operation"] {
	case clientRemove:
		rte := decodeRemoveClient(msg)
		err = es.svc.RemoveConfigHandler(ctx, rte.id)
	case clientConnect:
		cte := decodeConnectClient(msg)
		if cte.channelID == "" || len(cte.clientIDs) == 0 {
			return svcerr.ErrMalformedEntity
		}
		for _, clientID := range cte.clientIDs {
			if clientID == "" {
				return svcerr.ErrMalformedEntity
			}
			if err := es.svc.ConnectClientHandler(ctx, cte.channelID, clientID); err != nil {
				return err
			}
		}
	case clientDisconnect:
		dte := decodeDisconnectClient(msg)
		if dte.channelID == "" || len(dte.clientIDs) == 0 {
			return svcerr.ErrMalformedEntity
		}
		for _, clientID := range dte.clientIDs {
			if clientID == "" {
				return svcerr.ErrMalformedEntity
			}
		}

		for _, c := range dte.clientIDs {
			if err = es.svc.DisconnectClientHandler(ctx, dte.channelID, c); err != nil {
				return err
			}
		}
	case channelUpdate:
		uce := decodeUpdateChannel(msg)
		err = es.handleUpdateChannel(ctx, uce)
	case channelRemove:
		rce := decodeRemoveChannel(msg)
		err = es.svc.RemoveChannelHandler(ctx, rce.id)
	}
	if err != nil {
		return err
	}

	return nil
}

func decodeRemoveClient(event map[string]any) removeEvent {
	return removeEvent{
		id: events.Read(event, "id", ""),
	}
}

func decodeUpdateChannel(event map[string]any) updateChannelEvent {
	metadata := events.Read(event, "metadata", map[string]any{})

	return updateChannelEvent{
		id:        events.Read(event, "id", ""),
		name:      events.Read(event, "name", ""),
		metadata:  metadata,
		updatedAt: events.Read(event, "updated_at", time.Now()),
		updatedBy: events.Read(event, "updated_by", ""),
	}
}

func decodeRemoveChannel(event map[string]any) removeEvent {
	return removeEvent{
		id: events.Read(event, "id", ""),
	}
}

func decodeConnectClient(event map[string]any) connectionEvent {
	if events.Read(event, "memberKind", "") != memberKind && events.Read(event, "relation", "") != relation {
		return connectionEvent{}
	}

	return connectionEvent{
		channelID: events.Read(event, "group_id", ""),
		clientIDs: events.ReadStringSlice(event, "member_ids"),
	}
}

func decodeDisconnectClient(event map[string]any) connectionEvent {
	if events.Read(event, "memberKind", "") != memberKind && events.Read(event, "relation", "") != relation {
		return connectionEvent{}
	}

	return connectionEvent{
		channelID: events.Read(event, "group_id", ""),
		clientIDs: events.ReadStringSlice(event, "member_ids"),
	}
}

func (es *eventHandler) handleUpdateChannel(ctx context.Context, uce updateChannelEvent) error {
	channel := bootstrap.Channel{
		ID:        uce.id,
		Name:      uce.name,
		Metadata:  uce.metadata,
		UpdatedAt: uce.updatedAt,
		UpdatedBy: uce.updatedBy,
	}

	return es.svc.UpdateChannelHandler(ctx, channel)
}