124 lines
2.3 KiB
Go
124 lines
2.3 KiB
Go
package capture
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"m1k1o/neko/internal/capture/gst"
|
|
"m1k1o/neko/internal/types"
|
|
)
|
|
|
|
type BroacastManagerCtx struct {
|
|
logger zerolog.Logger
|
|
mu sync.Mutex
|
|
sampleChannel chan types.Sample
|
|
|
|
pipeline *gst.Pipeline
|
|
pipelineMu sync.Mutex
|
|
pipelineFn func(url string) (string, error)
|
|
|
|
url string
|
|
started bool
|
|
}
|
|
|
|
func broadcastNew(pipelineFn func(url string) (string, error), defaultUrl string) *BroacastManagerCtx {
|
|
logger := log.With().
|
|
Str("module", "capture").
|
|
Str("submodule", "broadcast").
|
|
Logger()
|
|
|
|
return &BroacastManagerCtx{
|
|
logger: logger,
|
|
pipelineFn: pipelineFn,
|
|
sampleChannel: make(chan types.Sample),
|
|
url: defaultUrl,
|
|
started: defaultUrl != "",
|
|
}
|
|
}
|
|
|
|
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 *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
|
|
pipelineStr, err := manager.pipelineFn(manager.url)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
manager.logger.Info().
|
|
Str("url", manager.url).
|
|
Str("src", pipelineStr).
|
|
Msgf("starting pipeline")
|
|
|
|
manager.pipeline, err = gst.CreatePipeline(pipelineStr, manager.sampleChannel)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
manager.pipeline.Play()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (manager *BroacastManagerCtx) destroyPipeline() {
|
|
manager.pipelineMu.Lock()
|
|
defer manager.pipelineMu.Unlock()
|
|
|
|
if manager.pipeline == nil {
|
|
return
|
|
}
|
|
|
|
manager.pipeline.Destroy()
|
|
manager.logger.Info().Msgf("destroying pipeline")
|
|
manager.pipeline = nil
|
|
}
|