Bug #315615 - Use EbuildIpcDaemon to implement has_version and best_version.

This provides performance benefits and also avoids permissions issues with
FEATURES=userpriv.

1 files changed, 55 insertions, 0 deletions

diff --git a/pym/portage/package/ebuild/_ipc/QueryCommand.py b/pym/portage/package/ebuild/_ipc/QueryCommand.py
new file mode 100644
index 000000000..684837456
--- /dev/null
+++ b/pym/portage/package/ebuild/_ipc/QueryCommand.py
@@ -0,0 +1,55 @@
+# Copyright 2010 Gentoo Foundation
+# Distributed under the terms of the GNU General Public License v2
+
+import portage
+from portage import os
+from portage.dep import Atom
+from portage.exception import InvalidAtom
+from portage.package.ebuild._ipc.IpcCommand import IpcCommand
+from portage.util import normalize_path
+from portage.versions import best
+
+class QueryCommand(IpcCommand):
+
+	__slots__ = ()
+
+	_db = None
+
+	def __init__(self):
+		IpcCommand.__init__(self)
+
+	def __call__(self, argv):
+		"""
+		@returns: tuple of (stdout, stderr, returncode)
+		"""
+		cmd, root, atom, use = argv
+
+		try:
+			atom = Atom(atom)
+		except InvalidAtom:
+			return ('', 'invalid atom: %s\n' % atom, 2)
+
+		use = frozenset(use.split())
+		atom = atom.evaluate_conditionals(use)
+
+		db = self._db
+		if db is None:
+			db = portage.db
+
+		root = normalize_path(root).rstrip(os.path.sep) + os.path.sep
+		if root not in db:
+			return ('', 'invalid ROOT: %s\n' % root, 2)
+
+		vardb = db[root]["vartree"].dbapi
+
+		if cmd == 'has_version':
+			if vardb.match(atom):
+				returncode = 0
+			else:
+				returncode = 1
+			return ('', '', returncode)
+		elif cmd == 'best_version':
+			m = best(vardb.match(atom))
+			return ('%s\n' % m, '', 0)
+		else:
+			return ('', 'invalid command: %s\n' % cmd, 2)