-
Notifications
You must be signed in to change notification settings - Fork 1
/
kinesis.go
86 lines (66 loc) · 1.97 KB
/
kinesis.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
package kinesis
import (
_ "embed"
"fmt"
"os"
"path"
"strings"
"github.com/joselitofilho/aws-terraform-generator/internal/generators"
"github.com/joselitofilho/aws-terraform-generator/internal/generators/config"
generatorserrs "github.com/joselitofilho/aws-terraform-generator/internal/generators/errors"
)
type Data struct {
Name string
KMSEncription bool
RetentionPeriod string
KMSKeyID string
}
type Kinesis struct {
configFileName string
output string
}
func NewKinesis(configFileName, output string) *Kinesis {
return &Kinesis{configFileName: configFileName, output: output}
}
func (k *Kinesis) Build() error {
yamlParser := config.NewYAML(k.configFileName)
yamlConfig, err := yamlParser.Parse()
if err != nil {
return fmt.Errorf("%w: %w", generatorserrs.ErrYAMLParse, err)
}
modPath := path.Join(k.output, "mod")
_ = os.MkdirAll(modPath, os.ModePerm)
tmplName := "kinesis-tf-template"
result := make([]string, 0, len(yamlConfig.Kinesis))
for i := range yamlConfig.Kinesis {
conf := yamlConfig.Kinesis[i]
data := Data{
Name: conf.Name,
KMSEncription: conf.KMSKeyID != "",
RetentionPeriod: conf.RetentionPeriod,
KMSKeyID: conf.KMSKeyID,
}
if len(conf.Files) > 0 {
filesConf := generators.CreateFilesMap(conf.Files)
err = generators.GenerateFiles(nil, filesConf, data, modPath)
if err != nil {
return fmt.Errorf("%w", err)
}
fmt.Printf("Kinesis '%s' has been generated successfully\n", conf.Name)
}
output, err := generators.Build(data, tmplName, string(kinesisTFTmpl))
if err != nil {
return fmt.Errorf("%w", err)
}
result = append(result, output)
}
if len(result) > 0 {
outputFile := path.Join(modPath, "kinesis.tf")
err := generators.GenerateFile(defaultTfTemplateFiles, tmplName, strings.Join(result, "\n"), outputFile, Data{})
if err != nil {
return fmt.Errorf("%w", err)
}
fmt.Println("Kinesis has been generated successfully")
}
return nil
}