matterbridge/vendor/github.com/SevereCloud/vksdk/v2/internal/transport.go

/*
Package internal unimportable
*/
package internal // import "github.com/SevereCloud/vksdk/v2/internal"

import (
	"context"
	"net/http"

	"github.com/SevereCloud/vksdk/v2"
)

// UserAgent module.
const UserAgent = "vksdk/" + vksdk.Version + " (+https://github.com/SevereCloud/vksdk)"

// ContextKey is just an empty struct. It exists so HTTPClient can be
// an immutable public variable with a unique type. It's immutable
// because nobody else can create a ContextKey, being unexported.
type ContextKey int

// ContextKey list.
const (
	HTTPClientKey ContextKey = iota
	UserAgentKey
	GroupIDKey
	EventIDKey
	LongPollTsKey
	CallbackRetryCounterKey
	CallbackRetryAfterKey
)

// ContextClient return *http.Client.
func ContextClient(ctx context.Context) *http.Client {
	if ctx != nil {
		if hc, ok := ctx.Value(HTTPClientKey).(*http.Client); ok {
			return hc
		}
	}

	return http.DefaultClient
}

// ContextUserAgent return User-Agent from context.
func ContextUserAgent(ctx context.Context) string {
	if ctx != nil {
		if hc, ok := ctx.Value(UserAgentKey).(string); ok {
			return hc
		}
	}

	return UserAgent
}

// DoRequest sends an HTTP request and returns an HTTP response.
//
// The provided ctx must be non-nil. If it is canceled or times out,
// ctx.Err() will be returned.
func DoRequest(ctx context.Context, req *http.Request) (*http.Response, error) {
	client := ContextClient(ctx)

	if req.UserAgent() == "" {
		req.Header.Set("User-Agent", ContextUserAgent(ctx))
	}

	resp, err := client.Do(req.WithContext(ctx))
	// If we got an error, and the context has been canceled,
	// the context's error is probably more useful.
	if err != nil {
		<-ctx.Done()
		err = ctx.Err()
	}

	return resp, err
}