buildtools/wafsamba/samba_wildcard.py

   1 #! /usr/bin/env python
   2
   3 # based on playground/evil in the waf svn tree
   4
   5 import os, datetime
   6 import Scripting, Utils, Options, Logs, Environment, fnmatch
   7 from Constants import *
   8 from samba_utils import *
   9
  10 def run_task(t, k):
  11     '''run a single build task'''
  12     ret = t.run()
  13     if ret:
  14         raise Utils.WafError("Failed to build %s: %u" % (k, ret))
  15
  16
  17 def run_named_build_task(cmd):
  18     '''run a named build task, matching the cmd name using fnmatch
  19     wildcards against inputs and outputs of all build tasks'''
  20     bld = fake_build_environment(info=False)
  21     found = False
  22     cwd_node = bld.root.find_dir(os.getcwd())
  23     top_node = bld.root.find_dir(bld.srcnode.abspath())
  24
  25     cmd = os.path.normpath(cmd)
  26
  27     # cope with builds of bin/*/*
  28     if os.path.islink(cmd):
  29         cmd = os_path_relpath(os.readlink(cmd), os.getcwd())
  30
  31     if cmd[0:12] == "bin/default/":
  32         cmd = cmd[12:]
  33
  34     for g in bld.task_manager.groups:
  35         for attr in ['outputs', 'inputs']:
  36             for t in g.tasks:
  37                 s = getattr(t, attr, [])
  38                 for k in s:
  39                     relpath1 = k.relpath_gen(cwd_node)
  40                     relpath2 = k.relpath_gen(top_node)
  41                     if (fnmatch.fnmatch(relpath1, cmd) or
  42                         fnmatch.fnmatch(relpath2, cmd)):
  43                         t.position = [0,0]
  44                         print(t.display())
  45                         run_task(t, k)
  46                         found = True
  47
  48
  49     if not found:
  50         raise Utils.WafError("Unable to find build target matching %s" % cmd)
  51
  52
  53 def rewrite_compile_targets():
  54     '''cope with the bin/ form of compile target'''
  55     if not Options.options.compile_targets:
  56         return
  57
  58     bld = fake_build_environment(info=False)
  59     targets = LOCAL_CACHE(bld, 'TARGET_TYPE')
  60     tlist = []
  61
  62     for t in Options.options.compile_targets.split(','):
  63         if not os.path.islink(t):
  64             tlist.append(t)
  65             continue
  66         link = os.readlink(t)
  67         list = link.split('/')
  68         for name in [list[-1], '/'.join(list[-2:])]:
  69             if name in targets:
  70                 tlist.append(name)
  71                 continue
  72     Options.options.compile_targets = ",".join(tlist)
  73
  74
  75
  76 def wildcard_main(missing_cmd_fn):
  77     '''this replaces main from Scripting, allowing us to override the
  78        behaviour for unknown commands
  79
  80        If a unknown command is found, then missing_cmd_fn() is called with
  81        the name of the requested command
  82        '''
  83     Scripting.commands = Options.arg_line[:]
  84
  85     # rewrite the compile targets to cope with the bin/xx form
  86     rewrite_compile_targets()
  87
  88     while Scripting.commands:
  89         x = Scripting.commands.pop(0)
  90
  91         ini = datetime.datetime.now()
  92         if x == 'configure':
  93             fun = Scripting.configure
  94         elif x == 'build':
  95             fun = Scripting.build
  96         else:
  97             fun = getattr(Utils.g_module, x, None)
  98
  99         # this is the new addition on top of main from Scripting.py
 100         if not fun:
 101             missing_cmd_fn(x)
 102             break
 103
 104         ctx = getattr(Utils.g_module, x + '_context', Utils.Context)()
 105
 106         if x in ['init', 'shutdown', 'dist', 'distclean', 'distcheck']:
 107             try:
 108                 fun(ctx)
 109             except TypeError:
 110                 fun()
 111         else:
 112             fun(ctx)
 113
 114         ela = ''
 115         if not Options.options.progress_bar:
 116             ela = ' (%s)' % Utils.get_elapsed_time(ini)
 117
 118         if x != 'init' and x != 'shutdown':
 119             Logs.info('%r finished successfully%s' % (x, ela))
 120
 121         if not Scripting.commands and x != 'shutdown':
 122             Scripting.commands.append('shutdown')
 123
 124
 125
 126
 127 def fake_build_environment(info=True, flush=False):
 128     """create all the tasks for the project, but do not run the build
 129     return the build context in use"""
 130     bld = getattr(Utils.g_module, 'build_context', Utils.Context)()
 131     bld = Scripting.check_configured(bld)
 132
 133     Options.commands['install'] = False
 134     Options.commands['uninstall'] = False
 135     Options.is_install = False
 136
 137     bld.is_install = 0 # False
 138
 139     try:
 140         proj = Environment.Environment(Options.lockfile)
 141     except IOError:
 142         raise Utils.WafError("Project not configured (run 'waf configure' first)")
 143
 144     bld.load_dirs(proj[SRCDIR], proj[BLDDIR])
 145     bld.load_envs()
 146
 147     if info:
 148         Logs.info("Waf: Entering directory `%s'" % bld.bldnode.abspath())
 149     bld.add_subdirs([os.path.split(Utils.g_module.root_path)[0]])
 150
 151     bld.pre_build()
 152     if flush:
 153         bld.flush()
 154     return bld
 155