Added dynamic capture capabilities to plugin API

- Added dynamic capture ingress and sniff endpoint
- Removed static capture example (API update)

example/plugins/debugger/main.go

@@ -3,7 +3,9 @@ package main
 import (
 	"fmt"
 	"net/http"
+	"sort"
 	"strconv"
+	"strings"
 
 	plugin "aroz.org/zoraxy/debugger/mod/zoraxy_plugin"
 )
@@ -39,6 +41,9 @@ func main() {
 		},
 		StaticCaptureIngress: "/s_capture",
 
+		DynamicCaptureSniff:   "/d_sniff",
+		DynamicCaptureIngress: "/d_capture",
+
 		UIPath: UI_PATH,
 
 		/*
@@ -51,9 +56,13 @@ func main() {
 		panic(err)
 	}
 
-	//Create a path handler for the capture paths
+	// Setup the path router
 	pathRouter := plugin.NewPathRouter()
 	pathRouter.SetDebugPrintMode(true)
+
+	/*
+		Static Routers
+	*/
 	pathRouter.RegisterPathHandler("/test_a", http.HandlerFunc(HandleCaptureA))
 	pathRouter.RegisterPathHandler("/test_b", http.HandlerFunc(HandleCaptureB))
 	pathRouter.SetDefaultHandler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
@@ -63,7 +72,46 @@ func main() {
 		w.Header().Set("Content-Type", "text/html")
 		w.Write([]byte("This request is captured by the default handler!<br>Request URI: " + r.URL.String()))
 	}))
-	pathRouter.RegisterHandle(STATIC_CAPTURE_INGRESS, http.DefaultServeMux)
+	pathRouter.RegisterStaticCaptureHandle(STATIC_CAPTURE_INGRESS, http.DefaultServeMux)
+
+	/*
+		Dynamic Captures
+	*/
+	pathRouter.RegisterDynamicSniffHandler("/d_sniff", http.DefaultServeMux, func(dsfr *plugin.DynamicSniffForwardRequest) plugin.SniffResult {
+		fmt.Println("Dynamic Capture Sniffed Request:")
+		fmt.Println("Request URI: " + dsfr.RequestURI)
+
+		//In this example, we want to capture all URI
+		//that start with /test_ and forward it to the dynamic capture handler
+		if strings.HasPrefix(dsfr.RequestURI, "/test_") {
+			reqUUID := dsfr.GetRequestUUID()
+			fmt.Println("Accepting request with UUID: " + reqUUID)
+			return plugin.SniffResultAccpet
+		}
+
+		return plugin.SniffResultSkip
+	})
+	pathRouter.RegisterDynamicCaptureHandle("/d_capture", http.DefaultServeMux, func(w http.ResponseWriter, r *http.Request) {
+		// This is the dynamic capture handler where it actually captures and handle the request
+		w.WriteHeader(http.StatusOK)
+		w.Write([]byte("Welcome to the dynamic capture handler!"))
+
+		// Print all the request info to the response writer
+		w.Write([]byte("\n\nRequest Info:\n"))
+		w.Write([]byte("Request URI: " + r.RequestURI + "\n"))
+		w.Write([]byte("Request Method: " + r.Method + "\n"))
+		w.Write([]byte("Request Headers:\n"))
+		headers := make([]string, 0, len(r.Header))
+		for key := range r.Header {
+			headers = append(headers, key)
+		}
+		sort.Strings(headers)
+		for _, key := range headers {
+			for _, value := range r.Header[key] {
+				w.Write([]byte(fmt.Sprintf("%s: %s\n", key, value)))
+			}
+		}
+	})
 
 	http.HandleFunc(UI_PATH+"/", RenderDebugUI)
 	fmt.Println("Debugger started at http://127.0.0.1:" + strconv.Itoa(runtimeCfg.Port))
@@ -72,21 +120,21 @@ func main() {
 
 // Handle the captured request
 func HandleCaptureA(w http.ResponseWriter, r *http.Request) {
-	for key, values := range r.Header {
+	/*for key, values := range r.Header {
 		for _, value := range values {
 			fmt.Printf("%s: %s\n", key, value)
 		}
-	}
+	}*/
 	w.Header().Set("Content-Type", "text/html")
 	w.Write([]byte("This request is captured by A handler!<br>Request URI: " + r.URL.String()))
 }
 
 func HandleCaptureB(w http.ResponseWriter, r *http.Request) {
-	for key, values := range r.Header {
+	/*for key, values := range r.Header {
 		for _, value := range values {
 			fmt.Printf("%s: %s\n", key, value)
 		}
-	}
+	}*/
 	w.Header().Set("Content-Type", "text/html")
 	w.Write([]byte("This request is captured by the B handler!<br>Request URI: " + r.URL.String()))
 }