#!/usr/bin/php -Cq
<?
// Written by Chris Chabot
// Published under the GNU GPL for licence
// You know what it is and where to find it..

function usage()
{
	global $argv;
	
	echo "Usage: {$argv[0]} [-d] [-r=path] package [package package]\n\n".
	     "  -r=path (optional):  Use this for RPM's --root\n".
		 "  -d (optional):       Include devel package(s)\n".
	     "  package              Name of installed package(s)\n".
		 "\n";
	exit(1);
}

function get_devel_list()
{
	global $devels, $root;
	$exec   = "rpm -qa $root --qf \"%{NAME}\n\" | grep devel";
	$output = `$exec`;
	$devels = explode("\n",$output);
}

function get_sizes($package)
{
	global $size_installed, $size_packages, $root;
	$exec     = "rpm -q $root --qf \"%{SIZE} %{SIGSIZE}\" ".escapeshellarg($package);
	$output   = `$exec`;
	$sizes    = explode(' ',$output);
	if (isset($sizes[0]) && is_numeric($sizes[0])) {
		$size_installed += (int) $sizes[0];
	}
	if (isset($sizes[1]) && is_numeric($sizes[1])) {
		$size_packages += (int) $sizes[1];
		echo " (".ceil($sizes[1] / 1024)." Kb)";
	}
}

function what_provides($p_requires)
{
	global $files_done, $requires, $root;
	$packages = array();
	reset($p_requires);
	while (list($require,) = each($p_requires)) {
		// Check if require has already been done
		if (!isset($files_done[$require])) {
			$exec     = "rpm -q $root --whatprovides \"$require\" --qf \"%{NAME}\"";
			$output   = `$exec`;
			if (strpos($output,'no package') === FALSE) {
					$output               = trim($output);
					$packages[$output]    = $output;
					$files_done[$require] = $output;
			} elseif (isset($files_done[$require])) {
				$packages[$output] = $files_done[$require];
			}  else {
				// echo "No package provides: \"$require\"\n";
			}
		}
	}
	return $packages;
}

function package_requires($package)
{
	global $root;
	$exec   = "rpm -q $root --requires ".escapeshellarg($package);
	$output = `$exec`;
	$output = explode("\n",$output);
	unset($output[count($output)-1]);
	$list   = array();
	while (list(,$require) = each($output)) {
		if (($pos = strpos($require,' ')) !== FALSE) {
			$require = trim(substr($require,0,$pos));
		}
		// Skip internal rpmlib tags which are not 'provided'
		if (!isset($list[$require]) && strpos($require,'rpmlib') === false && strpos($require,'perl') === false) {
			$list[$require] = TRUE;
		}
	}
	return $list;
}

function find_package_require($package)
{
	global $requires, $level, $devel, $devels, $packages;
	// Don't re-require self
	$packages++;
	$requires[$package] = $package;
	echo str_repeat('  ',$level).$package;
	get_sizes($package);
	echo "\n";
	$p_requires = package_requires($package);
	if (count($p_requires)) {
		$pkgs = what_provides($p_requires);
		if (count($pkgs)) {
			$level++;
			reset($pkgs);
			while (list(,$pkg) = each($pkgs)) {
				// Skip dep's we've already done
				if (!isset($requires[$pkg])) {
					$requires[$pkg] = $pkg;
					find_package_require($pkg);
				}
			}
			$level--;
		}
	}
	if ($devel) {
		reset($devels);
		while (list(,$pkg) = each($devels)) {
			if ($pkg == $package."-devel") {
				find_package_require($pkg);
				break;
			}
		}
	}
}

echo "\nDep Tree 0.1 by Chris Chabot.\nThis program may be freely redistributed under the terms of the GNU GPL.\n\n";
// Program initialisation
$requires       = array();
$files_done     = array();
$devels         = array();
$packages       = 0;
$level          = 0;
$max_level      = 9999;
$size_installed = 0;
$size_packages  = 0;
$devel          = false;
$root           = '';

if ($argc < 2) {
	usage();
} else {
	for ($i = 1 ; $i < $argc ; $i++) {
		if ($argv[$i] == '-d') {
			$devel = TRUE;
			get_devel_list();
		} elseif (substr($argv[$i],0,2) == '-r') {
			$root = substr($argv[$i],3);
			if (!file_exists($root)) {
				die("Invalid RPM root specified, $root doesn't exist. Aborting\n");
			}
			$root = " --root=$root ";
		}
	}
	for ($i = 1 ; $i < $argc ; $i++) {
		if (substr($argv[$i],0,1) != '-') {
			// If package was already evaluted as dependencie, don't re-process it
			if (!isset($requires[$argv[$i]])) {
				find_package_require($argv[$i]);
			}
		}
	}
}
echo "\n\n";
echo "Number of packages: $packages\n";
echo "Total size installed: ".ceil(($size_installed / 1024) / 1024)." Mb\n";
echo "Total packaged size:  ".ceil(($size_packages / 1024) / 1024)." Mb\n";
echo "\n";

?>