chore: Add JSON unmarshalling for duration in helpers package

Author: ShocOne

helpers/helpers.go

@@ -2,6 +2,7 @@
 package helpers
 
 import (
+	"encoding/json"
 	"fmt"
 	"os"
 	"path/filepath"
@@ -33,3 +34,25 @@ func SafeOpenFile(filePath string) (*os.File, error) {
 	// Open the file if the path is deemed safe
 	return os.Open(absPath)
 }
+
+// UnmarshalJSON parses the duration from JSON string.
+func (d *JSONDuration) UnmarshalJSON(b []byte) error {
+	var s string
+	if err := json.Unmarshal(b, &s); err != nil {
+		return err
+	}
+	duration, err := time.ParseDuration(s)
+	if err != nil {
+		return err
+	}
+	*d = JSONDuration(duration)
+	return nil
+}
+
+// Duration returns the time.Duration value.
+func (d JSONDuration) Duration() time.Duration {
+	return time.Duration(d)
+}
+
+// JSONDuration wraps time.Duration for custom JSON unmarshalling.
+type JSONDuration time.Duration

httpclient/client.go

@@ -95,9 +95,9 @@ type ConcurrencyConfig struct {
 
 // TimeoutConfig holds custom timeout settings.
 type TimeoutConfig struct {
-	CustomTimeout            time.Duration // Custom timeout for the HTTP client
-	TokenRefreshBufferPeriod time.Duration // Buffer period before token expiry to attempt token refresh
-	TotalRetryDuration       time.Duration // Total duration to attempt retries
+	CustomTimeout            time.Duration `json:"CustomTimeout,omitempty"`            // Custom timeout for the HTTP client
+	TokenRefreshBufferPeriod time.Duration `json:"TokenRefreshBufferPeriod,omitempty"` // Buffer period before token expiry to attempt token refresh
+	TotalRetryDuration       time.Duration `json:"TotalRetryDuration,omitempty"`       // Total duration to attempt retries
 }
 
 // RedirectConfig holds configuration related to redirect handling.