/
framework_datasource.go
81 lines (68 loc) · 1.84 KB
/
framework_datasource.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
package accountavailabilities
import (
"context"
"github.com/hashicorp/terraform-plugin-framework/datasource"
"github.com/hashicorp/terraform-plugin-framework/types"
"github.com/hashicorp/terraform-plugin-log/tflog"
"github.com/linode/linodego"
"github.com/linode/terraform-provider-linode/v2/linode/helper"
)
func NewDataSource() datasource.DataSource {
return &DataSource{
BaseDataSource: helper.NewBaseDataSource(
helper.BaseDataSourceConfig{
Name: "linode_account_availabilities",
Schema: &frameworkDataSourceSchema,
},
),
}
}
type DataSource struct {
helper.BaseDataSource
}
func (r *DataSource) Read(
ctx context.Context,
req datasource.ReadRequest,
resp *datasource.ReadResponse,
) {
tflog.Debug(ctx, "Read data.linode_account_availabilities")
var data AccountAvailabilityFilterModel
client := r.Meta.Client
resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
if resp.Diagnostics.HasError() {
return
}
id, d := filterConfig.GenerateID(data.Filters)
if d != nil {
resp.Diagnostics.Append(d)
return
}
data.ID = id
result, d := filterConfig.GetAndFilter(
ctx, client, data.Filters, listAvailabilities,
types.StringNull(), types.StringNull())
if d != nil {
resp.Diagnostics.Append(d)
return
}
if d := data.parseAvailabilities(
ctx,
helper.AnySliceToTyped[linodego.AccountAvailability](result),
); d.HasError() {
resp.Diagnostics.Append(d...)
return
}
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
}
func listAvailabilities(ctx context.Context, client *linodego.Client, filter string) ([]any, error) {
tflog.Trace(ctx, "client.ListAccountAvailabilities(...)", map[string]any{
"filter": filter,
})
logins, err := client.ListAccountAvailabilities(ctx, &linodego.ListOptions{
Filter: filter,
})
if err != nil {
return nil, err
}
return helper.TypedSliceToAny(logins), nil
}