components/binary-hooks

   1 #!/usr/bin/python3
   2
   3 ## live-build(7) - Live System Build Components
   4 ## Copyright (C) 2006-2014 Daniel Baumann <mail@daniel-baumann.ch>
   5 ##
   6 ## This program comes with ABSOLUTELY NO WARRANTY; for details see COPYING.
   7 ## This is free software, and you are welcome to redistribute it
   8 ## under certain conditions; see COPYING for details.
   9
  10
  11 import argparse
  12 import configparser
  13 import glob
  14 import os
  15 import shutil
  16 import subprocess
  17 import sys
  18
  19
  20 # TODO:
  21 #   * logfile output
  22 #   * lockfile handling
  23 #   * use gettext for i18n
  24
  25 def main():
  26         ## Parsing Arguments
  27         arguments = argparse.ArgumentParser(
  28                 prog            = 'lb binary-hooks',
  29                 usage           = '%(prog)s [arguments]',
  30                 description     = '''live-build contains the components to build a live system from a configuration directory.
  31                                      The binary-hooks command executes hook files after the binary stage.''',
  32                 epilog          = 'See \'man lb-binary-hooks\' for more information.',
  33                 formatter_class = argparse.ArgumentDefaultsHelpFormatter
  34         )
  35
  36         arguments.add_argument('--version',     help='show program\'s version number and exit', action='version', version='live-build 4')
  37         arguments.add_argument('--verbose',     help='set verbose option',                      action='store_true')
  38
  39         args = arguments.parse_args()
  40
  41         # --verbose
  42         verbose = args.verbose
  43
  44         ## Calling binary hooks
  45
  46         # stagefile
  47         if os.path.isfile('.build/binary-hooks'):
  48                 if verbose:
  49                         print('I: binary-hooks already done - nothing to do')
  50
  51                 sys.exit(0)
  52
  53         # dependencies
  54         if not os.path.isfile('.build/bootstrap'):
  55                 print('E: bootstrap stage missing - aborting', file=sys.stderr)
  56
  57                 if verbose:
  58                         print('I: use \'lb bootstrap\' to bootstrap system')
  59
  60                 sys.exit(1)
  61
  62         # hooks
  63         if not glob.glob('config/hooks/*.hook') and not glob.glob('config/hooks/*.hook.binary'):
  64                 if verbose:
  65                         print ('I: no binary hooks found at config/hooks/*.hook{,.binary} - nothing to do')
  66
  67                 sys.exit(0)
  68
  69         # bind mount configuration directory
  70         if verbose:
  71                 print('I: Mounting config to binary/live-build/config')
  72
  73         os.makedirs('binary/live-build/config', exist_ok=True)
  74
  75         mount = subprocess.check_call('mount -o bind config binary/live-build/config', shell=True)
  76         remount = subprocess.check_call('mount -o remount,ro,bind binary/live-build/config', shell=True)
  77
  78         # process hooks
  79         os.makedirs('binary/live-build', exist_ok=True)
  80
  81         hooks = glob.glob('config/hooks/*.hook') + glob.glob('config/hooks/*.hook.binary')
  82
  83         for hook in sorted(hooks):
  84                 if verbose:
  85                         print('I: Copying config/hooks/*.hook.binary to binary/live-build')
  86
  87                 shutil.copy(hook, os.path.join('binary/live-build/' + os.path.basename(hook)), follow_symlinks=True)
  88
  89                 if verbose:
  90                         print('I: Executing \' ' + hook + '\'')
  91
  92                 os.chmod(hook, 0o755)
  93                 exec_hook = subprocess.check_call('cd binary && live-build/' + os.path.basename(hook), shell=True)
  94                 os.remove('binary/live-build/' + os.path.basename(hook))
  95
  96         # unmount coniguration directory
  97         umount = subprocess.check_call('umount binary/live-build/config', shell=True)
  98
  99         os.rmdir('binary/live-build/config')
 100         os.rmdir('binary/live-build')
 101
 102         ## stagefile
 103         os.makedirs('.build', exist_ok=True)
 104         open('.build/binary-hooks', 'w').close()
 105
 106
 107 if __name__ == '__main__':
 108         main()