Files
olcrtc/internal/config/config.go
2026-05-14 04:21:11 +03:00

183 lines
5.8 KiB
Go

// Package config loads olcrtc runtime configuration from YAML files.
//
// The YAML schema mirrors [session.Config]. Fields left unset in the file
// remain at their zero value, allowing CLI flags to fill them in. Use
// [Apply] to merge a parsed [File] onto an existing [session.Config];
// non-zero fields in the session config (typically populated from CLI flags)
// take precedence over the YAML values.
//
//nolint:tagliatelle // YAML keys are the documented config file schema.
package config
import (
"errors"
"fmt"
"os"
"github.com/openlibrecommunity/olcrtc/internal/app/session"
"gopkg.in/yaml.v3"
)
// ErrConfigNotFound is returned when a config file path is set but the file does not exist.
var ErrConfigNotFound = errors.New("config file not found")
// File is the on-disk YAML schema.
type File struct {
Mode string `yaml:"mode"`
Link string `yaml:"link"`
Auth Auth `yaml:"auth"`
Room Room `yaml:"room"`
Crypto Crypto `yaml:"crypto"`
Net Net `yaml:"net"`
SOCKS SOCKS `yaml:"socks"`
Engine Engine `yaml:"engine"`
Video Video `yaml:"video"`
VP8 VP8 `yaml:"vp8"`
SEI SEI `yaml:"sei"`
Gen Gen `yaml:"gen"`
Data string `yaml:"data"`
Debug bool `yaml:"debug"`
FFmpeg string `yaml:"ffmpeg"`
}
// Auth selects the auth provider.
type Auth struct {
Provider string `yaml:"provider"` // telemost, jazz, wbstream, none
}
// Room identifies the conference room.
type Room struct {
ID string `yaml:"id"`
}
// Crypto holds the shared secret used to authenticate and encrypt the tunnel.
type Crypto struct {
Key string `yaml:"key"` // 64-char hex (32 bytes)
}
// Net groups network and transport selection.
type Net struct {
Transport string `yaml:"transport"` // datachannel, videochannel, seichannel, vp8channel
DNS string `yaml:"dns"`
}
// SOCKS bundles SOCKS5 listener and outbound-proxy settings.
type SOCKS struct {
Host string `yaml:"host"`
Port int `yaml:"port"`
User string `yaml:"user"`
Pass string `yaml:"pass"`
ProxyAddr string `yaml:"proxy_addr"`
ProxyPort int `yaml:"proxy_port"`
}
// Engine selects a direct SFU connection when Auth.Provider is "none".
type Engine struct {
Name string `yaml:"name"` // livekit, goolom, salutejazz
URL string `yaml:"url"`
Token string `yaml:"token"`
}
// Video tunes the videochannel transport.
type Video struct {
Width int `yaml:"width"`
Height int `yaml:"height"`
FPS int `yaml:"fps"`
Bitrate string `yaml:"bitrate"`
HW string `yaml:"hw"`
QRSize int `yaml:"qr_size"`
QRRecovery string `yaml:"qr_recovery"`
Codec string `yaml:"codec"`
TileModule int `yaml:"tile_module"`
TileRS int `yaml:"tile_rs"`
}
// VP8 tunes the vp8channel transport.
type VP8 struct {
FPS int `yaml:"fps"`
BatchSize int `yaml:"batch_size"`
}
// SEI tunes the seichannel transport.
type SEI struct {
FPS int `yaml:"fps"`
BatchSize int `yaml:"batch_size"`
FragmentSize int `yaml:"fragment_size"`
AckTimeoutMS int `yaml:"ack_timeout_ms"`
}
// Gen controls room-generation mode.
type Gen struct {
Amount int `yaml:"amount"`
}
// Load parses a YAML file from disk.
func Load(path string) (File, error) {
// #nosec G304 -- config path is an explicit CLI/user input.
data, err := os.ReadFile(path)
if err != nil {
if errors.Is(err, os.ErrNotExist) {
return File{}, fmt.Errorf("%w: %s", ErrConfigNotFound, path)
}
return File{}, fmt.Errorf("read config %s: %w", path, err)
}
var f File
if err := yaml.Unmarshal(data, &f); err != nil {
return File{}, fmt.Errorf("parse config %s: %w", path, err)
}
return f, nil
}
// Apply merges f onto dst. CLI-set fields (non-zero values in dst) win;
// YAML values fill in the rest.
func Apply(dst session.Config, f File) session.Config {
dst.Mode = pickString(dst.Mode, f.Mode)
dst.Link = pickString(dst.Link, f.Link)
dst.Transport = pickString(dst.Transport, f.Net.Transport)
dst.Auth = pickString(dst.Auth, f.Auth.Provider)
dst.Engine = pickString(dst.Engine, f.Engine.Name)
dst.URL = pickString(dst.URL, f.Engine.URL)
dst.Token = pickString(dst.Token, f.Engine.Token)
dst.RoomID = pickString(dst.RoomID, f.Room.ID)
dst.KeyHex = pickString(dst.KeyHex, f.Crypto.Key)
dst.SOCKSHost = pickString(dst.SOCKSHost, f.SOCKS.Host)
dst.SOCKSPort = pickInt(dst.SOCKSPort, f.SOCKS.Port)
dst.SOCKSUser = pickString(dst.SOCKSUser, f.SOCKS.User)
dst.SOCKSPass = pickString(dst.SOCKSPass, f.SOCKS.Pass)
dst.DNSServer = pickString(dst.DNSServer, f.Net.DNS)
dst.SOCKSProxyAddr = pickString(dst.SOCKSProxyAddr, f.SOCKS.ProxyAddr)
dst.SOCKSProxyPort = pickInt(dst.SOCKSProxyPort, f.SOCKS.ProxyPort)
dst.VideoWidth = pickInt(dst.VideoWidth, f.Video.Width)
dst.VideoHeight = pickInt(dst.VideoHeight, f.Video.Height)
dst.VideoFPS = pickInt(dst.VideoFPS, f.Video.FPS)
dst.VideoBitrate = pickString(dst.VideoBitrate, f.Video.Bitrate)
dst.VideoHW = pickString(dst.VideoHW, f.Video.HW)
dst.VideoQRSize = pickInt(dst.VideoQRSize, f.Video.QRSize)
dst.VideoQRRecovery = pickString(dst.VideoQRRecovery, f.Video.QRRecovery)
dst.VideoCodec = pickString(dst.VideoCodec, f.Video.Codec)
dst.VideoTileModule = pickInt(dst.VideoTileModule, f.Video.TileModule)
dst.VideoTileRS = pickInt(dst.VideoTileRS, f.Video.TileRS)
dst.VP8FPS = pickInt(dst.VP8FPS, f.VP8.FPS)
dst.VP8BatchSize = pickInt(dst.VP8BatchSize, f.VP8.BatchSize)
dst.SEIFPS = pickInt(dst.SEIFPS, f.SEI.FPS)
dst.SEIBatchSize = pickInt(dst.SEIBatchSize, f.SEI.BatchSize)
dst.SEIFragmentSize = pickInt(dst.SEIFragmentSize, f.SEI.FragmentSize)
dst.SEIAckTimeoutMS = pickInt(dst.SEIAckTimeoutMS, f.SEI.AckTimeoutMS)
dst.Amount = pickInt(dst.Amount, f.Gen.Amount)
return dst
}
func pickString(cli, yamlVal string) string {
if cli != "" {
return cli
}
return yamlVal
}
func pickInt(cli, yamlVal int) int {
if cli != 0 {
return cli
}
return yamlVal
}