From 93e735306e6c7d6e45f82e2a253745fca42694fc Mon Sep 17 00:00:00 2001
From: bakito <github@bakito.ch>
Date: Thu, 3 Feb 2022 19:36:30 +0100
Subject: [PATCH] add option to change redirect policy #59

---
 pkg/client/client.go | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/pkg/client/client.go b/pkg/client/client.go
index 7df3f60..3210b6d 100644
--- a/pkg/client/client.go
+++ b/pkg/client/client.go
@@ -6,7 +6,9 @@ import (
 	"fmt"
 	"net/http"
 	"net/url"
+	"os"
 	"path"
+	"strconv"
 
 	"github.com/bakito/adguardhome-sync/pkg/log"
 	"github.com/bakito/adguardhome-sync/pkg/types"
@@ -44,8 +46,16 @@ func New(config types.AdGuardInstance) (Client, error) {
 		cl = cl.SetBasicAuth(config.Username, config.Password)
 	}
 
-	// no redirect
-	cl.SetRedirectPolicy(resty.NoRedirectPolicy())
+	if v, ok := os.LookupEnv("REDIRECT_POLICY_NO_OF_REDIRECTS"); ok {
+		nbr, err := strconv.Atoi(v)
+		if err != nil {
+			return nil, fmt.Errorf("error parsing env var %q value must be an integer", "REDIRECT_POLICY_NO_OF_REDIRECTS")
+		}
+		cl.SetRedirectPolicy(resty.FlexibleRedirectPolicy(nbr))
+	} else {
+		// no redirect
+		cl.SetRedirectPolicy(resty.NoRedirectPolicy())
+	}
 
 	return &client{
 		host:   u.Host,