[apt.git] / cmdline / cacheset.cc

// -*- mode: cpp; mode: fold -*-
// Description								/*{{{*/
/* ######################################################################

   Simple wrapper around a std::set to provide a similar interface to
   a set of cache structures as to the complete set of all structures
   in the pkgCache. Currently only Package is supported.

   ##################################################################### */
									/*}}}*/
// Include Files							/*{{{*/
#include <apt-pkg/aptconfiguration.h>
#include <apt-pkg/error.h>
#include <apt-pkg/strutl.h>
#include <apt-pkg/versionmatch.h>

#include <apti18n.h>

#include "cacheset.h"

#include <vector>

#include <regex.h>
									/*}}}*/
namespace APT {
// FromTask - Return all packages in the cache from a specific task	/*{{{*/
PackageSet PackageSet::FromTask(pkgCacheFile &Cache, std::string pattern, CacheSetHelper &helper) {
	PackageSet pkgset;
	if (Cache.BuildCaches() == false || Cache.BuildDepCache() == false)
		return pkgset;

	size_t archfound = pattern.find_last_of(':');
	std::string arch = "native";
	if (archfound != std::string::npos) {
		arch = pattern.substr(archfound+1);
		pattern.erase(archfound);
	}

	if (pattern[pattern.length() -1] != '^')
		return pkgset;
	pattern.erase(pattern.length()-1);

	// get the records
	pkgRecords Recs(Cache);

	// build regexp for the task
	regex_t Pattern;
	char S[300];
	snprintf(S, sizeof(S), "^Task:.*[, ]%s([, ]|$)", pattern.c_str());
	if(regcomp(&Pattern,S, REG_EXTENDED | REG_NOSUB | REG_NEWLINE) != 0) {
		_error->Error("Failed to compile task regexp");
		return pkgset;
	}

	for (pkgCache::GrpIterator Grp = Cache->GrpBegin(); Grp.end() == false; ++Grp) {
		pkgCache::PkgIterator Pkg = Grp.FindPkg(arch);
		if (Pkg.end() == true)
			continue;
		pkgCache::VerIterator ver = Cache[Pkg].CandidateVerIter(Cache);
		if(ver.end() == true)
			continue;

		pkgRecords::Parser &parser = Recs.Lookup(ver.FileList());
		const char *start, *end;
		parser.GetRec(start,end);
		unsigned int const length = end - start;
		char buf[length];
		strncpy(buf, start, length);
		buf[length-1] = '\0';
		if (regexec(&Pattern, buf, 0, 0, 0) != 0)
			continue;

		pkgset.insert(Pkg);
	}
	regfree(&Pattern);

	if (pkgset.empty() == true)
		return helper.canNotFindTask(Cache, pattern);

	helper.showTaskSelection(pkgset, pattern);
	return pkgset;
}
									/*}}}*/
// FromRegEx - Return all packages in the cache matching a pattern	/*{{{*/
PackageSet PackageSet::FromRegEx(pkgCacheFile &Cache, std::string pattern, CacheSetHelper &helper) {
	PackageSet pkgset;
	static const char * const isregex = ".?+*|[^$";

	if (pattern.find_first_of(isregex) == std::string::npos)
		return pkgset;

	size_t archfound = pattern.find_last_of(':');
	std::string arch = "native";
	if (archfound != std::string::npos) {
		arch = pattern.substr(archfound+1);
		if (arch.find_first_of(isregex) == std::string::npos)
			pattern.erase(archfound);
		else
			arch = "native";
	}

	regex_t Pattern;
	int Res;
	if ((Res = regcomp(&Pattern, pattern.c_str() , REG_EXTENDED | REG_ICASE | REG_NOSUB)) != 0) {
		char Error[300];
		regerror(Res, &Pattern, Error, sizeof(Error));
		_error->Error(_("Regex compilation error - %s"), Error);
		return pkgset;
	}

	for (pkgCache::GrpIterator Grp = Cache.GetPkgCache()->GrpBegin(); Grp.end() == false; ++Grp)
	{
		if (regexec(&Pattern, Grp.Name(), 0, 0, 0) != 0)
			continue;
		pkgCache::PkgIterator Pkg = Grp.FindPkg(arch);
		if (Pkg.end() == true) {
			if (archfound == std::string::npos) {
				std::vector<std::string> archs = APT::Configuration::getArchitectures();
				for (std::vector<std::string>::const_iterator a = archs.begin();
				     a != archs.end() && Pkg.end() != true; ++a)
					Pkg = Grp.FindPkg(*a);
			}
			if (Pkg.end() == true)
				continue;
		}

		pkgset.insert(Pkg);
	}
	regfree(&Pattern);

	if (pkgset.empty() == true)
		return helper.canNotFindRegEx(Cache, pattern);

	helper.showRegExSelection(pkgset, pattern);
	return pkgset;
}
									/*}}}*/
// GroupedFromCommandLine - Return all versions specified on commandline/*{{{*/
std::map<unsigned short, PackageSet> PackageSet::GroupedFromCommandLine(
		pkgCacheFile &Cache, const char **cmdline,
		std::list<PackageSet::Modifier> const &mods,
		unsigned short const &fallback, CacheSetHelper &helper) {
	std::map<unsigned short, PackageSet> pkgsets;
	for (const char **I = cmdline; *I != 0; ++I) {
		unsigned short modID = fallback;
		std::string str = *I;
		for (std::list<PackageSet::Modifier>::const_iterator mod = mods.begin();
		     mod != mods.end(); ++mod) {
			size_t const alength = strlen(mod->Alias);
			switch(mod->Pos) {
			case PackageSet::Modifier::POSTFIX:
				if (str.compare(str.length() - alength, alength,
						mod->Alias, 0, alength) != 0)
					continue;
				str.erase(str.length() - alength);
				modID = mod->ID;
				break;
			case PackageSet::Modifier::PREFIX:
				continue;
			case PackageSet::Modifier::NONE:
				continue;
			}
			break;
		}
		pkgsets[modID].insert(PackageSet::FromString(Cache, str, helper));
	}
	return pkgsets;
}
									/*}}}*/
// FromCommandLine - Return all packages specified on commandline	/*{{{*/
PackageSet PackageSet::FromCommandLine(pkgCacheFile &Cache, const char **cmdline, CacheSetHelper &helper) {
	PackageSet pkgset;
	for (const char **I = cmdline; *I != 0; ++I) {
		PackageSet pset = FromString(Cache, *I, helper);
		pkgset.insert(pset.begin(), pset.end());
	}
	return pkgset;
}
									/*}}}*/
// FromString - Return all packages matching a specific string		/*{{{*/
PackageSet PackageSet::FromString(pkgCacheFile &Cache, std::string const &str, CacheSetHelper &helper) {
	std::string pkg = str;
	size_t archfound = pkg.find_last_of(':');
	std::string arch;
	if (archfound != std::string::npos) {
		arch = pkg.substr(archfound+1);
		pkg.erase(archfound);
	}

	pkgCache::PkgIterator Pkg;
	if (arch.empty() == true) {
		pkgCache::GrpIterator Grp = Cache.GetPkgCache()->FindGrp(pkg);
		if (Grp.end() == false)
			Pkg = Grp.FindPreferredPkg();
	} else
		Pkg = Cache.GetPkgCache()->FindPkg(pkg, arch);

	if (Pkg.end() == false) {
		PackageSet pkgset;
		pkgset.insert(Pkg);
		return pkgset;
	}
	PackageSet pset = FromTask(Cache, str, helper);
	if (pset.empty() == false)
		return pset;

	pset = FromRegEx(Cache, str, helper);
	if (pset.empty() == false)
		return pset;

	return helper.canNotFindPackage(Cache, str);
}
									/*}}}*/
// GroupedFromCommandLine - Return all versions specified on commandline/*{{{*/
std::map<unsigned short, VersionSet> VersionSet::GroupedFromCommandLine(
		pkgCacheFile &Cache, const char **cmdline,
		std::list<VersionSet::Modifier> const &mods,
		unsigned short const &fallback, CacheSetHelper &helper) {
	std::map<unsigned short, VersionSet> versets;
	for (const char **I = cmdline; *I != 0; ++I) {
		unsigned short modID = fallback;
		VersionSet::Version select = VersionSet::NEWEST;
		std::string str = *I;
		for (std::list<VersionSet::Modifier>::const_iterator mod = mods.begin();
		     mod != mods.end(); ++mod) {
			if (modID == fallback && mod->ID == fallback)
				select = mod->SelectVersion;
			size_t const alength = strlen(mod->Alias);
			switch(mod->Pos) {
			case VersionSet::Modifier::POSTFIX:
				if (str.compare(str.length() - alength, alength,
						mod->Alias, 0, alength) != 0)
					continue;
				str.erase(str.length() - alength);
				modID = mod->ID;
				select = mod->SelectVersion;
				break;
			case VersionSet::Modifier::PREFIX:
				continue;
			case VersionSet::Modifier::NONE:
				continue;
			}
			break;
		}
		versets[modID].insert(VersionSet::FromString(Cache, str, select , helper));
	}
	return versets;
}
									/*}}}*/
// FromCommandLine - Return all versions specified on commandline	/*{{{*/
APT::VersionSet VersionSet::FromCommandLine(pkgCacheFile &Cache, const char **cmdline,
		APT::VersionSet::Version const &fallback, CacheSetHelper &helper) {
	VersionSet verset;
	for (const char **I = cmdline; *I != 0; ++I) {
		VersionSet vset = VersionSet::FromString(Cache, *I, fallback, helper);
		verset.insert(vset.begin(), vset.end());
	}
	return verset;
}
									/*}}}*/
// FromString - Returns all versions spedcified by a string		/*{{{*/
APT::VersionSet VersionSet::FromString(pkgCacheFile &Cache, std::string pkg,
		APT::VersionSet::Version const &fallback, CacheSetHelper &helper) {
	std::string ver;
	bool verIsRel = false;
	size_t const vertag = pkg.find_last_of("/=");
	if (vertag != string::npos) {
		ver = pkg.substr(vertag+1);
		verIsRel = (pkg[vertag] == '/');
		pkg.erase(vertag);
	}
	PackageSet pkgset = PackageSet::FromString(Cache, pkg.c_str(), helper);
	VersionSet verset;
	for (PackageSet::const_iterator P = pkgset.begin();
	     P != pkgset.end(); ++P) {
		if (vertag == string::npos) {
			AddSelectedVersion(Cache, verset, P, fallback, helper);
			continue;
		}
		pkgCache::VerIterator V;
		if (ver == "installed")
			V = getInstalledVer(Cache, P, helper);
		else if (ver == "candidate")
			V = getCandidateVer(Cache, P, helper);
		else {
			pkgVersionMatch Match(ver, (verIsRel == true ? pkgVersionMatch::Release :
					pkgVersionMatch::Version));
			V = Match.Find(P);
			if (V.end() == true) {
				if (verIsRel == true)
					_error->Error(_("Release '%s' for '%s' was not found"),
							ver.c_str(), P.FullName(true).c_str());
				else
					_error->Error(_("Version '%s' for '%s' was not found"),
							ver.c_str(), P.FullName(true).c_str());
				continue;
			}
		}
		if (V.end() == true)
			continue;
		helper.showSelectedVersion(P, V, ver, verIsRel);
		verset.insert(V);
	}
	return verset;
}
									/*}}}*/
// AddSelectedVersion - add version from package based on fallback	/*{{{*/
void VersionSet::AddSelectedVersion(pkgCacheFile &Cache, VersionSet &verset,
		pkgCache::PkgIterator const &P, VersionSet::Version const &fallback,
		CacheSetHelper &helper) {
	pkgCache::VerIterator V;
	bool showErrors;
	switch(fallback) {
	case VersionSet::ALL:
		if (P->VersionList != 0)
			for (V = P.VersionList(); V.end() != true; ++V)
				verset.insert(V);
		else
			verset.insert(helper.canNotFindAllVer(Cache, P));
		break;
	case VersionSet::CANDANDINST:
		verset.insert(getInstalledVer(Cache, P, helper));
		verset.insert(getCandidateVer(Cache, P, helper));
		break;
	case VersionSet::CANDIDATE:
		verset.insert(getCandidateVer(Cache, P, helper));
		break;
	case VersionSet::INSTALLED:
		verset.insert(getInstalledVer(Cache, P, helper));
		break;
	case VersionSet::CANDINST:
		showErrors = helper.showErrors(false);
		V = getCandidateVer(Cache, P, helper);
		if (V.end() == true)
			V = getInstalledVer(Cache, P, helper);
		helper.showErrors(showErrors);
		if (V.end() == false)
			verset.insert(V);
		else
			verset.insert(helper.canNotFindInstCandVer(Cache, P));
		break;
	case VersionSet::INSTCAND:
		showErrors = helper.showErrors(false);
		V = getInstalledVer(Cache, P, helper);
		if (V.end() == true)
			V = getCandidateVer(Cache, P, helper);
		helper.showErrors(showErrors);
		if (V.end() == false)
			verset.insert(V);
		else
			verset.insert(helper.canNotFindInstCandVer(Cache, P));
		break;
	case VersionSet::NEWEST:
		if (P->VersionList != 0)
			verset.insert(P.VersionList());
		else
			verset.insert(helper.canNotFindNewestVer(Cache, P));
		break;
	}
}
									/*}}}*/
// getCandidateVer - Returns the candidate version of the given package	/*{{{*/
pkgCache::VerIterator VersionSet::getCandidateVer(pkgCacheFile &Cache,
		pkgCache::PkgIterator const &Pkg, CacheSetHelper &helper) {
	pkgCache::VerIterator Cand;
	if (Cache.IsDepCacheBuilt() == true)
		Cand = Cache[Pkg].CandidateVerIter(Cache);
	else {
		if (unlikely(Cache.BuildPolicy() == false))
			return pkgCache::VerIterator(*Cache);
		Cand = Cache.GetPolicy()->GetCandidateVer(Pkg);
	}
	if (Cand.end() == true)
		return helper.canNotFindCandidateVer(Cache, Pkg);
	return Cand;
}
									/*}}}*/
// getInstalledVer - Returns the installed version of the given package	/*{{{*/
pkgCache::VerIterator VersionSet::getInstalledVer(pkgCacheFile &Cache,
		pkgCache::PkgIterator const &Pkg, CacheSetHelper &helper) {
	if (Pkg->CurrentVer == 0)
		return helper.canNotFindInstalledVer(Cache, Pkg);
	return Pkg.CurrentVer();
}
									/*}}}*/
// canNotFindTask - handle the case no package is found for a task	/*{{{*/
PackageSet CacheSetHelper::canNotFindTask(pkgCacheFile &Cache, std::string pattern) {
	if (ShowError == true)
		_error->Error(_("Couldn't find task '%s'"), pattern.c_str());
	return PackageSet();
}
									/*}}}*/
// canNotFindRegEx - handle the case no package is found by a regex	/*{{{*/
PackageSet CacheSetHelper::canNotFindRegEx(pkgCacheFile &Cache, std::string pattern) {
	if (ShowError == true)
		_error->Error(_("Couldn't find any package by regex '%s'"), pattern.c_str());
	return PackageSet();
}
									/*}}}*/
// canNotFindPackage - handle the case no package is found from a string/*{{{*/
PackageSet CacheSetHelper::canNotFindPackage(pkgCacheFile &Cache, std::string const &str) {
	if (ShowError == true)
		_error->Error(_("Unable to locate package %s"), str.c_str());
	return PackageSet();
}
									/*}}}*/
// canNotFindAllVer							/*{{{*/
VersionSet CacheSetHelper::canNotFindAllVer(pkgCacheFile &Cache,
		pkgCache::PkgIterator const &Pkg) {
	if (ShowError == true)
		_error->Error(_("Can't select versions from package '%s' as it purely virtual"), Pkg.FullName(true).c_str());
	return VersionSet();
}
									/*}}}*/
// canNotFindInstCandVer						/*{{{*/
VersionSet CacheSetHelper::canNotFindInstCandVer(pkgCacheFile &Cache,
		pkgCache::PkgIterator const &Pkg) {
	if (ShowError == true)
		_error->Error(_("Can't select installed nor candidate version from package '%s' as it has neither of them"), Pkg.FullName(true).c_str());
	return VersionSet();
}
									/*}}}*/
// canNotFindNewestVer							/*{{{*/
pkgCache::VerIterator CacheSetHelper::canNotFindNewestVer(pkgCacheFile &Cache,
		pkgCache::PkgIterator const &Pkg) {
	if (ShowError == true)
		_error->Error(_("Can't select newest version from package '%s' as it is purely virtual"), Pkg.FullName(true).c_str());
	return pkgCache::VerIterator(*Cache);
}
									/*}}}*/
// canNotFindCandidateVer						/*{{{*/
pkgCache::VerIterator CacheSetHelper::canNotFindCandidateVer(pkgCacheFile &Cache,
		pkgCache::PkgIterator const &Pkg) {
	if (ShowError == true)
		_error->Error(_("Can't select candidate version from package %s as it has no candidate"), Pkg.FullName(true).c_str());
	return pkgCache::VerIterator(*Cache);
}
									/*}}}*/
// canNotFindInstalledVer						/*{{{*/
pkgCache::VerIterator CacheSetHelper::canNotFindInstalledVer(pkgCacheFile &Cache,
		pkgCache::PkgIterator const &Pkg) {
	if (ShowError == true)
		_error->Error(_("Can't select installed version from package %s as it is not installed"), Pkg.FullName(true).c_str());
	return pkgCache::VerIterator(*Cache);
}
									/*}}}*/
}
Commit	Line	Data
ffee1c2b DK	1	// -- mode: cpp; mode: fold --
	2	// Description /{{{/
	3	/* ######################################################################
	4
	5	Simple wrapper around a std::set to provide a similar interface to
7959c5ed DK	6	a set of cache structures as to the complete set of all structures
7959c5ed DK	7	in the pkgCache. Currently only Package is supported.
ffee1c2b DK	8
	9	##################################################################### */
	10	/}}}/
	11	// Include Files /{{{/
