Mohammed Naser | 3b3507d | 2023-11-28 21:13:39 -0500 | [diff] [blame] | 1 | package defaults |
| 2 | |
| 3 | import ( |
| 4 | "bytes" |
| 5 | _ "embed" |
Oleksandr K. | 99651a6 | 2024-10-30 04:41:51 +0100 | [diff] [blame] | 6 | "strings" |
Mohammed Naser | 3b3507d | 2023-11-28 21:13:39 -0500 | [diff] [blame] | 7 | |
| 8 | "github.com/goccy/go-yaml" |
| 9 | ) |
| 10 | |
| 11 | var ( |
| 12 | //go:embed vars/main.yml |
Mohammed Naser | 59853d4 | 2023-11-29 20:32:24 -0500 | [diff] [blame] | 13 | varsFile []byte |
Mohammed Naser | 3b3507d | 2023-11-28 21:13:39 -0500 | [diff] [blame] | 14 | ) |
| 15 | |
Oleksandr K. | 99651a6 | 2024-10-30 04:41:51 +0100 | [diff] [blame] | 16 | // Define a global variable for the release value. |
| 17 | var release = "main" |
| 18 | |
Mohammed Naser | 4c8e0cb | 2024-02-21 11:51:34 -0500 | [diff] [blame] | 19 | func GetImages() (map[string]string, error) { |
Oleksandr K. | 99651a6 | 2024-10-30 04:41:51 +0100 | [diff] [blame] | 20 | // Replace {{ release }} with the actual release value |
Jason Hall | 2647624 | 2024-02-19 12:56:23 -0600 | [diff] [blame] | 21 | modifiedVarsFile := []byte(strings.ReplaceAll(string(varsFile), "{{ atmosphere_release }}", release)) |
| 22 | |
| 23 | // Fix prefixes for images to allow tests to run |
| 24 | modifiedVarsFile = []byte(strings.ReplaceAll(string(modifiedVarsFile), "{{ atmosphere_image_prefix }}registry.atmosphere.dev", "harbor.atmosphere.dev")) |
| 25 | modifiedVarsFile = []byte(strings.ReplaceAll(string(modifiedVarsFile), "{{ atmosphere_image_prefix }}", "harbor.atmosphere.dev/")) |
Oleksandr K. | 99651a6 | 2024-10-30 04:41:51 +0100 | [diff] [blame] | 26 | |
Mohammed Naser | 4c8e0cb | 2024-02-21 11:51:34 -0500 | [diff] [blame] | 27 | path, err := yaml.PathString("$._atmosphere_images") |
| 28 | if err != nil { |
| 29 | return nil, err |
| 30 | } |
| 31 | |
| 32 | var images map[string]string |
Oleksandr K. | 99651a6 | 2024-10-30 04:41:51 +0100 | [diff] [blame] | 33 | if err := path.Read(bytes.NewReader(modifiedVarsFile), &images); err != nil { |
Mohammed Naser | 4c8e0cb | 2024-02-21 11:51:34 -0500 | [diff] [blame] | 34 | return nil, err |
| 35 | } |
| 36 | |
| 37 | return images, nil |
| 38 | } |
| 39 | |
Mohammed Naser | 3b3507d | 2023-11-28 21:13:39 -0500 | [diff] [blame] | 40 | func GetImageByKey(key string) (string, error) { |
| 41 | path, err := yaml.PathString("$._atmosphere_images." + key) |
| 42 | if err != nil { |
| 43 | return "", err |
| 44 | } |
| 45 | |
| 46 | var image string |
Mohammed Naser | 59853d4 | 2023-11-29 20:32:24 -0500 | [diff] [blame] | 47 | if err := path.Read(bytes.NewReader(varsFile), &image); err != nil { |
Mohammed Naser | 3b3507d | 2023-11-28 21:13:39 -0500 | [diff] [blame] | 48 | return "", err |
| 49 | } |
| 50 | |
| 51 | return image, nil |
| 52 | } |