samba-mirror/buildtools/wafsamba/samba_wildcard.py

#! /usr/bin/env python

# based on playground/evil in the waf svn tree

import os, datetime
import Scripting, Utils, Options, Logs, Environment, fnmatch
from Constants import *

def run_task(t, k):
	'''run a single build task'''
	ret = t.run()
	if ret:
		raise Utils.WafError("Failed to build %s: %u" % (k, ret))


def run_named_build_task(cmd):
	'''run a named build task, matching the cmd name using fnmatch
	wildcards against inputs and outputs of all build tasks'''
	bld = fake_build_environment()
	found = False
	cwd_node = bld.root.find_dir(os.getcwd())
	cmd = os.path.normpath(cmd)
	for g in bld.task_manager.groups:
		for attr in ['outputs', 'inputs']:
			for t in g.tasks:
				s = getattr(t, attr, [])
				for k in s:
					relpath = k.relpath_gen(cwd_node)
					if fnmatch.fnmatch(relpath, cmd):
						t.position= [0,0]
						print(t.display())
						run_task(t, k)
						found = True


	if not found:
		raise Utils.WafError("Unable to find build target matching %s" % cmd)



def wildcard_main(missing_cmd_fn):
	'''this replaces main from Scripting, allowing us to override the
	   behaviour for unknown commands

	   If a unknown command is found, then missing_cmd_fn() is called with
	   the name of the requested command
	   '''
	Scripting.commands = Options.arg_line[:]

	while Scripting.commands:
		x = Scripting.commands.pop(0)

		ini = datetime.datetime.now()
		if x == 'configure':
			fun = Scripting.configure
		elif x == 'build':
			fun = Scripting.build
		else:
			fun = getattr(Utils.g_module, x, None)

		# this is the new addition on top of main from Scripting.py
		if not fun:
			missing_cmd_fn(x)
			break

		ctx = getattr(Utils.g_module, x + '_context', Utils.Context)()

		if x in ['init', 'shutdown', 'dist', 'distclean', 'distcheck']:
			try:
				fun(ctx)
			except TypeError:
				fun()
		else:
			fun(ctx)

		ela = ''
		if not Options.options.progress_bar:
			ela = ' (%s)' % Utils.get_elapsed_time(ini)

		if x != 'init' and x != 'shutdown':
			Logs.info('%r finished successfully%s' % (x, ela))

		if not Scripting.commands and x != 'shutdown':
			Scripting.commands.append('shutdown')




def fake_build_environment():
	"""create all the tasks for the project, but do not run the build
	return the build context in use"""
	bld = getattr(Utils.g_module, 'build_context', Utils.Context)()
	bld = Scripting.check_configured(bld)

	Options.commands['install'] = False
	Options.commands['uninstall'] = False
	Options.is_install = False

	bld.is_install = 0 # False

	try:
		proj = Environment.Environment(Options.lockfile)
	except IOError:
		raise Utils.WafError("Project not configured (run 'waf configure' first)")

	bld.load_dirs(proj[SRCDIR], proj[BLDDIR])
	bld.load_envs()

	Logs.info("Waf: Entering directory `%s'" % bld.bldnode.abspath())
	bld.add_subdirs([os.path.split(Utils.g_module.root_path)[0]])

	bld.pre_build()
	bld.flush()
	return bld