packer-cn/builder/vmware/step_download_iso.go

91 lines
2.1 KiB
Go
Raw Normal View History

package vmware
import (
2013-06-10 02:00:07 -04:00
"crypto/md5"
2013-06-10 02:12:37 -04:00
"encoding/hex"
"fmt"
"github.com/mitchellh/multistep"
2013-06-12 20:41:44 -04:00
"github.com/mitchellh/packer/builder/common"
"github.com/mitchellh/packer/packer"
"log"
"time"
)
// This step downloads the ISO specified.
//
// Uses:
// cache packer.Cache
// config *config
// ui packer.Ui
//
// Produces:
// iso_path string
type stepDownloadISO struct{}
func (s stepDownloadISO) Run(state map[string]interface{}) multistep.StepAction {
cache := state["cache"].(packer.Cache)
config := state["config"].(*config)
ui := state["ui"].(packer.Ui)
2013-06-12 20:41:44 -04:00
checksum, err := hex.DecodeString(config.ISOMD5)
if err != nil {
2013-06-20 00:20:48 -04:00
err := fmt.Errorf("Error parsing checksum: %s", err)
state["error"] = err
ui.Error(err.Error())
2013-06-12 20:41:44 -04:00
return multistep.ActionHalt
}
log.Printf("Acquiring lock to download the ISO.")
cachePath := cache.Lock(config.ISOUrl)
defer cache.Unlock(config.ISOUrl)
2013-06-12 20:41:44 -04:00
downloadConfig := &common.DownloadConfig{
Url: config.ISOUrl,
TargetPath: cachePath,
CopyFile: false,
Hash: md5.New(),
Checksum: checksum,
}
2013-06-12 20:41:44 -04:00
download := common.NewDownloadClient(downloadConfig)
2013-06-12 20:41:44 -04:00
downloadCompleteCh := make(chan error, 1)
go func() {
ui.Say("Copying or downloading ISO. Progress will be reported periodically.")
cachePath, err = download.Get()
downloadCompleteCh <- err
}()
2013-06-10 02:00:07 -04:00
2013-06-12 20:41:44 -04:00
progressTicker := time.NewTicker(5 * time.Second)
defer progressTicker.Stop()
2013-06-10 02:00:07 -04:00
2013-06-12 20:41:44 -04:00
DownloadWaitLoop:
for {
select {
case err := <-downloadCompleteCh:
if err != nil {
2013-06-20 00:20:48 -04:00
err := fmt.Errorf("Error downloading ISO: %s", err)
state["error"] = err
ui.Error(err.Error())
return multistep.ActionHalt
}
2013-06-10 02:00:07 -04:00
2013-06-12 20:41:44 -04:00
break DownloadWaitLoop
case <-progressTicker.C:
ui.Say(fmt.Sprintf("Download progress: %d%%", download.PercentProgress()))
case <-time.After(1 * time.Second):
if _, ok := state[multistep.StateCancelled]; ok {
ui.Say("Interrupt received. Cancelling download...")
return multistep.ActionHalt
}
2013-06-10 02:00:07 -04:00
}
}
log.Printf("Path to ISO on disk: %s", cachePath)
state["iso_path"] = cachePath
return multistep.ActionContinue
}
func (stepDownloadISO) Cleanup(map[string]interface{}) {}