torjus/ezshare
1
0
Fork 0
Code Issues Pull Requests Projects Releases 2 Wiki Activity

Change valid-function for config

Browse Source
This commit is contained in:
Torjus Håkestad 2021-12-08 13:12:16 +01:00
parent 84fc2d6667
commit 6558c63096
2 changed files with 26 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
config/config.go
View File

@ -211,14 +211,14 @@ func (c *Config) UpdateFromEnv() error {
return nil
}
func (c *Config) ValidForServer() error {
func (sc *ServerConfig) Valid() error {
// Verify that grpc-endpoint is set
if c.Server.GRPCEndpoint == "" {
if sc.GRPCEndpoint == "" {
return fmt.Errorf("missing require config-value Server.GRPCEndpoint")
}
// Verify loglevel
switch strings.ToUpper(c.Server.LogLevel) {
switch strings.ToUpper(sc.LogLevel) {
case "DEBUG", "INFO", "WARN", "ERROR", "FATAL":
break
default:
@ -226,11 +226,11 @@ func (c *Config) ValidForServer() error {
}
// Verify datastore config
switch strings.ToLower(c.Server.DataStoreConfig.Type) {
switch strings.ToLower(sc.DataStoreConfig.Type) {
case "memory":
break
case "bolt":
if c.Server.DataStoreConfig.BoltStoreConfig == nil || c.Server.DataStoreConfig.BoltStoreConfig.Path == "" {
if sc.DataStoreConfig.BoltStoreConfig == nil || sc.DataStoreConfig.BoltStoreConfig.Path == "" {
return fmt.Errorf("server datastore is bolt, missing required config value Server.DataStore.Bolt.Path")
}
default:
@ -238,54 +238,54 @@ func (c *Config) ValidForServer() error {
}
// Verify filestore config
switch strings.ToLower(c.Server.FileStoreConfig.Type) {
switch strings.ToLower(sc.FileStoreConfig.Type) {
case "memory":
break
case "filesystem":
if c.Server.FileStoreConfig.FSStoreConfig == nil || c.Server.FileStoreConfig.FSStoreConfig.Dir == "" {
if sc.FileStoreConfig.FSStoreConfig == nil || sc.FileStoreConfig.FSStoreConfig.Dir == "" {
return fmt.Errorf("server datastore is bolt, missing required config value Server.FileStore.FSStore.Path")
}
case "bolt":
if c.Server.FileStoreConfig.BoltStoreConfig == nil || c.Server.FileStoreConfig.BoltStoreConfig.Path == "" {
if sc.FileStoreConfig.BoltStoreConfig == nil || sc.FileStoreConfig.BoltStoreConfig.Path == "" {
return fmt.Errorf("server datastore is bolt, missing required config value Server.DataStore.Bolt.Path")
}
}
// Verify grpc-config
if c.Server.GRPC.ListenAddr == "" {
if sc.GRPC.ListenAddr == "" {
return fmt.Errorf("missing required config-value Server.GRPC.ListenAddr")
}
if c.Server.GRPC.CACerts.CertificateKeyPath == "" {
if sc.GRPC.CACerts.CertificateKeyPath == "" {
// TODO: Maybe return custom error, so we can create certs if missing
return fmt.Errorf("missing require value Server.GRPC.CACerts.CertificateKeyPath")
}
if c.Server.GRPC.CACerts.CertificatePath == "" {
if sc.GRPC.CACerts.CertificatePath == "" {
// TODO: Maybe return custom error, so we can create certs if missing
return fmt.Errorf("missing require value Server.GRPC.CACerts.CertificatePath")
}
if c.Server.GRPC.Certs.CertificatePath == "" {
if sc.GRPC.Certs.CertificatePath == "" {
// TODO: Maybe return custom error, so we can create certs if missing
return fmt.Errorf("missing require value Server.GRPC.Certs.CertificatePath")
}
if c.Server.GRPC.Certs.CertificateKeyPath == "" {
if sc.GRPC.Certs.CertificateKeyPath == "" {
// TODO: Maybe return custom error, so we can create certs if missing
return fmt.Errorf("missing require value Server.GRPC.Certs.CertificateKeyPath")
}
return nil
}
func (c *Config) ValidForClient() error {
if c.Client.Certs.CertificateKeyPath == "" {
func (cc *ClientConfig) Valid() error {
if cc.Certs.CertificateKeyPath == "" {
return fmt.Errorf("missing required value Client.Certs.CertificateKeyPath")
}
if c.Client.Certs.CertificatePath == "" {
if cc.Certs.CertificatePath == "" {
return fmt.Errorf("missing required value Client.Certs.CertificatePath")
}
if c.Client.DefaultServer == "" {
if cc.DefaultServer == "" {
// TODO: Should probably have its own custom error
return fmt.Errorf("missing required value Client.DefaultServer")
}
if c.Client.ServerCertPath == "" {
if cc.ServerCertPath == "" {
// TODO: Should probably have its own custom error
return fmt.Errorf("missing required value Client.ServerCertPath")
}

18
config/config_test.go
View File

@ -189,20 +189,18 @@ func TestConfig(t *testing.T) {
if err != nil {
t.Fatalf("Error reading config: %s", err)
}
clientErr := cfg.ValidForClient()
serverErr := cfg.ValidForServer()
if c.ValidForClient && !(clientErr == nil) {
t.Errorf("Valid config ValidClientConfig returned wrong result: %s", clientErr)
if c.ValidForClient && !(cfg.Client.Valid() == nil) {
t.Errorf("Valid config ValidClientConfig returned wrong result: %s", cfg.Client.Valid())
}
if !c.ValidForClient && (clientErr == nil) {
t.Errorf("Invalid config ValidClientConfig returned wrong result: %s", clientErr)
if !c.ValidForClient && (cfg.Client.Valid() == nil) {
t.Errorf("Invalid config ValidClientConfig returned wrong result: %s", cfg.Client.Valid())
}
if c.ValidForServer && !(serverErr == nil) {
t.Errorf("Valid config ValidServerConfig returned wrong result: %s", clientErr)
if c.ValidForServer && !(cfg.Server.Valid() == nil) {
t.Errorf("Valid config ValidServerConfig returned wrong result: %s", cfg.Server.Valid())
}
if !c.ValidForServer && (serverErr == nil) {
t.Errorf("Invalid config ValidServerConfig returned wrong result: %s", clientErr)
if !c.ValidForServer && (cfg.Server.Valid() == nil) {
t.Errorf("Invalid config ValidServerConfig returned wrong result: %s", cfg.Server.Valid())
}
})
}