Cloudpods/backend/cmd/climc/shell/k8s/repos.go

// Copyright 2019 Yunion
//
// 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 k8s

import (
	"io"
	"os"
	"path/filepath"

	"github.com/cheggaaa/pb/v3"

	"yunion.io/x/jsonutils"
	"yunion.io/x/pkg/errors"

	"yunion.io/x/onecloud/pkg/mcclient"
	"yunion.io/x/onecloud/pkg/mcclient/modules/k8s"
	o "yunion.io/x/onecloud/pkg/mcclient/options/k8s"
)

func initRepo() {
	cmd := NewK8sResourceCmd(k8s.Repos)
	cmd.List(new(o.RepoListOptions))
	cmd.Show(new(o.RepoGetOptions))
	cmd.Create(new(o.RepoCreateOptions))
	cmd.Update(new(o.RepoUpdateOptions))
	cmd.Delete(new(o.RepoGetOptions))
	cmd.Perform("sync", new(o.RepoGetOptions))
	cmd.Perform("public", new(o.RepoPublicOptions))
	cmd.Perform("private", new(o.RepoGetOptions))

	type UploadOptions struct {
		REPO string `help:"The name or id of helm repository`
		FILE string `help:"Helm packaged file"`
	}
	R(new(UploadOptions), "k8s-repo-upload-chart", "Upload chart to the repository", func(s *mcclient.ClientSession, args *UploadOptions) error {
		f, err := os.Open(args.FILE)
		if err != nil {
			return err
		}
		defer f.Close()
		finfo, err := f.Stat()
		if err != nil {
			return err
		}
		size := finfo.Size()
		bar := pb.Full.Start64(size)
		barReader := bar.NewProxyReader(f)
		param := jsonutils.Marshal(args)
		param.(*jsonutils.JSONDict).Add(jsonutils.NewString(filepath.Base(args.FILE)), "chart_name")
		ret, err := k8s.Repos.UploadChart(s, args.REPO, param, barReader, size)
		if err != nil {
			return err
		}
		printObject(ret)
		return nil
	})

	type DownloadOptions struct {
		REPO    string `help:"The name or id of helm repository`
		CHART   string `help:"The name of chart"`
		Output  string `help:"Saved file path"`
		Version string `help:"The version of chart"`
	}
	R(new(DownloadOptions), "k8s-repo-download-chart", "Download helm chart to a file", func(s *mcclient.ClientSession, args *DownloadOptions) error {
		chartFileName, src, size, err := k8s.Repos.DownloadChart(s, args.REPO, args.CHART, args.Version)
		if err != nil {
			return errors.Wrap(err, "download chart")
		}
		output := args.Output
		if output == "" && chartFileName != "" {
			output = chartFileName
		}
		if output == "" {
			return errors.Errorf("--output filepath must provide")
		}
		f, err := os.Create(output)
		if err != nil {
			return errors.Wrapf(err, "create saved file: %q", args.Output)
		}
		defer f.Close()
		var sink io.Writer = f
		bar := pb.Full.Start64(size)
		barReader := bar.NewProxyReader(src)
		if _, err := io.Copy(sink, barReader); err != nil {
			return errors.Wrap(err, "save chart")
		}
		return nil
	})
}