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.
142 lines
3.5 KiB
142 lines
3.5 KiB
/*
|
|
Copyright The Helm Authors.
|
|
|
|
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 main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gofrs/flock"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
"sigs.k8s.io/yaml"
|
|
|
|
"helm.sh/helm/v3/cmd/helm/require"
|
|
"helm.sh/helm/v3/pkg/getter"
|
|
"helm.sh/helm/v3/pkg/repo"
|
|
)
|
|
|
|
type repoAddOptions struct {
|
|
name string
|
|
url string
|
|
username string
|
|
password string
|
|
noUpdate bool
|
|
|
|
certFile string
|
|
keyFile string
|
|
caFile string
|
|
|
|
repoFile string
|
|
repoCache string
|
|
}
|
|
|
|
func newRepoAddCmd(out io.Writer) *cobra.Command {
|
|
o := &repoAddOptions{}
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "add [NAME] [URL]",
|
|
Short: "add a chart repository",
|
|
Args: require.ExactArgs(2),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
o.name = args[0]
|
|
o.url = args[1]
|
|
o.repoFile = settings.RepositoryConfig
|
|
o.repoCache = settings.RepositoryCache
|
|
|
|
return o.run(out)
|
|
},
|
|
}
|
|
|
|
f := cmd.Flags()
|
|
f.StringVar(&o.username, "username", "", "chart repository username")
|
|
f.StringVar(&o.password, "password", "", "chart repository password")
|
|
f.BoolVar(&o.noUpdate, "no-update", false, "raise error if repo is already registered")
|
|
f.StringVar(&o.certFile, "cert-file", "", "identify HTTPS client using this SSL certificate file")
|
|
f.StringVar(&o.keyFile, "key-file", "", "identify HTTPS client using this SSL key file")
|
|
f.StringVar(&o.caFile, "ca-file", "", "verify certificates of HTTPS-enabled servers using this CA bundle")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func (o *repoAddOptions) run(out io.Writer) error {
|
|
//Ensure the file directory exists as it is required for file locking
|
|
err := os.MkdirAll(filepath.Dir(o.repoFile), os.ModePerm)
|
|
if err != nil && !os.IsExist(err) {
|
|
return err
|
|
}
|
|
|
|
// Acquire a file lock for process synchronization
|
|
fileLock := flock.New(strings.Replace(o.repoFile, filepath.Ext(o.repoFile), ".lock", 1))
|
|
lockCtx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
defer cancel()
|
|
locked, err := fileLock.TryLockContext(lockCtx, time.Second)
|
|
if err == nil && locked {
|
|
defer fileLock.Unlock()
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
b, err := ioutil.ReadFile(o.repoFile)
|
|
if err != nil && !os.IsNotExist(err) {
|
|
return err
|
|
}
|
|
|
|
var f repo.File
|
|
if err := yaml.Unmarshal(b, &f); err != nil {
|
|
return err
|
|
}
|
|
|
|
if o.noUpdate && f.Has(o.name) {
|
|
return errors.Errorf("repository name (%s) already exists, please specify a different name", o.name)
|
|
}
|
|
|
|
c := repo.Entry{
|
|
Name: o.name,
|
|
URL: o.url,
|
|
Username: o.username,
|
|
Password: o.password,
|
|
CertFile: o.certFile,
|
|
KeyFile: o.keyFile,
|
|
CAFile: o.caFile,
|
|
}
|
|
|
|
r, err := repo.NewChartRepository(&c, getter.All(settings))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := r.DownloadIndexFile(); err != nil {
|
|
return errors.Wrapf(err, "looks like %q is not a valid chart repository or cannot be reached", o.url)
|
|
}
|
|
|
|
f.Update(&c)
|
|
|
|
if err := f.WriteFile(o.repoFile, 0644); err != nil {
|
|
return err
|
|
}
|
|
fmt.Fprintf(out, "%q has been added to your repositories\n", o.name)
|
|
return nil
|
|
}
|