mirror of
https://github.com/livekit/livekit.git
synced 2026-10-24 02:02:55 +00:00
small refactor (#104)
This commit is contained in:
1 parent
0bfd05c3b8
commit
2fd61531d2
8 files changed
+74
-89
No files matched your search
+1
-35
@@ -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")
|
||||
|
||||
Reference in new issue
Block a user