add virtualbox builders

mayn · Aug 17, 2017 · e126c67 · e126c67
1 parent 0122bc8
commit e126c67
Show file tree

Hide file tree

Showing 5 changed files with 188 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,6 +3,8 @@
 
 
 ### FEATURES:
+* builder/VirtualboxIso: add support for packer's virtualbox iso builder
+* builder/VirtualboxOvf: add support for packer's virtualbox ovf builder
 * builder/VMwareIso: add support for packer's vmware iso builder
 * builder/VMwareVmx: add support for packer's vmware vmx builder
 

diff --git a/README.rst b/README.rst
@@ -90,6 +90,8 @@ Builders:
 - amazon-ebs
 - docker
 - file
+- virtualbox-iso
+- virtualbox-ovf
 - vmware-iso
 - vmware-vmx
 

diff --git a/src/packerlicious/builder.py b/src/packerlicious/builder.py
@@ -237,6 +237,133 @@ def validate(self):
             warnings.warn("Both source and content not specified, artifact will be empty.")
 
 
+class VirtualboxIso(PackerBuilder):
+    """
+    VirtualBox ISO Builder
+    https://www.packer.io/docs/builders/virtualbox-iso.html
+    """
+    resource_type = "virtualbox-iso"
+
+    # VirtualBox OVF checksum types
+    NONE = "none"
+    MD5 = "md5"
+    SHA1 = "sha1"
+    SHA256 = "sha256"
+    SHA512 = "sha512"
+    OVF = "ovf"
+    OVA = "ova"
+
+    props = {
+        'iso_checksum': (basestring, False),
+        'iso_checksum_type': (validator.string_list_item([NONE, MD5, SHA1, SHA256, SHA512]), True),
+        'iso_checksum_url': (basestring, False),
+        'iso_url': (basestring, True),
+        'boot_command': ([basestring], False),
+        'boot_wait': (basestring, False),
+        'disk_size': (int, False),
+        'export_opts': ([basestring], False),
+        'floppy_files': ([basestring], False),
+        'floppy_dirs': ([basestring], False),
+        'format': (validator.string_list_item([OVF, OVA]), False),
+        'guest_additions_mode': (validator.string_list_item(["upload", "attach", "disable"]), False),
+        'guest_additions_path': (basestring, False),
+        'guest_additions_sha256': (basestring, False),
+        'guest_additions_url': (basestring, False),
+        'guest_additions_url': (basestring, False),
+        'guest_os_type': (basestring, False),
+        'hard_drive_interface': (basestring, False),
+        'sata_port_count': (int, False),
+        'hard_drive_nonrotational': (validator.boolean, False),
+        'hard_drive_discard': (validator.boolean, False),
+        'headless': (validator.boolean, False),
+        'http_directory': (basestring, False),
+        'http_port_min': (int, False),
+        'http_port_max': (int, False),
+        'iso_interface': (basestring, False),
+        'iso_target_extension': (basestring, False),
+        'iso_target_path': (basestring, False),
+        'iso_urls': ([basestring], False),
+        'keep_registered': (validator.boolean, False),
+        'output_directory': (basestring, False),
+        'post_shutdown_delay': (basestring, False),
+        'shutdown_command': (basestring, False),
+        'shutdown_timeout': (basestring, False),
+        'skip_export': (validator.boolean, False),
+        'ssh_host_port_min': (int, False),
+        'ssh_host_port_max': (int, False),
+        'ssh_skip_nat_mapping': (validator.boolean, False),
+        'vboxmanage': ([[basestring]], False),
+        'vboxmanage_post': ([[basestring]], False),
+        'virtualbox_version_file': (basestring, False),
+        'vm_name': (basestring, False),
+        'vrdp_bind_address': (basestring, False),
+        'vrdp_port_min': (int, False),
+        'vrdp_port_max': (int, False),
+    }
+
+    def validate(self):
+        conds = [
+            'iso_checksum',
+            'iso_checksum_url',
+        ]
+        validator.mutually_exclusive(self.__class__.__name__, self.properties, conds)
+
+
+class VirtualboxOvf(PackerBuilder):
+    """
+    VirtualBox OVF Builder
+    https://www.packer.io/docs/builders/virtualbox-ovf.html
+    """
+    resource_type = "virtualbox-ovf"
+
+    # VirtualBox OVF checksum types
+    NONE = "none"
+    MD5 = "md5"
+    SHA1 = "sha1"
+    SHA256 = "sha256"
+    SHA512 = "sha512"
+    OVF = "ovf"
+    OVA = "ova"
+
+    props = {
+        'source_path': (basestring, True),
+        'boot_command': ([basestring], False),
+        'boot_wait': (basestring, False),
+        'checksum': (basestring, False),
+        'checksum_type': (validator.string_list_item([NONE, MD5, SHA1, SHA256, SHA512]), False),
+        'export_opts': ([basestring], False),
+        'floppy_files': ([basestring], False),
+        'floppy_dirs': ([basestring], False),
+        'format': (validator.string_list_item([OVF, OVA]), False),
+        'guest_additions_mode': (validator.string_list_item(["upload", "attach", "disable"]), False),
+        'guest_additions_path': (basestring, False),
+        'guest_additions_sha256': (basestring, False),
+        'guest_additions_url': (basestring, False),
+        'headless': (validator.boolean, False),
+        'http_directory': (basestring, False),
+        'http_port_min': (int, False),
+        'http_port_max': (int, False),
+        'import_flags': ([basestring], False),
+        'import_opts': (basestring, False),
+        'output_directory': (basestring, False),
+        'post_shutdown_delay': (basestring, False),
+        'shutdown_command': (basestring, False),
+        'shutdown_timeout': (basestring, False),
+        'skip_export': (validator.boolean, False),
+        'ssh_host_port_min': (int, False),
+        'ssh_host_port_max': (int, False),
+        'ssh_skip_nat_mapping': (validator.boolean, False),
+        'target_path': (basestring, False),
+        'vboxmanage': ([[basestring]], False),
+        'vboxmanage_post': ([[basestring]], False),
+        'virtualbox_version_file': (basestring, False),
+        'vm_name': (basestring, False),
+        'vrdp_bind_address': (basestring, False),
+        'vrdp_port_min': (int, False),
+        'vrdp_port_max': (int, False),
+    }
+
+
 class VMwareIso(PackerBuilder):
     """
     VMware ISO Builder
@@ -252,9 +379,9 @@ class VMwareIso(PackerBuilder):
     SHA512 = "sha512"
 
     props = {
-        'iso_checksum': (basestring, True),
+        'iso_checksum': (basestring, False),
         'iso_checksum_type': (validator.string_list_item([NONE, MD5, SHA1, SHA256, SHA512]), True),
-        'iso_checksum_url': (basestring, True),
+        'iso_checksum_url': (basestring, False),
         'iso_url': (basestring, True),
         'boot_command': ([basestring], False),
         'boot_wait': (basestring, False),
@@ -302,6 +429,13 @@ class VMwareIso(PackerBuilder):
         'vnc_port_max': (int, False),
     }
 
+    def validate(self):
+        conds = [
+            'iso_checksum',
+            'iso_checksum_url',
+        ]
+        validator.mutually_exclusive(self.__class__.__name__, self.properties, conds)
+
 
 class VMwareVmx(PackerBuilder):
     """