78c32596	12	#include <apt-pkg/aptconfiguration.h>
ffee1c2b	13	#include <apt-pkg/error.h>
ffee1c2b	14	#include <apt-pkg/strutl.h>
856d3b06	15	#include <apt-pkg/versionmatch.h>
ffee1c2b DK	16
	17	#include <apti18n.h>
	18
dc0f01f7 DK	19	#include "cacheset.h"
dc0f01f7 DK	20
78c32596 DK	21	#include <vector>
78c32596 DK	22
ffee1c2b DK	23	#include <regex.h>
	24	/}}}/
	25	namespace APT {
dc0f01f7	26	// FromTask - Return all packages in the cache from a specific task /{{{/
70e706ad	27	PackageSet PackageSet::FromTask(pkgCacheFile &Cache, std::string pattern, CacheSetHelper &helper) {
dc0f01f7 DK	28	PackageSet pkgset;
	29	if (Cache.BuildCaches() == false \|\| Cache.BuildDepCache() == false)
	30	return pkgset;
	31
	32	size_t archfound = pattern.find_last_of(':');
	33	std::string arch = "native";
	34	if (archfound != std::string::npos) {
	35	arch = pattern.substr(archfound+1);
	36	pattern.erase(archfound);
	37	}
	38
	39	if (pattern[pattern.length() -1] != '^')
	40	return pkgset;
	41	pattern.erase(pattern.length()-1);
	42
	43	// get the records
	44	pkgRecords Recs(Cache);
	45
	46	// build regexp for the task
	47	regex_t Pattern;
	48	char S[300];
	49	snprintf(S, sizeof(S), "^Task:.*[, ]%s([, ]\|$)", pattern.c_str());
	50	if(regcomp(&Pattern,S, REG_EXTENDED \| REG_NOSUB \| REG_NEWLINE) != 0) {
	51	_error->Error("Failed to compile task regexp");
	52	return pkgset;
	53	}
	54
	55	for (pkgCache::GrpIterator Grp = Cache->GrpBegin(); Grp.end() == false; ++Grp) {
	56	pkgCache::PkgIterator Pkg = Grp.FindPkg(arch);
	57	if (Pkg.end() == true)
	58	continue;
	59	pkgCache::VerIterator ver = Cache[Pkg].CandidateVerIter(Cache);
	60	if(ver.end() == true)
	61	continue;
	62
	63	pkgRecords::Parser &parser = Recs.Lookup(ver.FileList());
	64	const char start, end;
	65	parser.GetRec(start,end);
	66	unsigned int const length = end - start;
	67	char buf[length];
	68	strncpy(buf, start, length);
	69	buf[length-1] = '\0';
70e706ad DK	70	if (regexec(&Pattern, buf, 0, 0, 0) != 0)
	71	continue;
	72
	73	pkgset.insert(Pkg);
dc0f01f7	74	}
70e706ad	75	regfree(&Pattern);
dc0f01f7 DK	76
dc0f01f7 DK	77	if (pkgset.empty() == true)
70e706ad	78	return helper.canNotFindTask(Cache, pattern);
dc0f01f7	79
70e706ad	80	helper.showTaskSelection(pkgset, pattern);
dc0f01f7 DK	81	return pkgset;
	82	}
	83	/}}}/
