-
Notifications
You must be signed in to change notification settings - Fork 0
/
encode.go
111 lines (100 loc) · 2.38 KB
/
encode.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
// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0
// Originally copied from https://github.com/hashicorp/terraform-provider-kubernetes/blob/main/internal/framework/provider/functions/encode.go
package helpers
import (
"fmt"
"github.com/hashicorp/terraform-plugin-framework/attr"
"github.com/hashicorp/terraform-plugin-framework/types/basetypes"
)
func EncodeValue(v attr.Value) (any, error) {
if v.IsNull() {
return nil, nil
}
switch vv := v.(type) {
case basetypes.StringValue:
return vv.ValueString(), nil
case basetypes.NumberValue:
f, _ := vv.ValueBigFloat().Float64()
return f, nil
case basetypes.BoolValue:
return vv.ValueBool(), nil
case basetypes.ObjectValue:
return EncodeObject(vv)
case basetypes.TupleValue:
return EncodeTuple(vv)
case basetypes.MapValue:
return EncodeMap(vv)
case basetypes.ListValue:
return EncodeList(vv)
case basetypes.SetValue:
return EncodeSet(vv)
case basetypes.DynamicValue:
return EncodeValue(vv.UnderlyingValue())
default:
return nil, fmt.Errorf("tried to encode unsupported type: %T: %v", v, vv)
}
}
func EncodeSet(sv basetypes.SetValue) ([]any, error) {
elems := sv.Elements()
size := len(elems)
l := make([]any, size)
for i := 0; i < size; i++ {
var err error
l[i], err = EncodeValue(elems[i])
if err != nil {
return nil, err
}
}
return l, nil
}
func EncodeList(lv basetypes.ListValue) ([]any, error) {
elems := lv.Elements()
size := len(elems)
l := make([]any, size)
for i := 0; i < size; i++ {
var err error
l[i], err = EncodeValue(elems[i])
if err != nil {
return nil, err
}
}
return l, nil
}
func EncodeTuple(tv basetypes.TupleValue) ([]any, error) {
elems := tv.Elements()
size := len(elems)
l := make([]any, size)
for i := 0; i < size; i++ {
var err error
l[i], err = EncodeValue(elems[i])
if err != nil {
return nil, err
}
}
return l, nil
}
func EncodeObject(ov basetypes.ObjectValue) (map[string]any, error) {
attrs := ov.Attributes()
m := make(map[string]any, len(attrs))
for k, v := range attrs {
var err error
m[k], err = EncodeValue(v)
if err != nil {
return nil, err
}
}
return m, nil
}
func EncodeMap(mv basetypes.MapValue) (map[string]any, error) {
elems := mv.Elements()
m := make(map[string]any, len(elems))
for k, v := range elems {
var err error
m[k], err = EncodeValue(v)
if err != nil {
return nil, err
}
}
return m, nil
}