update example

.traefik.yml

@@ -11,6 +11,5 @@ testData:
     - userId1
     - userId2
   paths:
-    - prefix: /v1/users
-      value: testValue
-    - prefix: /v1/organizations
+    - base: /v1/users
+      path: /testValue

docker-compose.yaml

@@ -0,0 +1,16 @@
+version: '3'
+
+services:
+  reverse-proxy:
+    image: traefik:v3.0
+    command:
+      - --api.insecure=true
+      - --providers.docker
+      - "--experimental.localPlugins.usersblocker.moduleName=github.com/ditkrg/traefik-users-blocker-plugin"
+    ports:
+      - "80:80"
+      - "8080:8080"
+    volumes:
+      # So that Traefik can listen to the Docker events
+      - /var/run/docker.sock:/var/run/docker.sock
+      - ./:/plugins-local/src/github.com/ditkrg/traefik-users-blocker-plugin

main.go

@@ -4,12 +4,17 @@ import (
 	"context"
 	"fmt"
 	"net/http"
+	"os"
 	"strings"
 )
 
+type Rule struct {
+	AllowedSubPaths []string `json:"allowedSubPaths,omitempty"`
+}
+
 type Path struct {
-	Prefix      string `json:"prefix,omitempty"`
-	MustContain string `json:"mustContain,omitempty"`
+	Path string `json:"base,omitempty"`
+	Rule Rule   `json:"rule,omitempty"`
 }
 
 type Config struct {
@@ -37,8 +42,8 @@ func New(ctx context.Context, next http.Handler, config *Config, name string) (h
 	}
 
 	for _, path := range config.Paths {
-		if path.Prefix == "" {
-			return nil, fmt.Errorf("Paths.Prefix cannot be empty")
+		if path.Path == "" {
+			return nil, fmt.Errorf("Paths.Path cannot be empty")
 		}
 	}
 
@@ -53,6 +58,9 @@ func New(ctx context.Context, next http.Handler, config *Config, name string) (h
 func (a *UsersBlocker) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
 	userId := req.Header["X-Auth-User-Id"][0]
 
+	message := fmt.Sprintf("{requestPath: %s, userId: %s}\n", req.URL.Path, userId)
+	os.Stdout.WriteString(message)
+
 	var isUserBlocked bool
 
 	for _, id := range a.userId {
@@ -67,16 +75,29 @@ func (a *UsersBlocker) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
 	}
 
 	for _, path := range a.paths {
-		isPathBlocked := strings.HasPrefix(req.URL.Path, path.Prefix)
+		isPathMatched := strings.HasPrefix(req.URL.Path, path.Path)
 
-		if isPathBlocked && path.MustContain != "" {
-			isPathBlocked = !strings.Contains(req.URL.Path, path.MustContain)
+		if !isPathMatched {
+			a.next.ServeHTTP(rw, req)
+			return
 		}
 
-		if isPathBlocked {
-			http.Error(rw, "Forbidden", http.StatusForbidden)
+		if len(path.Rule.AllowedSubPaths) == 0 {
+			message := fmt.Sprintf("blocked path %s (matched with %s) for user %s", req.URL.Path, path.Path, userId)
+			os.Stdout.WriteString(message)
+			http.Error(rw, message, http.StatusForbidden)
 			return
 		}
+
+		for _, allowedSubPath := range path.Rule.AllowedSubPaths {
+			isAllowedSubPathMatched := strings.HasPrefix(req.URL.Path, path.Path+allowedSubPath)
+			if !isAllowedSubPathMatched {
+				message := fmt.Sprintf("blocked path %s (matched with %s) for user %s", req.URL.Path, path.Path+path.Path+allowedSubPath, userId)
+				os.Stdout.WriteString(message)
+				http.Error(rw, message, http.StatusForbidden)
+				return
+			}
+		}
 	}
 
 	a.next.ServeHTTP(rw, req)