mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
115 lines
3.3 KiB
Go
115 lines
3.3 KiB
Go
package config
|
|
|
|
import (
|
|
"github.com/pion/webrtc/v2"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
type Capture struct {
|
|
Display string
|
|
Device string
|
|
AudioCodec string
|
|
AudioParams string
|
|
VideoCodec string
|
|
VideoParams string
|
|
BroadcastPipeline string
|
|
}
|
|
|
|
func (Capture) Init(cmd *cobra.Command) error {
|
|
cmd.PersistentFlags().String("display", ":99.0", "XDisplay to capture")
|
|
if err := viper.BindPFlag("display", cmd.PersistentFlags().Lookup("display")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().String("device", "auto_null.monitor", "audio device to capture")
|
|
if err := viper.BindPFlag("device", cmd.PersistentFlags().Lookup("device")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().String("audio", "", "audio codec parameters to use for streaming")
|
|
if err := viper.BindPFlag("audio", cmd.PersistentFlags().Lookup("audio")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().String("video", "", "video codec parameters to use for streaming")
|
|
if err := viper.BindPFlag("video", cmd.PersistentFlags().Lookup("video")); err != nil {
|
|
return err
|
|
}
|
|
|
|
// video codecs
|
|
cmd.PersistentFlags().Bool("vp8", false, "use VP8 video codec")
|
|
if err := viper.BindPFlag("vp8", cmd.PersistentFlags().Lookup("vp8")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().Bool("vp9", false, "use VP9 video codec")
|
|
if err := viper.BindPFlag("vp9", cmd.PersistentFlags().Lookup("vp9")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().Bool("h264", false, "use H264 video codec")
|
|
if err := viper.BindPFlag("h264", cmd.PersistentFlags().Lookup("h264")); err != nil {
|
|
return err
|
|
}
|
|
|
|
// audio codecs
|
|
cmd.PersistentFlags().Bool("opus", false, "use Opus audio codec")
|
|
if err := viper.BindPFlag("opus", cmd.PersistentFlags().Lookup("opus")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().Bool("g722", false, "use G722 audio codec")
|
|
if err := viper.BindPFlag("g722", cmd.PersistentFlags().Lookup("g722")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().Bool("pcmu", false, "use PCMU audio codec")
|
|
if err := viper.BindPFlag("pcmu", cmd.PersistentFlags().Lookup("pcmu")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().Bool("pcma", false, "use PCMA audio codec")
|
|
if err := viper.BindPFlag("pcma", cmd.PersistentFlags().Lookup("pcma")); err != nil {
|
|
return err
|
|
}
|
|
|
|
// broadcast
|
|
cmd.PersistentFlags().String("broadcast_pipeline", "", "audio video codec parameters to use for broadcasting")
|
|
if err := viper.BindPFlag("broadcast_pipeline", cmd.PersistentFlags().Lookup("broadcast_pipeline")); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Capture) Set() {
|
|
videoCodec := webrtc.VP8
|
|
if viper.GetBool("vp8") {
|
|
videoCodec = webrtc.VP8
|
|
} else if viper.GetBool("vp9") {
|
|
videoCodec = webrtc.VP9
|
|
} else if viper.GetBool("h264") {
|
|
videoCodec = webrtc.H264
|
|
}
|
|
|
|
audioCodec := webrtc.Opus
|
|
if viper.GetBool("opus") {
|
|
audioCodec = webrtc.Opus
|
|
} else if viper.GetBool("g722") {
|
|
audioCodec = webrtc.G722
|
|
} else if viper.GetBool("pcmu") {
|
|
audioCodec = webrtc.PCMU
|
|
} else if viper.GetBool("pcma") {
|
|
audioCodec = webrtc.PCMA
|
|
}
|
|
|
|
s.Device = viper.GetString("device")
|
|
s.AudioCodec = audioCodec
|
|
s.AudioParams = viper.GetString("audio")
|
|
s.Display = viper.GetString("display")
|
|
s.VideoCodec = videoCodec
|
|
s.VideoParams = viper.GetString("video")
|
|
s.BroadcastPipeline = viper.GetString("broadcast_pipeline")
|
|
}
|