diff options
author | appilon <apilon@hashicorp.com> | 2019-02-27 16:43:31 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-02-27 16:43:31 -0500 |
commit | 844b5a68d8af4791755b8f0ad293cc99f5959183 (patch) | |
tree | 255c250a5c9d4801c74092d33b7337d8c14438ff /vendor/github.com/hashicorp/terraform/httpclient/useragent.go | |
parent | 303b299eeb6b06e939e35905e4b34cb410dd9dc3 (diff) | |
parent | 15c0b25d011f37e7c20aeca9eaf461f78285b8d9 (diff) | |
download | terraform-provider-statuscake-844b5a68d8af4791755b8f0ad293cc99f5959183.tar.gz terraform-provider-statuscake-844b5a68d8af4791755b8f0ad293cc99f5959183.tar.zst terraform-provider-statuscake-844b5a68d8af4791755b8f0ad293cc99f5959183.zip |
Merge pull request #27 from terraform-providers/go-modules-2019-02-22
[MODULES] Switch to Go Modules
Diffstat (limited to 'vendor/github.com/hashicorp/terraform/httpclient/useragent.go')
-rw-r--r-- | vendor/github.com/hashicorp/terraform/httpclient/useragent.go | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/vendor/github.com/hashicorp/terraform/httpclient/useragent.go b/vendor/github.com/hashicorp/terraform/httpclient/useragent.go new file mode 100644 index 0000000..5e28017 --- /dev/null +++ b/vendor/github.com/hashicorp/terraform/httpclient/useragent.go | |||
@@ -0,0 +1,40 @@ | |||
1 | package httpclient | ||
2 | |||
3 | import ( | ||
4 | "fmt" | ||
5 | "log" | ||
6 | "net/http" | ||
7 | "os" | ||
8 | "strings" | ||
9 | |||
10 | "github.com/hashicorp/terraform/version" | ||
11 | ) | ||
12 | |||
13 | const userAgentFormat = "Terraform/%s" | ||
14 | const uaEnvVar = "TF_APPEND_USER_AGENT" | ||
15 | |||
16 | func UserAgentString() string { | ||
17 | ua := fmt.Sprintf(userAgentFormat, version.Version) | ||
18 | |||
19 | if add := os.Getenv(uaEnvVar); add != "" { | ||
20 | add = strings.TrimSpace(add) | ||
21 | if len(add) > 0 { | ||
22 | ua += " " + add | ||
23 | log.Printf("[DEBUG] Using modified User-Agent: %s", ua) | ||
24 | } | ||
25 | } | ||
26 | |||
27 | return ua | ||
28 | } | ||
29 | |||
30 | type userAgentRoundTripper struct { | ||
31 | inner http.RoundTripper | ||
32 | userAgent string | ||
33 | } | ||
34 | |||
35 | func (rt *userAgentRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) { | ||
36 | if _, ok := req.Header["User-Agent"]; !ok { | ||
37 | req.Header.Set("User-Agent", rt.userAgent) | ||
38 | } | ||
39 | return rt.inner.RoundTrip(req) | ||
40 | } | ||