diff --git a/utils/strings.go b/utils/strings.go new file mode 100644 index 0000000..beacfe1 --- /dev/null +++ b/utils/strings.go @@ -0,0 +1,31 @@ +package utils + +import ( + "encoding/json" + "gopkg.in/yaml.v1" + "math/rand" + "strings" +) + +var letterRunes = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ") + +func RandStringRunes(n int) string { + b := make([]rune, n) + for i := range b { + b[i] = letterRunes[rand.Intn(len(letterRunes))] + } + return string(b) +} + +func ListOutput(rels []string, output string) []string { + switch strings.ToLower(output) { + case "yaml": + d, _ := yaml.Marshal(rels) + return []string{string(d)} + case "json": + d, _ := json.Marshal(rels) + return []string{string(d)} + default: + return rels + } +}