ffee1c2b	84	// FromRegEx - Return all packages in the cache matching a pattern /{{{/
70e706ad	85	PackageSet PackageSet::FromRegEx(pkgCacheFile &Cache, std::string pattern, CacheSetHelper &helper) {
ffee1c2b	86	PackageSet pkgset;
6e235c66	87	static const char * const isregex = ".?+*\|[^$";
ffee1c2b	88
6e235c66	89	if (pattern.find_first_of(isregex) == std::string::npos)
ffee1c2b DK	90	return pkgset;
ffee1c2b DK	91
6e235c66	92	size_t archfound = pattern.find_last_of(':');
dc0f01f7	93	std::string arch = "native";
6e235c66 DK	94	if (archfound != std::string::npos) {
	95	arch = pattern.substr(archfound+1);
	96	if (arch.find_first_of(isregex) == std::string::npos)
	97	pattern.erase(archfound);
	98	else
	99	arch = "native";
	100	}
	101
ffee1c2b DK	102	regex_t Pattern;
ffee1c2b DK	103	int Res;
6e235c66	104	if ((Res = regcomp(&Pattern, pattern.c_str() , REG_EXTENDED \| REG_ICASE \| REG_NOSUB)) != 0) {
ffee1c2b DK	105	char Error[300];
	106	regerror(Res, &Pattern, Error, sizeof(Error));
	107	_error->Error(_("Regex compilation error - %s"), Error);
	108	return pkgset;
	109	}
	110
856d3b06	111	for (pkgCache::GrpIterator Grp = Cache.GetPkgCache()->GrpBegin(); Grp.end() == false; ++Grp)
ffee1c2b DK	112	{
	113	if (regexec(&Pattern, Grp.Name(), 0, 0, 0) != 0)
	114	continue;
6e235c66	115	pkgCache::PkgIterator Pkg = Grp.FindPkg(arch);
78c32596	116	if (Pkg.end() == true) {
6e235c66 DK	117	if (archfound == std::string::npos) {
	118	std::vector<std::string> archs = APT::Configuration::getArchitectures();
	119	for (std::vector<std::string>::const_iterator a = archs.begin();
	120	a != archs.end() && Pkg.end() != true; ++a)
	121	Pkg = Grp.FindPkg(*a);
78c32596 DK	122	}
	123	if (Pkg.end() == true)
	124	continue;
	125	}
ffee1c2b	126
ffee1c2b DK	127	pkgset.insert(Pkg);
ffee1c2b DK	128	}
ffee1c2b DK	129	regfree(&Pattern);
ffee1c2b DK	130
70e706ad DK	131	if (pkgset.empty() == true)
	132	return helper.canNotFindRegEx(Cache, pattern);
	133
	134	helper.showRegExSelection(pkgset, pattern);
78c32596 DK	135	return pkgset;
	136	}
	137	/}}}/
