+#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++;
- }
- while (*pstr && *pstr != '-') {
- if (ver) {
- *ver++ = *pstr;
+ dbfile = getenv("ZPMDB");
+
+ 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;
}