register_phase_hook() {

	if [ -z "$3" ]; then

		echo "!!! register_phase_hook() called without enough parameters." >&2

		echo "!!! register_phase_hook <before|after> <phase|all> <cmd>" >&2

		return 1

	fi

	local x when phase cmd cmdargs

	when="$(echo $1 | tr 'a-z' 'A-Z')"; shift # uppercase when

	phase="$(echo $1 | tr 'A-Z' 'a-z')"; shift # lowercase phase name (to match real phase names)

	case "${when}" in

		BEFORE|AFTER)

			: # allowed

			;;

		*)

			echo "!!! register_phase_hook() called with invalid when parameter: $when" >&2

			return 1

			;;

	esac

	for x in $* ; do

		hasq $x $(eval 'echo $EBUILD_PHASE_HOOKS_'"${when}"'_'"${phase}") || \

			export EBUILD_PHASE_HOOKS_"${when}"_"${phase}"+="${x} "

	done

}

# TODO: deprecate this function? Should be easy to provide backwards

# compatibility:

# register_phase_hook before <phase> pre_<phase>

# register_phase_hook after <phase> post_<phase>

exec_ebuild_phase() {

	declare -F "$1" >/dev/null && qa_call $1

}

# Portage hooks

portage_hooks_pre_ebuild() {

	source "${HOOKS_SH_BINARY}" --do-pre-ebuild || return $?

}

portage_hooks_post_ebuild() {

	source "${HOOKS_SH_BINARY}" --do-post-ebuild || return $?

}

register_phase_hook before all portage_hooks_pre_ebuild

register_phase_hook after all portage_hooks_post_ebuild

#!/bin/bash

# Copyright 1999-2010 Gentoo Foundation

# Distributed under the terms of the GNU General Public License v2

# @MAINTAINER:

# jacobgodserv@gmail.com

# @BLURB: Executes hooks in the current directory.

# @DESCRIPTION:

# Part of the portage hooks system, this script is responsible for executing

# hooks within a prepared environment

# Only run hooks if it's requested in $FEATURES

if ! (source "${PORTAGE_BIN_PATH}/isolated-functions.sh" && hasq hooks $FEATURES) ; then

	return

fi

# TODO: unit testing does not cover this portion of hooks.sh

# This code is put here so it's easier to do one-liners elsewhere.

# This section is meant to be run by ebuild.sh

if [[ "$1" == "--do-pre-ebuild" || "$1" == "--do-post-ebuild" ]]; then

	if [[ "${EBUILD_PHASE}" == "" ]]; then

		# an in-between-phases moment; useless to hooks

		return

	fi

	oldwd="$(pwd)"

	if [[ "$1" == "--do-pre-ebuild" ]]; then

		hooks_dir="${PORTAGE_CONFIGROOT}/${HOOKS_PATH}/pre-ebuild.d"

	else

		hooks_dir="${PORTAGE_CONFIGROOT}/${HOOKS_PATH}/post-ebuild.d"

	fi

	[ -d "${hooks_dir}" ] && cd "${hooks_dir}"

	exit_code="$?"

	if [[ "${exit_code}" != "0" ]]; then

		# mimicks behavior in hooks.py

		# TODO: --verbose detection?

		:

		#debug-print "This hook path could not be found; ignored: ${hooks_dir}"

	else

		# Execute the hooks

		source "${HOOKS_SH_BINARY}" --action "${EBUILD_PHASE}" --target "${EBUILD}"

		exit_code="$?"

		if [[ "${exit_code}" != "0" ]]; then

			# mimicks behavior in hooks.py

			die "Hook directory ${HOOKS_PATH}/pre-ebuild.d failed with exit code ${exit_code}"

		fi

	fi

	cd "${oldwd}" || die "Could not return to the old ebuild directory after pre-ebuild hooks: ${oldwd}"

	return

fi

# Local variables listed here.

# Using the local keyword makes no difference since this script is being sourced

# so we'll have to unset them manually later. Be sure to keep the local_vars

# array up-to-date.

hook_files=( * )

hook_args=( "$@" )

hook_verbosity="0"

hook_local_vars=( "hook_files" "hook_args" "hook_verbosity" ) # variables unset for hooks

