neko/internal/config/capture.go

package config

import (
	"github.com/spf13/cobra"
	"github.com/spf13/viper"

	"demodesk/neko/internal/types/codec"
)

type Capture struct {
	Display      string
	Device       string
	AudioCodec   codec.RTP
	AudioParams  string
	VideoCodec   codec.RTP
	VideoParams  string

	BroadcastPipeline string
	Screencast bool
	ScreencastRate string
	ScreencastQuality string
	ScreencastPipeline 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
	}

	// screencast
	cmd.PersistentFlags().Bool("screencast", false, "enable screencast")
	if err := viper.BindPFlag("screencast", cmd.PersistentFlags().Lookup("screencast")); err != nil {
		return err
	}

	cmd.PersistentFlags().String("screencast_rate", "10/1", "set screencast frame rate")
	if err := viper.BindPFlag("screencast_rate", cmd.PersistentFlags().Lookup("screencast_rate")); err != nil {
		return err
	}

	cmd.PersistentFlags().String("screencast_quality", "60", "set screencast JPEG quality")
	if err := viper.BindPFlag("screencast_quality", cmd.PersistentFlags().Lookup("screencast_quality")); err != nil {
		return err
	}

	cmd.PersistentFlags().String("screencast_pipeline", "", "custom screencast pipeline")
	if err := viper.BindPFlag("screencast_pipeline", cmd.PersistentFlags().Lookup("screencast_pipeline")); err != nil {
		return err
	}

	return nil
}

func (s *Capture) Set() {
	var videoCodec codec.RTP
	if viper.GetBool("vp8") {
		videoCodec = codec.New(codec.VP8)
	} else if viper.GetBool("vp9") {
		videoCodec = codec.New(codec.VP9)
	} else if viper.GetBool("h264") {
		videoCodec = codec.New(codec.H264)
	} else {
		// default
		videoCodec = codec.New(codec.VP8)
	}

	var audioCodec codec.RTP
	if viper.GetBool("opus") {
		audioCodec = codec.New(codec.Opus)
	} else if viper.GetBool("g722") {
		audioCodec = codec.New(codec.G722)
	} else if viper.GetBool("pcmu") {
		audioCodec = codec.New(codec.PCMU)
	} else if viper.GetBool("pcma") {
		audioCodec = codec.New(codec.PCMA)
	} else {
		// default
		audioCodec = codec.New(codec.Opus)
	}

	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")
	s.Screencast = viper.GetBool("screencast")
	s.ScreencastRate = viper.GetString("screencast_rate")
	s.ScreencastQuality = viper.GetString("screencast_quality")
	s.ScreencastPipeline = viper.GetString("screencast_pipeline")
}