plugins/wasm-go/pkg/wrapper/request_wrapper.go

// Copyright (c) 2022 Alibaba Group Holding Ltd.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//	http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package wrapper

import (
	"net/url"
	"strconv"
	"strings"

	"github.com/higress-group/proxy-wasm-go-sdk/proxywasm"
)

func GetRequestScheme() string {
	scheme, err := proxywasm.GetHttpRequestHeader(":scheme")
	if err != nil {
		proxywasm.LogErrorf("get request scheme failed: %v", err)
		return ""
	}
	return scheme
}

func GetRequestHost() string {
	host, err := proxywasm.GetHttpRequestHeader(":authority")
	if err != nil {
		proxywasm.LogErrorf("get request host failed: %v", err)
		return ""
	}
	return host
}

func GetRequestPath() string {
	path, err := proxywasm.GetHttpRequestHeader(":path")
	if err != nil {
		proxywasm.LogErrorf("get request path failed: %v", err)
		return ""
	}
	return path
}

func GetRequestPathWithoutQuery() string {
	rawPath := GetRequestPath()
	if rawPath == "" {
		return ""
	}
	path, err := url.Parse(rawPath)
	if err != nil {
		proxywasm.LogErrorf("failed to parse request path '%s': %v", rawPath, err)
		return ""
	}
	return path.Path
}

func GetRequestMethod() string {
	method, err := proxywasm.GetHttpRequestHeader(":method")
	if err != nil {
		proxywasm.LogErrorf("get request method failed: %v", err)
		return ""
	}
	return method
}

func IsBinaryRequestBody() bool {
	contentType, _ := proxywasm.GetHttpRequestHeader("content-type")
	if strings.Contains(contentType, "octet-stream") ||
		strings.Contains(contentType, "grpc") {
		return true
	}
	encoding, _ := proxywasm.GetHttpRequestHeader("content-encoding")
	if encoding != "" {
		return true
	}
	return false
}

func IsBinaryResponseBody() bool {
	contentType, _ := proxywasm.GetHttpResponseHeader("content-type")
	if strings.Contains(contentType, "octet-stream") ||
		strings.Contains(contentType, "grpc") {
		return true
	}
	encoding, _ := proxywasm.GetHttpResponseHeader("content-encoding")
	if encoding != "" {
		return true
	}
	return false
}

func HasRequestBody() bool {
	contentTypeStr, _ := proxywasm.GetHttpRequestHeader("content-type")
	contentLengthStr, _ := proxywasm.GetHttpRequestHeader("content-length")
	transferEncodingStr, _ := proxywasm.GetHttpRequestHeader("transfer-encoding")
	proxywasm.LogDebugf("check has request body: contentType:%s, contentLengthStr:%s, transferEncodingStr:%s",
		contentTypeStr, contentLengthStr, transferEncodingStr)
	if contentTypeStr != "" {
		return true
	}
	if contentLengthStr != "" {
		contentLength, err := strconv.Atoi(contentLengthStr)
		if err == nil && contentLength > 0 {
			return true
		}
	}
	return strings.Contains(transferEncodingStr, "chunked")
}