refactor: update customhttp client logic

Author: xkilldash9x

pkg/customhttp/config.go

@@ -1,7 +1,9 @@
 package customhttp
 
 import (
+	"context"
 	"errors"
+	"net"
 	"net/http"
 	"net/http/cookiejar"
 	"net/url"
@@ -131,6 +133,9 @@ type ClientConfig struct {
 	// proxy settings.
 	DialerConfig *network.DialerConfig
 
+	// Optional custom dial context for advanced connection control.
+	DialContext func(ctx context.Context, network, addr string) (net.Conn, error)
+
 	// Specifies the cookie jar for the client. If nil, cookies are not managed.
 	CookieJar http.CookieJar
 
@@ -176,6 +181,7 @@ func NewBrowserClientConfig() *ClientConfig {
 
 	return &ClientConfig{
 		DialerConfig:        network.NewDialerConfig(),
+		DialContext:         nil, // Default to nil; can be set by user
 		CookieJar:           jar,
 		RequestTimeout:      30 * time.Second,
 		IdleConnTimeout:     90 * time.Second, // Standard browser idle timeout

pkg/customhttp/h1client.go

@@ -85,6 +85,12 @@ func (c *H1Client) Connect(ctx context.Context) error {
 
 	// Clone the dialer config to modify it safely.
 	dialerConfig := c.Config.DialerConfig.Clone()
+
+	// Use custom DialContext if provided in ClientConfig
+	if c.Config.DialContext != nil {
+		dialerConfig.DialContext = c.Config.DialContext
+	}
+
 	// Ensure NoDelay is set for responsiveness, crucial for H1.
 	dialerConfig.NoDelay = true
 

pkg/customhttp/h2client.go

@@ -257,6 +257,11 @@ func (c *H2Client) Connect(ctx context.Context) error {
 func (c *H2Client) dialH2Connection(ctx context.Context) (net.Conn, error) {
 	dialerConfig := c.Config.DialerConfig.Clone()
 
+	// Use custom DialContext if provided in ClientConfig
+	if c.Config.DialContext != nil {
+		dialerConfig.DialContext = c.Config.DialContext
+	}
+
 	if dialerConfig.TLSConfig == nil {
 		dialerConfig.TLSConfig = network.NewDialerConfig().TLSConfig.Clone()
 	}