105 lines
1.9 KiB
Go
105 lines
1.9 KiB
Go
package capture
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"m1k1o/neko/internal/capture/gst"
|
|
"m1k1o/neko/internal/types/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 = gst.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
|
|
}
|