118 lines
2.5 KiB
Go
118 lines
2.5 KiB
Go
package vagrant
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/hashicorp/packer/packer"
|
|
)
|
|
|
|
// Lowercase a ascii letter.
|
|
func lower(c byte) byte {
|
|
return c | ('a' - 'A')
|
|
}
|
|
|
|
// Convert a string that represents a qemu disk image size to megabytes.
|
|
//
|
|
// Valid units (case-insensitive):
|
|
//
|
|
// B (byte) 1B
|
|
// K (kilobyte) 1024B
|
|
// M (megabyte) 1024K
|
|
// G (gigabyte) 1024M
|
|
// T (terabyte) 1024G
|
|
// P (petabyte) 1024T
|
|
// E (exabyte) 1024P
|
|
//
|
|
// The default is M.
|
|
func sizeInMegabytes(size string) uint64 {
|
|
unit := size[len(size)-1]
|
|
|
|
if unit >= '0' && unit <= '9' {
|
|
unit = 'm'
|
|
} else {
|
|
size = size[:len(size)-1]
|
|
}
|
|
|
|
value, _ := strconv.ParseUint(size, 10, 64)
|
|
|
|
switch lower(unit) {
|
|
case 'b':
|
|
return value / 1024 / 1024
|
|
case 'k':
|
|
return value / 1024
|
|
case 'm':
|
|
return value
|
|
case 'g':
|
|
return value * 1024
|
|
case 't':
|
|
return value * 1024 * 1024
|
|
case 'p':
|
|
return value * 1024 * 1024 * 1024
|
|
case 'e':
|
|
return value * 1024 * 1024 * 1024 * 1024
|
|
default:
|
|
panic(fmt.Sprintf("Unknown size unit %c", unit))
|
|
}
|
|
}
|
|
|
|
type LibVirtProvider struct{}
|
|
|
|
func (p *LibVirtProvider) KeepInputArtifact() bool {
|
|
return false
|
|
}
|
|
func (p *LibVirtProvider) Process(ui packer.Ui, artifact packer.Artifact, dir string) (vagrantfile string, metadata map[string]interface{}, err error) {
|
|
diskName := artifact.State("diskName").(string)
|
|
|
|
// Copy the disk image into the temporary directory (as box.img)
|
|
for _, path := range artifact.Files() {
|
|
if strings.HasSuffix(path, "/"+diskName) {
|
|
ui.Message(fmt.Sprintf("Copying from artifact: %s", path))
|
|
dstPath := filepath.Join(dir, "box.img")
|
|
if err = CopyContents(dstPath, path); err != nil {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
format := artifact.State("diskType").(string)
|
|
origSize := sizeInMegabytes(artifact.State("diskSize").(string))
|
|
size := origSize / 1024 // In MB, want GB
|
|
if origSize%1024 > 0 {
|
|
// Make sure we don't make the size smaller
|
|
size++
|
|
}
|
|
domainType := artifact.State("domainType").(string)
|
|
|
|
// Convert domain type to libvirt driver
|
|
var driver string
|
|
switch domainType {
|
|
case "none", "tcg", "hvf":
|
|
driver = "qemu"
|
|
case "kvm":
|
|
driver = domainType
|
|
default:
|
|
return "", nil, fmt.Errorf("Unknown libvirt domain type: %s", domainType)
|
|
}
|
|
|
|
// Create the metadata
|
|
metadata = map[string]interface{}{
|
|
"provider": "libvirt",
|
|
"format": format,
|
|
"virtual_size": size,
|
|
}
|
|
|
|
vagrantfile = fmt.Sprintf(libvirtVagrantfile, driver)
|
|
return
|
|
}
|
|
|
|
var libvirtVagrantfile = `
|
|
Vagrant.configure("2") do |config|
|
|
config.vm.provider :libvirt do |libvirt|
|
|
libvirt.driver = "%s"
|
|
end
|
|
end
|
|
`
|