9cc83a6f DK	138	// GroupedFromCommandLine - Return all versions specified on commandline/{{{/
	139	std::map<unsigned short, PackageSet> PackageSet::GroupedFromCommandLine(
	140	pkgCacheFile &Cache, const char **cmdline,
	141	std::list<PackageSet::Modifier> const &mods,
70e706ad	142	unsigned short const &fallback, CacheSetHelper &helper) {
9cc83a6f DK	143	std::map<unsigned short, PackageSet> pkgsets;
	144	for (const char *I = cmdline; I != 0; ++I) {
	145	unsigned short modID = fallback;
	146	std::string str = *I;
	147	for (std::list<PackageSet::Modifier>::const_iterator mod = mods.begin();
	148	mod != mods.end(); ++mod) {
	149	size_t const alength = strlen(mod->Alias);
	150	switch(mod->Pos) {
	151	case PackageSet::Modifier::POSTFIX:
	152	if (str.compare(str.length() - alength, alength,
55c59998	153	mod->Alias, 0, alength) != 0)
9cc83a6f DK	154	continue;
	155	str.erase(str.length() - alength);
	156	modID = mod->ID;
	157	break;
	158	case PackageSet::Modifier::PREFIX:
	159	continue;
	160	case PackageSet::Modifier::NONE:
	161	continue;
	162	}
	163	break;
	164	}
70e706ad	165	pkgsets[modID].insert(PackageSet::FromString(Cache, str, helper));
9cc83a6f DK	166	}
	167	return pkgsets;
	168	}
	169	/}}}/
