Skip to content

Commit a5e4875

Browse files
authored
Merge pull request #46 from planetscale/set-user-agent
http: set user-agent
2 parents eba4591 + 09c4ee1 commit a5e4875

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

internal/provider/provider.go

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,14 @@ func (p *PlanetScaleProvider) Configure(ctx context.Context, req provider.Config
158158
return
159159
}
160160

161-
client := planetscale.NewClient(&http.Client{Transport: rt}, baseURL)
161+
client := planetscale.NewClient(
162+
&http.Client{
163+
Transport: roundTripperFunc(func(r *http.Request) (*http.Response, error) {
164+
r.Header.Set("User-Agent", "PlanetScale_Terraform_Provider/"+p.version+" (Terraform "+req.TerraformVersion+")")
165+
return rt.RoundTrip(r)
166+
}),
167+
}, baseURL,
168+
)
162169

163170
resp.DataSourceData = client
164171
resp.ResourceData = client

main.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import (
77
"context"
88
"flag"
99
"log"
10+
"os"
1011

1112
"github.com/hashicorp/terraform-plugin-framework/providerserver"
1213
"github.com/planetscale/terraform-provider-planetscale/internal/provider"
@@ -44,7 +45,8 @@ func main() {
4445
Debug: debug,
4546
}
4647

47-
err := providerserver.Serve(context.Background(), provider.New(version, debug), opts)
48+
debugProvider := os.Getenv("TF_PS_PROVIDER_DEBUG") != ""
49+
err := providerserver.Serve(context.Background(), provider.New(version, debug || debugProvider), opts)
4850

4951
if err != nil {
5052
log.Fatal(err.Error())

0 commit comments

Comments
 (0)