Merge pull request #116 from aykhans/feat/value-generator

Add value generators

Author: aykhans

Dockerfile

@@ -6,7 +6,7 @@ COPY go.mod go.sum ./
 RUN go mod download
 COPY . .
 
-RUN go build -ldflags "-s -w" -o dodo
+RUN CGO_ENABLED=0 go build -ldflags "-s -w" -o dodo
 
 FROM gcr.io/distroless/static-debian12:latest
 

README.md

@@ -12,10 +12,11 @@
 - [Usage](#usage)
     - [1. CLI Usage](#1-cli-usage)
     - [2. Config File Usage](#2-config-file-usage)
-        - [2.1 JSON Example](#21-json-example)
-        - [2.2 YAML/YML Example](#22-yamlyml-example)
+        - [2.1 YAML/YML Example](#21-yamlyml-example)
+        - [2.2 JSON Example](#22-json-example)
     - [3. CLI & Config File Combination](#3-cli--config-file-combination)
 - [Config Parameters Reference](#config-parameters-reference)
+- [Template Functions](#template-functions)
 
 ## Installation
 
@@ -73,7 +74,84 @@ docker run --rm -i aykhans/dodo -u https://example.com -m GET -d 10 -r 1000 -o 1
 
 Send 1000 GET requests to https://example.com with 10 parallel dodos (threads), each with a timeout of 800 milliseconds, within a maximum duration of 250 seconds:
 
-#### 2.1 JSON Example
+#### 2.1 YAML/YML Example
+
+```yaml
+method: "GET"
+url: "https://example.com"
+yes: false
+timeout: "800ms"
+dodos: 10
+requests: 1000
+duration: "250s"
+skip_verify: false
+
+params:
+    # A random value will be selected from the list for first "key1" param on each request
+    # And always "value" for second "key1" param on each request
+    # e.g. "?key1=value2&key1=value"
+    - key1: ["value1", "value2", "value3", "value4"]
+    - key1: "value"
+
+    # A random value will be selected from the list for param "key2" on each request
+    # e.g. "?key2=value2"
+    - key2: ["value1", "value2"]
+
+headers:
+    # A random value will be selected from the list for first "key1" header on each request
+    # And always "value" for second "key1" header on each request
+    # e.g. "key1: value3", "key1: value"
+    - key1: ["value1", "value2", "value3", "value4"]
+    - key1: "value"
+
+    # A random value will be selected from the list for header "key2" on each request
+    # e.g. "key2: value2"
+    - key2: ["value1", "value2"]
+
+cookies:
+    # A random value will be selected from the list for first "key1" cookie on each request
+    # And always "value" for second "key1" cookie on each request
+    # e.g. "key1=value4; key1=value"
+    - key1: ["value1", "value2", "value3", "value4"]
+    - key1: "value"
+
+    # A random value will be selected from the list for cookie "key2" on each request
+    # e.g. "key2=value1"
+    - key2: ["value1", "value2"]
+
+body: "body-text"
+# OR
+# A random body value will be selected from the list for each request
+body:
+    - "body-text1"
+    - "body-text2"
+    - "body-text3"
+
+proxy: "http://example.com:8080"
+# OR
+# A random proxy will be selected from the list for each request
+proxy:
+    - "http://example.com:8080"
+    - "http://username:[email protected]:8080"
+    - "socks5://example.com:8080"
+    - "socks5h://example.com:8080"
+```
+
+```sh
+dodo -f /path/config.yaml
+# OR
+dodo -f https://example.com/config.yaml
+```
+
+With Docker:
+
+```sh
+docker run --rm -i -v /path/to/config.yaml:/config.yaml aykhans/dodo -f /config.yaml
+# OR
+docker run --rm -i aykhans/dodo -f https://example.com/config.yaml
+```
+
+#### 2.2 JSON Example
 
 ```jsonc
 {
@@ -153,83 +231,6 @@ docker run --rm -i -v /path/to/config.json:/config.json aykhans/dodo
 docker run --rm -i aykhans/dodo -f https://example.com/config.json
 ```
 
-#### 2.2 YAML/YML Example
-
-```yaml
-method: "GET"
-url: "https://example.com"
-yes: false
-timeout: "800ms"
-dodos: 10
-requests: 1000
-duration: "250s"
-skip_verify: false
-
-params:
-    # A random value will be selected from the list for first "key1" param on each request
-    # And always "value" for second "key1" param on each request
-    # e.g. "?key1=value2&key1=value"
-    - key1: ["value1", "value2", "value3", "value4"]
-    - key1: "value"
-
-    # A random value will be selected from the list for param "key2" on each request
-    # e.g. "?key2=value2"
-    - key2: ["value1", "value2"]
-
-headers:
-    # A random value will be selected from the list for first "key1" header on each request
-    # And always "value" for second "key1" header on each request
-    # e.g. "key1: value3", "key1: value"
-    - key1: ["value1", "value2", "value3", "value4"]
-    - key1: "value"
-
-    # A random value will be selected from the list for header "key2" on each request
-    # e.g. "key2: value2"
-    - key2: ["value1", "value2"]
-
-cookies:
-    # A random value will be selected from the list for first "key1" cookie on each request
-    # And always "value" for second "key1" cookie on each request
-    # e.g. "key1=value4; key1=value"
-    - key1: ["value1", "value2", "value3", "value4"]
-    - key1: "value"
-
-    # A random value will be selected from the list for cookie "key2" on each request
-    # e.g. "key2=value1"
-    - key2: ["value1", "value2"]
-
-body: "body-text"
-# OR
-# A random body value will be selected from the list for each request
-body:
-    - "body-text1"
-    - "body-text2"
-    - "body-text3"
-
-proxy: "http://example.com:8080"
-# OR
-# A random proxy will be selected from the list for each request
-proxy:
-    - "http://example.com:8080"
-    - "http://username:[email protected]:8080"
-    - "socks5://example.com:8080"
-    - "socks5h://example.com:8080"
-```
-
-```sh
-dodo -f /path/config.yaml
-# OR
-dodo -f https://example.com/config.yaml
-```
-
-With Docker:
-
-```sh
-docker run --rm -i -v /path/to/config.yaml:/config.yaml aykhans/dodo -f /config.yaml
-# OR
-docker run --rm -i aykhans/dodo -f https://example.com/config.yaml
-```
-
 ### 3. CLI & Config File Combination
 
 CLI arguments override config file values:
@@ -264,3 +265,55 @@ If `Headers`, `Params`, `Cookies`, `Body`, or `Proxy` fields have multiple value
 | Body            | body        | -body        | -b             | String OR [String]             | Request body or list of request bodies                      | -       |
 | Proxy           | proxies     | -proxy       | -x             | String OR [String]             | Proxy URL or list of proxy URLs                             | -       |
 | Skip Verify     | skip_verify | -skip-verify |                | Boolean                        | Skip SSL/TLS certificate verification                       | false   |
+
+## Template Functions
+
+Dodo supports template functions in `Headers`, `Params`, `Cookies`, and `Body` fields. These functions allow you to generate dynamic values for each request.
+
+You can use Go template syntax to include dynamic values in your requests. Here's how to use template functions:
+
+In CLI config:
+
+```sh
+dodo -u https://example.com -r 1 \
+    -header "User-Agent:{{ fakeit_UserAgent }}" \ # e.g. "User-Agent: Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko)"
+    -param "username={{ strings_ToUpper fakeit_Username }}" \ # e.g. "username=JOHN BOB"
+    -cookie "token={{ fakeit_Password true true true true true 10 }}" \ # e.g. token=1234567890abcdef1234567890abcdef
+    -body '{"email":"{{ fakeit_Email }}", "password":"{{ fakeit_Password true true true true true 10 }}"}' # e.g. {"email":"[email protected]", "password":"12rw4d-78d"}
+```
+
+In YAML/YML config:
+
+```yaml
+headers:
+    - User-Agent: "{{ fakeit_UserAgent }}" # e.g. "User-Agent: Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko)"
+    - "Random-Header-{{fakeit_FirstName}}": "static_value" # e.g. "Random-Header-John: static_value"
+
+cookies:
+    - token: "Bearer {{ fakeit_UUID }}" # e.g. "token=Bearer 1234567890abcdef1234567890abcdef"
+
+params:
+    - id: "{{ fakeit_Uint }}" # e.g. "id=1234567890"
+    - username: "{{ fakeit_Username }}" # e.g. "username=John Doe"
+
+body:
+    - '{ "username": "{{ fakeit_Username }}", "password": "{{ fakeit_Password }}" }' # e.g. { "username": "john.doe", "password": "password123" }
+    - '{ "email": "{{ fakeit_Email }}", "phone": "{{ fakeit_Phone }}" }' # e.g. { "email": "[email protected]", "phone": "1234567890" }
+    - '{{ body_FormData (dict_Str "username" fakeit_Username "password" "secret123") }}' # Creates multipart form data for form submissions, automatically sets the appropriate Content-Type header.
+```
+
+In JSON config:
+
+```jsonc
+{
+    "headers": [
+        { "User-Agent": "{{ fakeit_UserAgent }}" }, // e.g. "User-Agent: Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko)"
+    ],
+    "body": [
+        "{ \"username\": \"{{ strings_RemoveSpaces fakeit_Username }}\", \"password\": \"{{ fakeit_Password }}\" }", // e.g. { "username": "johndoe", "password": "password123" }
+        "{{ body_FormData (dict_Str \"username\" fakeit_Username \"password\" \"12345\") }}", // Creates multipart form data for form submissions, automatically sets the appropriate Content-Type header.
+    ],
+}
+```
+
+For the full list of template functions over 200 functions, refer to the `NewFuncMap` function in `utils/templates.go`.

Taskfile.yaml

@@ -32,7 +32,7 @@ tasks:
 
     lint: golangci-lint run
 
-    build: go build -ldflags "-s -w" -o "dodo"
+    build: CGO_ENABLED=0 go build -ldflags "-s -w" -o "dodo"
 
     build-all:
         silent: true

config/config.go

@@ -1,12 +1,15 @@
 package config
 
 import (
+	"bytes"
 	"errors"
 	"fmt"
+	"math/rand"
 	"net/url"
 	"os"
 	"slices"
 	"strings"
+	"text/template"
 	"time"
 
 	"github.com/aykhans/dodo/types"
@@ -15,7 +18,7 @@ import (
 )
 
 const (
-	VERSION             string        = "0.6.5"
+	VERSION             string        = "0.7.0"
 	DefaultUserAgent    string        = "Dodo/" + VERSION
 	DefaultMethod       string        = "GET"
 	DefaultTimeout      time.Duration = time.Second * 10
@@ -201,6 +204,98 @@ func (config *Config) Validate() []error {
 		}
 	}
 
+	funcMap := *utils.NewFuncMapGenerator(
+		rand.New(
+			rand.NewSource(
+				time.Now().UnixNano(),
+			),
+		),
+	).GetFuncMap()
+
+	for _, header := range config.Headers {
+		t, err := template.New("default").Funcs(funcMap).Parse(header.Key)
+		if err != nil {
+			errs = append(errs, fmt.Errorf("header key (%s) parse error: %v", header.Key, err))
+		} else {
+			var buf bytes.Buffer
+			if err = t.Execute(&buf, nil); err != nil {
+				errs = append(errs, fmt.Errorf("header key (%s) parse error: %v", header.Key, err))
+			}
+		}
+
+		for _, value := range header.Value {
+			t, err := template.New("default").Funcs(funcMap).Parse(value)
+			if err != nil {
+				errs = append(errs, fmt.Errorf("header value (%s) parse error: %v", value, err))
+			} else {
+				var buf bytes.Buffer
+				if err = t.Execute(&buf, nil); err != nil {
+					errs = append(errs, fmt.Errorf("header value (%s) parse error: %v", value, err))
+				}
+			}
+		}
+	}
+
+	for _, cookie := range config.Cookies {
+		t, err := template.New("default").Funcs(funcMap).Parse(cookie.Key)
+		if err != nil {
+			errs = append(errs, fmt.Errorf("cookie key (%s) parse error: %v", cookie.Key, err))
+		} else {
+			var buf bytes.Buffer
+			if err = t.Execute(&buf, nil); err != nil {
+				errs = append(errs, fmt.Errorf("cookie key (%s) parse error: %v", cookie.Key, err))
+			}
+		}
+
+		for _, value := range cookie.Value {
+			t, err := template.New("default").Funcs(funcMap).Parse(value)
+			if err != nil {
+				errs = append(errs, fmt.Errorf("cookie value (%s) parse error: %v", value, err))
+			} else {
+				var buf bytes.Buffer
+				if err = t.Execute(&buf, nil); err != nil {
+					errs = append(errs, fmt.Errorf("cookie value (%s) parse error: %v", value, err))
+				}
+			}
+		}
+	}
+
+	for _, param := range config.Params {
+		t, err := template.New("default").Funcs(funcMap).Parse(param.Key)
+		if err != nil {
+			errs = append(errs, fmt.Errorf("param key (%s) parse error: %v", param.Key, err))
+		} else {
+			var buf bytes.Buffer
+			if err = t.Execute(&buf, nil); err != nil {
+				errs = append(errs, fmt.Errorf("param key (%s) parse error: %v", param.Key, err))
+			}
+		}
+
+		for _, value := range param.Value {
+			t, err := template.New("default").Funcs(funcMap).Parse(value)
+			if err != nil {
+				errs = append(errs, fmt.Errorf("param value (%s) parse error: %v", value, err))
+			} else {
+				var buf bytes.Buffer
+				if err = t.Execute(&buf, nil); err != nil {
+					errs = append(errs, fmt.Errorf("param value (%s) parse error: %v", value, err))
+				}
+			}
+		}
+	}
+
+	for _, body := range config.Body {
+		t, err := template.New("default").Funcs(funcMap).Parse(body)
+		if err != nil {
+			errs = append(errs, fmt.Errorf("body (%s) parse error: %v", body, err))
+		} else {
+			var buf bytes.Buffer
+			if err = t.Execute(&buf, nil); err != nil {
+				errs = append(errs, fmt.Errorf("body (%s) parse error: %v", body, err))
+			}
+		}
+	}
+
 	return errs
 }
 

go.mod

@@ -3,6 +3,7 @@ module github.com/aykhans/dodo
 go 1.24.2
 
 require (
+	github.com/brianvoe/gofakeit/v7 v7.2.1
 	github.com/jedib0t/go-pretty/v6 v6.6.7
 	github.com/valyala/fasthttp v1.62.0
 	gopkg.in/yaml.v3 v3.0.1

go.sum

@@ -1,5 +1,7 @@
 github.com/andybalholm/brotli v1.1.1 h1:PR2pgnyFznKEugtsUo0xLdDop5SKXd5Qf5ysW+7XdTA=
 github.com/andybalholm/brotli v1.1.1/go.mod h1:05ib4cKhjx3OQYUY22hTVd34Bc8upXjOLL2rKwwZBoA=
+github.com/brianvoe/gofakeit/v7 v7.2.1 h1:AGojgaaCdgq4Adzrd2uWdbGNDyX6MWNhHdQBraNfOHI=
+github.com/brianvoe/gofakeit/v7 v7.2.1/go.mod h1:QXuPeBw164PJCzCUZVmgpgHJ3Llj49jSLVkKPMtxtxA=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/jedib0t/go-pretty/v6 v6.6.7 h1:m+LbHpm0aIAPLzLbMfn8dc3Ht8MW7lsSO4MPItz/Uuo=