🔨 ProxySlice -> Proxy struct

config/config.go

@@ -11,7 +11,7 @@ import (
 const (
 	VERSION                 = "0.0.1"
 	DefaultUserAgent        = "Dodo/" + VERSION
-	ProxyCheckURL           = "https://google.com"
+	ProxyCheckURL           = "https://www.google.com"
 	DefaultMethod           = "GET"
 	DefaultTimeout          = 10000 // Milliseconds (10 seconds)
 	DefaultDodosCount       = 1
@@ -23,8 +23,6 @@ type IConfig interface {
 	MergeConfigs(newConfig IConfig) IConfig
 }
 
-type ProxySlice []map[string]string
-
 type DodoConfig struct {
 	Method       string
 	URL          string
@@ -34,7 +32,7 @@ type DodoConfig struct {
 	Params       map[string]string
 	Headers      map[string]string
 	Cookies      map[string]string
-	Proxies      ProxySlice
+	Proxies      []Proxy
 	Body         string
 }
 
@@ -98,12 +96,18 @@ func (config *Config) SetDefaults() {
 	}
 }
 
+type Proxy struct {
+	URL      string `json:"url" validate:"required,proxy_url"`
+	Username string `json:"username"`
+	Password string `json:"password"`
+}
+
 type JSONConfig struct {
 	Config
 	Params  map[string]string `json:"params"`
 	Headers map[string]string `json:"headers"`
 	Cookies map[string]string `json:"cookies"`
-	Proxies ProxySlice        `json:"proxies" validate:"url_map_slice"`
+	Proxies []Proxy           `json:"proxies" validate:"dive"`
 	Body    string            `json:"body"`
 }
 

requests/requests.go

@@ -268,13 +268,13 @@ func printProgress(wg *sync.WaitGroup, total int, message string, countSlice *[]
 	pw.Stop()
 }
 
-func getClientFunc(proxies config.ProxySlice, timeout time.Duration, dodosCount int) func() http.Client {
+func getClientFunc(proxies []config.Proxy, timeout time.Duration, dodosCount int) func() http.Client {
 	if len(proxies) > 0 {
 		activeProxyClientsArray := make([][]http.Client, dodosCount)
 		proxiesCount := len(proxies)
 		var wg sync.WaitGroup
 		wg.Add(dodosCount + 1)
-		var proxiesSlice config.ProxySlice
+		var proxiesSlice []config.Proxy
 
 		countSlice := make([]int, dodosCount)
 		go printProgress(&wg, proxiesCount, "Searching for active proxies🌐", &countSlice)
@@ -330,7 +330,7 @@ func getClientFunc(proxies config.ProxySlice, timeout time.Duration, dodosCount
 }
 
 func findActiveProxyClients(
-	proxies config.ProxySlice,
+	proxies []config.Proxy,
 	timeout time.Duration,
 	activeProxyClients *[]http.Client,
 	counter *int,
@@ -365,12 +365,12 @@ func findActiveProxyClients(
 	}
 }
 
-func getTransport(proxy map[string]string) (*http.Transport, error) {
-	proxyURL, err := url.Parse(proxy["url"])
+func getTransport(proxy config.Proxy) (*http.Transport, error) {
+	proxyURL, err := url.Parse(proxy.URL)
 	if err != nil {
 		return nil, err
 	}
-	if _, ok := proxy["username"]; !ok {
+	if proxy.Username != "" {
 		transport := &http.Transport{
 			Proxy: http.ProxyURL(proxyURL),
 		}
@@ -382,7 +382,7 @@ func getTransport(proxy map[string]string) (*http.Transport, error) {
 			&url.URL{
 				Scheme: proxyURL.Scheme,
 				Host:   proxyURL.Host,
-				User:   url.UserPassword(proxy["username"], proxy["password"]),
+				User:   url.UserPassword(proxy.Username, proxy.Password),
 			},
 		),
 	}

validation/validator.go

@@ -4,7 +4,6 @@ import (
 	"reflect"
 	"strings"
 
-	"github.com/aykhans/dodo/config"
 	"github.com/go-playground/validator/v10"
 	"golang.org/x/net/http/httpguts"
 )
@@ -39,16 +38,19 @@ func NewValidator() *validator.Validate {
 		},
 	)
 	validation.RegisterValidation(
-		"url_map_slice",
+		"proxy_url",
 		func(fl validator.FieldLevel) bool {
-			proxies := fl.Field().Interface().(config.ProxySlice)
-			for _, proxy := range proxies {
-				if _, ok := proxy["url"]; !ok {
+			url := fl.Field().String()
+			if url == "" {
 				return false
 			}
-				if err := validation.Var(proxy["url"], "url"); err != nil {
+			if err := validation.Var(url, "url"); err != nil {
 				return false
 			}
+			if !(url[:7] == "http://" ||
+				url[:9] == "socks5://" ||
+				url[:10] == "socks5h://") {
+				return false
 			}
 			return true
 		},