func (r *Repo) DefaultImage() string { if !core.IsTemplateFile("Capstanfile") { return "" } pwd, err := os.Getwd() if err != nil { return "" } image := filepath.Base(pwd) return image }
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 } }