import tasks.packages
import tasks.boot
import tasks.image
from bootstrapvz.common.tasks import volume
from bootstrapvz.common.tasks import loopback
from bootstrapvz.common.tasks import partitioning
from bootstrapvz.common.tasks import filesystem
from bootstrapvz.common.tasks import bootstrap
from bootstrapvz.common.tasks import security
from bootstrapvz.common.tasks import network
from bootstrapvz.common.tasks import initd
from bootstrapvz.common.tasks import cleanup
from bootstrapvz.common.tasks import workspace


def initialize():
	pass


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)

	if data['volume']['partitions']['type'] == 'none' and data['system']['bootloader'] != 'extlinux':
			error('Only extlinux can boot from unpartitioned disks', ['system', 'bootloader'])


def resolve_tasks(taskset, manifest):
	from bootstrapvz.common import task_groups
	taskset.update(task_groups.base_set)
	taskset.update(task_groups.volume_set)
	taskset.update(task_groups.mounting_set)
	taskset.update(task_groups.get_apt_set(manifest))
	taskset.update(task_groups.locale_set)

	taskset.update(task_groups.bootloader_set.get(manifest.system['bootloader']))

	if manifest.volume['partitions']['type'] != 'none':
		taskset.update(task_groups.partitioning_set)

	if manifest.system.get('hostname', False):
		taskset.add(network.SetHostname)
	else:
		taskset.add(network.RemoveHostname)

	taskset.update([tasks.packages.DefaultPackages,
	                 loopback.Create,
	                 security.EnableShadowConfig,
	                 network.RemoveDNSInfo,
	                 network.ConfigureNetworkIF,
	                 initd.AddSSHKeyGeneration,
	                 initd.InstallInitScripts,
	                 tasks.packages.Waagent,
	                 tasks.boot.ConfigureGrub,
	                 cleanup.ClearMOTD,
	                 cleanup.CleanTMP,
	                 tasks.image.ConvertToVhd,
	                 ])

	if manifest.bootstrapper.get('tarball', False):
		taskset.add(bootstrap.MakeTarball)

	taskset.update(task_groups.get_fs_specific_set(manifest.volume['partitions']))

	if 'boot' in manifest.volume['partitions']:
		taskset.update(task_groups.boot_partition_set)


def resolve_rollback_tasks(taskset, manifest, counter_task):
	counter_task(loopback.Create, volume.Delete)
	counter_task(filesystem.CreateMountDir, filesystem.DeleteMountDir)
	counter_task(partitioning.MapPartitions, partitioning.UnmapPartitions)
	counter_task(filesystem.MountRoot, filesystem.UnmountRoot)
	counter_task(volume.Attach, volume.Detach)
	counter_task(workspace.CreateWorkspace, workspace.DeleteWorkspace)