forked from rook/rook
Divesh Angula
9e08afb015
- refactoring basic tests so they can be reused with different rook installations
110 lines
3.6 KiB
Go
110 lines
3.6 KiB
Go
/*
|
|
Copyright 2016 The Rook Authors. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package utils
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/rook/rook/pkg/util/exec"
|
|
"github.com/rook/rook/pkg/util/sys"
|
|
"time"
|
|
)
|
|
|
|
//HelmHelper is wrapper for running helm commands
|
|
type HelmHelper struct {
|
|
executor *exec.CommandExecutor
|
|
}
|
|
|
|
//NewHelmHelper creates a instance of HelmHelper
|
|
func NewHelmHelper() *HelmHelper {
|
|
executor := &exec.CommandExecutor{}
|
|
return &HelmHelper{executor: executor}
|
|
|
|
}
|
|
|
|
//Execute is wrapper for executing helm commands
|
|
func (h *HelmHelper) Execute(args ...string) (string, error) {
|
|
result, err := h.executor.ExecuteCommandWithOutput(false, "", "helm", args...)
|
|
if err != nil {
|
|
logger.Errorf("Errors Encountered while executing helm command %v: %v", result, err)
|
|
return result, fmt.Errorf("Failed to run helm commands on args %v : %v , err -> %v", args, result, err)
|
|
|
|
}
|
|
return result, nil
|
|
|
|
}
|
|
|
|
//GetLocalRookHelmChartVersion returns helm chart version for a give chart
|
|
func (h *HelmHelper) GetLocalRookHelmChartVersion(chartName string) (string, error) {
|
|
cmdArgs := []string{"search", chartName}
|
|
result, err := h.Execute(cmdArgs...)
|
|
if err != nil {
|
|
logger.Errorf("cannot find helm chart %v : %v", chartName, err)
|
|
return "", fmt.Errorf("Failed to find helm chart %v : %v", chartName, err)
|
|
}
|
|
|
|
if strings.Contains(result, "No results found") {
|
|
return "", fmt.Errorf("Failed to find helm chart %v ", chartName)
|
|
}
|
|
cd := strings.Replace(sys.Grep(result, chartName), "\t", " ", 2)
|
|
|
|
return sys.Awk(cd, 2), nil
|
|
}
|
|
|
|
//InstallLocalRookHelmChart installs a give helm chart
|
|
func (h *HelmHelper) InstallLocalRookHelmChart(chartName string, deployName string, chartVersion string, namespace string) error {
|
|
cmdArgs := []string{"install", chartName, "--name", deployName, "--version", chartVersion}
|
|
if namespace != "" {
|
|
cmdArgs = append(cmdArgs, "--namespace", namespace)
|
|
}
|
|
var result string
|
|
var err error
|
|
inc := 0
|
|
for inc < RetryLoop {
|
|
result, err = h.Execute(cmdArgs...)
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
logger.Infof("helm install for %s failed %v, err ->%v", chartName, result, err)
|
|
ls, _ := h.Execute([]string{"ls"}...)
|
|
logger.Infof("Helm ls result : %v", ls)
|
|
ss, _ := h.Execute([]string{"search"}...)
|
|
logger.Infof("Helm search result : %v", ss)
|
|
rl, _ := h.Execute([]string{"repo", "list"}...)
|
|
logger.Infof("Helm repo list result : %v", rl)
|
|
|
|
inc++
|
|
time.Sleep(RetryInterval * time.Second)
|
|
}
|
|
|
|
logger.Errorf("cannot install helm chart with name : %v, version: %v, namespace: %v - %v , err: %v", chartName, chartVersion, namespace, result, err)
|
|
return fmt.Errorf("cannot install helm chart with name : %v, version: %v, namespace: %v - %v, err: %v", chartName, chartVersion, namespace, result, err)
|
|
}
|
|
|
|
//DeleteLocalRookHelmChart uninstalls a give helm deploy
|
|
func (h *HelmHelper) DeleteLocalRookHelmChart(deployName string) error {
|
|
cmdArgs := []string{"delete", "--purge", deployName}
|
|
_, err := h.Execute(cmdArgs...)
|
|
if err != nil {
|
|
logger.Errorf("cannot delete helm chart with name %v : %v", deployName, err)
|
|
return fmt.Errorf("Failed to delete helm chart with name %v : %v", deployName, err)
|
|
}
|
|
|
|
return nil
|
|
}
|