feat: add enabled config option

Signed-off-by: nscuro <[email protected]>
tkhq · Oct 5, 2021 · 14a26eb · 14a26eb
1 parent 11bb06c
commit 14a26eb
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/.traefik.yml b/.traefik.yml
@@ -3,5 +3,7 @@ type: middleware
 import: github.com/nscuro/traefik-plugin-geoblock
 summary: Description of what my plugin is doing
 testData:
+  enabled: true
   database_file: IP2LOCATION-LITE-DB1.IPV6.BIN
-  allowed_countries: [ "CH", "DE" ]
+  allowed_countries: [ "CH", "DE" ]
+  allow_private: true
diff --git a/README.md b/README.md
@@ -21,6 +21,7 @@ http:
     geoblock:
       plugin:
         geoblock:
+          enabled: true
           database_file: /plugins-local/src/github.com/nscuro/traefik-plugin-geoblock/IP2LOCATION-LITE-DB1.IPV6.BIN
           allowed_countries: [ "AT", "CH", "DE" ]
           allow_private: true

diff --git a/plugin.go b/plugin.go
@@ -12,6 +12,7 @@ import (
 )
 
 type Config struct {
+	Enabled          bool     `yaml:"enabled"`
 	DatabaseFilePath string   `yaml:"database_file"`
 	AllowedCountries []string `yaml:"allowed_countries"`
 	AllowPrivate     bool     `yaml:"allow_private"`
@@ -43,6 +44,11 @@ func New(_ context.Context, next http.Handler, cfg *Config, name string) (http.H
 }
 
 func (p Plugin) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
+	if !p.cfg.Enabled {
+		p.next.ServeHTTP(rw, req)
+		return
+	}
+
 	ips := p.GetRemoteIPs(req)
 
 	for _, ip := range ips {