for (( i = 0 ; i < ${#hook_args[@]} ; i++ )); do

	if [[ "${hook_args[$i]}" == "--verbose" ]]; then

		hook_verbosity="1"

	fi

done

for (( i = 0 ; i < ${#hook_files[@]} ; i++ )); do

	hook="${hook_files[$i]}"

	if [[ ! -e "${hook}" ]]; then

		continue

	elif [[ ! -f "${hook}" ]]; then

		[ "${hook_verbosity}" -gt 0 ] && ewarn "Only files are recognized in a hook directory: ${hook}"

		continue

	fi

	[ "${hook_verbosity}" -gt 0 ] && einfo "Executing hook ${hook}..."

	# We use eval so the hook_args gets expanded before it is unset

	( eval unset "${hook_local_vars[@]}" '&&' source "${hook}" "${hook_args[@]}" )

	exit_code="$?"

	if [[ "${exit_code}" != "0" ]]; then

		eerror "Hook $(pwd)/${hook} returned with exit code ${exit_code}"

		exit "${exit_code}"

	fi

done

unset "${hook_local_vars[@]}"

&config_hooks;

<chapter id='config-hooks'>

	<title id="config-hooks.title">Hooks Configuration</title>

	<sect1 id='config-hooks-execution'>

		<title id="config-hooks-execution.title">Hooks Execution</title>

		<para>Hooks are only executed if <quote>hooks</quote> is set in

		FEATURES.</para>

		<para>

		If a hook directory exists, the bash scripts within each one

		wil either be executed before or after that particular phase, in

		alphabetical order. Each one will receive the environment of an

		ebuild, so they are capable of inherit, einfo, and other common

		commands (if you find them useful). For non-ebuild hooks, avoid

		commands that may trigger changes in the filesystem!

		</para>

		<para>

		Ebuild hooks are executed within ebuild.sh, so they receive the

		same sandbox limitations as ebuilds.

		</para>

		<para>

		A hook script is expected to understand the following usage:

		<cmdsynopsis>

			<command>/bin/bash <replaceable>...</replaceable></command><sbr/>

			<arg>--opt <replaceable>portage arguments, always translated to long form, given by user at the prompt, such as "--verbose" or "--newuse"</replaceable></arg><sbr/>

			<arg>--action <replaceable>a single action being performed by portage, such as "depclean", "sync", or an ebuild phase</replaceable></arg><sbr/>

			<arg>--target <replaceable>the thing to perform the action with or on</replaceable></arg>

		</cmdsynopsis>

		</para>

		<para>

		Some hook types have slightly different usage. See <quote>

		<link linkend='config-hooks-locations' endterm="config-hooks-locations.title"/></quote> for more

		information.

		</para>

	</sect1>

	<sect1 id='config-hooks-locations'>

		<title id="config-hooks-locations.title">Hooks Locations</title>		

		<para>

		The following hook directories are supported. Each directory

		corresponds to a specific type, such as <quote>ebuild</quote> or

		<quote>run</quote>. The standard hook script usage applies given

		in <link linkend='config-hooks-execution' endterm="config-hooks-execution.title"/>,

		except wherever	described differently below.

		</para>

		<itemizedlist>

			<listitem><para><filename>/etc/portage/hooks/pre-ebuild.d/</filename> - executed before every ebuild phase execution, within ebuild.sh itself. Never receives --opt, and --target is set to the full path of the ebuild.</para></listitem>

			<listitem><para><filename>/etc/portage/hooks/post-ebuild.d/</filename> - executed after every ebuild phase execution. Never receives --opt, and --target is set to the full path of the ebuild.</para></listitem>

			<listitem><para><filename>/etc/portage/hooks/pre-run.d/</filename> - executed before portage considers most things, including proper permissions and validity of parsed arguments.</para></listitem>

			<listitem><para><filename>/etc/portage/hooks/post-run.d/</filename> - executed after portage is done. It should run regardless of any errors or signals sent, but this cannot be guaranteed for certain scenarios (such as when the KILL signal is received). No information is available concerning the reason portage is exiting. This is a limitation of python itself.</para></listitem>

			<listitem><para><filename>/etc/portage/hooks/pre-sync.d/</filename> - executed before portage synchronizes the portage tree.</para></listitem>

			<listitem><para><filename>/etc/portage/hooks/post-sync.d/</filename> - executed after portage has <emphasis>successfully</emphasis> synchronized the portage tree. If you want to catch a sync failure, use post-run.</para></listitem>

		</itemizedlist>

	</sect1>

	<sect1 id='config-hooks-skeleton-hook'>

		<title id="config-hooks-skeleton-hook.title">Skeleton Hook</title>

		<para>

		Most hooks will parse the options at the beginning and look for

		specific things. This skeleton hook provides that functionality

		to get you started.

		</para>

		<para>

		It's highly recommended that --verbose, --debug, and --quiet be

		utilized for suppressing or adding to <quote>regular</quote>

		output. The following skeleton hook already has example code in

		place to handle these flags.

		</para>

		<programlisting>

		#!/bin/bash

		verbose_redirect="/dev/null"

		debug_redirect="/dev/null"

		while [[ "$1" != "" ]]; do

			if [[ "$1" == "--opt" ]]; then

				if [[ "$2" == "--verbose" ]]; then

					verbose_redirect="/dev/tty"

				fi

				if [[ "$2" == "--debug" ]]; then

					debug_redirect="/dev/tty"

				fi

				if [[ "$2" == "--quiet" ]]; then

					verbose_redirect="/dev/null"

					debug_redirect="/dev/null"

				fi

			elif [[ "$1" == "--action" ]]; then

				: # do nothing

			elif [[ "$1" == "--target" ]]; then

				: # do nothing

			else

				ewarn "Unknown hook option: $1 $2" &> "${verbose_redirect}"

			fi

			shift 2

		done

		einfo "This is an example hook." &> "${verbose_redirect}"

		einfo "This is debug output." &> "${debug_redirect}"

		</programlisting>

	</sect1>

</chapter>

	<!ENTITY config_hooks SYSTEM "config/hooks.docbook">

.BR /etc/portage/hooks/

portage pre/post hooks

.TP

.BR /etc/portage/hooks/

.RS

In this directory, portage hooks are executed before each ebuild phase,

before and after synchronization, and before and after portage runs

themselves. Please see the DocBook documentation for detailed

information.

.RE

.TP

from portage.hooks import HookDirectory

	HookDirectory(phase='pre-sync', settings=settings, myopts=myopts, myaction=myaction).execute()

	HookDirectory(phase='post-sync', settings=settings, myopts=myopts, myaction=myaction).execute()

import portage.hooks

import portage.process

	# Portage configured; let's let hooks run before we do anything more

	portage.hooks.HookDirectory(phase='pre-run', settings=settings, myopts=myopts, myaction=myaction, mytargets=myfiles).execute()

	settings, trees, mtimedb = load_emerge_config() # once more, since pre-run might've done something

	# Have post-run hooks executed whenever portage quits

	portage.process.atexit_register(portage.hooks.HookDirectory(phase='post-run', settings=settings, myopts=myopts, myaction=myaction, mytargets=myfiles).execute)

HOOKS_PATH               = USER_CONFIG_PATH + "/hooks"

HOOKS_SH_BINARY          = PORTAGE_BIN_PATH + "/hooks.sh"

# Copyright 1998-2010 Gentoo Foundation

# Distributed under the terms of the GNU General Public License v2

# $Id$

from portage.const import BASH_BINARY, HOOKS_PATH, HOOKS_SH_BINARY, PORTAGE_BIN_PATH

from portage import os

from portage import check_config_instance

from portage import normalize_path

from portage.exception import PortageException

from portage.exception import InvalidLocation

from portage.output import EOutput

from process import spawn

from shutil import rmtree

from tempfile import mkdtemp

class HookDirectory(object):

	def __init__ (self, phase, settings, myopts=None, myaction=None, mytargets=None):

		self.myopts = myopts

		self.myaction = myaction

		self.mytargets = mytargets

		check_config_instance(settings)

		self.settings = settings

		self.path = os.path.join(settings["PORTAGE_CONFIGROOT"], HOOKS_PATH, phase + '.d')

		self.output = EOutput()

	def execute (self, path=None):

		if "hooks" not in self.settings['FEATURES']:

			return

		if not path:

			path = self.path

		path = normalize_path(path)

		if not os.path.exists(path):

			if self.myopts and "--debug" in self.myopts:

				# behavior mimicked by hook.sh

				self.output.ewarn('This hook path could not be found; ignored: ' + path)

			return

		if os.path.isdir(path):

			command=[HOOKS_SH_BINARY]

			if self.myopts:

				for myopt in self.myopts:

					command.extend(['--opt', myopt])

			if self.myaction:

				command.extend(['--action', self.myaction])

			if self.mytargets:

				for mytarget in self.mytargets:

					command.extend(['--target', mytarget])

			command=[BASH_BINARY, '-c', 'cd "'+path+'" && source "' + PORTAGE_BIN_PATH + '/isolated-functions.sh" && source ' + ' '.join(command)]

			if self.myopts and "--verbose" in self.myopts:

				self.output.einfo('Executing hooks directory "' + self.path + '"...')

			code = spawn(mycommand=command, env=self.settings.environ())

			if code: # if failure

				# behavior mimicked by hook.sh

				raise PortageException('!!! Hook directory %s failed with exit code %s' % (self.path, code))

		else:

			raise InvalidLocation('This hook path ought to be a directory: ' + path)

	def merge_to_env (self, existingenv, path):

		path = normalize_path(path)

		if not os.path.isdir(path):

			raise InvalidLocation('This environment path is not a directory: ' + path)

		for parent, dirs, files in os.walk(path):

			for varname in files:

				file = open(os.path.join(path, varname), 'r')

				# read the file, remove the very last newline, and make the escaped double-quotes just plain double-quotes (since only bash needs them to be escaped, not python)

				vardata = file.read()[:-1].replace('\"','"').strip('"')

				existingenv[varname] = vardata

				existingenv.backup_changes(varname)

		return existingenv

			self["HOOKS_PATH"] = HOOKS_PATH

			self.backup_changes("HOOKS_PATH")

			self["HOOKS_SH_BINARY"] = HOOKS_SH_BINARY

			self.backup_changes("HOOKS_SH_BINARY")

	'portage.package.ebuild.const:HOOKS_SH_BINARY',

from portage.hooks import HookDirectory

# tests/portage/hooks/__init__.py -- Portage Unit Test functionality

# Copyright 2010 Gentoo Foundation

# Distributed under the terms of the GNU General Public License v2

# $Id$

# test_HookDirectory.py -- Portage Unit Testing Functionality

# Copyright 2010 Gentoo Foundation

# Distributed under the terms of the GNU General Public License v2

# $Id$

from portage import os

from portage.hooks import HookDirectory

from portage.package.ebuild.config import config

from portage.tests import TestCase

from tempfile import mkdtemp

from shutil import rmtree

class HookDirectoryTestCase(TestCase):

	def testHookDirectory(self):

		"""

		Tests to be sure a hook loads and reads the right settings

		Based on test_PackageKeywordsFile.py

		"""

		self.tmp_dir_path = self.BuildTmp('/etc/portage/hooks/test.d')

		try:

			settings = config()

			settings["PORTAGE_CONFIGROOT"] = self.tmp_dir_path

			settings["FEATURES"] += " hooks"

			hooks = HookDirectory(phase='test', settings=settings)

			hooks.execute()

			self.assert_(settings["hookonlytest"] == "")

		finally:

			rmtree(self.tmp_dir_path)

	def BuildTmp(self, tmp_subdir):

		tmp_dir = mkdtemp()

		hooks_dir = tmp_dir + '/' + tmp_subdir

		os.makedirs(hooks_dir)

		f = open(hooks_dir+'/1-testhook', 'w')

		f.write('#!/bin/bash\n')

		f.write('export hookonlytest="portage cannot see me!"\n')

		f.write('exit 0\n')

		f.close()

		f = open(hooks_dir+'/2-testhook', 'w')

		f.write('#!/bin/bash\n')

		f.write('if [[ "${hookonlytest}" != "" ]]; then echo "Unexpected hookonlytest value: ${hookonlytest}"; exit 1; fi\n');

		f.write('exit 0\n')

		f.close()

		return tmp_dir