neko-rooms/internal/types/room.go

package types

import (
	"fmt"
	"strings"
	"strconv"
	"time"
)

type RoomEntry struct {
	ID             string    `json:"id"`
	URL            string    `json:"url"`
	Name           string    `json:"name"`
	MaxConnections uint16    `json:"max_connections"`
	Image          string    `json:"image"`
	Running        bool      `json:"running"`
	Status         string    `json:"status"`
	Created        time.Time `json:"created"`
}

type RoomSettings struct {
	Name           string `json:"name"`
	MaxConnections uint16 `json:"max_connections"`

	UserPass  string `json:"user_pass"`
	AdminPass string `json:"admin_pass"`

	BroadcastPipeline string `json:"broadcast_pipeline,omitempty"`

	Screen        string `json:"screen"`
	VideoCodec    string `json:"video_codec,omitempty"`
	VideoBitrate  int    `json:"video_bitrate,omitempty"`
	VideoPipeline string `json:"video_pipeline,omitempty"`
	VideoMaxFPS   int    `json:"video_max_fps"`

	AudioCodec    string `json:"audio_codec,omitempty"`
	AudioBitrate  int    `json:"audio_bitrate,omitempty"`
	AudioPipeline string `json:"audio_pipeline,omitempty"`
}

func (settings *RoomSettings) ToEnv() []string {
	env := []string{
		fmt.Sprintf("NEKO_PASSWORD=%s", settings.UserPass),
		fmt.Sprintf("NEKO_PASSWORD_ADMIN=%s", settings.AdminPass),
		fmt.Sprintf("NEKO_SCREEN=%s", settings.Screen),
		fmt.Sprintf("NEKO_MAX_FPS=%d", settings.VideoMaxFPS),
	}

	if settings.BroadcastPipeline != "" {
		env = append(env, fmt.Sprintf("NEKO_BROADCAST_PIPELINE=%s", settings.BroadcastPipeline))
	}

	if settings.VideoCodec == "VP8" || settings.VideoCodec == "VP9" || settings.VideoCodec == "H264" {
		env = append(env, fmt.Sprintf("NEKO_%s=true", strings.ToUpper(settings.VideoCodec)))
	}

	if settings.VideoBitrate != 0 {
		env = append(env, fmt.Sprintf("NEKO_VIDEO_BITRATE=%d", settings.VideoBitrate))
	}

	if settings.VideoPipeline != "" {
		env = append(env, fmt.Sprintf("NEKO_VIDEO=%s", settings.VideoPipeline))
	}

	if settings.AudioCodec == "OPUS" || settings.AudioCodec == "G722" || settings.AudioCodec == "PCMU" || settings.AudioCodec == "PCMA" {
		env = append(env, fmt.Sprintf("NEKO_%s=true", strings.ToUpper(settings.AudioCodec)))
	}

	if settings.AudioBitrate != 0 {
		env = append(env, fmt.Sprintf("NEKO_AUDIO_BITRATE=%d", settings.AudioBitrate))
	}

	if settings.AudioPipeline != "" {
		env = append(env, fmt.Sprintf("NEKO_AUDIO=%s", settings.AudioPipeline))
	}

	return env
}

func (settings *RoomSettings) FromEnv(envs []string) error {
	var err error
	for _, env := range envs {
		r := strings.SplitAfterN(env, "=", 2)
		key := r[0]
		val := r[1]

		switch key {
		case "NEKO_PASSWORD=":
			settings.UserPass = val
		case "NEKO_PASSWORD_ADMIN=":
			settings.AdminPass = val
		case "NEKO_SCREEN=":
			settings.Screen = val
		case "NEKO_MAX_FPS=":
			settings.VideoMaxFPS, err = strconv.Atoi(val)
		case "NEKO_BROADCAST_PIPELINE=":
			settings.BroadcastPipeline = val
		case "NEKO_VP8=":
			if ok, _ := strconv.ParseBool(val); !ok {
				settings.VideoCodec = "VP8"
			}
		case "NEKO_VP9=":
			if ok, _ := strconv.ParseBool(val); !ok {
				settings.VideoCodec = "VP9"
			}
		case "NEKO_H264=":
			if ok, _ := strconv.ParseBool(val); !ok {
				settings.VideoCodec = "H264"
			}
		case "NEKO_VIDEO_BITRATE=":
			settings.VideoBitrate, err = strconv.Atoi(val)
		case "NEKO_VIDEO=":
			settings.VideoPipeline = val
		case "NEKO_OPUS=":
			if ok, _ := strconv.ParseBool(val); ok {
				settings.AudioCodec = "OPUS"
			}
		case "NEKO_G722=":
			if ok, _ := strconv.ParseBool(val); ok {
				settings.AudioCodec = "G722"
			}
		case "NEKO_PCMU=":
			if ok, _ := strconv.ParseBool(val); ok {
				settings.AudioCodec = "PCMU"
			}
		case "NEKO_PCMA=":
			if ok, _ := strconv.ParseBool(val); ok {
				settings.AudioCodec = "PCMA"
			}
		case "NEKO_AUDIO_BITRATE=":
			settings.AudioBitrate, err = strconv.Atoi(val)
		case "NEKO_AUDIO=":
			settings.AudioPipeline = val
		}

		if err != nil {
			return err
		}
	}

	return nil
}

type RoomManager interface {
	List() ([]RoomEntry, error)

	Create(settings RoomSettings) (string, error)
	GetEntry(id string) (*RoomEntry, error)
	GetSettings(id string) (*RoomSettings, error)
	Remove(id string) error

	Start(id string) error
	Stop(id string) error
	Restart(id string) error
}