mirror of
https://github.com/livekit/livekit.git
synced 2026-04-26 13:07:39 +00:00
104 lines
2.2 KiB
Go
104 lines
2.2 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/mitchellh/go-homedir"
|
|
"github.com/urfave/cli/v2"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Config struct {
|
|
Port uint32 `yaml:"port"`
|
|
RTC RTCConfig `yaml:"rtc"`
|
|
Redis RedisConfig `yaml:"redis"`
|
|
KeyFile string `yaml:"key_file"`
|
|
Keys map[string]string `yaml:"keys"`
|
|
LogLevel string `yaml:"log_level"`
|
|
|
|
Development bool `yaml:"development"`
|
|
}
|
|
|
|
type RTCConfig struct {
|
|
ICEPortRangeStart uint16 `yaml:"port_range_start"`
|
|
ICEPortRangeEnd uint16 `yaml:"port_range_end"`
|
|
StunServers []string `yaml:"stun_servers"`
|
|
UseExternalIP bool `yaml:"use_external_ip"`
|
|
|
|
MaxBitrate uint64 `yaml:"max_bandwidth"`
|
|
MaxBufferTime int `yaml:"max_buffer_time"`
|
|
}
|
|
|
|
type RedisConfig struct {
|
|
Address string `yaml:"address"`
|
|
Password string `yaml:"password"`
|
|
}
|
|
|
|
func NewConfig(confString string) (*Config, error) {
|
|
// start with defaults
|
|
conf := &Config{
|
|
Port: 7880,
|
|
RTC: RTCConfig{
|
|
ICEPortRangeStart: 8000,
|
|
ICEPortRangeEnd: 10000,
|
|
StunServers: []string{
|
|
"stun.l.google.com:19302",
|
|
},
|
|
},
|
|
Redis: RedisConfig{},
|
|
Keys: map[string]string{},
|
|
}
|
|
if confString != "" {
|
|
yaml.Unmarshal([]byte(confString), conf)
|
|
}
|
|
return conf, nil
|
|
}
|
|
|
|
func (conf *Config) HasRedis() bool {
|
|
return conf.Redis.Address != ""
|
|
}
|
|
|
|
func (conf *Config) UpdateFromCLI(c *cli.Context) error {
|
|
if c.IsSet("dev") {
|
|
conf.Development = c.Bool("dev")
|
|
}
|
|
if c.IsSet("key-file") {
|
|
conf.KeyFile = c.String("key-file")
|
|
}
|
|
if c.IsSet("keys") {
|
|
if err := conf.unmarshalKeys(c.String("keys")); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if c.IsSet("redis-host") {
|
|
conf.Redis.Address = c.String("redis-host")
|
|
}
|
|
if c.IsSet("redis-password") {
|
|
conf.Redis.Password = c.String("redis-password")
|
|
}
|
|
// expand env vars in filenames
|
|
file, err := homedir.Expand(os.ExpandEnv(conf.KeyFile))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
conf.KeyFile = file
|
|
|
|
return nil
|
|
}
|
|
|
|
func (conf *Config) unmarshalKeys(keys string) error {
|
|
temp := make(map[string]interface{})
|
|
if err := yaml.Unmarshal([]byte(keys), temp); err != nil {
|
|
return err
|
|
}
|
|
|
|
conf.Keys = make(map[string]string, len(temp))
|
|
|
|
for key, val := range temp {
|
|
if secret, ok := val.(string); ok {
|
|
conf.Keys[key] = secret
|
|
}
|
|
}
|
|
return nil
|
|
}
|