stillbox/pkg/gordio/sinks/database.go

package sinks

import (
	"context"
	"fmt"

	"dynatron.me/x/stillbox/internal/common"
	"dynatron.me/x/stillbox/pkg/gordio/calls"
	"dynatron.me/x/stillbox/pkg/gordio/database"

	"github.com/rs/zerolog/log"
)

type DatabaseSink struct {
}

func NewDatabaseSink() *DatabaseSink {
	return &DatabaseSink{}
}

func (s *DatabaseSink) Call(ctx context.Context, call *calls.Call) error {
	db := database.FromCtx(ctx)

	dbCall, err := db.AddCall(ctx, s.toAddCallParams(call))
	if err != nil {
		return fmt.Errorf("add call: %w", err)
	}

	log.Debug().Str("id", dbCall.String()).Int("system", call.System).Int("tgid", call.Talkgroup).Msg("stored")

	return nil
}

func (s *DatabaseSink) SinkType() string {
	return "database"
}

func (s *DatabaseSink) toAddCallParams(call *calls.Call) database.AddCallParams {
	return database.AddCallParams{
		Submitter:   call.Submitter.Int32Ptr(),
		System:      call.System,
		Talkgroup:   call.Talkgroup,
		CallDate:    call.DateTime,
		AudioName:   common.PtrOrNull(call.AudioName),
		AudioBlob:   call.Audio,
		AudioType:   common.PtrOrNull(call.AudioType),
		Frequency:   call.Frequency,
		Frequencies: call.Frequencies,
		Patches:     call.Patches,
		TgLabel:     call.TalkgroupLabel,
		TgTag:       call.TalkgroupTag,
		TgGroup:     call.TalkgroupGroup,
		Source:      call.Source,
	}
}