diff --git a/tests/packerlicious/test_builder_virtualbox.py b/tests/packerlicious/test_builder_virtualbox.py
@@ -0,0 +1,35 @@
+import pytest
+
+import packerlicious.builder as builder
+
+
+class TestVirtualBoxOvfBuilder(object):
+
+    def test_required_fields_missing(self):
+        b = builder.VirtualboxOvf()
+
+        with pytest.raises(ValueError) as excinfo:
+            b.to_dict()
+        assert 'required' in str(excinfo.value)
+
+
+class TestVirtualBoxIsoBuilder(object):
+
+    def test_required_fields_missing(self):
+        b = builder.VirtualboxIso()
+
+        with pytest.raises(ValueError) as excinfo:
+            b.to_dict()
+        assert 'required' in str(excinfo.value)
+
+    def test_is_checksum_mutually_exclusive(self):
+        b = builder.VirtualboxIso(
+            iso_url="/url/to/iso",
+            iso_checksum_type=builder.VirtualboxIso.MD5,
+            iso_checksum="my_checksum",
+            iso_checksum_url="my_checksum_url",
+        )
+
+        with pytest.raises(ValueError) as excinfo:
+            b.to_dict()
+        assert 'VirtualboxIso: only one of the following can be specified: iso_checksum, iso_checksum_url' == str(excinfo.value)
diff --git a/tests/packerlicious/test_builder_vmware.py b/tests/packerlicious/test_builder_vmware.py
@@ -12,6 +12,19 @@ def test_required_fields_missing(self):
             b.to_dict()
         assert 'required' in str(excinfo.value)
 
+    def test_is_checksum_mutually_exclusive(self):
+        b = builder.VMwareIso(
+            iso_url="/url/to/iso",
+            iso_checksum_type=builder.VirtualboxIso.MD5,
+            iso_checksum="my_checksum",
+            iso_checksum_url="my_checksum_url",
+        )
+
+        with pytest.raises(ValueError) as excinfo:
+            b.to_dict()
+        assert 'VMwareIso: only one of the following can be specified: iso_checksum, iso_checksum_url' == str(
+            excinfo.value)
+
 
 class TestVMwareVmxBuilder(object):