mirror of https://github.com/linka-cloud/d2vm.git
102 lines
3.0 KiB
Go
102 lines
3.0 KiB
Go
// Copyright 2022 Linka Cloud 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 main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/c2h5oh/datasize"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
|
|
"go.linka.cloud/d2vm"
|
|
"go.linka.cloud/d2vm/pkg/docker"
|
|
"go.linka.cloud/d2vm/pkg/exec"
|
|
)
|
|
|
|
var (
|
|
pull = false
|
|
|
|
convertCmd = &cobra.Command{
|
|
Use: "convert [docker image]",
|
|
Short: "Convert Docker image to vm image",
|
|
Args: cobra.ExactArgs(1),
|
|
SilenceUsage: true,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
img := args[0]
|
|
tag := "latest"
|
|
if parts := strings.Split(img, ":"); len(parts) > 1 {
|
|
img, tag = parts[0], parts[1]
|
|
}
|
|
size, err := parseSize(size)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if _, err := os.Stat(output); err == nil || !os.IsNotExist(err) {
|
|
if !force {
|
|
return fmt.Errorf("%s already exists", output)
|
|
}
|
|
}
|
|
if debug {
|
|
exec.Run = exec.RunStdout
|
|
}
|
|
if _, err := os.Stat(output); err == nil || !os.IsNotExist(err) {
|
|
if !force {
|
|
return fmt.Errorf("%s already exists", output)
|
|
}
|
|
}
|
|
found := false
|
|
if !pull {
|
|
o, _, err := docker.CmdOut(cmd.Context(), "image", "ls", "--format={{ .Repository }}:{{ .Tag }}", img)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
found = strings.TrimSuffix(o, "\n") == fmt.Sprintf("%s:%s", img, tag)
|
|
if found {
|
|
logrus.Infof("using local image %s:%s", img, tag)
|
|
}
|
|
}
|
|
if pull || !found {
|
|
logrus.Infof("pulling image %s", img)
|
|
if err := docker.Cmd(cmd.Context(), "image", "pull", img); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return d2vm.Convert(cmd.Context(), img, size, password, output, format)
|
|
},
|
|
}
|
|
)
|
|
|
|
func parseSize(s string) (int64, error) {
|
|
var v datasize.ByteSize
|
|
if err := v.UnmarshalText([]byte(s)); err != nil {
|
|
return 0, err
|
|
}
|
|
return int64(v), nil
|
|
}
|
|
|
|
func init() {
|
|
convertCmd.Flags().BoolVar(&pull, "pull", false, "Always pull docker image")
|
|
convertCmd.Flags().StringVarP(&format, "output-format", "O", format, "The output image format, supported formats: "+strings.Join(d2vm.OutputFormats(), " "))
|
|
convertCmd.Flags().StringVarP(&output, "output", "o", output, "The output image")
|
|
convertCmd.Flags().StringVarP(&password, "password", "p", "root", "The Root user password")
|
|
convertCmd.Flags().StringVarP(&size, "size", "s", "10G", "The output image size")
|
|
convertCmd.Flags().BoolVarP(&debug, "debug", "d", false, "Enable Debug output")
|
|
convertCmd.Flags().BoolVarP(&force, "force", "f", false, "Override output qcow2 image")
|
|
rootCmd.AddCommand(convertCmd)
|
|
}
|