mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 09:50:37 +00:00
59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
|
from bootstrapvz.base import Task
|
||
|
|
from bootstrapvz.common import phases
|
||
|
|
from bootstrapvz.common.tasks import initd
|
||
|
|
from bootstrapvz.common.tasks import packages
|
||
|
|
from bootstrapvz.common.tools import log_check_call
|
||
|
|
from bootstrapvz.common.tools import rel_path
|
||
|
|
from bootstrapvz.common.tools import sed_i
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
|
||
|
|
ASSETS_DIR = rel_path(__file__, 'assets')
|
||
|
|
|
||
|
|
|
||
|
|
class InstallGrowpart(Task):
|
||
|
|
description = 'Adding necessary packages for growpart.'
|
||
|
|
phase = phases.preparation
|
||
|
|
successors = [packages.AddManifestPackages]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def run(cls, info):
|
||
|
|
# Use the cloud-guest-utils package from jessie-backports which has
|
||
|
|
# several significant bug fixes from the mainline growpart script.
|
||
|
|
target = None
|
||
|
|
from bootstrapvz.common.releases import jessie
|
||
|
|
if info.manifest.release == jessie:
|
||
|
|
target = '{system.release}-backports'
|
||
|
|
info.packages.add('cloud-guest-utils', target)
|
||
|
|
|
||
|
|
|
||
|
|
class InstallExpandRootScripts(Task):
|
||
|
|
description = 'Installing scripts for expand-root.'
|
||
|
|
phase = phases.system_modification
|
||
|
|
successors = [initd.InstallInitScripts]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def run(cls, info):
|
||
|
|
expand_root_script = os.path.join(ASSETS_DIR, 'expand-root.sh')
|
||
|
|
expand_root_service = os.path.join(ASSETS_DIR, 'expand-root.service')
|
||
|
|
|
||
|
|
expand_root_script_dest = os.path.join(info.root, 'usr/bin/expand-root.sh')
|
||
|
|
expand_root_service_dest = os.path.join(info.root, 'lib/systemd/system/expand-root.service')
|
||
|
|
|
||
|
|
filesystem_type = info.manifest.plugins['expand_root'].get('filesystem_type')
|
||
|
|
root_device = info.manifest.plugins['expand_root'].get('root_device')
|
||
|
|
root_partition = info.manifest.plugins['expand_root'].get('root_partition')
|
||
|
|
|
||
|
|
# Copy files over
|
||
|
|
shutil.copy(expand_root_script, expand_root_script_dest)
|
||
|
|
os.chmod(expand_root_script_dest, 0750)
|
||
|
|
shutil.copy(expand_root_service, expand_root_service_dest)
|
||
|
|
|
||
|
|
# Expand out options into expand-root.sh script.
|
||
|
|
opts = '%s %s %s' % (root_device, root_partition, filesystem_type)
|
||
|
|
sed_i(expand_root_service_dest, r'^ExecStart=/usr/bin/expand-root.sh.*$',
|
||
|
|
'ExecStart=/usr/bin/expand-root.sh %s' % opts)
|
||
|
|
|
||
|
|
# Enable systemd service
|
||
|
|
log_check_call(['chroot', info.root, 'systemctl', 'enable', 'expand-root.service'])
|