mirror of https://github.com/helm/helm
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
131 lines
3.3 KiB
131 lines
3.3 KiB
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
"google.golang.org/grpc"
|
|
|
|
"k8s.io/helm/pkg/helm"
|
|
)
|
|
|
|
const (
|
|
homeEnvVar = "HELM_HOME"
|
|
hostEnvVar = "HELM_HOST"
|
|
)
|
|
|
|
var (
|
|
helmHome string
|
|
tillerHost string
|
|
tillerNamespace string
|
|
)
|
|
|
|
// flagDebug is a signal that the user wants additional output.
|
|
var flagDebug bool
|
|
|
|
var globalUsage = `The Kubernetes package manager
|
|
|
|
To begin working with Helm, run the 'helm init' command:
|
|
|
|
$ helm init
|
|
|
|
This will install Tiller to your running Kubernetes cluster.
|
|
It will also set up any necessary local configuration.
|
|
|
|
Common actions from this point include:
|
|
|
|
- helm search: search for charts
|
|
- helm fetch: download a chart to your local directory to view
|
|
- helm install: upload the chart to Kubernetes
|
|
- helm list: list releases of charts
|
|
|
|
Environment:
|
|
$HELM_HOME Set an alternative location for Helm files. By default, these are stored in ~/.helm
|
|
$HELM_HOST Set an alternative Tiller host. The format is host:port (default ":44134").
|
|
`
|
|
|
|
// RootCommand is the top-level command for Helm.
|
|
var RootCommand = &cobra.Command{
|
|
Use: "helm",
|
|
Short: "The Helm package manager for Kubernetes.",
|
|
Long: globalUsage,
|
|
PersistentPostRun: teardown,
|
|
}
|
|
|
|
func init() {
|
|
home := os.Getenv(homeEnvVar)
|
|
if home == "" {
|
|
home = "$HOME/.helm"
|
|
}
|
|
thost := os.Getenv(hostEnvVar)
|
|
p := RootCommand.PersistentFlags()
|
|
p.StringVar(&helmHome, "home", home, "location of your Helm config. Overrides $HELM_HOME.")
|
|
p.StringVar(&tillerHost, "host", thost, "address of tiller. Overrides $HELM_HOST.")
|
|
p.StringVarP(&tillerNamespace, "namespace", "", "", "kubernetes namespace")
|
|
p.BoolVarP(&flagDebug, "debug", "", false, "enable verbose output")
|
|
}
|
|
|
|
func main() {
|
|
if err := RootCommand.Execute(); err != nil {
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func setupConnection(c *cobra.Command, args []string) error {
|
|
if tillerHost == "" {
|
|
tunnel, err := newTillerPortForwarder(tillerNamespace)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
tillerHost = fmt.Sprintf(":%d", tunnel.Local)
|
|
if flagDebug {
|
|
fmt.Printf("Created tunnel using local port: '%d'\n", tunnel.Local)
|
|
}
|
|
}
|
|
|
|
// Set up the gRPC config.
|
|
helm.Config.ServAddr = tillerHost
|
|
if flagDebug {
|
|
fmt.Printf("Server: %q\n", helm.Config.ServAddr)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func teardown(c *cobra.Command, args []string) {
|
|
if tunnel != nil {
|
|
tunnel.Close()
|
|
}
|
|
}
|
|
|
|
func checkArgsLength(expectedNum, actualNum int, requiredArgs ...string) error {
|
|
if actualNum != expectedNum {
|
|
arg := "arguments"
|
|
if expectedNum == 1 {
|
|
arg = "argument"
|
|
}
|
|
return fmt.Errorf("This command needs %v %s: %s", expectedNum, arg, strings.Join(requiredArgs, ", "))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// requireInit is a PreRunE implementation for validating that $HELM_HOME is configured.
|
|
func requireInit(cmd *cobra.Command, args []string) error {
|
|
err := requireHome()
|
|
if err != nil {
|
|
return fmt.Errorf("%s (try running 'helm init')", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// prettyError unwraps or rewrites certain errors to make them more user-friendly.
|
|
func prettyError(err error) error {
|
|
// This is ridiculous. Why is 'grpc.rpcError' not exported? The least they
|
|
// could do is throw an interface on the lib that would let us get back
|
|
// the desc. Instead, we have to pass ALL errors through this.
|
|
return errors.New(grpc.ErrorDesc(err))
|
|
}
|