small refactor (#104)

This commit is contained in:
David Colburn authored and GitHub committed 2021-08-30 18:22:11 -05:00
1 parent 0bfd05c3b8
commit 2fd61531d2
8 files changed
+74 -89

No files matched your search

+1 -35
View File
@@ -11,8 +11,6 @@ import (
"syscall"
"time"
"github.com/livekit/protocol/auth"
"github.com/pkg/errors"
"github.com/urfave/cli/v2"
"github.com/livekit/livekit-server/pkg/config"
@@ -187,19 +185,12 @@ func startServer(c *cli.Context) error {
}
}
// require a key provider
keyProvider, err := createKeyProvider(conf)
if err != nil {
return err
}
logger.Infow("configured key provider", "numKeys", keyProvider.NumKeys())
currentNode, err := routing.NewLocalNode(conf)
if err != nil {
return err
}
server, err := service.InitializeServer(conf, keyProvider, currentNode)
server, err := service.InitializeServer(conf, currentNode)
if err != nil {
return err
}
@@ -216,31 +207,6 @@ func startServer(c *cli.Context) error {
return server.Start()
}
func createKeyProvider(conf *config.Config) (auth.KeyProvider, error) {
// prefer keyfile if set
if conf.KeyFile != "" {
if st, err := os.Stat(conf.KeyFile); err != nil {
return nil, err
} else if st.Mode().Perm() != 0600 {
return nil, fmt.Errorf("key file must have permission set to 600")
}
f, err := os.Open(conf.KeyFile)
if err != nil {
return nil, err
}
defer func() {
_ = f.Close()
}()
return auth.NewFileBasedKeyProviderFromReader(f)
}
if len(conf.Keys) == 0 {
return nil, errors.New("one of key-file or keys must be provided in order to support a secure installation")
}
return auth.NewFileBasedKeyProviderFromMap(conf.Keys), nil
}
func getConfigString(c *cli.Context) (string, error) {
configFile := c.String("config")
configBody := c.String("config-body")