Merge pull request #1438 from rneugeba/moby

Support multiple "run" backends for the CLI

src/cmd/moby/build.go

@@ -4,73 +4,42 @@ import (
 	"archive/tar"
 	"bytes"
 	"errors"
+	"flag"
 	"fmt"
 	"io"
 	"io/ioutil"
+	"os"
 	"path/filepath"
 
 	log "github.com/Sirupsen/logrus"
 	"github.com/docker/moby/src/initrd"
 )
 
-func untarKernel(buf *bytes.Buffer, bzimageName, ktarName string) (*bytes.Buffer, *bytes.Buffer, error) {
-	tr := tar.NewReader(buf)
-
-	var bzimage, ktar *bytes.Buffer
-
-	for {
-		hdr, err := tr.Next()
-		if err == io.EOF {
-			break
-		}
-		if err != nil {
-			log.Fatalln(err)
-		}
-		switch hdr.Name {
-		case bzimageName:
-			bzimage = new(bytes.Buffer)
-			_, err := io.Copy(bzimage, tr)
-			if err != nil {
-				return nil, nil, err
-			}
-		case ktarName:
-			ktar = new(bytes.Buffer)
-			_, err := io.Copy(bzimage, tr)
-			if err != nil {
-				return nil, nil, err
-			}
-		default:
-			continue
-		}
+// Process the build arguments and execute build
+func build(args []string) {
+	buildCmd := flag.NewFlagSet("build", flag.ExitOnError)
+	buildCmd.Usage = func() {
+		fmt.Printf("USAGE: %s build [options] [file.yml]\n\n", os.Args[0])
+		fmt.Printf("'file.yml' defaults to 'moby.yml' if not specified.\n\n")
+		fmt.Printf("Options:\n")
+		buildCmd.PrintDefaults()
 	}
+	buildName := buildCmd.String("name", "", "Name to use for output files")
+	buildPull := buildCmd.Bool("pull", false, "Always pull images")
 
-	if ktar == nil || bzimage == nil {
-		return nil, nil, errors.New("did not find bzImage and kernel.tar in tarball")
-	}
+	buildCmd.Parse(args)
+	remArgs := buildCmd.Args()
 
-	return bzimage, ktar, nil
-}
-
-func containersInitrd(containers []*bytes.Buffer) (*bytes.Buffer, error) {
-	w := new(bytes.Buffer)
-	iw := initrd.NewWriter(w)
-	defer iw.Close()
-	for _, file := range containers {
-		_, err := initrd.Copy(iw, file)
-		if err != nil {
-			return nil, err
-		}
-	}
-
-	return w, nil
-}
-
-func build(name string, pull bool, args []string) {
 	conf := "moby.yml"
-	if len(args) > 0 {
-		conf = args[0]
+	if len(remArgs) > 0 {
+		conf = remArgs[0]
 	}
 
+	buildInternal(*buildName, *buildPull, conf)
+}
+
+// Perform the actual build process
+func buildInternal(name string, pull bool, conf string) {
 	if name == "" {
 		name = filepath.Base(conf)
 		ext := filepath.Ext(conf)
@@ -198,3 +167,55 @@ func build(name string, pull bool, args []string) {
 		log.Fatalf("Error writing outputs: %v", err)
 	}
 }
+
+func untarKernel(buf *bytes.Buffer, bzimageName, ktarName string) (*bytes.Buffer, *bytes.Buffer, error) {
+	tr := tar.NewReader(buf)
+
+	var bzimage, ktar *bytes.Buffer
+
+	for {
+		hdr, err := tr.Next()
+		if err == io.EOF {
+			break
+		}
+		if err != nil {
+			log.Fatalln(err)
+		}
+		switch hdr.Name {
+		case bzimageName:
+			bzimage = new(bytes.Buffer)
+			_, err := io.Copy(bzimage, tr)
+			if err != nil {
+				return nil, nil, err
+			}
+		case ktarName:
+			ktar = new(bytes.Buffer)
+			_, err := io.Copy(bzimage, tr)
+			if err != nil {
+				return nil, nil, err
+			}
+		default:
+			continue
+		}
+	}
+
+	if ktar == nil || bzimage == nil {
+		return nil, nil, errors.New("did not find bzImage and kernel.tar in tarball")
+	}
+
+	return bzimage, ktar, nil
+}
+
+func containersInitrd(containers []*bytes.Buffer) (*bytes.Buffer, error) {
+	w := new(bytes.Buffer)
+	iw := initrd.NewWriter(w)
+	defer iw.Close()
+	for _, file := range containers {
+		_, err := initrd.Copy(iw, file)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	return w, nil
+}

src/cmd/moby/main.go

@@ -40,33 +40,6 @@ func main() {
 	flagQuiet := flag.Bool("q", false, "Quiet execution")
 	flagVerbose := flag.Bool("v", false, "Verbose execution")
 
-	buildCmd := flag.NewFlagSet("build", flag.ExitOnError)
-	buildCmd.Usage = func() {
-		fmt.Printf("USAGE: %s build [options] [file.yml]\n\n", os.Args[0])
-		fmt.Printf("'file.yml' defaults to 'moby.yml' if not specified.\n\n")
-		fmt.Printf("Options:\n")
-		buildCmd.PrintDefaults()
-	}
-	buildName := buildCmd.String("name", "", "Name to use for output files")
-	buildPull := buildCmd.Bool("pull", false, "Always pull images")
-
-	runCmd := flag.NewFlagSet("run", flag.ExitOnError)
-	runCmd.Usage = func() {
-		fmt.Printf("USAGE: %s run [options] [prefix]\n\n", os.Args[0])
-		fmt.Printf("'prefix' specifies the path to the VM image.\n")
-		fmt.Printf("It defaults to './moby'.\n")
-		fmt.Printf("\n")
-		fmt.Printf("Options:\n")
-		runCmd.PrintDefaults()
-		fmt.Printf("\n")
-		fmt.Printf("If 'data' is supplied or if 'background' is selected\n")
-		fmt.Printf("some per VM state is kept in a sub-directory in the ~/.moby\n")
-	}
-	runCPUs := runCmd.Int("cpus", 1, "Number of CPUs")
-	runMem := runCmd.Int("mem", 1024, "Amount of memory in MB")
-	runDiskSz := runCmd.Int("disk-size", 0, "Size of Disk in MB")
-	runDisk := runCmd.String("disk", "", "Path to disk image to used")
-
 	// Set up logging
 	log.SetFormatter(new(infoFormatter))
 	log.SetLevel(log.InfoLevel)
@@ -93,11 +66,9 @@ func main() {
 
 	switch args[0] {
 	case "build":
-		buildCmd.Parse(args[1:])
-		build(*buildName, *buildPull, buildCmd.Args())
+		build(args[1:])
 	case "run":
-		runCmd.Parse(args[1:])
-		run(*runCPUs, *runMem, *runDiskSz, *runDisk, runCmd.Args())
+		run(args[1:])
 	case "help":
 		flag.Usage()
 	default:

src/cmd/moby/run.go

@@ -0,0 +1,45 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"runtime"
+
+	log "github.com/Sirupsen/logrus"
+)
+
+func runUsage() {
+	fmt.Printf("USAGE: %s run [backend] [options] [prefix]\n\n", os.Args[0])
+
+	fmt.Printf("'backend' specifies the run backend.\n")
+	fmt.Printf("If not specified the platform specific default will be used\n")
+	fmt.Printf("Supported backends are (default platform in brackets):\n")
+	fmt.Printf("  hyperkit [macOS]\n")
+	fmt.Printf("\n")
+	fmt.Printf("'options' are the backend specific options.\n")
+	fmt.Printf("See 'moby run [backend] --help' for details.\n\n")
+	fmt.Printf("'prefix' specifies the path to the VM image.\n")
+	fmt.Printf("It defaults to './moby'.\n")
+}
+
+func run(args []string) {
+	if len(args) < 1 {
+		runUsage()
+		os.Exit(1)
+	}
+
+	switch args[0] {
+	case "help", "-h", "-help", "--help":
+		runUsage()
+		os.Exit(0)
+	case "hyperkit":
+		runHyperKit(args[1:])
+	default:
+		switch runtime.GOOS {
+		case "darwin":
+			runHyperKit(args)
+		default:
+			log.Errorf("There currently is no default 'run' backend for your platform.")
+		}
+	}
+}

src/cmd/moby/run_darwin.go

@@ -1,51 +0,0 @@
-// +build darwin
-
-package main
-
-import (
-	"io/ioutil"
-	"os"
-	"os/user"
-
-	log "github.com/Sirupsen/logrus"
-	"github.com/docker/hyperkit/go"
-)
-
-func run(cpus, mem, diskSz int, disk string, args []string) {
-	prefix := "moby"
-	if len(args) > 0 {
-		prefix = args[0]
-	}
-
-	cmdline, err := ioutil.ReadFile(prefix + "-cmdline")
-	if err != nil {
-		log.Fatalf("Cannot open cmdline file: %v", err)
-	}
-
-	if diskSz != 0 && disk == "" {
-		disk = prefix + "-disk.img"
-	}
-
-	h, err := hyperkit.New("", "", "auto", disk)
-	if err != nil {
-		log.Fatalln("Error creating hyperkit: ", err)
-	}
-
-	h.Kernel = prefix + "-bzImage"
-	h.Initrd = prefix + "-initrd.img"
-	h.CPUs = cpus
-	h.Memory = mem
-	h.DiskSize = diskSz
-
-	err = h.Run(string(cmdline))
-	if err != nil {
-		log.Fatalf("Cannot run hyperkit: %v", err)
-	}
-}
-
-func getHome() string {
-	if usr, err := user.Current(); err == nil {
-		return usr.HomeDir
-	}
-	return os.Getenv("HOME")
-}

src/cmd/moby/run_fallback.go

@@ -1,11 +0,0 @@
-// +build !darwin
-
-package main
-
-import (
-	log "github.com/Sirupsen/logrus"
-)
-
-func run(cpus, mem, diskSz int, userData string, args []string) {
-	log.Fatalf("'run' is not supported yet on your OS")
-}

src/cmd/moby/run_hyperkit.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"io/ioutil"
+	"os"
+
+	log "github.com/Sirupsen/logrus"
+	"github.com/docker/hyperkit/go"
+)
+
+// Process the run arguments and execute run
+func runHyperKit(args []string) {
+	hyperkitCmd := flag.NewFlagSet("hyperkit", flag.ExitOnError)
+	hyperkitCmd.Usage = func() {
+		fmt.Printf("USAGE: %s run hyperkit [options] [prefix]\n\n", os.Args[0])
+		fmt.Printf("'prefix' specifies the path to the VM image.\n")
+		fmt.Printf("It defaults to './moby'.\n")
+		fmt.Printf("\n")
+		fmt.Printf("Options:\n")
+		hyperkitCmd.PrintDefaults()
+	}
+	runCPUs := hyperkitCmd.Int("cpus", 1, "Number of CPUs")
+	runMem := hyperkitCmd.Int("mem", 1024, "Amount of memory in MB")
+	runDiskSz := hyperkitCmd.Int("disk-size", 0, "Size of Disk in MB")
+	runDisk := hyperkitCmd.String("disk", "", "Path to disk image to used")
+
+	hyperkitCmd.Parse(args)
+	remArgs := hyperkitCmd.Args()
+
+	prefix := "moby"
+	if len(remArgs) > 0 {
+		prefix = remArgs[0]
+	}
+
+	runHyperKitInternal(*runCPUs, *runMem, *runDiskSz, *runDisk, prefix)
+}
+
+func runHyperKitInternal(cpus, mem, diskSz int, disk, prefix string) {
+	cmdline, err := ioutil.ReadFile(prefix + "-cmdline")
+	if err != nil {
+		log.Fatalf("Cannot open cmdline file: %v", err)
+	}
+
+	if diskSz != 0 && disk == "" {
+		disk = prefix + "-disk.img"
+	}
+
+	h, err := hyperkit.New("", "", "auto", disk)
+	if err != nil {
+		log.Fatalln("Error creating hyperkit: ", err)
+	}
+
+	h.Kernel = prefix + "-bzImage"
+	h.Initrd = prefix + "-initrd.img"
+	h.CPUs = cpus
+	h.Memory = mem
+	h.DiskSize = diskSz
+
+	err = h.Run(string(cmdline))
+	if err != nil {
+		log.Fatalf("Cannot run hyperkit: %v", err)
+	}
+}