feat(gateway): load dns provider from env variable instead of hardcoded value

Author: henrybarreto

.env

@@ -50,6 +50,10 @@ SHELLHUB_TUNNELS=false
 # NOTICE: If empty, the [SHELLHUB_DOMAIN] will be used.
 SHELLHUB_TUNNELS_DOMAIN=
 
+# The DNS provider used to generate wildcard SSL certificate.
+# NOTICE: If empty, the gateway will be assumed as "digitalocean".
+SHELLHUB_TUNNELS_DNS_PROVIDER=digitalocean
+
 # The token used to generate wildcard SSL certificate using DNS method for tunnels' URL.
 # Currently, only DigitalOcean is supported.
 # NOTICE: Required if SHELLHUB_AUTO_SSL is defined.

docker-compose.yml

@@ -98,6 +98,7 @@ services:
       - SHELLHUB_DOMAIN=${SHELLHUB_DOMAIN}
       - SHELLHUB_TUNNELS=${SHELLHUB_TUNNELS}
       - SHELLHUB_TUNNELS_DOMAIN=${SHELLHUB_TUNNELS_DOMAIN}
+      - SHELLHUB_TUNNELS_DNS_PROVIDER=${SHELLHUB_TUNNELS_DNS_PROVIDER}
       - SHELLHUB_TUNNELS_DNS_PROVIDER_TOKEN=${SHELLHUB_TUNNELS_DNS_PROVIDER_TOKEN}
       - SHELLHUB_VERSION=${SHELLHUB_VERSION}
       - SHELLHUB_SSH_PORT=${SHELLHUB_SSH_PORT}

gateway/config.go

@@ -11,19 +11,20 @@ import (
 
 // GatewayConfig holds the configuration settings for the gateway.
 type GatewayConfig struct {
-	Env                     string `env:"SHELLHUB_ENV"`
-	Domain                  string `env:"SHELLHUB_DOMAIN,required" validate:"hostname"`
-	Tunnels                 bool   `env:"SHELLHUB_TUNNELS,default=false"`
-	TunnelsDomain           string `env:"SHELLHUB_TUNNELS_DOMAIN"`
-	TunnelsDNSProviderToken string `env:"SHELLHUB_TUNNELS_DNS_PROVIDER_TOKEN"`
-	WorkerProcesses         string `env:"WORKER_PROCESSES,default=auto"`
-	MaxWorkerOpenFiles      int    `env:"MAX_WORKER_OPEN_FILES,default=0"`
-	MaxWorkerConnections    int    `env:"MAX_WORKER_CONNECTIONS,default=16384"`
-	BacklogSize             int    `env:"BACKLOG_SIZE"`
-	EnableAutoSSL           bool   `env:"SHELLHUB_AUTO_SSL"`
-	EnableProxyProtocol     bool   `env:"SHELLHUB_PROXY"`
-	EnableEnterprise        bool   `env:"SHELLHUB_ENTERPRISE"`
-	EnableCloud             bool   `env:"SHELLHUB_CLOUD"`
+	Env                     string      `env:"SHELLHUB_ENV"`
+	Domain                  string      `env:"SHELLHUB_DOMAIN,required" validate:"hostname"`
+	Tunnels                 bool        `env:"SHELLHUB_TUNNELS,default=false"`
+	TunnelsDomain           string      `env:"SHELLHUB_TUNNELS_DOMAIN"`
+	TunnelsDNSProvider      DNSProvider `env:"SHELLHUB_TUNNELS_DNS_PROVIDER,default=digitalocean"`
+	TunnelsDNSProviderToken string      `env:"SHELLHUB_TUNNELS_DNS_PROVIDER_TOKEN"`
+	WorkerProcesses         string      `env:"WORKER_PROCESSES,default=auto"`
+	MaxWorkerOpenFiles      int         `env:"MAX_WORKER_OPEN_FILES,default=0"`
+	MaxWorkerConnections    int         `env:"MAX_WORKER_CONNECTIONS,default=16384"`
+	BacklogSize             int         `env:"BACKLOG_SIZE"`
+	EnableAutoSSL           bool        `env:"SHELLHUB_AUTO_SSL"`
+	EnableProxyProtocol     bool        `env:"SHELLHUB_PROXY"`
+	EnableEnterprise        bool        `env:"SHELLHUB_ENTERPRISE"`
+	EnableCloud             bool        `env:"SHELLHUB_CLOUD"`
 }
 
 var validate = validator.New()

gateway/main.go

@@ -50,13 +50,14 @@ func main() {
 			}
 
 			log.WithFields(log.Fields{
-				"domain": domain,
-				"token":  halfString(config.TunnelsDNSProviderToken),
+				"domain":   domain,
+				"provider": config.TunnelsDNSProvider,
+				"token":    halfString(config.TunnelsDNSProviderToken),
 			}).Info("tunnels info")
 
 			certBot.Config.Tunnels = &Tunnels{
 				Domain:   domain,
-				Provider: DigitalOceanDNSProvider,
+				Provider: config.TunnelsDNSProvider,
 				Token:    config.TunnelsDNSProviderToken,
 			}
 		}