|
| 1 | +package main |
| 2 | + |
| 3 | +import ( |
| 4 | + "builder/installer" |
| 5 | + "encoding/json" |
| 6 | + "flag" |
| 7 | + "fmt" |
| 8 | + "net/url" |
| 9 | + "os" |
| 10 | + "regexp" |
| 11 | + "strings" |
| 12 | + |
| 13 | + "github.com/roemer/gover" |
| 14 | +) |
| 15 | + |
| 16 | +////////// |
| 17 | +// Configuration |
| 18 | +////////// |
| 19 | + |
| 20 | +const versionsIndexUrl = "https://go.dev/dl/?mode=json&include=all" |
| 21 | +const latestVersionUrl = "https://go.dev/VERSION?m=text" |
| 22 | + |
| 23 | +var versionRegexp *regexp.Regexp = regexp.MustCompile(`(?m:)^go(?P<d1>\d+)(?:\.(?P<d2>\d+))?(?:\.(?P<d3>\d+))?(?:(?P<s4>[a-z]+)(?P<d5>\d+)?)?$`) |
| 24 | + |
| 25 | +////////// |
| 26 | +// Main |
| 27 | +////////// |
| 28 | + |
| 29 | +func main() { |
| 30 | + if err := runMain(); err != nil { |
| 31 | + fmt.Printf("Error: %v\n", err) |
| 32 | + os.Exit(1) |
| 33 | + } |
| 34 | +} |
| 35 | + |
| 36 | +func runMain() error { |
| 37 | + // Handle the flags |
| 38 | + version := flag.String("version", "latest", "The version of Go to install.") |
| 39 | + downloadRegistryBase := flag.String("downloadRegistryBase", "", "The download registry to use for Go binaries.") |
| 40 | + downloadRegistryPath := flag.String("downloadRegistryPath", "", "The download registry path to use for Go binaries.") |
| 41 | + flag.Parse() |
| 42 | + |
| 43 | + fill(downloadRegistryBase, "https://dl.google.com", "") |
| 44 | + fill(downloadRegistryPath, "/go", "") |
| 45 | + |
| 46 | + // Create and process the feature |
| 47 | + feature := installer.NewFeature("Go", true, |
| 48 | + &goComponent{ |
| 49 | + ComponentBase: installer.NewComponentBase("Go", *version), |
| 50 | + DownloadRegistryBase: *downloadRegistryBase, |
| 51 | + DownloadRegistryPath: *downloadRegistryPath, |
| 52 | + }) |
| 53 | + return feature.Process() |
| 54 | +} |
| 55 | + |
| 56 | +func fill(passedValue *string, defaultValue string, key string) { |
| 57 | + if *passedValue == "" { |
| 58 | + // TODO: Get value from env?file? |
| 59 | + // Otherwise set to default value |
| 60 | + *passedValue = defaultValue |
| 61 | + } |
| 62 | +} |
| 63 | + |
| 64 | +func buildUrl(base string, parts ...string) (string, error) { |
| 65 | + return url.JoinPath(base, parts...) |
| 66 | +} |
| 67 | + |
| 68 | +////////// |
| 69 | +// Implementation |
| 70 | +////////// |
| 71 | + |
| 72 | +type goComponent struct { |
| 73 | + *installer.ComponentBase |
| 74 | + DownloadRegistryBase string |
| 75 | + DownloadRegistryPath string |
| 76 | +} |
| 77 | + |
| 78 | +func (c *goComponent) GetLatestVersion() (*gover.Version, error) { |
| 79 | + versionFileContent, err := installer.Tools.Download.AsString(latestVersionUrl) |
| 80 | + if err != nil { |
| 81 | + return nil, err |
| 82 | + } |
| 83 | + // Only use the first line |
| 84 | + lines := strings.Split(strings.ReplaceAll(versionFileContent, "\r\n", "\n"), "\n") |
| 85 | + if len(lines) == 0 { |
| 86 | + return nil, fmt.Errorf("no version found in go latest") |
| 87 | + } |
| 88 | + version, err := gover.ParseVersionFromRegex(lines[0], versionRegexp) |
| 89 | + if err != nil { |
| 90 | + return nil, err |
| 91 | + } |
| 92 | + return version, err |
| 93 | +} |
| 94 | + |
| 95 | +func (c *goComponent) GetAllVersions() ([]*gover.Version, error) { |
| 96 | + versionFileContent, err := installer.Tools.Download.AsBytes(versionsIndexUrl) |
| 97 | + if err != nil { |
| 98 | + return nil, err |
| 99 | + } |
| 100 | + var jsonData []map[string]interface{} |
| 101 | + if err := json.Unmarshal(versionFileContent, &jsonData); err != nil { |
| 102 | + return nil, err |
| 103 | + } |
| 104 | + versions := []*gover.Version{} |
| 105 | + for _, entry := range jsonData { |
| 106 | + versionString := entry["version"].(string) |
| 107 | + version, err := gover.ParseVersionFromRegex(versionString, versionRegexp) |
| 108 | + if err != nil { |
| 109 | + return nil, err |
| 110 | + } |
| 111 | + versions = append(versions, version) |
| 112 | + } |
| 113 | + |
| 114 | + return versions, nil |
| 115 | +} |
| 116 | + |
| 117 | +func (c *goComponent) InstallVersion(version *gover.Version) error { |
| 118 | + // Download the file |
| 119 | + fileName := fmt.Sprintf("%s.linux-amd64.tar.gz", version.Raw) |
| 120 | + downloadUrl, err := buildUrl(c.DownloadRegistryBase, c.DownloadRegistryPath, fileName) |
| 121 | + if err != nil { |
| 122 | + return err |
| 123 | + } |
| 124 | + if err := installer.Tools.Download.ToFile(downloadUrl, fileName, "Go"); err != nil { |
| 125 | + return err |
| 126 | + } |
| 127 | + // Extract it |
| 128 | + if err := installer.Tools.Compression.ExtractTarGz(fileName, "/usr/local", false); err != nil { |
| 129 | + return err |
| 130 | + } |
| 131 | + // Cleanup |
| 132 | + if err := os.Remove(fileName); err != nil { |
| 133 | + return err |
| 134 | + } |
| 135 | + return nil |
| 136 | +} |
0 commit comments