Radish alpha
H
HardenedBSD-pkg
rad:z3QDZAW2FAfuLvihrhiyDC9fAD8G9
HardenedBSD Package Manager
Radicle
Git
Commits › 05199de
Start doing installations from multiple repositories (!)
Marin Atanasov Nikolov committed 14 years ago
commit 05199de43b72fe192293726f49b5c1448a453aed
parent e3fdf10
1 file changed +120 -28
modified pkg/install.c
@@ -1,4 +1,5 @@
#include <sys/types.h>
+ 
#include <sys/param.h>



#include <err.h>

#include <libgen.h>
@@ -22,12 +23,19 @@ usage_install(void)
int

exec_install(int argc, char **argv)

{
- 
	struct pkg *pkg = NULL;
- 
	struct pkgdb *db = NULL;
+ 
	char dbfile[MAXPATHLEN];
+
+ 
	int retcode = EPKG_OK;
+ 
	int i, multi_repos = 0;
+ 


	struct pkg_jobs *jobs = NULL;

	struct pkg_jobs_entry *je = NULL;
- 
	int retcode = EPKG_OK;
- 
	int i;
+
+ 
	struct pkg_repos *repos = NULL;
+ 
	struct pkg_repos_entry *re = NULL;
+
+ 
	struct pkg *pkg = NULL, *tmp = NULL;
+ 
	struct pkgdb *db = NULL;



	if (argc < 2) {

		usage_install();
@@ -39,45 +47,129 @@ exec_install(int argc, char **argv)
		return (EX_NOPERM);

	}
- 
	if (pkgdb_open(&db, PKGDB_REMOTE, "repo.sqlite") != EPKG_OK) {
- 
		return (EX_IOERR);
- 
	}
-
- 


	/* create a jobs object */

	if (pkg_jobs_new(&jobs) != EPKG_OK) {

		retcode = EPKG_FATAL;

		goto cleanup;

	}
- 
	/* create a jobs entry */
- 
	if (pkg_jobs_new_entry(jobs, &je, PKG_JOBS_INSTALL, db) != EPKG_OK) {
- 
		retcode = EPKG_FATAL;
- 
		goto cleanup;
- 
	}
-
- 
	for (i = 1; i < argc; i++) {
- 
		if ((pkg = pkgdb_query_remote(db, argv[i])) == NULL) {
- 
			retcode = EPKG_FATAL;
- 
			goto cleanup;
+ 
	/*
+ 
	 * Honor PACKAGESITE if specified
+ 
	 * Working on a single repo database
+ 
	 */
+ 
	if (pkg_config("PACKAGESITE") != NULL) {
+ 
                if (pkgdb_open(&db, PKGDB_REMOTE, "repo.sqlite") != EPKG_OK) {
+ 
                        warnx("cannot open repository database: %s/repo.sqlite\n", pkg_config("PKG_DBDIR"));
+ 
                        retcode = EPKG_FATAL;
+ 
                        goto cleanup;
+ 
                }
+
+ 
                /* create a jobs entry */
+ 
                if (pkg_jobs_new_entry(jobs, &je, PKG_JOBS_INSTALL, db) != EPKG_OK) {
+ 
                        retcode = EPKG_FATAL;
+ 
                        goto cleanup;
+ 
                }
+
+ 
		for (i = 1; i < argc; i++) {
+ 
			if ((pkg = pkgdb_query_remote(db, argv[i])) == NULL) {
+ 
				retcode = EPKG_FATAL;
+ 
				goto cleanup;
+ 
			}
+
+ 
			/* pkg_jobs_resolv() will be enough here for jobs resolving :) */
+ 
			pkg_jobs_add(je, pkg);

		}
-
- 
		pkg_jobs_add(je, pkg);
- 
	}
+ 
	} else {
+ 
		/* MULTI_REPOS_INSTALL */
+
+ 
		multi_repos = 1;
+
+ 
                fprintf(stderr, "\n");
+ 
                warnx("/!\\     Working on multiple repositories     /!\\");
+ 
                warnx("/!\\  This is an unsupported preview feature  /!\\");
+ 
                warnx("/!\\     It can kill kittens and puppies      /!\\");
+ 
                fprintf(stderr, "\n");
+
+ 
                if (pkg_repos_new(&repos) != EPKG_OK) {
+ 
                        retcode = EPKG_FATAL;
+ 
                        goto cleanup;
+ 
                }
+
+ 
                if (pkg_repos_load(repos) != EPKG_OK) {
+ 
                        retcode = EPKG_FATAL;
+ 
                        goto cleanup;
+ 
                }
+
+ 
                while (pkg_repos_next(repos, &re) == EPKG_OK) {
+ 
                        snprintf(dbfile, MAXPATHLEN, "%s.sqlite", pkg_repos_get_name(re));
+
+ 
                        if (pkgdb_open(&db, PKGDB_REMOTE, dbfile) != EPKG_OK) {
+ 
                                warnx("cannot open repository database: %s/%s\n",
+ 
                                                pkg_config("PKG_DBDIR"), dbfile);
+ 
                                retcode = EPKG_FATAL;
+ 
                                goto cleanup;
+ 
                        }
+
+ 
                        /* create a jobs entry for each db connection */
+ 
                        if (pkg_jobs_new_entry(jobs, &je, PKG_JOBS_INSTALL, db) != EPKG_OK) {
+ 
                                retcode = EPKG_FATAL;
+ 
                                goto cleanup;
+ 
                        }
+
+ 
        		for (i = 1; i < argc; i++) {
+ 
				if ((pkg = pkgdb_query_remote(db, argv[i])) == NULL) {
+ 
					retcode = EPKG_FATAL;
+ 
					goto cleanup;
+ 
				}
+
+ 
				/*
+ 
				 * check if the job already exists in other job entries
+ 
				 * If it exists we add additional repo to the package.
+ 
				 * Otherwise we have a new package job.
+ 
				 */
+ 
				tmp = NULL;
+ 
				if (pkg_jobs_exists(jobs, pkg, &tmp) == EPKG_OK)
+ 
					pkg_jobs_add(je, pkg);
+ 
				else
+ 
					pkg_repos_add_in_pkg(tmp, re);
+ 
			}
+ 
                }
+ 
	} /* !MULTI_REPOS_INSTALL */



	/* print a summary before applying the jobs */
- 
	pkg = NULL;

	printf("The following packages will be installed:\n");
- 
	while (pkg_jobs_entry(je, &pkg) == EPKG_OK) {
- 
		printf("%s-%s\n", pkg_get(pkg, PKG_NAME), pkg_get(pkg, PKG_VERSION));
+
+ 
	je = NULL; /* starts with the first job entry */
+ 
	while (pkg_jobs(jobs, &je) == EPKG_OK) {
+ 
		pkg = NULL; /* start with the first package in a job entry */
+ 
		while (pkg_jobs_entry(je, &pkg) == EPKG_OK) {
+ 
			printf("\t%s-%s", pkg_get(pkg, PKG_NAME), pkg_get(pkg, PKG_VERSION));
+
+ 
			if (multi_repos == 1) {
+ 
				printf(" [ found in repos: ");
+
+ 
				re = NULL;
+ 
				while (pkg_repos_next_in_pkg(pkg, &re) == EPKG_OK)
+ 
					printf("%s ", pkg_repos_get_name(re));
+
+ 
				printf(" ]");
+ 
			}
+
+ 
			printf("\n");
+ 
		}

	}
- 
	retcode = pkg_jobs_apply(je, 0);
+ 
	je = NULL;
+ 
	while (pkg_jobs(jobs, &je) == EPKG_OK)
+ 
		retcode = pkg_jobs_apply(je, 0);



	cleanup:
- 
	pkgdb_close(db);
+
+ 
	if (multi_repos == 1)
+ 
		pkg_repos_free(repos);
+
+ 
	/* db connections are closed by pkg_jobs_free() */

	pkg_jobs_free(jobs);



	return (retcode == EPKG_OK ? EX_OK : 1);

}
-