feat: support replacements

pkg/controller/config.go

@@ -99,42 +99,54 @@ type PackageInfo interface {
 	RenderAsset(pkg *Package) (string, error)
 	GetLink() string
 	GetDescription() string
+	GetReplacements() map[string]string
 }
 
 type mergedPackageInfo struct {
-	Name        string
-	Type        string
-	RepoOwner   string `yaml:"repo_owner"`
-	RepoName    string `yaml:"repo_name"`
-	Asset       *text.Template
+	Name                 string
+	Type                 string
+	RepoOwner            string `yaml:"repo_owner"`
+	RepoName             string `yaml:"repo_name"`
+	Asset                *text.Template
+	ArchiveType          string `yaml:"archive_type"`
+	Files                []*File
+	URL                  *text.Template
+	Description          string
+	Link                 string
+	Replacements         map[string]string
+	ArchiveTypeOverrides []*ArchiveTypeOverride `yaml:"archive_type_overrides"`
+}
+
+type ArchiveTypeOverride struct {
+	GOOS        string
 	ArchiveType string `yaml:"archive_type"`
-	Files       []*File
-	URL         *text.Template
-	Description string
-	Link        string
 }
 
 func (pkgInfo *mergedPackageInfo) GetPackageInfo() (PackageInfo, error) {
 	switch pkgInfo.Type {
 	case pkgInfoTypeGitHubRelease:
 		return &GitHubReleasePackageInfo{
-			Name:        pkgInfo.Name,
-			RepoOwner:   pkgInfo.RepoOwner,
-			RepoName:    pkgInfo.RepoName,
-			Asset:       pkgInfo.Asset,
-			ArchiveType: pkgInfo.ArchiveType,
-			Files:       pkgInfo.Files,
-			Link:        pkgInfo.Link,
-			Description: pkgInfo.Description,
+			Name:                 pkgInfo.Name,
+			RepoOwner:            pkgInfo.RepoOwner,
+			RepoName:             pkgInfo.RepoName,
+			Asset:                pkgInfo.Asset,
+			ArchiveType:          pkgInfo.ArchiveType,
+			ArchiveTypeOverrides: pkgInfo.ArchiveTypeOverrides,
+			Files:                pkgInfo.Files,
+			Link:                 pkgInfo.Link,
+			Description:          pkgInfo.Description,
+			Replacements:         pkgInfo.Replacements,
 		}, nil
 	case pkgInfoTypeHTTP:
 		return &HTTPPackageInfo{
-			Name:        pkgInfo.Name,
-			ArchiveType: pkgInfo.ArchiveType,
-			URL:         pkgInfo.URL,
-			Files:       pkgInfo.Files,
-			Link:        pkgInfo.Link,
-			Description: pkgInfo.Description,
+			Name:                 pkgInfo.Name,
+			ArchiveType:          pkgInfo.ArchiveType,
+			ArchiveTypeOverrides: pkgInfo.ArchiveTypeOverrides,
+			URL:                  pkgInfo.URL,
+			Files:                pkgInfo.Files,
+			Link:                 pkgInfo.Link,
+			Description:          pkgInfo.Description,
+			Replacements:         pkgInfo.Replacements,
 		}, nil
 	default:
 		return nil, logerr.WithFields(errInvalidType, logrus.Fields{ //nolint:wrapcheck
@@ -156,6 +168,10 @@ func (file *File) RenderSrc(pkg *Package, pkgInfo PackageInfo) (string, error) {
 		"OS":          runtime.GOOS,
 		"Arch":        runtime.GOARCH,
 		"File":        file,
+		"Replacements": map[string]interface{}{
+			"OS":   replace(runtime.GOOS, pkgInfo.GetReplacements()),
+			"Arch": replace(runtime.GOARCH, pkgInfo.GetReplacements()),
+		},
 	})
 }
 

pkg/controller/github_release_package.go

@@ -9,14 +9,16 @@ import (
 )
 
 type GitHubReleasePackageInfo struct {
-	Name        string `validate:"required"`
-	ArchiveType string `yaml:"archive_type"`
-	Link        string
-	Description string
-	Files       []*File `validate:"required,dive"`
+	Name                 string `validate:"required"`
+	ArchiveType          string
+	Link                 string
+	Description          string
+	Files                []*File `validate:"required,dive"`
+	Replacements         map[string]string
+	ArchiveTypeOverrides []*ArchiveTypeOverride
 
-	RepoOwner string         `yaml:"repo_owner" validate:"required"`
-	RepoName  string         `yaml:"repo_name" validate:"required"`
+	RepoOwner string
+	RepoName  string
 	Asset     *text.Template `validate:"required"`
 }
 
@@ -37,9 +39,18 @@ func (pkgInfo *GitHubReleasePackageInfo) GetDescription() string {
 }
 
 func (pkgInfo *GitHubReleasePackageInfo) GetArchiveType() string {
+	for _, arcTypeOverride := range pkgInfo.ArchiveTypeOverrides {
+		if arcTypeOverride.GOOS == runtime.GOOS {
+			return arcTypeOverride.ArchiveType
+		}
+	}
 	return pkgInfo.ArchiveType
 }
 
+func (pkgInfo *GitHubReleasePackageInfo) GetReplacements() map[string]string {
+	return pkgInfo.Replacements
+}
+
 func (pkgInfo *GitHubReleasePackageInfo) GetPkgPath(rootDir string, pkg *Package) (string, error) {
 	assetName, err := pkgInfo.RenderAsset(pkg)
 	if err != nil {
@@ -76,5 +87,9 @@ func (pkgInfo *GitHubReleasePackageInfo) RenderAsset(pkg *Package) (string, erro
 		"PackageInfo": pkgInfo,
 		"OS":          runtime.GOOS,
 		"Arch":        runtime.GOARCH,
+		"Replacements": map[string]interface{}{
+			"OS":   replace(runtime.GOOS, pkgInfo.Replacements),
+			"Arch": replace(runtime.GOARCH, pkgInfo.Replacements),
+		},
 	})
 }

pkg/controller/http_package.go

@@ -10,11 +10,13 @@ import (
 )
 
 type HTTPPackageInfo struct {
-	Name        string `validate:"required"`
-	ArchiveType string `yaml:"archive_type"`
-	Description string
-	Link        string
-	Files       []*File `validate:"required,dive"`
+	Name                 string `validate:"required"`
+	ArchiveType          string
+	Description          string
+	Link                 string
+	Files                []*File `validate:"required,dive"`
+	Replacements         map[string]string
+	ArchiveTypeOverrides []*ArchiveTypeOverride
 
 	URL *text.Template `validate:"required"`
 }
@@ -36,9 +38,18 @@ func (pkgInfo *HTTPPackageInfo) GetDescription() string {
 }
 
 func (pkgInfo *HTTPPackageInfo) GetArchiveType() string {
+	for _, arcTypeOverride := range pkgInfo.ArchiveTypeOverrides {
+		if arcTypeOverride.GOOS == runtime.GOOS {
+			return arcTypeOverride.ArchiveType
+		}
+	}
 	return pkgInfo.ArchiveType
 }
 
+func (pkgInfo *HTTPPackageInfo) GetReplacements() map[string]string {
+	return pkgInfo.Replacements
+}
+
 func (pkgInfo *HTTPPackageInfo) GetPkgPath(rootDir string, pkg *Package) (string, error) {
 	uS, err := pkgInfo.URL.Execute(map[string]interface{}{
 		"Package":     pkg,
@@ -84,6 +95,10 @@ func (pkgInfo *HTTPPackageInfo) RenderURL(pkg *Package) (string, error) {
 		"PackageInfo": pkgInfo,
 		"OS":          runtime.GOOS,
 		"Arch":        runtime.GOARCH,
+		"Replacements": map[string]interface{}{
+			"OS":   replace(runtime.GOOS, pkgInfo.Replacements),
+			"Arch": replace(runtime.GOARCH, pkgInfo.Replacements),
+		},
 	})
 	if err != nil {
 		return "", fmt.Errorf("render URL: %w", err)

pkg/controller/replacements.go

@@ -0,0 +1,9 @@
+package controller
+
+func replace(key string, replacements map[string]string) string {
+	a := replacements[key]
+	if a == "" {
+		return key
+	}
+	return a
+}