mirror of
				https://github.com/aykhans/dodo.git
				synced 2025-11-03 22:19:58 +00:00 
			
		
		
		
	Compare commits
	
		
			6 Commits
		
	
	
		
			b9dd14a588
			...
			ff09a3365e
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| ff09a3365e | |||
| c83246abe4 | |||
| efc4c62001 | |||
| 
						 | 
					f567774e3a | ||
| d2bd60e3ff | |||
| 0fe782c768 | 
@@ -59,9 +59,9 @@ You can find an example config structure in the [config.json](https://github.com
 | 
				
			|||||||
    "method": "GET",
 | 
					    "method": "GET",
 | 
				
			||||||
    "url": "https://example.com",
 | 
					    "url": "https://example.com",
 | 
				
			||||||
    "no_proxy_check": false,
 | 
					    "no_proxy_check": false,
 | 
				
			||||||
    "timeout": 10000,
 | 
					    "timeout": 2000,
 | 
				
			||||||
    "dodos_count": 1,
 | 
					    "dodos_count": 10,
 | 
				
			||||||
    "request_count": 1,
 | 
					    "request_count": 1000,
 | 
				
			||||||
    "params": {},
 | 
					    "params": {},
 | 
				
			||||||
    "headers": {},
 | 
					    "headers": {},
 | 
				
			||||||
    "cookies": {},
 | 
					    "cookies": {},
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -1,9 +1,9 @@
 | 
				
			|||||||
package config
 | 
					package config
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"fmt"
 | 
					 | 
				
			||||||
	"net/url"
 | 
						"net/url"
 | 
				
			||||||
	"os"
 | 
						"os"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
	"time"
 | 
						"time"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	. "github.com/aykhans/dodo/types"
 | 
						. "github.com/aykhans/dodo/types"
 | 
				
			||||||
@@ -42,7 +42,18 @@ func (config *RequestConfig) Print() {
 | 
				
			|||||||
	t.SetOutputMirror(os.Stdout)
 | 
						t.SetOutputMirror(os.Stdout)
 | 
				
			||||||
	t.SetStyle(table.StyleLight)
 | 
						t.SetStyle(table.StyleLight)
 | 
				
			||||||
	t.SetColumnConfigs([]table.ColumnConfig{
 | 
						t.SetColumnConfigs([]table.ColumnConfig{
 | 
				
			||||||
		{Number: 2, WidthMax: 50},
 | 
							{
 | 
				
			||||||
 | 
								Number: 2,
 | 
				
			||||||
 | 
								WidthMaxEnforcer: func(col string, maxLen int) string {
 | 
				
			||||||
 | 
									lines := strings.Split(col, "\n")
 | 
				
			||||||
 | 
									for i, line := range lines {
 | 
				
			||||||
 | 
										if len(line) > maxLen {
 | 
				
			||||||
 | 
											lines[i] = line[:maxLen-3] + "..."
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									return strings.Join(lines, "\n")
 | 
				
			||||||
 | 
								},
 | 
				
			||||||
 | 
								WidthMax: 50},
 | 
				
			||||||
	})
 | 
						})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	newHeaders := make(map[string][]string)
 | 
						newHeaders := make(map[string][]string)
 | 
				
			||||||
@@ -56,23 +67,23 @@ func (config *RequestConfig) Print() {
 | 
				
			|||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"URL", config.URL})
 | 
						t.AppendRow(table.Row{"URL", config.URL})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Timeout", fmt.Sprintf("%dms", config.Timeout/time.Millisecond)})
 | 
						t.AppendRow(table.Row{"Timeout", config.Timeout})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Dodos", config.DodosCount})
 | 
						t.AppendRow(table.Row{"Dodos", config.DodosCount})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Requests", config.RequestCount})
 | 
						t.AppendRow(table.Row{"Requests", config.RequestCount})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Params", utils.MarshalJSON(config.Params, 3)})
 | 
						t.AppendRow(table.Row{"Params", string(utils.PrettyJSONMarshal(config.Params, 3, "", "  "))})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Headers", utils.MarshalJSON(newHeaders, 3)})
 | 
						t.AppendRow(table.Row{"Headers", string(utils.PrettyJSONMarshal(newHeaders, 3, "", "  "))})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Cookies", utils.MarshalJSON(config.Cookies, 3)})
 | 
						t.AppendRow(table.Row{"Cookies", string(utils.PrettyJSONMarshal(config.Cookies, 3, "", "  "))})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Proxies Count", len(config.Proxies)})
 | 
						t.AppendRow(table.Row{"Proxies Count", string(utils.PrettyJSONMarshal(config.Proxies, 3, "", "  "))})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Proxy Check", !config.NoProxyCheck})
 | 
						t.AppendRow(table.Row{"Proxy Check", !config.NoProxyCheck})
 | 
				
			||||||
	t.AppendSeparator()
 | 
						t.AppendSeparator()
 | 
				
			||||||
	t.AppendRow(table.Row{"Body", utils.MarshalJSON(config.Body, 3)})
 | 
						t.AppendRow(table.Row{"Body", string(utils.PrettyJSONMarshal(config.Body, 3, "", "  "))})
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	t.Render()
 | 
						t.Render()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										2
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										2
									
								
								go.mod
									
									
									
									
									
								
							@@ -6,7 +6,7 @@ require (
 | 
				
			|||||||
	github.com/go-playground/validator/v10 v10.23.0
 | 
						github.com/go-playground/validator/v10 v10.23.0
 | 
				
			||||||
	github.com/jedib0t/go-pretty/v6 v6.6.5
 | 
						github.com/jedib0t/go-pretty/v6 v6.6.5
 | 
				
			||||||
	github.com/valyala/fasthttp v1.58.0
 | 
						github.com/valyala/fasthttp v1.58.0
 | 
				
			||||||
	golang.org/x/net v0.32.0
 | 
						golang.org/x/net v0.33.0
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
require (
 | 
					require (
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										4
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										4
									
								
								go.sum
									
									
									
									
									
								
							@@ -35,8 +35,8 @@ github.com/xyproto/randomstring v1.0.5 h1:YtlWPoRdgMu3NZtP45drfy1GKoojuR7hmRcnhZ
 | 
				
			|||||||
github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
 | 
					github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
 | 
				
			||||||
golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
 | 
					golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
 | 
				
			||||||
golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=
 | 
					golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=
 | 
				
			||||||
golang.org/x/net v0.32.0 h1:ZqPmj8Kzc+Y6e0+skZsuACbx+wzMgo5MQsJh9Qd6aYI=
 | 
					golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I=
 | 
				
			||||||
golang.org/x/net v0.32.0/go.mod h1:CwU0IoeOlnQQWJ6ioyFrfRuomB8GKF6KbYXZVyeXNfs=
 | 
					golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4=
 | 
				
			||||||
golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=
 | 
					golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=
 | 
				
			||||||
golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 | 
					golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 | 
				
			||||||
golang.org/x/term v0.27.0 h1:WP60Sv1nlK1T6SupCHbXzSaN0b9wUmsPoRS9b61A23Q=
 | 
					golang.org/x/term v0.27.0 h1:WP60Sv1nlK1T6SupCHbXzSaN0b9wUmsPoRS9b61A23Q=
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										102
									
								
								utils/convert.go
									
									
									
									
									
								
							
							
						
						
									
										102
									
								
								utils/convert.go
									
									
									
									
									
								
							@@ -6,49 +6,77 @@ import (
 | 
				
			|||||||
	"reflect"
 | 
						"reflect"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func MarshalJSON(v any, maxSliceSize uint) string {
 | 
					type TruncatedMarshaller struct {
 | 
				
			||||||
	rv := reflect.ValueOf(v)
 | 
						Value    interface{}
 | 
				
			||||||
	if rv.Kind() == reflect.Slice && rv.Len() == 0 {
 | 
						MaxItems int
 | 
				
			||||||
		return "[]"
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	data, err := json.MarshalIndent(truncateLists(v, int(maxSliceSize)), "", "  ")
 | 
					 | 
				
			||||||
	if err != nil {
 | 
					 | 
				
			||||||
		return "{}"
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return string(data)
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func truncateLists(v interface{}, maxItems int) interface{} {
 | 
					func (t TruncatedMarshaller) MarshalJSON() ([]byte, error) {
 | 
				
			||||||
	rv := reflect.ValueOf(v)
 | 
						val := reflect.ValueOf(t.Value)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	switch rv.Kind() {
 | 
						if val.Kind() != reflect.Slice && val.Kind() != reflect.Array {
 | 
				
			||||||
	case reflect.Slice, reflect.Array:
 | 
							return json.Marshal(t.Value)
 | 
				
			||||||
		if rv.Len() > maxItems {
 | 
					 | 
				
			||||||
			newSlice := reflect.MakeSlice(rv.Type(), maxItems, maxItems)
 | 
					 | 
				
			||||||
			reflect.Copy(newSlice, rv.Slice(0, maxItems))
 | 
					 | 
				
			||||||
			newSlice = reflect.Append(newSlice, reflect.ValueOf(fmt.Sprintf("...(%d more)", rv.Len()-maxItems)))
 | 
					 | 
				
			||||||
			return newSlice.Interface()
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						length := val.Len()
 | 
				
			||||||
 | 
						if length <= t.MaxItems {
 | 
				
			||||||
 | 
							return json.Marshal(t.Value)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						truncated := make([]interface{}, t.MaxItems+1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for i := 0; i < t.MaxItems; i++ {
 | 
				
			||||||
 | 
							truncated[i] = val.Index(i).Interface()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						remaining := length - t.MaxItems
 | 
				
			||||||
 | 
						truncated[t.MaxItems] = fmt.Sprintf("+%d", remaining)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return json.Marshal(truncated)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func PrettyJSONMarshal(v interface{}, maxItems int, prefix, indent string) []byte {
 | 
				
			||||||
 | 
						truncated := processValue(v, maxItems)
 | 
				
			||||||
 | 
						d, _ := json.MarshalIndent(truncated, prefix, indent)
 | 
				
			||||||
 | 
						return d
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func processValue(v interface{}, maxItems int) interface{} {
 | 
				
			||||||
 | 
						val := reflect.ValueOf(v)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						switch val.Kind() {
 | 
				
			||||||
	case reflect.Map:
 | 
						case reflect.Map:
 | 
				
			||||||
		newMap := reflect.MakeMap(rv.Type())
 | 
							newMap := make(map[string]interface{})
 | 
				
			||||||
		for _, key := range rv.MapKeys() {
 | 
							iter := val.MapRange()
 | 
				
			||||||
			newMap.SetMapIndex(key, reflect.ValueOf(truncateLists(rv.MapIndex(key).Interface(), maxItems)))
 | 
							for iter.Next() {
 | 
				
			||||||
		}
 | 
								k := iter.Key().String()
 | 
				
			||||||
		return newMap.Interface()
 | 
								newMap[k] = processValue(iter.Value().Interface(), maxItems)
 | 
				
			||||||
	case reflect.Struct:
 | 
					 | 
				
			||||||
		newStruct := reflect.New(rv.Type()).Elem()
 | 
					 | 
				
			||||||
		for i := 0; i < rv.NumField(); i++ {
 | 
					 | 
				
			||||||
			newStruct.Field(i).Set(reflect.ValueOf(truncateLists(rv.Field(i).Interface(), maxItems)))
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		return newStruct.Interface()
 | 
					 | 
				
			||||||
	case reflect.Ptr:
 | 
					 | 
				
			||||||
		if rv.IsNil() {
 | 
					 | 
				
			||||||
			return nil
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		return truncateLists(rv.Elem().Interface(), maxItems)
 | 
					 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
							return newMap
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						case reflect.Slice, reflect.Array:
 | 
				
			||||||
 | 
							return TruncatedMarshaller{Value: v, MaxItems: maxItems}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						case reflect.Struct:
 | 
				
			||||||
 | 
							newMap := make(map[string]interface{})
 | 
				
			||||||
 | 
							t := val.Type()
 | 
				
			||||||
 | 
							for i := 0; i < t.NumField(); i++ {
 | 
				
			||||||
 | 
								field := t.Field(i)
 | 
				
			||||||
 | 
								if field.IsExported() {
 | 
				
			||||||
 | 
									jsonTag := field.Tag.Get("json")
 | 
				
			||||||
 | 
									if jsonTag == "-" {
 | 
				
			||||||
 | 
										continue
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									fieldName := field.Name
 | 
				
			||||||
 | 
									if jsonTag != "" {
 | 
				
			||||||
 | 
										fieldName = jsonTag
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									newMap[fieldName] = processValue(val.Field(i).Interface(), maxItems)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return newMap
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
		return v
 | 
							return v
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user