78c32596	170	// FromCommandLine - Return all packages specified on commandline /{{{/
70e706ad	171	PackageSet PackageSet::FromCommandLine(pkgCacheFile &Cache, const char **cmdline, CacheSetHelper &helper) {
78c32596	172	PackageSet pkgset;
9f1f17cc	173	for (const char *I = cmdline; I != 0; ++I) {
70e706ad	174	PackageSet pset = FromString(Cache, *I, helper);
856d3b06 DK	175	pkgset.insert(pset.begin(), pset.end());
	176	}
	177	return pkgset;
	178	}
	179	/}}}/
	180	// FromString - Return all packages matching a specific string /{{{/
70e706ad	181	PackageSet PackageSet::FromString(pkgCacheFile &Cache, std::string const &str, CacheSetHelper &helper) {
fe870feb DK	182	std::string pkg = str;
	183	size_t archfound = pkg.find_last_of(':');
	184	std::string arch;
	185	if (archfound != std::string::npos) {
	186	arch = pkg.substr(archfound+1);
	187	pkg.erase(archfound);
	188	}
	189
	190	pkgCache::PkgIterator Pkg;
	191	if (arch.empty() == true) {
	192	pkgCache::GrpIterator Grp = Cache.GetPkgCache()->FindGrp(pkg);
	193	if (Grp.end() == false)
	194	Pkg = Grp.FindPreferredPkg();
	195	} else
	196	Pkg = Cache.GetPkgCache()->FindPkg(pkg, arch);
	197
	198	if (Pkg.end() == false) {
856d3b06 DK	199	PackageSet pkgset;
	200	pkgset.insert(Pkg);
	201	return pkgset;
	202	}
70e706ad	203	PackageSet pset = FromTask(Cache, str, helper);
dc0f01f7 DK	204	if (pset.empty() == false)
	205	return pset;
	206
70e706ad	207	pset = FromRegEx(Cache, str, helper);
dc0f01f7 DK	208	if (pset.empty() == false)
	209	return pset;
	210
70e706ad	211	return helper.canNotFindPackage(Cache, str);
856d3b06 DK	212	}
856d3b06 DK	213	/}}}/
55c59998 DK	214	// GroupedFromCommandLine - Return all versions specified on commandline/{{{/
	215	std::map<unsigned short, VersionSet> VersionSet::GroupedFromCommandLine(
	216	pkgCacheFile &Cache, const char **cmdline,
	217	std::list<VersionSet::Modifier> const &mods,
70e706ad	218	unsigned short const &fallback, CacheSetHelper &helper) {
55c59998 DK	219	std::map<unsigned short, VersionSet> versets;
	220	for (const char *I = cmdline; I != 0; ++I) {
	221	unsigned short modID = fallback;
	222	VersionSet::Version select = VersionSet::NEWEST;
	223	std::string str = *I;
	224	for (std::list<VersionSet::Modifier>::const_iterator mod = mods.begin();
	225	mod != mods.end(); ++mod) {
	226	if (modID == fallback && mod->ID == fallback)
	227	select = mod->SelectVersion;
	228	size_t const alength = strlen(mod->Alias);
	229	switch(mod->Pos) {
	230	case VersionSet::Modifier::POSTFIX:
	231	if (str.compare(str.length() - alength, alength,
	232	mod->Alias, 0, alength) != 0)
	233	continue;
	234	str.erase(str.length() - alength);
	235	modID = mod->ID;
	236	select = mod->SelectVersion;
	237	break;
	238	case VersionSet::Modifier::PREFIX:
	239	continue;
	240	case VersionSet::Modifier::NONE:
	241	continue;
	242	}
	243	break;
	244	}
70e706ad	245	versets[modID].insert(VersionSet::FromString(Cache, str, select , helper));
55c59998 DK	246	}
	247	return versets;
	248	}
	249	/}}}/
