// loadPlugins loads plugins into the command list. // // This follows a different pattern than the other commands because it has // to inspect its environment and then add commands to the base command // as it finds them. func loadPlugins(baseCmd *cobra.Command, home helmpath.Home, out io.Writer) { plugdirs := os.Getenv(pluginEnvVar) if plugdirs == "" { plugdirs = home.Plugins() } found, err := findPlugins(plugdirs) if err != nil { fmt.Fprintf(os.Stderr, "failed to load plugins: %s", err) return } // Now we create commands for all of these. for _, plug := range found { plug := plug md := plug.Metadata if md.Usage == "" { md.Usage = fmt.Sprintf("the %q plugin", md.Name) } c := &cobra.Command{ Use: md.Name, Short: md.Usage, Long: md.Description, RunE: func(cmd *cobra.Command, args []string) error { k, u := manuallyProcessArgs(args) if err := cmd.Parent().ParseFlags(k); err != nil { return err } // Call setupEnv before PrepareCommand because // PrepareCommand uses os.ExpandEnv and expects the // setupEnv vars. setupEnv(md.Name, plug.Dir, plugdirs, home) main, argv := plug.PrepareCommand(u) prog := exec.Command(main, argv...) prog.Env = os.Environ() prog.Stdout = out prog.Stderr = os.Stderr if err := prog.Run(); err != nil { if eerr, ok := err.(*exec.ExitError); ok { os.Stderr.Write(eerr.Stderr) return fmt.Errorf("plugin %q exited with error", md.Name) } return err } return nil }, // This passes all the flags to the subcommand. DisableFlagParsing: true, } if md.UseTunnel { c.PersistentPreRunE = func(cmd *cobra.Command, args []string) error { // Parse the parent flag, but not the local flags. k, _ := manuallyProcessArgs(args) if err := c.Parent().ParseFlags(k); err != nil { return err } return setupConnection(cmd, args) } } // TODO: Make sure a command with this name does not already exist. baseCmd.AddCommand(c) } }
// ensureHome checks to see if $HELM_HOME exists // // If $HELM_HOME does not exist, this function will create it. func ensureHome(home helmpath.Home, out io.Writer) error { configDirectories := []string{home.String(), home.Repository(), home.Cache(), home.LocalRepository(), home.Plugins(), home.Starters()} for _, p := range configDirectories { if fi, err := os.Stat(p); err != nil { fmt.Fprintf(out, "Creating %s \n", p) if err := os.MkdirAll(p, 0755); err != nil { return fmt.Errorf("Could not create %s: %s", p, err) } } else if !fi.IsDir() { return fmt.Errorf("%s must be a directory", p) } } repoFile := home.RepositoryFile() if fi, err := os.Stat(repoFile); err != nil { fmt.Fprintf(out, "Creating %s \n", repoFile) r := repo.NewRepoFile() r.Add(&repo.Entry{ Name: stableRepository, URL: stableRepositoryURL, Cache: "stable-index.yaml", }, &repo.Entry{ Name: localRepository, URL: localRepositoryURL, Cache: "local-index.yaml", }) if err := r.WriteFile(repoFile, 0644); err != nil { return err } cif := home.CacheIndex(stableRepository) if err := repo.DownloadIndexFile(stableRepository, stableRepositoryURL, cif); err != nil { fmt.Fprintf(out, "WARNING: Failed to download %s: %s (run 'helm repo update')\n", stableRepository, err) } } else if fi.IsDir() { return fmt.Errorf("%s must be a file, not a directory", repoFile) } if r, err := repo.LoadRepositoriesFile(repoFile); err == repo.ErrRepoOutOfDate { fmt.Fprintln(out, "Updating repository file format...") if err := r.WriteFile(repoFile, 0644); err != nil { return err } } localRepoIndexFile := home.LocalRepository(localRepoIndexFilePath) if fi, err := os.Stat(localRepoIndexFile); err != nil { fmt.Fprintf(out, "Creating %s \n", localRepoIndexFile) i := repo.NewIndexFile() if err := i.WriteFile(localRepoIndexFile, 0644); err != nil { return err } //TODO: take this out and replace with helm update functionality os.Symlink(localRepoIndexFile, home.CacheIndex("local")) } else if fi.IsDir() { return fmt.Errorf("%s must be a file, not a directory", localRepoIndexFile) } fmt.Fprintf(out, "$HELM_HOME has been configured at %s.\n", helmHome) return nil }
// ensureTestHome creates a home directory like ensureHome, but without remote references. // // t is used only for logging. func ensureTestHome(home helmpath.Home, t *testing.T) error { configDirectories := []string{home.String(), home.Repository(), home.Cache(), home.LocalRepository(), home.Plugins(), home.Starters()} for _, p := range configDirectories { if fi, err := os.Stat(p); err != nil { if err := os.MkdirAll(p, 0755); err != nil { return fmt.Errorf("Could not create %s: %s", p, err) } } else if !fi.IsDir() { return fmt.Errorf("%s must be a directory", p) } } repoFile := home.RepositoryFile() if fi, err := os.Stat(repoFile); err != nil { rf := repo.NewRepoFile() rf.Add(&repo.Entry{ Name: "charts", URL: "http://example.com/foo", Cache: "charts-index.yaml", }, &repo.Entry{ Name: "local", URL: "http://localhost.com:7743/foo", Cache: "local-index.yaml", }) if err := rf.WriteFile(repoFile, 0644); err != nil { return err } } else if fi.IsDir() { return fmt.Errorf("%s must be a file, not a directory", repoFile) } if r, err := repo.LoadRepositoriesFile(repoFile); err == repo.ErrRepoOutOfDate { t.Log("Updating repository file format...") if err := r.WriteFile(repoFile, 0644); err != nil { return err } } localRepoIndexFile := home.LocalRepository(localRepoIndexFilePath) if fi, err := os.Stat(localRepoIndexFile); err != nil { i := repo.NewIndexFile() if err := i.WriteFile(localRepoIndexFile, 0644); err != nil { return err } //TODO: take this out and replace with helm update functionality os.Symlink(localRepoIndexFile, home.CacheIndex("local")) } else if fi.IsDir() { return fmt.Errorf("%s must be a file, not a directory", localRepoIndexFile) } t.Logf("$HELM_HOME has been configured at %s.\n", helmHome) return nil }