from base import Task
from common import phases
from plugins.packages.tasks import InstallRemotePackages


class SetUsername(Task):
	description = 'Setting username in cloud.cfg'
	phase = phases.system_modification
	predecessors = [InstallRemotePackages]

	def run(self, info):
		from common.tools import sed_i
		import os.path
		cloud_cfg = os.path.join(info.root, 'etc/cloud/cloud.cfg')
		username = info.manifest.plugins['cloud_init']['username']
		search = '^     name: debian$'
		replace = ('     name: {username}\n'
		           '     sudo: ALL=(ALL) NOPASSWD:ALL\n'
		           '     shell: /bin/bash').format(username=username)
		sed_i(cloud_cfg, search, replace)