v2fly/transport/internet/headers/http/config.go

package http

import (
	"strings"

	"v2ray.com/core/common/dice"
)

func pickString(arr []string) string {
	n := len(arr)
	switch n {
	case 0:
		return ""
	case 1:
		return arr[0]
	default:
		return arr[dice.Roll(n)]
	}
}

func (v *RequestConfig) PickUri() string {
	return pickString(v.Uri)
}

func (v *RequestConfig) PickHeaders() []string {
	n := len(v.Header)
	if n == 0 {
		return nil
	}
	headers := make([]string, n)
	for idx, headerConfig := range v.Header {
		headerName := headerConfig.Name
		headerValue := pickString(headerConfig.Value)
		headers[idx] = headerName + ": " + headerValue
	}
	return headers
}

func (v *RequestConfig) GetVersionValue() string {
	if v == nil || v.Version == nil {
		return "1.1"
	}
	return v.Version.Value
}

func (v *RequestConfig) GetMethodValue() string {
	if v == nil || v.Method == nil {
		return "GET"
	}
	return v.Method.Value
}

func (v *RequestConfig) GetFullVersion() string {
	return "HTTP/" + v.GetVersionValue()
}

func (v *ResponseConfig) HasHeader(header string) bool {
	cHeader := strings.ToLower(header)
	for _, tHeader := range v.Header {
		if strings.ToLower(tHeader.Name) == cHeader {
			return true
		}
	}
	return false
}

func (v *ResponseConfig) PickHeaders() []string {
	n := len(v.Header)
	if n == 0 {
		return nil
	}
	headers := make([]string, n)
	for idx, headerConfig := range v.Header {
		headerName := headerConfig.Name
		headerValue := pickString(headerConfig.Value)
		headers[idx] = headerName + ": " + headerValue
	}
	return headers
}

func (v *ResponseConfig) GetVersionValue() string {
	if v == nil || v.Version == nil {
		return "1.1"
	}
	return v.Version.Value
}

func (v *ResponseConfig) GetFullVersion() string {
	return "HTTP/" + v.GetVersionValue()
}

func (v *ResponseConfig) GetStatusValue() *Status {
	if v == nil || v.Status == nil {
		return &Status{
			Code:   "200",
			Reason: "OK",
		}
	}
	return v.Status
}