summaryrefslogtreecommitdiff
path: root/scripts/build/binary_hooks
blob: 16a3e81f7daf0e7cd627f95c7681a3ec33f86553 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#!/usr/bin/python3

## live-build(7) - Live System Build Components
## Copyright (C) 2006-2013 Daniel Baumann <mail@daniel-baumann.ch>
##
## This program comes with ABSOLUTELY NO WARRANTY; for details see COPYING.
## This is free software, and you are welcome to redistribute it
## under certain conditions; see COPYING for details.


import argparse
import configparser
import glob
import os
import shutil
import subprocess
import sys


# TODO:
#   * logfile output
#   * lockfile handling
#   * use gettext for i18n

def main():
	## Parsing Arguments
	arguments = argparse.ArgumentParser(
		prog            = 'lb binary_hooks',
		usage           = '%(prog)s [arguments]',
		description     = '''live-build contains the components to build a live system from a configuration directory.
		                     The binary_hooks command executes hooks after the binary stage.''',
		epilog          = 'See \'man lb_binary_hooks\' for more information.',
		version         = 'live-build 4',
		formatter_class = argparse.ArgumentDefaultsHelpFormatter
	)

	arguments.add_argument('--verbose',     help='set verbose option',         action='store_true')

	args = arguments.parse_args()

	# --verbose
	verbose = args.verbose

	## Executing hooks

	# stagefile
	if os.path.isfile('.build/binary_hooks'):
		if verbose:
			print('I: binary_hooks already done - nothing to do')

		sys.exit(0)

	# dependencies
	if not os.path.isfile('.build/bootstrap'):
		print('E: bootstrap stage missing - aborting', file=sys.stderr)

		if verbose:
			print('I: use \'lb bootstrap\' to bootstrap system')

		sys.exit(1)

	# hooks
	if not glob.glob('config/hooks/*.hook') and not glob.glob('config/hooks/*.hook.binary'):
		if verbose:
			print ('I: no binary hooks found at config/hooks/*.hook{,.binary} - nothing to do')

		sys.exit(0)

	# bind mount configuration directory
	if verbose:
		print('I: Mounting config to binary/live-build/config')

	os.makedirs('binary/live-build/config', exist_ok=True)

	mount = subprocess.call('mount -o bind config binary/live-build/config', shell=True)
	remount = subprocess.call('mount -o remount,ro,bind binary/live-build/config', shell=True)

	# process hooks
	os.makedirs('binary/live-build', exist_ok=True)

	hooks = glob.glob('config/hooks/*.hook') + glob.glob('config/hooks/*.hook.binary')

	for hook in hooks:
		if verbose:
			print('I: Copying config/hooks/*.hook.binary to binary/live-build')

		os.link(hook, os.path.join('binary/live-build/' + os.path.basename(hook)), follow_symlinks=True)

		if verbose:
			print('I: Executing \' ' + hook + '\'')

		os.chmod(hook, 0o755)
		exec_hook = subprocess.call('cd binary && live-build/' + os.path.basename(hook), shell=True)
		os.remove('binary/live-build/' + os.path.basename(hook))

	# unmount coniguration directory
	umount = subprocess.call('umount binary/live-build/config', shell=True)

	os.rmdir('binary/live-build/config')
	os.rmdir('binary/live-build')

	## stagefile
	os.makedirs('.build', exist_ok=True)
	open('.build/binary_hooks', 'w').close()


if __name__ == '__main__':
	main()