Rename APIRegister to PublishRegister

[nonrtric/plt/sme.git] / capifcore / internal / helmmanagement / helm.go

diff --git a/capifcore/internal/helmmanagement/helm.go b/capifcore/internal/helmmanagement/helm.go
index cfed6eb..99c27cc 100644 (file)
--- a/capifcore/internal/helmmanagement/helm.go
+++ b/capifcore/internal/helmmanagement/helm.go
@@ -22,7 +22,7 @@ package helmmanagement
 
 import (
        "fmt"
-       "io/ioutil"
+       "io/fs"
        "os"
        "path/filepath"
 
@@ -41,32 +41,33 @@ import (
 
 //go:generate mockery --name HelmManager
 type HelmManager interface {
-       AddToRepo(repoName, url string) error
+       SetUpRepo(repoName, url string) error
        InstallHelmChart(namespace, repoName, chartName, releaseName string) error
        UninstallHelmChart(namespace, chartName string)
 }
 
 type helmManagerImpl struct {
        settings *cli.EnvSettings
+       repo     *repo.ChartRepository
 }
 
-func NewHelmManager() *helmManagerImpl {
+func NewHelmManager(s *cli.EnvSettings) *helmManagerImpl {
        return &helmManagerImpl{
-               settings: cli.New(),
+               settings: s,
        }
 }
 
-func (hm *helmManagerImpl) AddToRepo(repoName, url string) error {
+func (hm *helmManagerImpl) SetUpRepo(repoName, url string) error {
        repoFile := hm.settings.RepositoryConfig
 
        //Ensure the file directory exists as it is required for file locking
        err := os.MkdirAll(filepath.Dir(repoFile), os.ModePerm)
-       if err != nil && !os.IsExist(err) {
+       if err != nil && !errors.Is(err, fs.ErrNotExist) {
                return err
        }
 
-       b, err := ioutil.ReadFile(repoFile)
-       if err != nil && !os.IsNotExist(err) {
+       b, err := os.ReadFile(repoFile)
+       if err != nil {
                return err
        }
 
@@ -76,7 +77,7 @@ func (hm *helmManagerImpl) AddToRepo(repoName, url string) error {
        }
 
        if f.Has(repoName) {
-               fmt.Printf("repository name (%s) already exists\n", repoName)
+               log.Debugf("repository name (%s) already exists\n", repoName)
                return nil
        }
 
@@ -85,9 +86,12 @@ func (hm *helmManagerImpl) AddToRepo(repoName, url string) error {
                URL:  url,
        }
 
-       r, err := repo.NewChartRepository(&c, getter.All(hm.settings))
-       if err != nil {
-               return err
+       r := hm.repo
+       if r == nil {
+               r, err = repo.NewChartRepository(&c, getter.All(hm.settings))
+               if err != nil {
+                       return err
+               }
        }
 
        if _, err := r.DownloadIndexFile(); err != nil {
@@ -100,7 +104,7 @@ func (hm *helmManagerImpl) AddToRepo(repoName, url string) error {
        if err := f.WriteFile(repoFile, 0644); err != nil {
                return err
        }
-       fmt.Printf("%q has been added to your repositories\n", repoName)
+       log.Debugf("%q has been added to your repositories\n", repoName)
        return nil
 }
 
@@ -138,14 +142,16 @@ func (hm *helmManagerImpl) InstallHelmChart(namespace, repoName, chartName, rele
 func (hm *helmManagerImpl) UninstallHelmChart(namespace, chartName string) {
        actionConfig, err := getActionConfig(namespace)
        if err != nil {
-               fmt.Println(err)
+               log.Error("unable to get action config: ", err)
+               return
        }
 
        iCli := action.NewUninstall(actionConfig)
 
        resp, err := iCli.Run(chartName)
        if err != nil {
-               fmt.Println(err)
+               log.Error("Unable to uninstall chart ", chartName, err)
+               return
        }
        log.Debug("Successfully uninstalled chart: ", resp.Release.Name)
 }
@@ -164,11 +170,8 @@ func getActionConfig(namespace string) (*action.Configuration, error) {
                if envvar := os.Getenv("KUBECONFIG"); len(envvar) > 0 {
                        kubeconfigPath = envvar
                }
-               if err := actionConfig.Init(kube.GetConfig(kubeconfigPath, "", namespace), namespace, os.Getenv("HELM_DRIVER"),
-                       func(format string, v ...interface{}) {
-                               fmt.Sprintf(format, v)
-                       }); err != nil {
-                       fmt.Println(err)
+               if err := actionConfig.Init(kube.GetConfig(kubeconfigPath, "", namespace), namespace, os.Getenv("HELM_DRIVER"), log.Debugf); err != nil {
+                       log.Error(err)
                }
        } else {
                // Create the ConfigFlags struct instance with initialized values from ServiceAccount
@@ -177,10 +180,8 @@ func getActionConfig(namespace string) (*action.Configuration, error) {
                kubeConfig.BearerToken = &config.BearerToken
                kubeConfig.CAFile = &config.CAFile
                kubeConfig.Namespace = &namespace
-               if err := actionConfig.Init(kubeConfig, namespace, os.Getenv("HELM_DRIVER"), func(format string, v ...interface{}) {
-                       fmt.Sprintf(format, v)
-               }); err != nil {
-                       fmt.Println(err)
+               if err := actionConfig.Init(kubeConfig, namespace, os.Getenv("HELM_DRIVER"), log.Debugf); err != nil {
+                       log.Error(err)
                }
        }
        return actionConfig, err