feat(proxymanager): add Rotate method

dwisiswant0 · dwisiswant0 · commit 96de3aa0b4e5 · 2024-09-03T20:25:36.000+07:00
Signed-off-by: Dwi Siswanto &lt;git@dw1.io&gt;
diff --git a/internal/proxymanager/utils.go b/internal/proxymanager/utils.go
@@ -4,6 +4,7 @@ import (
 	"math/rand"
 
 	"github.com/fsnotify/fsnotify"
+	"github.com/kitabisa/mubeng/pkg/helper"
 )
 
 // NextProxy will navigate the next proxy to use
@@ -23,6 +24,26 @@ func (p *ProxyManager) RandomProxy() string {
 	return p.Proxies[rand.Intn(len(p.Proxies))]
 }
 
+// Rotate proxy based on method
+//
+// Valid methods are "sequent" and "random", default return empty string.
+func (p *ProxyManager) Rotate(method string) string {
+	var proxy string
+
+	switch method {
+	case "sequent":
+		proxy = p.NextProxy()
+	case "random":
+		proxy = p.RandomProxy()
+	}
+
+	if proxy != "" {
+		proxy = helper.EvalFunc(proxy)
+	}
+
+	return proxy
+}
+
 // Watch proxy file from events
 func (p *ProxyManager) Watch() (*fsnotify.Watcher, error) {
 	watcher, err := fsnotify.NewWatcher()