856d3b06 DK	250	// FromCommandLine - Return all versions specified on commandline /{{{/
856d3b06 DK	251	APT::VersionSet VersionSet::FromCommandLine(pkgCacheFile &Cache, const char **cmdline,
70e706ad	252	APT::VersionSet::Version const &fallback, CacheSetHelper &helper) {
856d3b06	253	VersionSet verset;
9f1f17cc	254	for (const char *I = cmdline; I != 0; ++I) {
70e706ad	255	VersionSet vset = VersionSet::FromString(Cache, *I, fallback, helper);
55c59998 DK	256	verset.insert(vset.begin(), vset.end());
	257	}
	258	return verset;
	259	}
	260	/}}}/
	261	// FromString - Returns all versions spedcified by a string /{{{/
	262	APT::VersionSet VersionSet::FromString(pkgCacheFile &Cache, std::string pkg,
70e706ad	263	APT::VersionSet::Version const &fallback, CacheSetHelper &helper) {
55c59998 DK	264	std::string ver;
	265	bool verIsRel = false;
	266	size_t const vertag = pkg.find_last_of("/=");
	267	if (vertag != string::npos) {
	268	ver = pkg.substr(vertag+1);
	269	verIsRel = (pkg[vertag] == '/');
	270	pkg.erase(vertag);
	271	}
70e706ad	272	PackageSet pkgset = PackageSet::FromString(Cache, pkg.c_str(), helper);
55c59998 DK	273	VersionSet verset;
	274	for (PackageSet::const_iterator P = pkgset.begin();
	275	P != pkgset.end(); ++P) {
	276	if (vertag == string::npos) {
70e706ad	277	AddSelectedVersion(Cache, verset, P, fallback, helper);
55c59998	278	continue;
856d3b06	279	}
55c59998 DK	280	pkgCache::VerIterator V;
55c59998 DK	281	if (ver == "installed")
70e706ad	282	V = getInstalledVer(Cache, P, helper);
55c59998	283	else if (ver == "candidate")
70e706ad	284	V = getCandidateVer(Cache, P, helper);
55c59998 DK	285	else {
	286	pkgVersionMatch Match(ver, (verIsRel == true ? pkgVersionMatch::Release :
	287	pkgVersionMatch::Version));
	288	V = Match.Find(P);
	289	if (V.end() == true) {
	290	if (verIsRel == true)
	291	_error->Error(_("Release '%s' for '%s' was not found"),
	292	ver.c_str(), P.FullName(true).c_str());
	293	else
	294	_error->Error(_("Version '%s' for '%s' was not found"),
	295	ver.c_str(), P.FullName(true).c_str());
84910ad5 DK	296	continue;
84910ad5 DK	297	}
78c32596	298	}
55c59998 DK	299	if (V.end() == true)
55c59998 DK	300	continue;
70e706ad	301	helper.showSelectedVersion(P, V, ver, verIsRel);
55c59998	302	verset.insert(V);
78c32596	303	}
856d3b06 DK	304	return verset;
	305	}
	306	/}}}/
