stillbox/pkg/calls/call.go

package calls

import (
	"context"
	"fmt"
	"time"

	"dynatron.me/x/stillbox/internal/audio"
	"dynatron.me/x/stillbox/pkg/auth"
	"dynatron.me/x/stillbox/pkg/database"
	"dynatron.me/x/stillbox/pkg/pb"
	"dynatron.me/x/stillbox/pkg/talkgroups"

	"github.com/google/uuid"
	"google.golang.org/protobuf/types/known/timestamppb"
)

type CallDuration time.Duration

func (d CallDuration) Duration() time.Duration {
	return time.Duration(d)
}

func (d CallDuration) MsInt32Ptr() *int32 {
	if time.Duration(d) == 0 {
		return nil
	}

	i := int32(time.Duration(d).Milliseconds())
	return &i
}

func (d CallDuration) Seconds() int32 {
	return int32(time.Duration(d).Seconds())
}

type Call struct {
	ID             uuid.UUID    `form:"-"`
	Audio          []byte       `form:"audio" filenameField:"AudioName"`
	AudioName      string       `form:"audioName"`
	AudioType      string       `form:"audioType"`
	Duration       CallDuration `form:"-"`
	DateTime       time.Time    `form:"dateTime"`
	Frequencies    []int        `form:"frequencies"`
	Frequency      int          `form:"frequency"`
	Patches        []int        `form:"patches"`
	Source         int          `form:"source"`
	Sources        []int        `form:"sources"`
	System         int          `form:"system"`
	Submitter      *auth.UserID `form:"-"`
	SystemLabel    string       `form:"systemLabel"`
	Talkgroup      int          `form:"talkgroup"`
	TalkgroupGroup *string      `form:"talkgroupGroup"`
	TalkgroupLabel *string      `form:"talkgroupLabel"`
	TGAlphaTag     *string      `form:"talkgroupTag"` // not 1:1

	shouldStore bool `form:"-"`
}

func (c *Call) String() string {
	return fmt.Sprintf("%s to %d from %d", c.AudioName, c.Talkgroup, c.Source)
}

func (c *Call) ShouldStore() bool {
	return c.shouldStore
}

func Make(call *Call, dontStore bool) (*Call, error) {
	err := call.computeLength()
	if err != nil {
		return nil, err
	}

	call.shouldStore = dontStore
	call.ID = uuid.New()

	return call, nil
}

func toInt64Slice(s []int) []int64 {
	n := make([]int64, len(s))
	for i := range s {
		n[i] = int64(s[i])
	}

	return n
}

func toInt32Slice(s []int) []int32 {
	n := make([]int32, len(s))
	for i := range s {
		n[i] = int32(s[i])
	}

	return n
}

func (c *Call) ToPB() *pb.Call {
	return &pb.Call{
		Id:          c.ID.String(),
		AudioName:   c.AudioName,
		AudioType:   c.AudioType,
		DateTime:    timestamppb.New(c.DateTime),
		System:      int32(c.System),
		Talkgroup:   int32(c.Talkgroup),
		Source:      int32(c.Source),
		Frequency:   int64(c.Frequency),
		Frequencies: toInt64Slice(c.Frequencies),
		Patches:     toInt32Slice(c.Patches),
		Sources:     toInt32Slice(c.Sources),
		Duration:    c.Duration.MsInt32Ptr(),
		Audio:       c.Audio,
	}
}

func (c *Call) LearnTG(ctx context.Context, db database.Store) (learnedId int, err error) {
	err = db.AddTalkgroupWithLearnedFlag(ctx, int32(c.System), int32(c.Talkgroup))
	if err != nil {
		return 0, fmt.Errorf("addTalkgroupWithLearnedFlag: %w", err)
	}

	return db.AddLearnedTalkgroup(ctx, database.AddLearnedTalkgroupParams{
		SystemID: c.System,
		TGID:     c.Talkgroup,
		Name:     c.TalkgroupLabel,
		AlphaTag: c.TGAlphaTag,
		TGGroup:  c.TalkgroupGroup,
	})
}

func (c *Call) computeLength() (err error) {
	var td time.Duration

	switch c.AudioType {
	case "audio/mpeg":
		td, err = audio.MP3Duration(c.Audio)
		if err != nil {
			return fmt.Errorf("mp3: %w", err)
		}
	case "audio/wav":
		td, err = audio.WAVDuration(c.Audio)
		if err != nil {
			return fmt.Errorf("wav: %w", err)
		}
	default:
		return fmt.Errorf("length not implemented for mime type %s", c.AudioType)
	}

	c.Duration = CallDuration(td)

	return nil
}

func (c *Call) TalkgroupTuple() talkgroups.ID {
	return talkgroups.TG(c.System, c.Talkgroup)
}