New plugin: puppet

Install your own puppet configuration and run a manifest

plugins/puppet/__init__.py

@@ -0,0 +1,16 @@
+import tasks
+
+
+def validate_manifest(data, validator, error):
+	import os.path
+	schema_path = os.path.normpath(os.path.join(os.path.dirname(__file__), 'manifest-schema.json'))
+	validator(data, schema_path)
+
+
+def resolve_tasks(taskset, manifest):
+	taskset.add(tasks.CheckPaths)
+	taskset.add(tasks.AddPackages)
+	if 'assets' in manifest.plugins['puppet']:
+		taskset.add(tasks.CopyPuppetAssets)
+	if 'manifest' in manifest.plugins['puppet']:
+		taskset.add(tasks.ApplyPuppetManifest)

plugins/puppet/manifest-schema.json

@@ -0,0 +1,27 @@
+{
+	"$schema": "http://json-schema.org/draft-04/schema#",
+	"title": "Puppet plugin manifest",
+	"type": "object",
+	"properties": {
+		"plugins": {
+			"type": "object",
+			"properties": {
+				"puppet": {
+					"type": "object",
+					"properties": {
+						"manifest": { "$ref": "#/definitions/absolute_path" },
+						"assets": { "$ref": "#/definitions/absolute_path" }
+					},
+					"minProperties": 1,
+					"additionalProperties": false
+				}
+			}
+		}
+	},
+	"definitions": {
+		"absolute_path": {
+			"type":    "string",
+			"pattern": "^/[^\\0]+$"
+		}
+	}
+}

plugins/puppet/tasks.py

@@ -0,0 +1,93 @@
+from base import Task
+from common import phases
+from common.tasks import apt
+from common.tasks import network
+import os
+
+
+class CheckPaths(Task):
+	description = 'Checking whether manifest and assets paths exist'
+	phase = phases.preparation
+
+	@classmethod
+	def run(cls, info):
+		from common.exceptions import TaskError
+		assets = info.manifest.plugins['puppet']['assets']
+		if not os.path.exists(assets):
+			msg = 'The assets directory {assets} does not exist.'.format(assets=assets)
+			raise TaskError(msg)
+		if not os.path.isdir(assets):
+			msg = 'The assets path {assets} does not point to a directory.'.format(assets=assets)
+			raise TaskError(msg)
+
+		manifest = info.manifest.plugins['puppet']['manifest']
+		if not os.path.exists(manifest):
+			msg = 'The manifest file {manifest} does not exist.'.format(manifest=manifest)
+			raise TaskError(msg)
+		if not os.path.isfile(manifest):
+			msg = 'The manifest path {manifest} does not point to a file.'.format(manifest=manifest)
+			raise TaskError(msg)
+
+
+class AddPackages(Task):
+	description = 'Add puppet package'
+	phase = phases.preparation
+	predecessors = [apt.AddDefaultSources]
+
+	@classmethod
+	def run(cls, info):
+		info.packages.add('puppet')
+
+
+class CopyPuppetAssets(Task):
+	description = 'Copying puppet assets'
+	phase = phases.system_modification
+
+	@classmethod
+	def run(cls, info):
+		from shutil import copy
+		puppet_path = os.path.join(info.root, 'etc/puppet')
+		puppet_assets = info.manifest.plugins['puppet']['assets']
+		for abs_prefix, dirs, files in os.walk(puppet_assets):
+			prefix = os.path.normpath(os.path.relpath(abs_prefix, puppet_assets))
+			for path in dirs:
+				full_path = os.path.join(puppet_path, prefix, path)
+				if os.path.exists(full_path):
+					if os.path.isdir(full_path):
+						continue
+					else:
+						os.remove(full_path)
+				os.mkdir(full_path)
+			for path in files:
+				copy(os.path.join(abs_prefix, path),
+				     os.path.join(puppet_path, prefix, path))
+
+
+class ApplyPuppetManifest(Task):
+	description = 'Applying puppet manifest'
+	phase = phases.system_modification
+	predecessors = [CopyPuppetAssets]
+	successors = [network.RemoveHostname, network.RemoveDNSInfo]
+
+	@classmethod
+	def run(cls, info):
+		with open(os.path.join(info.root, 'etc/hostname')) as handle:
+			hostname = handle.read().strip()
+		with open(os.path.join(info.root, 'etc/hosts'), 'a') as handle:
+			handle.write('127.0.0.1\t{hostname}\n'.format(hostname=hostname))
+
+		from shutil import copy
+		pp_manifest = info.manifest.plugins['puppet']['manifest']
+		manifest_rel_dst = os.path.join('tmp', os.path.basename(pp_manifest))
+		manifest_dst = os.path.join(info.root, manifest_rel_dst)
+		copy(pp_manifest, manifest_dst)
+
+		manifest_path = os.path.join('/', manifest_rel_dst)
+		from common.tools import log_check_call
+		log_check_call(['/usr/sbin/chroot', info.root,
+		                '/usr/bin/puppet', 'apply', manifest_path])
+		os.remove(manifest_dst)
+
+		from common.tools import sed_i
+		hosts_path = os.path.join(info.root, 'etc/hosts')
+		sed_i(hosts_path, '127.0.0.1\s*{hostname}\n?'.format(hostname=hostname), '')