feat: uses options instead of functional arguments for NewTransaction…

…WithOptions.

experimental/waf.go

@@ -4,33 +4,14 @@
 package experimental
 
 import (
-	"context"
-	"strings"
-
 	"github.com/corazawaf/coraza/v3/internal/corazawaf"
 	"github.com/corazawaf/coraza/v3/types"
 )
 
-type Option func(*corazawaf.Options)
-
-// WithID sets the transaction ID
-func WithID(id string) Option {
-	return func(o *corazawaf.Options) {
-		o.ID = strings.TrimSpace(id)
-	}
-}
-
-// WithContext sets the transaction context, this is useful for passing
-// a context into the logger.
-// The transaction lifecycle isn't tied to the context lifecycle.
-func WithContext(ctx context.Context) Option {
-	return func(o *corazawaf.Options) {
-		o.Context = ctx
-	}
-}
+type Options = corazawaf.Options
 
 // WAFWithOptions is an interface that allows to create transactions
 // with options
 type WAFWithOptions interface {
-	NewTransactionWithOptions(opts ...Option) types.Transaction
+	NewTransactionWithOptions(Options) types.Transaction
 }

experimental/waf_test.go

@@ -0,0 +1,32 @@
+// Copyright 2024 Juan Pablo Tosso and the OWASP Coraza contributors
+// SPDX-License-Identifier: Apache-2.0
+
+package experimental_test
+
+import (
+	"fmt"
+
+	"github.com/corazawaf/coraza/v3"
+	"github.com/corazawaf/coraza/v3/experimental"
+)
+
+func ExampleWAFWithOptions_NewTransactionWithOptions() {
+	waf, err := coraza.NewWAF(coraza.NewWAFConfig())
+	if err != nil {
+		panic(err)
+	}
+
+	oWAF, ok := waf.(experimental.WAFWithOptions)
+	if !ok {
+		panic("WAF does not implement WAFWithOptions")
+	}
+
+	tx := oWAF.NewTransactionWithOptions(experimental.Options{
+		ID: "abc123",
+	})
+
+	fmt.Println("Transaction ID:", tx.ID())
+
+	// Output:
+	// Transaction ID: abc123
+}

http/middleware.go

@@ -124,7 +124,9 @@ func WrapHandler(waf coraza.WAF, h http.Handler) http.Handler {
 
 	if ctxwaf, ok := waf.(experimental.WAFWithOptions); ok {
 		newTX = func(r *http.Request) types.Transaction {
-			return ctxwaf.NewTransactionWithOptions(experimental.WithContext(r.Context()))
+			return ctxwaf.NewTransactionWithOptions(experimental.Options{
+				Context: r.Context(),
+			})
 		}
 	}
 

internal/corazawaf/waf.go

@@ -147,6 +147,8 @@ func (w *WAF) NewTransaction() *Transaction {
 	})
 }
 
+// NewTransactionWithOptions Creates a new initialized transaction for this WAF
+// instance with the provided options
 func (w *WAF) NewTransactionWithOptions(opts Options) *Transaction {
 	if opts.ID == "" {
 		opts.ID = stringutils.RandomString(19)

waf.go

@@ -141,10 +141,6 @@ func (w wafWrapper) NewTransactionWithID(id string) types.Transaction {
 }
 
 // NewTransaction implements the same method on WAF.
-func (w wafWrapper) NewTransactionWithOptions(opts ...experimental.Option) types.Transaction {
-	o := corazawaf.Options{}
-	for _, opt := range opts {
-		opt(&o)
-	}
-	return w.waf.NewTransactionWithOptions(o)
+func (w wafWrapper) NewTransactionWithOptions(opts experimental.Options) types.Transaction {
+	return w.waf.NewTransactionWithOptions(opts)
 }