]> git.cameronkatri.com Git - mandoc.git/blobdiff - apropos.c
catch up with the changed interface of mparse_alloc(), see mandoc.h rev 1.100
[mandoc.git] / apropos.c
index f70243a2e7247a74164bfd2a129dfd83ddef10b9..9c3cae96afa030ad1fa68e498fde65ac371e1048 100644 (file)
--- a/apropos.c
+++ b/apropos.c
@@ -1,6 +1,6 @@
-/*     $Id: apropos.c,v 1.17 2011/11/20 21:36:00 kristaps Exp $ */
+/*     $Id: apropos.c,v 1.30 2012/03/24 02:18:51 kristaps Exp $ */
 /*
- * Copyright (c) 2011 Kristaps Dzonsons <kristaps@bsd.lv>
+ * Copyright (c) 2011, 2012 Kristaps Dzonsons <kristaps@bsd.lv>
  * Copyright (c) 2011 Ingo Schwarze <schwarze@openbsd.org>
  *
  * Permission to use, copy, modify, and distribute this software for any
 #ifdef HAVE_CONFIG_H
 #include "config.h"
 #endif
+#include <sys/param.h>
 
 #include <assert.h>
-#include <ctype.h>
 #include <getopt.h>
-#include <limits.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <unistd.h>
 
 #include "apropos_db.h"
 #include "mandoc.h"
+#include "manpath.h"
 
-/*
- * FIXME: add support for manpath(1), which everybody but OpenBSD and
- * NetBSD seem to use.
- */
-#define MAN_CONF_FILE  "/etc/man.conf"
-#define MAN_CONF_KEY   "_whatdb"
-
-/*
- * List of paths to be searched for manual databases.
- */
-struct manpaths {
-       int       sz;
-       char    **paths;
-};
+#define        SINGLETON(_res, _sz) \
+       ((_sz) && (_res)[0].matched && \
+        (1 == (_sz) || 0 == (_res)[1].matched))
+#define        EMPTYSET(_res, _sz) \
+       ((0 == (_sz)) || 0 == (_res)[0].matched)
 
 static int      cmp(const void *, const void *);
 static void     list(struct res *, size_t, void *);
-static void     manpath_add(struct manpaths *, const char *);
-static void     manpath_parse(struct manpaths *, char *);
-static void     manpath_parseconf(struct manpaths *);
 static void     usage(void);
 
 static char    *progname;
