78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
|
package vagrantcloud
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/mitchellh/multistep"
|
||
|
|
"github.com/mitchellh/packer/packer"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Version struct {
|
||
|
|
Version string `json:"version"`
|
||
|
|
Number uint `json:"number"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type NewVersion struct {
|
||
|
|
Version string `json:"version"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type stepCreateVersion struct {
|
||
|
|
number uint // number of the version, if needed in cleanup
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *stepCreateVersion) Run(state multistep.StateBag) multistep.StepAction {
|
||
|
|
client := state.Get("client").(*VagrantCloudClient)
|
||
|
|
ui := state.Get("ui").(packer.Ui)
|
||
|
|
config := state.Get("config").(Config)
|
||
|
|
box := state.Get("box").(*Box)
|
||
|
|
|
||
|
|
path := fmt.Sprintf("box/%s/versions", box.Tag)
|
||
|
|
|
||
|
|
// Wrap the version in a version object for the API
|
||
|
|
wrapper := make(map[string]interface{})
|
||
|
|
wrapper["version"] = NewVersion{Version: config.Version}
|
||
|
|
|
||
|
|
ui.Say(fmt.Sprintf("Creating version: %s", config.Version))
|
||
|
|
|
||
|
|
resp, err := client.Post(path, wrapper)
|
||
|
|
version := &Version{}
|
||
|
|
|
||
|
|
if err != nil || (resp.StatusCode != 200) {
|
||
|
|
cloudErrors := &VagrantCloudErrors{};
|
||
|
|
err = decodeBody(resp, cloudErrors);
|
||
|
|
state.Put("error", fmt.Errorf("Error creating version: %s", cloudErrors.FormatErrors()))
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
|
||
|
|
if err = decodeBody(resp, version); err != nil {
|
||
|
|
state.Put("error", fmt.Errorf("Error parsing version response: %s", err))
|
||
|
|
return multistep.ActionHalt
|
||
|
|
}
|
||
|
|
|
||
|
|
// Save the number for cleanup
|
||
|
|
s.number = version.Number
|
||
|
|
|
||
|
|
state.Put("version", version)
|
||
|
|
|
||
|
|
return multistep.ActionContinue
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *stepCreateVersion) Cleanup(state multistep.StateBag) {
|
||
|
|
// If we didn't save the version number, it likely doesn't exist
|
||
|
|
if (s.number == 0) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
client := state.Get("client").(*VagrantCloudClient)
|
||
|
|
ui := state.Get("ui").(packer.Ui)
|
||
|
|
box := state.Get("box").(*Box)
|
||
|
|
|
||
|
|
path := fmt.Sprintf("box/%s/version/%s", box.Tag, s.number)
|
||
|
|
|
||
|
|
// No need for resp from the cleanup DELETE
|
||
|
|
_, err := client.Delete(path)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
ui.Error(fmt.Sprintf("Error destroying version: %s", err))
|
||
|
|
}
|
||
|
|
}
|