mirror of https://github.com/linka-cloud/d2vm.git
74 lines
2.6 KiB
Go
74 lines
2.6 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 (
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
|
|
"go.linka.cloud/d2vm"
|
|
"go.linka.cloud/d2vm/pkg/docker"
|
|
)
|
|
|
|
var (
|
|
file = "Dockerfile"
|
|
tag = uuid.New().String()
|
|
networkManager string
|
|
buildArgs []string
|
|
buildCmd = &cobra.Command{
|
|
Use: "build [context directory]",
|
|
Short: "Build a vm image from Dockerfile",
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
// TODO(adphi): resolve context path
|
|
if runtime.GOOS != "linux" {
|
|
return docker.RunD2VM(cmd.Context(), d2vm.Image, d2vm.Version, cmd.Name(), os.Args[2:]...)
|
|
}
|
|
size, err := parseSize(size)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if file == "" {
|
|
file = filepath.Join(args[0], "Dockerfile")
|
|
}
|
|
logrus.Infof("building docker image from %s", file)
|
|
if err := docker.Build(cmd.Context(), tag, file, args[0], buildArgs...); err != nil {
|
|
return err
|
|
}
|
|
return d2vm.Convert(cmd.Context(), tag, size, password, output, cmdLineExtra, d2vm.NetworkManager(networkManager))
|
|
},
|
|
}
|
|
)
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(buildCmd)
|
|
|
|
buildCmd.Flags().StringVarP(&file, "file", "f", "", "Name of the Dockerfile")
|
|
buildCmd.Flags().StringArrayVar(&buildArgs, "build-arg", nil, "Set build-time variables")
|
|
|
|
buildCmd.Flags().StringVarP(&output, "output", "o", output, "The output image, the extension determine the image format, raw will be used if none. Supported formats: "+strings.Join(d2vm.OutputFormats(), " "))
|
|
buildCmd.Flags().StringVarP(&password, "password", "p", "root", "Root user password")
|
|
buildCmd.Flags().StringVarP(&size, "size", "s", "10G", "The output image size")
|
|
buildCmd.Flags().BoolVar(&force, "force", false, "Override output image")
|
|
buildCmd.Flags().StringVar(&cmdLineExtra, "append-to-cmdline", "", "Extra kernel cmdline arguments to append to the generated one")
|
|
buildCmd.Flags().StringVar(&networkManager, "network-manager", "", "Network manager to use for the image: none, netplan, ifupdown")
|
|
}
|