Merge rpool_get.c into rpool.c.

lib/rpool.c

@@ -33,6 +33,21 @@
 
 #include "xbps_api_impl.h"
 
+struct rpool_fpkg {
+	xbps_array_t revdeps;
+	xbps_dictionary_t pkgd;
+	const char *pattern;
+	const char *bestpkgver;
+	bool best;
+};
+
+typedef enum {
+	BEST_PKG = 1,
+	VIRTUAL_PKG,
+	REAL_PKG,
+	REVDEPS_PKG
+} pkg_repo_type_t;
+
 static SIMPLEQ_HEAD(rpool_head, xbps_repo) rpool_queue =
     SIMPLEQ_HEAD_INITIALIZER(rpool_queue);
 
@@ -106,3 +121,205 @@ xbps_rpool_foreach(struct xbps_handle *xhp,
 
 	return rv;
 }
+
+static int
+find_virtualpkg_cb(struct xbps_repo *repo, void *arg, bool *done)
+{
+	struct rpool_fpkg *rpf = arg;
+
+	rpf->pkgd = xbps_repo_get_virtualpkg(repo, rpf->pattern);
+	if (rpf->pkgd) {
+		/* found */
+		*done = true;
+		return 0;
+	}
+	/* not found */
+	return 0;
+}
+
+static int
+find_pkg_cb(struct xbps_repo *repo, void *arg, bool *done)
+{
+	struct rpool_fpkg *rpf = arg;
+
+	rpf->pkgd = xbps_repo_get_pkg(repo, rpf->pattern);
+	if (rpf->pkgd) {
+		/* found */
+		*done = true;
+		return 0;
+	}
+	/* Not found */
+	return 0;
+}
+
+static int
+find_pkg_revdeps_cb(struct xbps_repo *repo, void *arg, bool *done _unused)
+{
+	struct rpool_fpkg *rpf = arg;
+	xbps_array_t revdeps = NULL;
+	const char *pkgver;
+
+	revdeps = xbps_repo_get_pkg_revdeps(repo, rpf->pattern);
+	if (xbps_array_count(revdeps)) {
+		/* found */
+		if (rpf->revdeps == NULL)
+			rpf->revdeps = xbps_array_create();
+		for (unsigned int i = 0; i < xbps_array_count(revdeps); i++) {
+			xbps_array_get_cstring_nocopy(revdeps, i, &pkgver);
+			xbps_array_add_cstring_nocopy(rpf->revdeps, pkgver);
+		}
+		xbps_object_release(revdeps);
+	}
+	return 0;
+}
+
+static int
+find_best_pkg_cb(struct xbps_repo *repo, void *arg, bool *done _unused)
+{
+	struct rpool_fpkg *rpf = arg;
+	xbps_dictionary_t pkgd;
+	const char *repopkgver;
+
+	pkgd = xbps_repo_get_pkg(repo, rpf->pattern);
+	if (pkgd == NULL) {
+		if (errno && errno != ENOENT)
+			return errno;
+
+		xbps_dbg_printf(repo->xhp,
+		    "[rpool] Package '%s' not found in repository "
+		    "'%s'.\n", rpf->pattern, repo->uri);
+		return 0;
+	}
+	xbps_dictionary_get_cstring_nocopy(pkgd,
+	    "pkgver", &repopkgver);
+	if (rpf->bestpkgver == NULL) {
+		xbps_dbg_printf(repo->xhp,
+		    "[rpool] Found match '%s' (%s).\n",
+		    repopkgver, repo->uri);
+		rpf->pkgd = pkgd;
+		rpf->bestpkgver = repopkgver;
+		return 0;
+	}
+	/*
+	 * Compare current stored version against new
+	 * version from current package in repository.
+	 */
+	if (xbps_cmpver(repopkgver, rpf->bestpkgver) == 1) {
+		xbps_dbg_printf(repo->xhp,
+		    "[rpool] Found best match '%s' (%s).\n",
+		    repopkgver, repo->uri);
+		rpf->pkgd = pkgd;
+		rpf->bestpkgver = repopkgver;
+	}
+	return 0;
+}
+
+static xbps_object_t
+repo_find_pkg(struct xbps_handle *xhp,
+	      const char *pkg,
+	      pkg_repo_type_t type)
+{
+	struct rpool_fpkg rpf;
+	int rv = 0;
+
+	assert(xhp);
+	assert(pkg);
+
+	rpf.pattern = pkg;
+	rpf.pkgd = NULL;
+	rpf.revdeps = NULL;
+	rpf.bestpkgver = NULL;
+
+	switch (type) {
+	case BEST_PKG:
+		/*
+		 * Find best pkg version.
+		 */
+		rv = xbps_rpool_foreach(xhp, find_best_pkg_cb, &rpf);
+		break;
+	case VIRTUAL_PKG:
+		/*
+		 * Find virtual pkg.
+		 */
+		rv = xbps_rpool_foreach(xhp, find_virtualpkg_cb, &rpf);
+		break;
+	case REAL_PKG:
+		/*
+		 * Find real pkg.
+		 */
+		rv = xbps_rpool_foreach(xhp, find_pkg_cb, &rpf);
+		break;
+	case REVDEPS_PKG:
+		/*
+		 * Find revdeps for pkg.
+		 */
+		rv = xbps_rpool_foreach(xhp, find_pkg_revdeps_cb, &rpf);
+	}
+	if (rv != 0) {
+		errno = rv;
+		return NULL;
+	}
+	if (type == REVDEPS_PKG)
+		return rpf.revdeps;
+
+	return rpf.pkgd;
+}
+
+xbps_dictionary_t
+xbps_rpool_get_virtualpkg(struct xbps_handle *xhp, const char *pkg)
+{
+	return repo_find_pkg(xhp, pkg, VIRTUAL_PKG);
+}
+
+xbps_dictionary_t
+xbps_rpool_get_pkg(struct xbps_handle *xhp, const char *pkg)
+{
+	if (!xbps_pkgpattern_version(pkg) && !xbps_pkg_version(pkg))
+		return repo_find_pkg(xhp, pkg, BEST_PKG);
+
+	return repo_find_pkg(xhp, pkg, REAL_PKG);
+}
+
+xbps_array_t
+xbps_rpool_get_pkg_revdeps(struct xbps_handle *xhp, const char *pkg)
+{
+	return repo_find_pkg(xhp, pkg, REVDEPS_PKG);
+}
+
+xbps_dictionary_t
+xbps_rpool_get_pkg_plist(struct xbps_handle *xhp,
+			 const char *pkg,
+			 const char *plistf)
+{
+	xbps_dictionary_t pkgd = NULL, plistd = NULL;
+	char *url;
+
+	assert(pkg != NULL);
+	assert(plistf != NULL);
+	/*
+	 * Iterate over the the repository pool and search for a plist file
+	 * in the binary package matching `pattern'. The plist file will be
+	 * internalized to a proplib dictionary.
+	 *
+	 * The first repository that has it wins and the loop is stopped.
+	 * This will work locally and remotely, thanks to libarchive and
+	 * libfetch!
+	 */
+	if (((pkgd = xbps_rpool_get_pkg(xhp, pkg)) == NULL) &&
+	    ((pkgd = xbps_rpool_get_virtualpkg(xhp, pkg)) == NULL))
+		goto out;
+
+	url = xbps_repository_pkg_path(xhp, pkgd);
+	if (url == NULL) {
+		errno = EINVAL;
+		goto out;
+	}
+	plistd = xbps_get_pkg_plist_from_binpkg(url, plistf);
+	free(url);
+
+out:
+	if (plistd == NULL)
+		errno = ENOENT;
+
+	return plistd;
+}