neko/server/internal/capture/broadcast.go

package capture

import (
	"sync"

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

	"m1k1o/neko/internal/capture/gst"
	"m1k1o/neko/internal/config"
)

type BroadcastManager struct {
	mu       sync.Mutex
	logger   zerolog.Logger
	pipeline *gst.Pipeline
	capture  *config.Capture
	config   *config.Broadcast
	enabled  bool
	url      string
}

func NewBroadcast(capture *config.Capture, config *config.Broadcast) *BroadcastManager {
	return &BroadcastManager{
		logger:  log.With().Str("module", "broadcast").Logger(),
		capture: capture,
		config:  config,
		enabled: config.Enabled,
		url:     config.URL,
	}
}

func (manager *BroadcastManager) Shutdown() error {
	manager.Destroy()
	return nil
}

func (manager *BroadcastManager) Start() error {
	if !manager.enabled || manager.IsActive() {
		return nil
	}

	var err error
	manager.pipeline, err = CreateRTMPPipeline(
		manager.capture.Device,
		manager.capture.Display,
		manager.config.Pipeline,
		manager.url,
	)

	if err != nil {
		manager.pipeline = nil
		return err
	}

	manager.logger.Info().
		Str("audio_device", manager.capture.Device).
		Str("video_display", manager.capture.Display).
		Str("rtmp_pipeline_src", manager.pipeline.Src).
		Msgf("RTMP pipeline is starting...")

	manager.pipeline.Play()
	return nil
}

func (manager *BroadcastManager) Stop() {
	if !manager.IsActive() {
		return
	}

	manager.pipeline.Stop()
	manager.pipeline = nil
}

func (manager *BroadcastManager) IsActive() bool {
	return manager.pipeline != nil
}

func (manager *BroadcastManager) Create(url string) error {
	manager.mu.Lock()
	defer manager.mu.Unlock()

	manager.url = url
	manager.enabled = true

	err := manager.Start()
	if err != nil {
		manager.enabled = false
	}

	return err
}

func (manager *BroadcastManager) Destroy() {
	manager.mu.Lock()
	defer manager.mu.Unlock()

	manager.Stop()
	manager.enabled = false
}

func (manager *BroadcastManager) GetUrl() string {
	return manager.url
}