package capture import ( "errors" "github.com/rs/zerolog" "github.com/rs/zerolog/log" "m1k1o/neko/internal/config" "m1k1o/neko/internal/types" ) type CaptureManagerCtx struct { logger zerolog.Logger desktop types.DesktopManager // sinks broadcast *BroacastManagerCtx audio *StreamSinkManagerCtx video *StreamSinkManagerCtx } func New(desktop types.DesktopManager, config *config.Capture) *CaptureManagerCtx { logger := log.With().Str("module", "capture").Logger() return &CaptureManagerCtx{ logger: logger, desktop: desktop, // sinks broadcast: broadcastNew(func(url string) (string, error) { return NewBroadcastPipeline(config.AudioDevice, config.Display, config.BroadcastPipeline, url) }, config.BroadcastUrl), audio: streamSinkNew(config.AudioCodec, func() (string, error) { return NewAudioPipeline(config.AudioCodec, config.AudioDevice, config.AudioPipeline, config.AudioBitrate) }, "audio"), video: streamSinkNew(config.VideoCodec, func() (string, error) { return NewVideoPipeline(config.VideoCodec, config.Display, config.VideoPipeline, config.VideoMaxFPS, config.VideoBitrate, config.VideoHWEnc) }, "video"), } } func (manager *CaptureManagerCtx) Start() { if manager.broadcast.Started() { if err := manager.broadcast.createPipeline(); err != nil { manager.logger.Panic().Err(err).Msg("unable to create broadcast pipeline") } } manager.desktop.OnBeforeScreenSizeChange(func() { if manager.video.Started() { manager.video.destroyPipeline() } if manager.broadcast.Started() { manager.broadcast.destroyPipeline() } }) manager.desktop.OnAfterScreenSizeChange(func() { if manager.video.Started() { err := manager.video.createPipeline() if err != nil && !errors.Is(err, types.ErrCapturePipelineAlreadyExists) { manager.logger.Panic().Err(err).Msg("unable to recreate video pipeline") } } if manager.broadcast.Started() { err := manager.broadcast.createPipeline() if err != nil && !errors.Is(err, types.ErrCapturePipelineAlreadyExists) { manager.logger.Panic().Err(err).Msg("unable to recreate broadcast pipeline") } } }) } func (manager *CaptureManagerCtx) Shutdown() error { manager.logger.Info().Msgf("shutdown") manager.broadcast.shutdown() manager.audio.shutdown() manager.video.shutdown() return nil } func (manager *CaptureManagerCtx) Broadcast() types.BroadcastManager { return manager.broadcast } func (manager *CaptureManagerCtx) Audio() types.StreamSinkManager { return manager.audio } func (manager *CaptureManagerCtx) Video() types.StreamSinkManager { return manager.video }