Set the hostname on vagrant boxes

plugins/vagrant/__init__.py

@@ -10,13 +10,16 @@ def validate_manifest(data, validator, error):
 def resolve_tasks(taskset, manifest):
 	from common.tasks import security
 	from common.tasks import loopback
+	from common.tasks import network
 	taskset.discard(security.DisableSSHPasswordAuthentication)
 	taskset.discard(loopback.MoveImage)
+	taskset.discard(network.RemoveHostname)
 
 	from common.tasks import volume
 	taskset.update([tasks.CheckBoxPath,
 	                tasks.CreateVagrantBoxDir,
 	                tasks.AddPackages,
+	                tasks.SetHostname,
 	                tasks.CreateVagrantUser,
 	                tasks.PasswordlessSudo,
 	                tasks.SetRootPassword,

plugins/vagrant/manifest-schema.json

@@ -18,6 +18,21 @@
 				}
 			},
 			"required": ["backing"]
+		},
+		"plugins": {
+			"type": "object",
+			"properties": {
+				"vagrant": {
+					"type": "object",
+					"properties": {
+						"hostname": {
+							"type": "string",
+							"pattern": "^\\S+$"
+						}
+					},
+					"required": ["hostname"]
+				}
+			}
 		}
 	}
 }

plugins/vagrant/tasks.py

@@ -48,6 +48,22 @@ class AddPackages(Task):
 		info.packages.add('nfs-client')
 
 
+class SetHostname(Task):
+	description = 'Writing hostname into the hostname file'
+	phase = phases.system_modification
+
+	@classmethod
+	def run(cls, info):
+		hostname = info.manifest.plugins['vagrant']['hostname'].format(**info.manifest_vars)
+		hostname_file_path = os.path.join(info.root, 'etc/hostname')
+		with open(hostname_file_path, 'w') as hostname_file:
+			hostname_file.write(hostname)
+
+		hosts_path = os.path.join(info.root, 'etc/hosts')
+		from common.tools import sed_i
+		sed_i(hosts_path, '^127.0.0.1\tlocalhost$', '127.0.0.1\tlocalhost\n127.0.0.1\t' + hostname)
+
+
 class CreateVagrantUser(Task):
 	description = 'Creating the vagrant user'
 	phase = phases.system_modification