/
provider.go
112 lines (95 loc) · 2.66 KB
/
provider.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package provider
import (
"context"
"fmt"
"os"
"github.com/gjabell/terraform-provider-borgbase/gql"
"github.com/hashicorp/terraform-plugin-framework/datasource"
"github.com/hashicorp/terraform-plugin-framework/provider"
"github.com/hashicorp/terraform-plugin-framework/provider/schema"
"github.com/hashicorp/terraform-plugin-framework/resource"
"github.com/hashicorp/terraform-plugin-framework/types"
)
const borgBaseApi = "https://api.borgbase.com/graphql"
const apiTokenEnvVar = "BORGBASE_API_TOKEN"
// Ensure BorgBaseProvider satisfies various provider interfaces.
var _ provider.Provider = &BorgBaseProvider{}
type BorgBaseProvider struct {
// version is set to the provider version on release, "dev" when the
// provider is built and ran locally, and "test" when running acceptance
// testing.
version string
}
type BorgBaseProviderModel struct {
ApiToken types.String `tfsdk:"api_token"`
}
func (p *BorgBaseProvider) Metadata(
ctx context.Context,
req provider.MetadataRequest,
resp *provider.MetadataResponse,
) {
resp.TypeName = "borgbase"
resp.Version = p.version
}
func (p *BorgBaseProvider) Schema(
ctx context.Context,
req provider.SchemaRequest,
resp *provider.SchemaResponse,
) {
resp.Schema = schema.Schema{
Attributes: map[string]schema.Attribute{
"api_token": schema.StringAttribute{
MarkdownDescription: "BorgBase API token",
Optional: true,
Sensitive: true,
},
},
}
}
func (p *BorgBaseProvider) Configure(
ctx context.Context,
req provider.ConfigureRequest,
resp *provider.ConfigureResponse,
) {
var data BorgBaseProviderModel
resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
if resp.Diagnostics.HasError() {
return
}
apiToken := os.Getenv(apiTokenEnvVar)
if data.ApiToken.ValueString() != "" {
apiToken = data.ApiToken.ValueString()
}
if apiToken == "" {
resp.Diagnostics.AddError("Missing API token",
fmt.Sprintf("A BorgBase API token must be provided in the provider "+
"configuration block in the api_token attribute or in the %s env var.",
apiTokenEnvVar))
}
client := gql.NewClient(borgBaseApi, apiToken)
resp.DataSourceData = client
resp.ResourceData = client
}
func (p *BorgBaseProvider) Resources(
ctx context.Context,
) []func() resource.Resource {
return []func() resource.Resource{
NewBorgRepoResource,
NewSshKeyResource,
}
}
func (p *BorgBaseProvider) DataSources(
ctx context.Context,
) []func() datasource.DataSource {
return []func() datasource.DataSource{
NewBorgRepoDataSource,
NewSshKeyDataSource,
}
}
func New(version string) func() provider.Provider {
return func() provider.Provider {
return &BorgBaseProvider{
version: version,
}
}
}