func main() {
repo := util.NewRepo()
app := cli.NewApp()
app.Name = "capstan"
app.Version = VERSION
app.Usage = "pack, ship, and run applications in light-weight VMs"
app.Commands = []cli.Command{
{
Name: "info",
Usage: "show disk image information",
Action: func(c *cli.Context) {
if len(c.Args()) != 1 {
fmt.Println("usage: capstan info [image-file]")
return
}
image := c.Args()[0]
err := cmd.Info(image)
if err != nil {
fmt.Println(err.Error())
}
},
},
{
Name: "import",
Usage: "import an image to the local repository",
Flags: []cli.Flag{
cli.StringFlag{Name: "v", Value: "", Usage: "image version"},
cli.StringFlag{Name: "c", Value: "", Usage: "image creation date"},
cli.StringFlag{Name: "d", Value: "", Usage: "image description"},
cli.StringFlag{Name: "b", Value: "", Usage: "image build command"},
},
Action: func(c *cli.Context) {
if len(c.Args()) != 2 {
fmt.Println("usage: capstan import [image-name] [image-file]")
return
}
err := repo.ImportImage(c.Args()[0], c.Args()[1], c.String("v"), c.String("c"), c.String("d"), c.String("b"))
if err != nil {
fmt.Println(err.Error())
}
},
},
{
Name: "pull",
Usage: "pull an image from a repository",
Flags: []cli.Flag{
cli.StringFlag{Name: "p", Value: hypervisor.Default(), Usage: "hypervisor: qemu|vbox|vmw|gce"},
},
Action: func(c *cli.Context) {
if len(c.Args()) != 1 {
fmt.Println("usage: capstan pull [image-name]")
return
}
hypervisor := c.String("p")
if !isValidHypervisor(hypervisor) {
fmt.Printf("error: '%s' is not a supported hypervisor\n", c.String("p"))
return
}
err := cmd.Pull(repo, hypervisor, c.Args().First())
if err != nil {
fmt.Println(err.Error())
}
},
},
{
Name: "rmi",
Usage: "delete an image from a repository",
Action: func(c *cli.Context) {
if len(c.Args()) != 1 {
fmt.Println("usage: capstan rmi [image-name]")
return
}
err := repo.RemoveImage(c.Args().First())
if err != nil {
fmt.Println(err.Error())
}
},
},
{
Name: "run",
Usage: "launch a VM. You may pass the image name as the first argument.",
Flags: []cli.Flag{
cli.StringFlag{Name: "i", Value: "", Usage: "image_name"},
cli.StringFlag{Name: "p", Value: hypervisor.Default(), Usage: "hypervisor: qemu|vbox|vmw|gce"},
cli.StringFlag{Name: "m", Value: "1G", Usage: "memory size"},
cli.IntFlag{Name: "c", Value: 2, Usage: "number of CPUs"},
cli.StringFlag{Name: "n", Value: "nat", Usage: "networking: nat|bridge|tap"},
cli.BoolFlag{Name: "v", Usage: "verbose mode"},
cli.StringFlag{Name: "b", Value: "", Usage: "networking device (bridge or tap): e.g., virbr0, vboxnet0, tap0"},
cli.StringSliceFlag{Name: "f", Value: new(cli.StringSlice), Usage: "port forwarding rules"},
cli.StringFlag{Name: "gce-upload-dir", Value: "", Usage: "Directory to upload local image to: e.g., gs://osvimg"},
cli.StringFlag{Name: "mac", Value: "", Usage: "MAC address. If not specified, the MAC address will be generated automatically."},
},
Action: func(c *cli.Context) {
config := &cmd.RunConfig{
InstanceName: c.Args().First(),
ImageName: c.String("i"),
Hypervisor: c.String("p"),
Verbose: c.Bool("v"),
Memory: c.String("m"),
Cpus: c.Int("c"),
Networking: c.String("n"),
Bridge: c.String("b"),
NatRules: nat.Parse(c.StringSlice("f")),
GCEUploadDir: c.String("gce-upload-dir"),
MAC: c.String("mac"),
}
if !isValidHypervisor(config.Hypervisor) {
fmt.Printf("error: '%s' is not a supported hypervisor\n", config.Hypervisor)
return
}
err := cmd.Run(repo, config)
if err != nil {
fmt.Println(err.Error())
}
},
},
{
Name: "build",
Usage: "build an image",
Flags: []cli.Flag{
cli.StringFlag{Name: "p", Value: hypervisor.Default(), Usage: "hypervisor: qemu|vbox|vmw|gce"},
cli.StringFlag{Name: "m", Value: "512M", Usage: "memory size"},
cli.BoolFlag{Name: "v", Usage: "verbose mode"},
},
Action: func(c *cli.Context) {
imageName := c.Args().First()
if len(c.Args()) != 1 {
imageName = repo.DefaultImage()
}
if imageName == "" {
fmt.Println("usage: capstan build [image-name]")
return
}
hypervisor := c.String("p")
if !isValidHypervisor(hypervisor) {
fmt.Printf("error: '%s' is not a supported hypervisor\n", c.String("p"))
return
}
image := &core.Image{
Name: imageName,
Hypervisor: hypervisor,
}
template, err := core.ReadTemplateFile("Capstanfile")
if err != nil {
fmt.Println(err.Error())
return
}
if err := cmd.Build(repo, image, template, c.Bool("v"), c.String("m")); err != nil {
fmt.Println(err.Error())
return
}
},
},
{
Name: "images",
ShortName: "i",
Usage: "list images",
Action: func(c *cli.Context) {
repo.ListImages()
},
},
{
Name: "search",
Usage: "search a remote images",
Action: func(c *cli.Context) {
image := ""
if len(c.Args()) > 0 {
image = c.Args()[0]
}
util.ListImagesRemote(image)
},
},
{
Name: "instances",
ShortName: "I",
Usage: "list instances",
Action: func(c *cli.Context) {
cmd.Instances()
},
},
{
Name: "stop",
Usage: "stop an instance",
Action: func(c *cli.Context) {
if len(c.Args()) != 1 {
fmt.Println("usage: capstan stop [instance_name]")
return
}
instance := c.Args()[0]
cmd.Stop(instance)
},
},
{
Name: "delete",
Usage: "delete an instance",
Action: func(c *cli.Context) {
if len(c.Args()) != 1 {
fmt.Println("usage: capstan delete [instance_name]")
return
}
instance := c.Args()[0]
cmd.Delete(instance)
},
},
}
app.Run(os.Args)
}
func Run(repo *util.Repo, config *RunConfig) error {
var path string
var cmd *exec.Cmd
// Start an existing instance
if config.ImageName == "" && config.InstanceName != "" {
instanceName, instancePlatform := util.SearchInstance(config.InstanceName)
if instanceName != "" {
defer fmt.Println("")
fmt.Printf("Created instance: %s\n", instanceName)
// Do not set RawTerm for gce
if instancePlatform != "gce" {
util.RawTerm()
defer util.ResetTerm()
}
var err error
switch instancePlatform {
case "qemu":
c, err := qemu.LoadConfig(instanceName)
if err != nil {
return err
}
cmd, err = qemu.LaunchVM(c)
case "vbox":
c, err := vbox.LoadConfig(instanceName)
if err != nil {
return err
}
cmd, err = vbox.LaunchVM(c)
case "vmw":
c, err := vmw.LoadConfig(instanceName)
if err != nil {
return err
}
cmd, err = vmw.LaunchVM(c)
case "gce":
c, err := gce.LoadConfig(instanceName)
if err != nil {
return err
}
cmd, err = gce.LaunchVM(c)
}
if err != nil {
return err
}
if cmd != nil {
return cmd.Wait()
}
return nil
} else {
// The InstanceName is actually a ImageName
// so, cmd like "capstan run cloudius/osv" will work
config.ImageName = config.InstanceName
config.InstanceName = strings.Replace(config.InstanceName, "/", "-", -1)
return Run(repo, config)
}
} else if config.ImageName != "" && config.InstanceName != "" {
// Both ImageName and InstanceName are specified
if f, err := os.Stat(config.ImageName); (f != nil && f.IsDir()) || os.IsNotExist(err) {
if repo.ImageExists(config.Hypervisor, config.ImageName) {
path = repo.ImagePath(config.Hypervisor, config.ImageName)
} else if image.IsCloudImage(config.ImageName) {
path = config.ImageName
} else {
remote, err := util.IsRemoteImage(repo.URL, config.ImageName)
if err != nil {
return err
}
if remote {
err := Pull(repo, config.Hypervisor, config.ImageName)
if err != nil {
return err
}
path = repo.ImagePath(config.Hypervisor, config.ImageName)
} else {
return fmt.Errorf("%s: no such image", config.ImageName)
}
}
if config.Hypervisor == "gce" && !image.IsCloudImage(config.ImageName) {
str, err := ioutil.ReadFile(path)
if err != nil {
return err
}
path = strings.Replace(string(str), "\n", "", -1)
}
} else {
if strings.HasSuffix(config.ImageName, ".jar") {
config, err = buildJarImage(repo, config)
if err != nil {
return err
}
path = repo.ImagePath(config.Hypervisor, config.ImageName)
} else {
path = config.ImageName
}
}
deleteInstance(config.InstanceName)
} else if config.ImageName == "" && config.InstanceName == "" {
// Valid only when Capstanfile is present
config.ImageName = repo.DefaultImage()
config.InstanceName = config.ImageName
if config.ImageName == "" {
return fmt.Errorf("No Capstanfile found, unable to run.")
}
if !repo.ImageExists(config.Hypervisor, config.ImageName) {
if !core.IsTemplateFile("Capstanfile") {
return fmt.Errorf("%s: no such image", config.ImageName)
}
image := &core.Image{
Name: config.ImageName,
Hypervisor: config.Hypervisor,
}
template, err := core.ReadTemplateFile("Capstanfile")
if err != nil {
return err
}
if err := Build(repo, image, template, config.Verbose, config.Memory); err != nil {
return err
}
}
path = repo.ImagePath(config.Hypervisor, config.ImageName)
deleteInstance(config.InstanceName)
} else {
// Cmdline option is not valid
usage()
return nil
}
format, err := image.Probe(path)
if err != nil {
return err
}
if format == image.Unknown {
return fmt.Errorf("%s: image format not recognized, unable to run it.", path)
}
size, err := util.ParseMemSize(config.Memory)
if err != nil {
return err
}
defer fmt.Println("")
id := config.InstanceName
fmt.Printf("Created instance: %s\n", id)
// Do not set RawTerm for gce
if config.Hypervisor != "gce" {
util.RawTerm()
defer util.ResetTerm()
}
switch config.Hypervisor {
case "qemu":
dir := filepath.Join(util.ConfigDir(), "instances/qemu", id)
bridge := config.Bridge
if bridge == "" {
bridge = "virbr0"
}
config := &qemu.VMConfig{
Name: id,
Image: path,
Verbose: true,
Memory: size,
Cpus: config.Cpus,
Networking: config.Networking,
Bridge: bridge,
NatRules: config.NatRules,
BackingFile: true,
InstanceDir: dir,
Monitor: filepath.Join(dir, "osv.monitor"),
ConfigFile: filepath.Join(dir, "osv.config"),
MAC: config.MAC,
}
cmd, err = qemu.LaunchVM(config)
case "vbox":
if format != image.VDI && format != image.VMDK {
return fmt.Errorf("%s: image format of %s is not supported, unable to run it.", config.Hypervisor, path)
}
dir := filepath.Join(util.ConfigDir(), "instances/vbox", id)
bridge := config.Bridge
if bridge == "" {
bridge = "vboxnet0"
}
config := &vbox.VMConfig{
Name: id,
Dir: filepath.Join(util.ConfigDir(), "instances/vbox"),
Image: path,
Memory: size,
Cpus: config.Cpus,
Networking: config.Networking,
Bridge: bridge,
NatRules: config.NatRules,
ConfigFile: filepath.Join(dir, "osv.config"),
MAC: config.MAC,
}
cmd, err = vbox.LaunchVM(config)
case "gce":
if format != image.GCE_TARBALL && format != image.GCE_GS {
return fmt.Errorf("%s: image format of %s is not supported, unable to run it.", config.Hypervisor, path)
}
dir := filepath.Join(util.ConfigDir(), "instances/gce", id)
c := &gce.VMConfig{
Name: id,
Image: id,
Network: "default",
MachineType: "n1-standard-1",
Zone: "us-central1-a",
ConfigFile: filepath.Join(dir, "osv.config"),
InstanceDir: dir,
}
if format == image.GCE_TARBALL {
c.CloudStoragePath = strings.TrimSuffix(config.GCEUploadDir, "/") + "/" + id + ".tar.gz"
c.Tarball = path
} else {
c.CloudStoragePath = path
c.Tarball = ""
}
cmd, err = gce.LaunchVM(c)
case "vmw":
if format != image.VMDK {
return fmt.Errorf("%s: image format of %s is not supported, unable to run it.", config.Hypervisor, path)
}
dir := filepath.Join(util.ConfigDir(), "instances/vmw", id)
config := &vmw.VMConfig{
Name: id,
Dir: dir,
Image: filepath.Join(dir, "osv.vmdk"),
Memory: size,
Cpus: config.Cpus,
NatRules: config.NatRules,
VMXFile: filepath.Join(dir, "osv.vmx"),
InstanceDir: dir,
OriginalVMDK: path,
ConfigFile: filepath.Join(dir, "osv.config"),
}
cmd, err = vmw.LaunchVM(config)
default:
err = fmt.Errorf("%s: is not a supported hypervisor", config.Hypervisor)
}
if err != nil {
return err
}
if cmd != nil {
return cmd.Wait()
} else {
return nil
}
}