/
provider.go
118 lines (99 loc) · 3.07 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
113
114
115
116
117
118
package provider
import (
"context"
"os"
"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"
)
// Ensure ShieldooProvider satisfies various provider interfaces.
var _ provider.Provider = &ShieldooProvider{}
// ShieldooProvider defines the provider implementation.
type ShieldooProvider 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
}
// SshieldooProviderModel describes the provider data model.
type ShieldooProviderModel struct {
Endpoint types.String `tfsdk:"endpoint"`
ApiKey types.String `tfsdk:"apikey"`
}
func (p *ShieldooProvider) Metadata(ctx context.Context, req provider.MetadataRequest, resp *provider.MetadataResponse) {
resp.TypeName = "shieldoo"
resp.Version = p.version
}
func (p *ShieldooProvider) Schema(ctx context.Context, req provider.SchemaRequest, resp *provider.SchemaResponse) {
resp.Schema = schema.Schema{
Attributes: map[string]schema.Attribute{
"endpoint": schema.StringAttribute{
MarkdownDescription: "Shieldoo API endpoint",
Optional: true,
},
"apikey": schema.StringAttribute{
MarkdownDescription: "Shieldoo API Key",
Optional: true,
Sensitive: true,
},
},
}
}
func (p *ShieldooProvider) Configure(ctx context.Context, req provider.ConfigureRequest, resp *provider.ConfigureResponse) {
var data ShieldooProviderModel
apiKey := os.Getenv("SHIELDOO_API_KEY")
endpoint := os.Getenv("SHIELDOO_ENDPOINT")
resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
if resp.Diagnostics.HasError() {
return
}
if data.Endpoint.ValueString() != "" {
endpoint = data.Endpoint.ValueString()
}
if data.ApiKey.ValueString() != "" {
apiKey = data.ApiKey.ValueString()
}
// Configuration values are now available.
if endpoint == "" {
resp.Diagnostics.AddError(
"endpoint is required",
"Please set the endpoint in the provider configuration block.",
)
return
}
if apiKey == "" {
resp.Diagnostics.AddError(
"apikey is required",
"Please set the apikey in the provider configuration block.",
)
return
}
// Example client configuration for data sources and resources
client := &ShieldooClient{
uri: endpoint,
apiKey: apiKey,
}
resp.DataSourceData = client
resp.ResourceData = client
}
func (p *ShieldooProvider) Resources(ctx context.Context) []func() resource.Resource {
return []func() resource.Resource{
NewFirewallResource,
NewServerResource,
}
}
func (p *ShieldooProvider) DataSources(ctx context.Context) []func() datasource.DataSource {
return []func() datasource.DataSource{
NewFirewallDataSource,
NewServerDataSource,
}
}
func New(version string) func() provider.Provider {
return func() provider.Provider {
return &ShieldooProvider{
version: version,
}
}
}