#!/usr/bin/python # Copyright (c) 2015, cPanel, Inc. # All rights reserved. # # Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: # # 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. # # 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the # documentation and/or other materials provided with the distribution. # # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, # INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE # DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR # SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, # WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. import os, glob, re, tempfile from yum.plugins import PluginYumExit, TYPE_CORE, TYPE_INTERACTIVE requires_api_version = '2.3' plugin_type = (TYPE_CORE, TYPE_INTERACTIVE) def_base_dir = '/etc/yum/universal-hooks'; # conduit object info http://sourcecodebrowser.com/yum/2.4.0/namespaceyum_1_1plugins.html # TODO: help/docs the way yum plugins do them def _run_dir(dir, conduit, args = ''): dir.rstrip('*'); dir.rstrip('/'); if not os.path.isdir(dir): return None # TODO/YAGNI?: if yum called w/ --quiet: hide output from system() && do not call conduit.info() # TODO/YAGNI?: if yum called w/ --verbose also output pre/post "running $cmd" region markers # TODO: under dry run do nto run scripts just note that they would have been for script in sorted(glob.glob(dir + "/*")): if (os.access(script, os.X_OK)): # TODO/YAGNI?: if exit is ??? raise PluginYumExit("!!!! " + script + " said it was time to stop"); if (len(args)): exit = os.system(script + ' ' + args) if(exit != 0): conduit.info(2, "!!!! \"" + script + ' ' + args + "\" did not exit cleanly: " + str(exit)) else: exit = os.system(script) if(exit != 0): conduit.info(2, "!!!! " + script + " did not exit cleanly: " + str(exit)) else: conduit.info(2, "!!!! " + script + ' is not executable') def _run_pkg_dirs(base_dir, conduit, slot): ts = conduit.getTsInfo() # setup __WILDCARD__ data for the slot wc_slot_dir = base_dir + "/multi_pkgs/" + slot wildcard_list = {}; for path in glob.glob(wc_slot_dir + "/*"): if os.path.isdir(path): path = os.path.basename(os.path.normpath(path)) regx = path; regx = regx.replace("__WILDCARD__",".*") regx = re.compile("^" + regx + "$") wildcard_list[path] = regx wildcard_to_run = {}; # Get a temp file for writing package names to pkgs_file_path = tempfile.NamedTemporaryFile() members_seen = {} members = ts.getMembers() for member in list(set(members)): if members_seen.has_key(member.name): continue members_seen[member.name] = 1 # TODO/YAGNI?: set state to a normalized 'not_installed' 'updatable' 'installed' and pass as third arg to _run_dir() # This is helpful so scripts don't have to decifer the meaning of obtuse values. # It is also very tricky as the various state values (member.current_state, member.output_state, member.po.state, member.ts_state) are crazy, e.g.: # Doing a reinstall member.current_state was 70 which means not installed per http://yum.baseurl.org/api/yum-3.2.26/yum.constants-module.html (which can be brought in via 'from yum.constants import *'). pkg = member.name pkgs_file_path.write(pkg + "\n") pkgs_file_path.flush() _run_dir(base_dir + "/pkgs/" + pkg + "/" + slot, conduit) # note any __WILDCARD__ that need to run for pkg for wc in wildcard_list: if wildcard_list[wc].search(pkg): wildcard_to_run[wc] = 1 # call _run_dir on any __WILDCARD__ paths that match the pkg for wc_dir in wildcard_to_run: _run_dir(wc_slot_dir + "/" + wc_dir, conduit, "--pkg_list=" + pkgs_file_path.name ) # Close our packages file handle pkgs_file_path.close() def config_hook(conduit): """ Called first as plugins are initialised. Plugins that need to extend Yum's configuration files or command line options should do so during this slot. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/config", conduit) def postconfig_hook(conduit): """ Called immediately after Yum's config object is initialised. Useful for extending variables or modifying items in the config, for example the $ variables that are used in repo configuration. Note: Only available in yum 3.1.7 or later """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/postconfig", conduit) def init_hook(conduit): """ Called early in Yum's initialisation. May be used for general plugin related initialisation. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/init", conduit) def predownload_hook(conduit): """ Called just before Yum starts downloads of packages. Plugins may access information about the packages to be downloaded here. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/predownload", conduit) def postdownload_hook(conduit): """ Called just after Yum finishes package downloads. Plugins may access error information about the packages just downloaded. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/postdownload", conduit) def prereposetup_hook(conduit): """ Called just before Yum initialises its repository information. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/prereposetup", conduit) def postreposetup_hook(conduit): """ Called just after Yum initialises its repository information. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/postreposetup", conduit); def exclude_hook(conduit): """ Called after package inclusion and exclusions are processed. Plugins may modify package exclusions here. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/exclude", conduit); def preresolve_hook(conduit): """ Called before Yum begins package resolution. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/preresolve", conduit); def postresolve_hook(conduit): """ Called just after Yum finishes package resolution. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/postresolve", conduit); def pretrans_hook(conduit): """ Called before Yum begins the RPM update transation. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_pkg_dirs(base_dir, conduit, 'pretrans'); _run_dir(base_dir + "/pretrans", conduit); def posttrans_hook(conduit): """ Called just after Yum has finished the RPM update transation. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_pkg_dirs(base_dir, conduit, 'posttrans') _run_dir(base_dir + "/posttrans", conduit) def close_hook(conduit): """ Called as Yum is performing a normal exit. Plugins may wish to perform cleanup functions here. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/close", conduit); def clean_hook(conduit): """ Called during Yum's cleanup. This slot will be executed when Yum is run with the parameters 'clean all' or 'clean plugins'. """ base_dir = conduit.confString('main','base_dir',def_base_dir) _run_dir(base_dir + "/clean", conduit);