@@ -57,12 +46,17 @@ static      char    *progname;
 int
 main(int argc, char *argv[])
 {
-       int              i, ch, rc;
+       int              ch, rc, whatis, usecat;
+       struct res      *res;
        struct manpaths  paths;
-       size_t           terms;
+       const char      *prog;
+       pid_t            pid;
+       char             path[PATH_MAX];
+       int              fds[2];
+       size_t           terms, ressz, sz;
        struct opts      opts;
        struct expr     *e;
-       char            *defpaths, *auxpaths;
+       char            *defpaths, *auxpaths, *conf_file, *cp;
        extern int       optind;
        extern char     *optarg;
 
@@ -72,15 +66,24 @@ main(int argc, char *argv[])
        else
                ++progname;
 
+       whatis = 0 == strncmp(progname, "whatis", 6);
+
        memset(&paths, 0, sizeof(struct manpaths));
        memset(&opts, 0, sizeof(struct opts));
 
+       usecat = 0;
+       ressz = 0;
+       res = NULL;
        auxpaths = defpaths = NULL;
+       conf_file = NULL;
        e = NULL;
-       rc = 0;
+       path[0] = '\0';
 
-       while (-1 != (ch = getopt(argc, argv, "M:m:S:s:"))) 
+       while (-1 != (ch = getopt(argc, argv, "C:M:m:S:s:")))
                switch (ch) {
+               case ('C'):
+                       conf_file = optarg;
+                       break;
                case ('M'):
                        defpaths = optarg;
                        break;
@@ -95,72 +98,130 @@ main(int argc, char *argv[])
                        break;
                default:
                        usage();
-                       goto out;
+                       return(EXIT_FAILURE);
                }
 
        argc -= optind;
        argv += optind;
 
-       if (0 == argc) {
-               rc = 1;
-               goto out;
-       }
-
-       if (NULL != getenv("MANPATH"))
-               defpaths = getenv("MANPATH");
+       if (0 == argc) 
+               return(EXIT_SUCCESS);
 
-       if (NULL == defpaths)
-               manpath_parseconf(&paths);
-       else
-               manpath_parse(&paths, defpaths);
+       rc = 0;
 
-       manpath_parse(&paths, auxpaths);
+       manpath_parse(&paths, conf_file, defpaths, auxpaths);
 
-       if (NULL == (e = exprcomp(argc, argv, &terms))) {
+       e = whatis ? termcomp(argc, argv, &terms) :
+                    exprcomp(argc, argv, &terms);
+               
+       if (NULL == e) {
                fprintf(stderr, "%s: Bad expression\n", progname);
                goto out;
        }
 
        rc = apropos_search
-               (paths.sz, paths.paths, 
-                &opts, e, terms, NULL, list);
+               (paths.sz, paths.paths, &opts, 
+                e, terms, NULL, &ressz, &res, list);
 
-       if (0 == rc) 
-               fprintf(stderr, "%s: Error reading "
-                               "manual database\n", progname);
+       terms = 1;
 
-out:
-       for (i = 0; i < paths.sz; i++)
-               free(paths.paths[i]);
+       if (0 == rc) {
+               fprintf(stderr, "%s: Bad database\n", progname);
+               goto out;
+       } else if ( ! isatty(STDOUT_FILENO) || EMPTYSET(res, ressz))
+               goto out;
+
+       if ( ! SINGLETON(res, ressz)) {
+               printf("Which manpage would you like [1]? ");
+               fflush(stdout);
+               if (NULL != (cp = fgetln(stdin, &sz)) && 
+                               sz > 1 && '\n' == cp[--sz]) {
+                       if ((ch = atoi(cp)) <= 0)
+                               goto out;
+                       terms = (size_t)ch;
+               }
+       }
 
-       free(paths.paths);
+       if (--terms < ressz && res[terms].matched) {
+               chdir(paths.paths[res[terms].volume]);
+               strlcpy(path, res[terms].file, PATH_MAX);
+               usecat = RESTYPE_CAT == res[terms].type;
+       }
+out:
+       manpath_free(&paths);
+       resfree(res, ressz);
        exprfree(e);
 
-       return(rc ? EXIT_SUCCESS : EXIT_FAILURE);
+       if ('\0' == path[0])
+               return(rc ? EXIT_SUCCESS : EXIT_FAILURE);
+
+       if (-1 == pipe(fds)) {
+               perror(NULL);
+               exit(EXIT_FAILURE);
+       }
+
+       if (-1 == (pid = fork())) {
+               perror(NULL);
+               exit(EXIT_FAILURE);
+       } else if (pid > 0) {
+               dup2(fds[0], STDIN_FILENO);
+               close(fds[1]);
+               prog = NULL != getenv("MANPAGER") ? 
+                       getenv("MANPAGER") :
+                       (NULL != getenv("PAGER") ? 
+                        getenv("PAGER") : "more");
+               execlp(prog, prog, (char *)NULL);
+               perror(prog);
+               return(EXIT_FAILURE);
+       }
+
+       dup2(fds[1], STDOUT_FILENO);
+       close(fds[0]);
+       prog = usecat ? "cat" : "mandoc";
+       execlp(prog, prog, path, (char *)NULL);
+       perror(prog);
+       return(EXIT_FAILURE);
 }
 
 /* ARGSUSED */
 static void
 list(struct res *res, size_t sz, void *arg)
 {
-       int              i;
+       size_t           i;
 
        qsort(res, sz, sizeof(struct res), cmp);
 
-       for (i = 0; i < (int)sz; i++)
-               printf("%s(%s%s%s) - %s\n", res[i].title, 
-                               res[i].cat, 
-                               *res[i].arch ? "/" : "",
-                               *res[i].arch ? res[i].arch : "",
-                               res[i].desc);
+       if (EMPTYSET(res, sz) || SINGLETON(res, sz))
+               return;
+
+       if ( ! isatty(STDOUT_FILENO))
+               for (i = 0; i < sz && res[i].matched; i++)
+                       printf("%s(%s%s%s) - %.70s\n", 
+                                       res[i].title, res[i].cat,
+                                       *res[i].arch ? "/" : "",
+                                       *res[i].arch ? res[i].arch : "",
+                                       res[i].desc);
+       else
+               for (i = 0; i < sz && res[i].matched; i++)
+                       printf("[%zu] %s(%s%s%s) - %.70s\n", i + 1,
+                                       res[i].title, res[i].cat,
+                                       *res[i].arch ? "/" : "",
+                                       *res[i].arch ? res[i].arch : "",
+                                       res[i].desc);
 }
 
 static int
 cmp(const void *p1, const void *p2)
 {
+       const struct res *r1 = p1;
+       const struct res *r2 = p2;
+
+       if (0 == r1->matched)
+               return(1);
+       else if (0 == r2->matched)
+               return(1);
 
-       return(strcmp(((const struct res *)p1)->title,
-                               ((const struct res *)p2)->title));
+       return(strcasecmp(r1->title, r2->title));
 }
 
 static void
@@ -168,109 +229,11 @@ usage(void)
 {
 
        fprintf(stderr, "usage: %s "
-                       "[-M dirs] "
-                       "[-m dirs] "
+                       "[-C file] "
+                       "[-M manpath] "
+                       "[-m manpath] "
                        "[-S arch] "
                        "[-s section] "
-                       "expression...\n", progname);
-}
-
-/*
- * Parse a FULL pathname from a colon-separated list of arrays.
- */
-static void
-manpath_parse(struct manpaths *dirs, char *path) 
-{
-       char    *dir;
-
-       if (NULL == path)
-               return;
-
-       for (dir = strtok(path, ":"); dir; dir = strtok(NULL, ":"))
-               manpath_add(dirs, dir);
-}
-
-/*
- * Add a directory to the array, ignoring bad directories.
- * Grow the array one-by-one for simplicity's sake.
- */
-static void
-manpath_add(struct manpaths *dirs, const char *dir) 
-{
-       char             buf[PATH_MAX];
-       char            *cp;
-       int              i;
-
-       if (NULL == (cp = realpath(dir, buf)))
-               return;
-
-       for (i = 0; i < dirs->sz; i++)
-               if (0 == strcmp(dirs->paths[i], dir))
-                       return;
-
-       dirs->paths = mandoc_realloc
-               (dirs->paths, 
-                ((size_t)dirs->sz + 1) * sizeof(char *));
-
-       dirs->paths[dirs->sz++] = mandoc_strdup(cp);
-}
-
-static void
-manpath_parseconf(struct manpaths *dirs) 
-{
-       FILE            *stream;
-#ifdef USE_MANPATH
-       char            *buf;
-       size_t           sz, bsz;
-
-       stream = popen("manpath", "r");
-       if (NULL == stream)
-               return;
-
-       buf = NULL;
-       bsz = 0;
-
-       do {
-               buf = mandoc_realloc(buf, bsz + 1024);
-               sz = fread(buf + (int)bsz, 1, 1024, stream);
-               bsz += sz;
-       } while (sz > 0);
-
-       assert(bsz && '\n' == buf[bsz - 1]);
-       buf[bsz - 1] = '\0';
-
-       manpath_parse(dirs, buf);
-       free(buf);
-       pclose(stream);
-#else
-       char            *p, *q;
-       size_t           len, keysz;
-
-       keysz = strlen(MAN_CONF_KEY);
-       assert(keysz > 0);
-
-       if (NULL == (stream = fopen(MAN_CONF_FILE, "r")))
-               return;
-
-       while (NULL != (p = fgetln(stream, &len))) {
-               if (0 == len || '\n' == p[--len])
-                       break;
-               p[len] = '\0';
-               while (isspace((unsigned char)*p))
-                       p++;
-               if (strncmp(MAN_CONF_KEY, p, keysz))
-                       continue;
-               p += keysz;
-               while (isspace(*p))
-                       p++;
-               if ('\0' == *p)
-                       continue;
-               if (NULL == (q = strrchr(p, '/')))
-                       continue;
-               *q = '\0';
-               manpath_add(dirs, p);
-       }
-
-       fclose(stream);
-#endif
+                       "expression ...\n",
+                       progname);
 }