refactor capture with broadcast.

server/internal/capture/broadcast.go

@@ -7,98 +7,115 @@ import (
 	"github.com/rs/zerolog/log"
 
 	"m1k1o/neko/internal/capture/gst"
-	"m1k1o/neko/internal/config"
+	"m1k1o/neko/internal/types"
 )
 
-type BroadcastManager struct {
-	mu       sync.Mutex
-	logger   zerolog.Logger
-	pipeline *gst.Pipeline
-	capture  *config.Capture
-	config   *config.Broadcast
-	enabled  bool
-	url      string
+type BroacastManagerCtx struct {
+	logger zerolog.Logger
+	mu     sync.Mutex
+
+	pipeline   *gst.Pipeline
+	pipelineMu sync.Mutex
+	pipelineFn func(url string) (*gst.Pipeline, error)
+
+	url     string
+	started bool
 }
 
-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 broadcastNew(pipelineFn func(url string) (*gst.Pipeline, error), url string, started bool) *BroacastManagerCtx {
+	logger := log.With().
+		Str("module", "capture").
+		Str("submodule", "broadcast").
+		Logger()
+
+	return &BroacastManagerCtx{
+		logger:     logger,
+		pipelineFn: pipelineFn,
+		url:        url,
+		started:    started,
 	}
 }
 
-func (manager *BroadcastManager) Shutdown() error {
-	manager.Destroy()
+func (manager *BroacastManagerCtx) shutdown() {
+	manager.logger.Info().Msgf("shutdown")
+
+	manager.destroyPipeline()
+}
+
+func (manager *BroacastManagerCtx) Start(url string) error {
+	manager.mu.Lock()
+	defer manager.mu.Unlock()
+
+	err := manager.createPipeline()
+	if err != nil {
+		return err
+	}
+
+	manager.url = url
+	manager.started = true
 	return nil
 }
 
-func (manager *BroadcastManager) Start() error {
-	if !manager.enabled || manager.IsActive() {
-		return nil
+func (manager *BroacastManagerCtx) Stop() {
+	manager.mu.Lock()
+	defer manager.mu.Unlock()
+
+	manager.started = false
+	manager.destroyPipeline()
+}
+
+func (manager *BroacastManagerCtx) Started() bool {
+	manager.mu.Lock()
+	defer manager.mu.Unlock()
+
+	return manager.started
+}
+
+func (manager *BroacastManagerCtx) Url() string {
+	manager.mu.Lock()
+	defer manager.mu.Unlock()
+
+	return manager.url
+}
+
+func (manager *BroacastManagerCtx) createPipeline() error {
+	manager.pipelineMu.Lock()
+	defer manager.pipelineMu.Unlock()
+
+	if manager.pipeline != nil {
+		return types.ErrCapturePipelineAlreadyExists
 	}
 
 	var err error
-	manager.pipeline, err = CreateRTMPPipeline(
-		manager.capture.Device,
-		manager.capture.Display,
-		manager.config.Pipeline,
-		manager.url,
-	)
 
+	manager.logger.Info().
+		Str("url", manager.url).
+		Msgf("creating pipeline")
+
+	manager.pipeline, err = manager.pipelineFn(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...")
+		Str("url", manager.url).
+		Str("src", manager.pipeline.Src).
+		Msgf("created pipeline")
 
 	manager.pipeline.Play()
+
 	return nil
 }
 
-func (manager *BroadcastManager) Stop() {
-	if !manager.IsActive() {
+func (manager *BroacastManagerCtx) destroyPipeline() {
+	manager.pipelineMu.Lock()
+	defer manager.pipelineMu.Unlock()
+
+	if manager.pipeline == nil {
 		return
 	}
 
-	manager.pipeline.Stop()
+	manager.pipeline.Destroy()
+	manager.logger.Info().Msgf("destroying pipeline")
 	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
-}