downloading icons when running make charts

main.go

@@ -290,9 +290,19 @@ func generateCharts(c *cli.Context) {
 		logrus.Infof("No packages found.")
 		return
 	}
+	var releaseOptions options.ReleaseOptions
+	if c.Command.Name == "charts" {
+		repoRoot := getRepoRoot()
+		repoFs := filesystem.GetFilesystem(repoRoot)
+		var err error
+		releaseOptions, err = options.LoadReleaseOptionsFromFile(repoFs, "release.yaml")
+		if err != nil {
+			logrus.Fatalf("Unable to unmarshall release.yaml: %s", err)
+		}
+	}
 	chartsScriptOptions := parseScriptOptions()
 	for _, p := range packages {
-		if err := p.GenerateCharts(chartsScriptOptions.OmitBuildMetadataOnExport); err != nil {
+		if err := p.GenerateCharts(chartsScriptOptions.OmitBuildMetadataOnExport, releaseOptions); err != nil {
 			logrus.Fatal(err)
 		}
 	}

pkg/charts/additionalchart.go

@@ -214,11 +214,11 @@ func (c *AdditionalChart) GeneratePatch(rootFs, pkgFs billy.Filesystem) error {
 }
 
 // GenerateChart generates the chart and stores it in the assets and charts directory
-func (c *AdditionalChart) GenerateChart(rootFs, pkgFs billy.Filesystem, packageVersion *int, version *semver.Version, omitBuildMetadataOnExport bool) error {
+func (c *AdditionalChart) GenerateChart(rootFs, pkgFs billy.Filesystem, packageVersion *int, version *semver.Version, omitBuildMetadataOnExport bool, releaseOptions options.ReleaseOptions) error {
 	if c.upstreamChartVersion == nil {
 		return fmt.Errorf("cannot generate chart since it has never been prepared: upstreamChartVersion is not set")
 	}
-	if err := helm.ExportHelmChart(rootFs, pkgFs, c.WorkingDir, packageVersion, version, *c.upstreamChartVersion, omitBuildMetadataOnExport); err != nil {
+	if err := helm.ExportHelmChart(rootFs, pkgFs, c.WorkingDir, packageVersion, version, *c.upstreamChartVersion, omitBuildMetadataOnExport, releaseOptions); err != nil {
 		return fmt.Errorf("encountered error while trying to export Helm chart for %s: %s", c.WorkingDir, err)
 	}
 	return nil

pkg/charts/chart.go

@@ -2,6 +2,7 @@ package charts
 
 import (
 	"fmt"
+	"github.com/rancher/charts-build-scripts/pkg/options"
 
 	"github.com/blang/semver"
 	"github.com/go-git/go-billy/v5"
@@ -103,11 +104,11 @@ func (c *Chart) GeneratePatch(rootFs, pkgFs billy.Filesystem) error {
 }
 
 // GenerateChart generates the chart and stores it in the assets and charts directory
-func (c *Chart) GenerateChart(rootFs, pkgFs billy.Filesystem, packageVersion *int, version *semver.Version, omitBuildMetadataOnExport bool) error {
+func (c *Chart) GenerateChart(rootFs, pkgFs billy.Filesystem, packageVersion *int, version *semver.Version, omitBuildMetadataOnExport bool, releaseOptions options.ReleaseOptions) error {
 	if c.upstreamChartVersion == nil {
 		return fmt.Errorf("cannot generate chart since it has never been prepared: upstreamChartVersion is not set")
 	}
-	if err := helm.ExportHelmChart(rootFs, pkgFs, c.WorkingDir, packageVersion, version, *c.upstreamChartVersion, omitBuildMetadataOnExport); err != nil {
+	if err := helm.ExportHelmChart(rootFs, pkgFs, c.WorkingDir, packageVersion, version, *c.upstreamChartVersion, omitBuildMetadataOnExport, releaseOptions); err != nil {
 		return fmt.Errorf("encountered error while trying to export Helm chart for %s: %s", c.WorkingDir, err)
 	}
 	return nil

pkg/charts/package.go

@@ -2,6 +2,7 @@ package charts
 
 import (
 	"fmt"
+	"github.com/rancher/charts-build-scripts/pkg/options"
 	"path/filepath"
 
 	"github.com/blang/semver"
@@ -86,7 +87,7 @@ func (p *Package) GeneratePatch() error {
 }
 
 // GenerateCharts creates Helm chart archives for each chart after preparing it
-func (p *Package) GenerateCharts(omitBuildMetadataOnExport bool) error {
+func (p *Package) GenerateCharts(omitBuildMetadataOnExport bool, releaseOptions options.ReleaseOptions) error {
 	if p.DoNotRelease {
 		logrus.Infof("Skipping package marked doNotRelease")
 		return nil
@@ -95,12 +96,12 @@ func (p *Package) GenerateCharts(omitBuildMetadataOnExport bool) error {
 		return fmt.Errorf("encountered error while trying to prepare package: %s", err)
 	}
 	// Add PackageVersion to format
-	err := p.Chart.GenerateChart(p.rootFs, p.fs, p.PackageVersion, p.Version, omitBuildMetadataOnExport)
+	err := p.Chart.GenerateChart(p.rootFs, p.fs, p.PackageVersion, p.Version, omitBuildMetadataOnExport, releaseOptions)
 	if err != nil {
 		return fmt.Errorf("encountered error while exporting main chart: %s", err)
 	}
 	for _, additionalChart := range p.AdditionalCharts {
-		err = additionalChart.GenerateChart(p.rootFs, p.fs, p.PackageVersion, p.Version, omitBuildMetadataOnExport)
+		err = additionalChart.GenerateChart(p.rootFs, p.fs, p.PackageVersion, p.Version, omitBuildMetadataOnExport, releaseOptions)
 		if err != nil {
 			return fmt.Errorf("encountered error while exporting %s: %s", additionalChart.WorkingDir, err)
 		}

pkg/helm/export.go

@@ -2,7 +2,11 @@ package helm
 
 import (
 	"fmt"
+	"github.com/rancher/charts-build-scripts/pkg/icons"
+	"github.com/rancher/charts-build-scripts/pkg/options"
+	"helm.sh/helm/v3/pkg/chartutil"
 	"math"
+	"net/url"
 	"os"
 	"path/filepath"
 
@@ -26,7 +30,7 @@ var (
 
 // ExportHelmChart creates a Helm chart archive and an unarchived Helm chart at RepositoryAssetDirpath and RepositoryChartDirPath
 // helmChartPath is a relative path (rooted at the package level) that contains the chart.
-func ExportHelmChart(rootFs, fs billy.Filesystem, helmChartPath string, packageVersion *int, version *semver.Version, upstreamChartVersion string, omitBuildMetadata bool) error {
+func ExportHelmChart(rootFs, fs billy.Filesystem, helmChartPath string, packageVersion *int, version *semver.Version, upstreamChartVersion string, omitBuildMetadata bool, releaseOptions options.ReleaseOptions) error {
 	// Try to load the chart to see if it can be exported
 	absHelmChartPath := filesystem.GetAbsPath(fs, helmChartPath)
 	chart, err := helmLoader.Load(absHelmChartPath)
@@ -59,6 +63,27 @@ func ExportHelmChart(rootFs, fs billy.Filesystem, helmChartPath string, packageV
 	}
 	chartVersion := chartVersionSemver.String()
 
+	if releaseOptions != nil && releaseOptions.Contains(chart.Metadata.Name, chartVersion) {
+		//checking if icon is pointing to a valid http/https url
+		u, err := url.Parse(chart.Metadata.Icon)
+		if err == nil && (u.Scheme == "http" || u.Scheme == "https") {
+			logrus.Infof("Chart icon is pointing to a remote url. Downloading it...")
+			// download icon and change the icon property to point to it
+			p, err := icons.Download(rootFs, chart.Metadata)
+			if err == nil { // managed to download the icon and save it locally
+				chart.Metadata.Icon = fmt.Sprintf("file://%s", p)
+			} else {
+				logrus.Errorf("failed to download icon for chart %s, err: %s", chart.Name(), err)
+			}
+		}
+	}
+
+	chartYamlPath := fmt.Sprintf("%s/Chart.yaml", absHelmChartPath)
+	err = chartutil.SaveChartfile(chartYamlPath, chart.Metadata)
+	if err != nil {
+		return err
+	}
+
 	// Assets are indexed by chart name, independent of which package that chart is contained within
 	chartAssetsDirpath := filepath.Join(path.RepositoryAssetsDir, chart.Metadata.Name)
 	// All generated charts are indexed by chart name and version

pkg/icons/icons.go

@@ -0,0 +1,45 @@
+package icons
+
+import (
+	"fmt"
+	"io"
+	"mime"
+	"net/http"
+
+	"github.com/rancher/charts-build-scripts/pkg/path"
+
+	"github.com/go-git/go-billy/v5"
+	"github.com/sirupsen/logrus"
+	"helm.sh/helm/v3/pkg/chart"
+)
+
+// Download receives a chart metadata and the filesystem pointing to the root of the project.
+// From the metadata, gets the icon and name of the chart.
+// It downloads the icon, infers the type using the content-type header from the response
+// and saves the file locally to path.RepositoryLogosDir using the name of the chart as the file name.
+func Download(rootFs billy.Filesystem, metadata *chart.Metadata) (string, error) {
+	icon, err := http.Get(metadata.Icon)
+	if err != nil {
+		logrus.Errorf(err.Error())
+		return "", fmt.Errorf("err: %w", err)
+	}
+
+	byType, err := mime.ExtensionsByType(icon.Header.Get("Content-Type"))
+	if err != nil || len(byType) == 0 || icon.StatusCode != http.StatusOK {
+		return "", fmt.Errorf("invalid icon")
+	}
+	path := fmt.Sprintf("%s/%s%s", path.RepositoryLogosDir, metadata.Name, byType[0])
+	create, err := rootFs.Create(path)
+	if err != nil {
+		logrus.Errorf(err.Error())
+		return "", fmt.Errorf("err: %w", err)
+	}
+	defer create.Close()
+	_, err = io.Copy(create, icon.Body)
+	defer icon.Body.Close()
+	if err != nil {
+		logrus.Errorf(err.Error())
+		return "", fmt.Errorf("err: %w", err)
+	}
+	return path, nil
+}

pkg/path/path.go

@@ -46,4 +46,7 @@ const (
 
 	// DefaultCachePath represents the default place to put a cache on pulled values
 	DefaultCachePath = ".charts-build-scripts/.cache"
+
+	// RepositoryLogosDir is a directory on your Staging/Live branch that contains the files with the logos of each chart
+	RepositoryLogosDir = "assets/logos"
 )