84910ad5	307	// AddSelectedVersion - add version from package based on fallback /{{{/
70e706ad	308	void VersionSet::AddSelectedVersion(pkgCacheFile &Cache, VersionSet &verset,
84910ad5	309	pkgCache::PkgIterator const &P, VersionSet::Version const &fallback,
70e706ad	310	CacheSetHelper &helper) {
84910ad5	311	pkgCache::VerIterator V;
70e706ad	312	bool showErrors;
84910ad5 DK	313	switch(fallback) {
	314	case VersionSet::ALL:
	315	if (P->VersionList != 0)
	316	for (V = P.VersionList(); V.end() != true; ++V)
	317	verset.insert(V);
84910ad5	318	else
70e706ad	319	verset.insert(helper.canNotFindAllVer(Cache, P));
84910ad5 DK	320	break;
84910ad5 DK	321	case VersionSet::CANDANDINST:
70e706ad DK	322	verset.insert(getInstalledVer(Cache, P, helper));
70e706ad DK	323	verset.insert(getCandidateVer(Cache, P, helper));
84910ad5 DK	324	break;
84910ad5 DK	325	case VersionSet::CANDIDATE:
70e706ad	326	verset.insert(getCandidateVer(Cache, P, helper));
84910ad5 DK	327	break;
84910ad5 DK	328	case VersionSet::INSTALLED:
70e706ad	329	verset.insert(getInstalledVer(Cache, P, helper));
84910ad5 DK	330	break;
84910ad5 DK	331	case VersionSet::CANDINST:
70e706ad DK	332	showErrors = helper.showErrors(false);
70e706ad DK	333	V = getCandidateVer(Cache, P, helper);
84910ad5	334	if (V.end() == true)
70e706ad DK	335	V = getInstalledVer(Cache, P, helper);
70e706ad DK	336	helper.showErrors(showErrors);
84910ad5 DK	337	if (V.end() == false)
84910ad5 DK	338	verset.insert(V);
84910ad5	339	else
70e706ad	340	verset.insert(helper.canNotFindInstCandVer(Cache, P));
84910ad5 DK	341	break;
84910ad5 DK	342	case VersionSet::INSTCAND:
70e706ad DK	343	showErrors = helper.showErrors(false);
70e706ad DK	344	V = getInstalledVer(Cache, P, helper);
84910ad5	345	if (V.end() == true)
70e706ad DK	346	V = getCandidateVer(Cache, P, helper);
70e706ad DK	347	helper.showErrors(showErrors);
84910ad5 DK	348	if (V.end() == false)
84910ad5 DK	349	verset.insert(V);
84910ad5	350	else
70e706ad	351	verset.insert(helper.canNotFindInstCandVer(Cache, P));
84910ad5 DK	352	break;
	353	case VersionSet::NEWEST:
	354	if (P->VersionList != 0)
	355	verset.insert(P.VersionList());
84910ad5	356	else
70e706ad	357	verset.insert(helper.canNotFindNewestVer(Cache, P));
84910ad5 DK	358	break;
84910ad5 DK	359	}
84910ad5 DK	360	}
84910ad5 DK	361	/}}}/
856d3b06 DK	362	// getCandidateVer - Returns the candidate version of the given package /{{{/
856d3b06 DK	363	pkgCache::VerIterator VersionSet::getCandidateVer(pkgCacheFile &Cache,
70e706ad	364	pkgCache::PkgIterator const &Pkg, CacheSetHelper &helper) {
a8ef7efd DK	365	pkgCache::VerIterator Cand;
	366	if (Cache.IsDepCacheBuilt() == true)
	367	Cand = Cache[Pkg].CandidateVerIter(Cache);
	368	else {
	369	if (unlikely(Cache.BuildPolicy() == false))
	370	return pkgCache::VerIterator(*Cache);
	371	Cand = Cache.GetPolicy()->GetCandidateVer(Pkg);
	372	}
70e706ad DK	373	if (Cand.end() == true)
70e706ad DK	374	return helper.canNotFindCandidateVer(Cache, Pkg);
856d3b06 DK	375	return Cand;
	376	}
	377	/}}}/
	378	// getInstalledVer - Returns the installed version of the given package /{{{/
	379	pkgCache::VerIterator VersionSet::getInstalledVer(pkgCacheFile &Cache,
70e706ad DK	380	pkgCache::PkgIterator const &Pkg, CacheSetHelper &helper) {
	381	if (Pkg->CurrentVer == 0)
	382	return helper.canNotFindInstalledVer(Cache, Pkg);
856d3b06	383	return Pkg.CurrentVer();
ffee1c2b DK	384	}
ffee1c2b DK	385	/}}}/
70e706ad DK	386	// canNotFindTask - handle the case no package is found for a task /{{{/
	387	PackageSet CacheSetHelper::canNotFindTask(pkgCacheFile &Cache, std::string pattern) {
	388	if (ShowError == true)
	389	_error->Error(_("Couldn't find task '%s'"), pattern.c_str());
	390	return PackageSet();
	391	}
	392	/}}}/
	393	// canNotFindRegEx - handle the case no package is found by a regex /{{{/
	394	PackageSet CacheSetHelper::canNotFindRegEx(pkgCacheFile &Cache, std::string pattern) {
	395	if (ShowError == true)
	396	_error->Error(_("Couldn't find any package by regex '%s'"), pattern.c_str());
	397	return PackageSet();
	398	}
	399	/}}}/
	400	// canNotFindPackage - handle the case no package is found from a string/{{{/
	401	PackageSet CacheSetHelper::canNotFindPackage(pkgCacheFile &Cache, std::string const &str) {
	402	if (ShowError == true)
	403	_error->Error(_("Unable to locate package %s"), str.c_str());
	404	return PackageSet();
	405	}
	406	/}}}/
	407	// canNotFindAllVer /{{{/
	408	VersionSet CacheSetHelper::canNotFindAllVer(pkgCacheFile &Cache,
	409	pkgCache::PkgIterator const &Pkg) {
	410	if (ShowError == true)
	411	_error->Error(_("Can't select versions from package '%s' as it purely virtual"), Pkg.FullName(true).c_str());
	412	return VersionSet();
	413	}
	414	/}}}/
	415	// canNotFindInstCandVer /{{{/
	416	VersionSet CacheSetHelper::canNotFindInstCandVer(pkgCacheFile &Cache,
	417	pkgCache::PkgIterator const &Pkg) {
	418	if (ShowError == true)
	419	_error->Error(_("Can't select installed nor candidate version from package '%s' as it has neither of them"), Pkg.FullName(true).c_str());
	420	return VersionSet();
	421	}
	422	/}}}/
	423	// canNotFindNewestVer /{{{/
	424	pkgCache::VerIterator CacheSetHelper::canNotFindNewestVer(pkgCacheFile &Cache,
	425	pkgCache::PkgIterator const &Pkg) {
	426	if (ShowError == true)
	427	_error->Error(_("Can't select newest version from package '%s' as it is purely virtual"), Pkg.FullName(true).c_str());
	428	return pkgCache::VerIterator(*Cache);
	429	}
	430	/}}}/
	431	// canNotFindCandidateVer /{{{/
	432	pkgCache::VerIterator CacheSetHelper::canNotFindCandidateVer(pkgCacheFile &Cache,
	433	pkgCache::PkgIterator const &Pkg) {
	434	if (ShowError == true)
	435	_error->Error(_("Can't select candidate version from package %s as it has no candidate"), Pkg.FullName(true).c_str());
	436	return pkgCache::VerIterator(*Cache);
	437	}
	438	/}}}/
	439	// canNotFindInstalledVer /{{{/
	440	pkgCache::VerIterator CacheSetHelper::canNotFindInstalledVer(pkgCacheFile &Cache,
	441	pkgCache::PkgIterator const &Pkg) {
	442	if (ShowError == true)
	443	_error->Error(_("Can't select installed version from package %s as it is not installed"), Pkg.FullName(true).c_str());
	444	return pkgCache::VerIterator(*Cache);
	445	}
	446	/}}}/
ffee1c2b	447	}