Update main.go

synced with dev

main.go

@@ -353,19 +353,21 @@ func resolveDomain(domain string) (string, error) {
 }
 
 var (
-	endpoint      string
-	id            string
-	secret        string
-	mtu           string
-	mtuInt        int
-	dns           string
-	privateKey    wgtypes.Key
-	err           error
-	logLevel      string
-	updownScript  string
-	tlsPrivateKey string
-	dockerSocket  string
-	healthFile    string // NEW
+	endpoint      				string
+	id            				string
+	secret        				string
+	mtu           				string
+	mtuInt        				int
+	dns           				string
+	privateKey    				wgtypes.Key
+	err           				error
+	logLevel      				string
+	updownScript  				string
+	tlsPrivateKey 				string
+	dockerSocket  				string
+	dockerEnforceNetworkValidation     	string
+	dockerEnforceNetworkValidationBool 	bool
+	healthFile    				string // NEW
 )
 
 func main() {
@@ -379,6 +381,7 @@ func main() {
 	updownScript = os.Getenv("UPDOWN_SCRIPT")
 	tlsPrivateKey = os.Getenv("TLS_CLIENT_CERT")
 	dockerSocket = os.Getenv("DOCKER_SOCKET")
+	dockerEnforceNetworkValidation = os.Getenv("DOCKER_ENFORCE_NETWORK_VALIDATION")
 	healthFile = os.Getenv("HEALTH_FILE") // NEW
 
 	if endpoint == "" {
@@ -408,6 +411,9 @@ func main() {
 	if dockerSocket == "" {
 		flag.StringVar(&dockerSocket, "docker-socket", "", "Path to Docker socket (typically /var/run/docker.sock)")
 	}
+	if dockerEnforceNetworkValidation == "" {
+		flag.StringVar(&dockerEnforceNetworkValidation, "docker-enforce-network-validation", "false", "Enforce validation of container on newt network (true or false)")
+	}
 	// CLI flag for healthFile if not set by env
 	if healthFile == "" {
 		flag.StringVar(&healthFile, "health-file", "", "Path to health file (if unset, health file won’t be written)")