X-Git-Url: https://pd.if.org/git/?a=blobdiff_plain;f=zpm-findpkg.c;h=4ee3f120e9131d84efd401fd5c9b9dd1b8ce6981;hb=4b448fbc77510c589cc7f367c05bef63acc07858;hp=95085d5590f3516ba21911189a2f3c5696cd40a8;hpb=8551ba212214a075b6dbbb8081465b711274df79;p=zpackage

diff --git a/zpm-findpkg.c b/zpm-findpkg.c
index 95085d5..4ee3f12 100644
--- a/zpm-findpkg.c
+++ b/zpm-findpkg.c
@@ -1,128 +1,86 @@
+#define _POSIX_C_SOURCE 2 
+
 #include <stdlib.h>
 #include <stdio.h>
 #include <ctype.h>
+#include <unistd.h>
 
 #include "zpm.h"
 
-static int found = 0;
+void usage(void) {
+	fprintf(stderr, "zpm-findpkg [-I] [-s <status> ...] [-S <status>] [package]\n");
+}
 
-static int prow(void *f, int ncols, char **vals, char **cols) {
-	FILE *out = f;
-	int i;
+int main(int ac, char **av){
+	int opt, argn;
+	struct zpm zpm;
+	char *dbfile = 0, *pkgstr = 0, *pkgid = 0;
 
-	if (cols == 0) {
-		fprintf(stderr, "sqlite can't get column names\n");
-	}
-	for (i=3;i<ncols;i++) {
-		if (i>3) fprintf(out, "\t");
-		fprintf(out, "%s", vals[i]);
-	}
-	fprintf(out, "\n");
-	found++;
-	return 0;
-}
+	char *sql;
+	sqlite3_str *include;
+	sqlite3_str *exclude;
+	sqlite3_str *query;
 
-void parse_package(char *pstr, char *name, char *ver, int *rel) {
-	if (name) *name = 0;
-	if (ver) *ver = 0;
-	if (rel) *rel = -1;
-
-	/* string - ver - rel */
-	/* rel is all digits */
-	/* possible forms:
-	 * ^(.+)-([0-9][^-]*)-([\d+])$
-	 * ^(.+)-([0-9][^-]*)$
-	 * ^(.+)$
-	 * The main problem in parsing is that the package name itself
-	 * can contain a '-', so you can't just split on '-'
-	 * Also, the version can be just digits.
-	 */
-
-	/* everything up to the first '-' is in the name */
-	while (*pstr) {
-		if (*pstr == '-' && isdigit(*(pstr+1))) {
-			break;
-		}
-		if (name) {
-			*name++ = *pstr;
-		}
-		pstr++;
-	}
-	if (name) *name = 0;
-	if (*pstr == '-') {
-		pstr++;
+	dbfile = getenv("ZPMDB");
+	if (!dbfile) {
+		dbfile = "/var/lib/zpm/local.db";
 	}
-	while (*pstr && *pstr != '-') {
-		if (ver) {
-			*ver++ = *pstr;
+
+	include = sqlite3_str_new(NULL);
+	exclude = sqlite3_str_new(NULL);
+
+	while ((opt = getopt(ac, av, "f:s:S:I")) != -1) {
+		switch (opt) {
+			case 'f': dbfile = optarg; break;
+			case 's': sqlite3_str_appendf(include,",%Q", optarg);
+				  break;
+			case 'S': sqlite3_str_appendf(exclude,",%Q", optarg);
+				  break;
+			case 'I': sqlite3_str_appendall(include,",'installed'");
+				  break;
+			default:
+				  usage();
+				  exit(EXIT_FAILURE);
+				  break;
 		}
-		pstr++;
-	}
-	if (ver) *ver = 0;
-	if (*pstr == '-') {
-		pstr++;
-	}
-	if (rel && *pstr) {
-		*rel = atoi(pstr);
 	}
-}
-
-int main(int ac, char **av){
-	struct zpm pkg;
-	char *dbfile, *pkgstr;
-	char *select = "select package, version, release, package||'-'||version||'-'||release as pkgid from packages";
-	char *group = "group by package having max( version||'-'||release collate vercmp) order by length(package), package, version||'-'||release collate vercmp";
-
-//	char *select = "select package, version, release, package||'-'||version||'-'||release as pkgid from packages";
-//	char *order = "order by package, version collate vercmp, cast(release as integer)";
-	char sql[2048];
+	argn = optind;
 
-	if (ac < 2) {
-		fprintf(stderr, "usage: db path\n");
+	if (!dbfile) {
+		fprintf(stderr, "must specify db\n");
 		return 1;
 	}
 
-	dbfile = getenv("ZPMDB");
+	query = sqlite3_str_new(NULL);
 
-	if (ac > 1) {
-		dbfile = av[1];
-	}
+	if (zpm_open(&zpm, dbfile)) {
+		char *excludes, *includes;
+
+		excludes = sqlite3_str_value(exclude);
+		includes = sqlite3_str_value(include);
 
-	/* given a package name, get the packages */
-	/* no package name, get all */
-
-	if (zpm_open(&pkg, dbfile)) {
-		char *errmsg;
-
-		if (ac >= 2) {
-			int release;
-			char version[32];
-			char package[32];
-
-			pkgstr = av[2];
-
-			parse_package(pkgstr, package, version, &release);
-			if (release != -1) {
-				/* all three */
-			sprintf(sql, "%s where package = '%s' and version = "
-					"'%s' and release = %d %s;", select,
-					package, version, release, group);
-			} else if (*version != 0) {
-			sprintf(sql, "%s where package = '%s' and version = "
-					"'%s' %s;", select,
-					package, version, group);
-
-			} else {
-
-			sprintf(sql, "%s where package = '%s' %s;",
-				       	select, av[2], group);
-			}
-		} else {
-			sprintf(sql, "%s %s;", select, group);
+		if (includes) {
+			sqlite3_str_appendf(query, "status in (%s)", includes+1);
+		}
+		if (excludes) {
+			sqlite3_str_appendf(query, "%sstatus not in (%s)",
+				       includes ? " and " : "",
+			       	       excludes+1);
+		}
+
+		if (ac >= argn) {
+			pkgstr = av[argn];
 		}
 
-		zpm_exec(&pkg, sql, prow, stdout, &errmsg);
+		sql = sqlite3_str_value(query);
+
+		pkgid = zpm_findpkg(&zpm, pkgstr, sql);
+
+		sqlite3_free(sql);
+		if (pkgid) {
+			printf("%s\n", pkgid);
+		}
+		zpm_close(&zpm);
 	}
-	zpm_close(&pkg);
-	return found ? 0 : 1;
+	return pkgid ? 0 : 1;
 }