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)
	}
}