]> git.cameronkatri.com Git - mandoc.git/blobdiff - mandocdb.c
Keep words after .Ic together in a single argument.
[mandoc.git] / mandocdb.c
index e4346b69eb473862090ee0e317d26ed6d112efc1..e796385948ed1dbea8558290f6aa7c900c3135b8 100644 (file)
@@ -1,4 +1,4 @@
-/*     $Id: mandocdb.c,v 1.105 2014/01/06 13:54:17 schwarze Exp $ */
+/*     $Id: mandocdb.c,v 1.113 2014/01/19 22:48:16 schwarze Exp $ */
 /*
  * Copyright (c) 2011, 2012 Kristaps Dzonsons <kristaps@bsd.lv>
  * Copyright (c) 2011, 2012, 2013, 2014 Ingo Schwarze <schwarze@openbsd.org>
@@ -48,6 +48,9 @@
 #include "manpath.h"
 #include "mansearch.h"
 
+extern int mansearch_keymax;
+extern const char *const mansearch_keynames[];
+
 #define        SQL_EXEC(_v) \
        if (SQLITE_OK != sqlite3_exec(db, (_v), NULL, NULL, NULL)) \
                fprintf(stderr, "%s\n", sqlite3_errmsg(db))
@@ -154,8 +157,7 @@ static      int      parse_mdoc_Nd(struct mpage *, const struct mdoc_node *);
 static int      parse_mdoc_Nm(struct mpage *, const struct mdoc_node *);
 static int      parse_mdoc_Sh(struct mpage *, const struct mdoc_node *);
 static int      parse_mdoc_Xr(struct mpage *, const struct mdoc_node *);
-static void     putkey(const struct mpage *,
-                       const char *, uint64_t);
+static void     putkey(const struct mpage *, char *, uint64_t);
 static void     putkeys(const struct mpage *,
                        const char *, size_t, uint64_t);
 static void     putmdockey(const struct mpage *,
@@ -700,11 +702,19 @@ filescan(const char *file)
                exitcode = (int)MANDOCLEVEL_BADARG;
                say(file, NULL);
                return;
-       } else if (OP_TEST != op && strstr(buf, basedir) != buf) {
+       }
+
+       if (strstr(buf, basedir) == buf)
+               start = buf + strlen(basedir) + 1;
+       else if (OP_TEST == op)
+               start = buf;
+       else {
                exitcode = (int)MANDOCLEVEL_BADARG;
                say("", "%s: outside base directory", buf);
                return;
-       } else if (-1 == stat(buf, &st)) {
+       }
+
+       if (-1 == stat(buf, &st)) {
                exitcode = (int)MANDOCLEVEL_BADARG;
                say(file, NULL);
                return;
@@ -713,7 +723,7 @@ filescan(const char *file)
                say(file, "Not a regular file");
                return;
        }
-       start = buf + strlen(basedir);
+
        mlink = mandoc_calloc(1, sizeof(struct mlink));
        strlcpy(mlink->file, start, sizeof(mlink->file));
 
@@ -946,12 +956,13 @@ mlink_check(struct mpage *mpage, struct mlink *mlink)
 static void
 mpages_merge(struct mchars *mc, struct mparse *mp)
 {
+       char                     any[] = "any";
        struct ohash_info        str_info;
        struct mpage            *mpage;
        struct mlink            *mlink;
        struct mdoc             *mdoc;
        struct man              *man;
-       const char              *cp;
+       char                    *cp;
        int                      match;
        unsigned int             pslot;
        enum mandoclevel         lvl;
@@ -961,6 +972,9 @@ mpages_merge(struct mchars *mc, struct mparse *mp)
        str_info.hfree = hash_free;
        str_info.key_offset = offsetof(struct str, key);
 
+       if (0 == nodb)
+               SQL_EXEC("BEGIN TRANSACTION");
+
        mpage = ohash_first(&mpages, &pslot);
        while (NULL != mpage) {
                mlinks_undupe(mpage);
@@ -1014,7 +1028,7 @@ mpages_merge(struct mchars *mc, struct mparse *mp)
                }
                putkey(mpage, mpage->sec, TYPE_sec);
                putkey(mpage, '\0' == *mpage->arch ?
-                   "any" : mpage->arch, TYPE_arch);
+                   any : mpage->arch, TYPE_arch);
 
                for (mlink = mpage->mlinks; mlink; mlink = mlink->next) {
                        if ('\0' != *mlink->dsec)
@@ -1022,7 +1036,7 @@ mpages_merge(struct mchars *mc, struct mparse *mp)
                        if ('\0' != *mlink->fsec)
                                putkey(mpage, mlink->fsec, TYPE_sec);
                        putkey(mpage, '\0' == *mlink->arch ?
-                           "any" : mlink->arch, TYPE_arch);
+                           any : mlink->arch, TYPE_arch);
                        putkey(mpage, mlink->name, TYPE_Nm);
                }
 
@@ -1051,6 +1065,9 @@ mpages_merge(struct mchars *mc, struct mparse *mp)
                ohash_delete(&strings);
                mpage = ohash_next(&mpages, &pslot);
        }
+
+       if (0 == nodb)
+               SQL_EXEC("END TRANSACTION");
 }
 
 static void
@@ -1169,10 +1186,15 @@ parse_cat(struct mpage *mpage)
  * Put a type/word pair into the word database for this particular file.
  */
 static void
-putkey(const struct mpage *mpage, const char *value, uint64_t type)
+putkey(const struct mpage *mpage, char *value, uint64_t type)
 {
+       char     *cp;
 
        assert(NULL != value);
+       if (TYPE_arch == type)
+               for (cp = value; *cp; cp++)
+                       if (isupper((unsigned char)*cp))
+                               *cp = _tolower((unsigned char)*cp);
        putkeys(mpage, value, strlen(value), type);
 }
 
@@ -1288,6 +1310,15 @@ parse_man(struct mpage *mpage, const struct man_node *n)
                                byte = start[sz];
                                start[sz] = '\0';
 
+                               /*
+                                * Assume a stray trailing comma in the
+                                * name list if a name begins with a dash.
+                                */
+
+                               if ('-' == start[0] ||
+                                   ('\\' == start[0] && '-' == start[1]))
+                                       break;
+
                                putkey(mpage, start, TYPE_Nm);
 
                                if (' ' == byte) {
@@ -1417,7 +1448,7 @@ parse_mdoc_Fd(struct mpage *mpage, const struct mdoc_node *n)
 static int
 parse_mdoc_Fn(struct mpage *mpage, const struct mdoc_node *n)
 {
-       const char      *cp;
+       char    *cp;
 
        if (NULL == (n = n->child) || MDOC_TEXT != n->type)
                return(0);
@@ -1539,12 +1570,24 @@ putkeys(const struct mpage *mpage,
        const char *cp, size_t sz, uint64_t v)
 {
        struct str      *s;
-       unsigned int     slot;
        const char      *end;
+       uint64_t         mask;
+       unsigned int     slot;
+       int              i;
 
        if (0 == sz)
                return;
 
+       if (verb > 1) {
+               for (i = 0, mask = 1;
+                    i < mansearch_keymax;
+                    i++, mask <<= 1)
+                       if (mask & v)
+                               break;
+               say(mpage->mlinks->file, "Adding key %s=%*s",
+                   mansearch_keynames[i], sz, cp);
+       }
+
        end = cp + sz;
        slot = ohash_qlookupi(&strings, cp, &end);
        s = ohash_find(&strings, slot);
@@ -1746,8 +1789,6 @@ dbadd(const struct mpage *mpage, struct mchars *mc)
        if (nodb)
                return;
 
-       SQL_EXEC("BEGIN TRANSACTION");
-
        i = 1;
        SQL_BIND_INT(stmts[STMT_INSERT_PAGE], i, FORM_SRC == mpage->form);
        SQL_STEP(stmts[STMT_INSERT_PAGE]);
@@ -1779,8 +1820,6 @@ dbadd(const struct mpage *mpage, struct mchars *mc)
                        free(key->rendered);
                free(key);
        }
-
-       SQL_EXEC("END TRANSACTION");
 }
 
 static void
@@ -1791,20 +1830,31 @@ dbprune(void)
        size_t           i;
        unsigned int     slot;
 
-       if (nodb)
-               return;
+       if (0 == nodb)
+               SQL_EXEC("BEGIN TRANSACTION");
 
-       mpage = ohash_first(&mpages, &slot);
-       while (NULL != mpage) {
+       for (mpage = ohash_first(&mpages, &slot); NULL != mpage;
+            mpage = ohash_next(&mpages, &slot)) {
                mlink = mpage->mlinks;
-               i = 1;
-               SQL_BIND_TEXT(stmts[STMT_DELETE_PAGE], i, mlink->file);
-               SQL_STEP(stmts[STMT_DELETE_PAGE]);
-               sqlite3_reset(stmts[STMT_DELETE_PAGE]);
                if (verb)
-                       say(mlink->file, "Deleted from database");
-               mpage = ohash_next(&mpages, &slot);
+                       say(mlink->file, "Deleting from database");
+               if (nodb)
+                       continue;
+               for ( ; NULL != mlink; mlink = mlink->next) {
+                       i = 1;
+                       SQL_BIND_TEXT(stmts[STMT_DELETE_PAGE],
+                           i, mlink->dsec);
+                       SQL_BIND_TEXT(stmts[STMT_DELETE_PAGE],
+                           i, mlink->arch);
+                       SQL_BIND_TEXT(stmts[STMT_DELETE_PAGE],
+                           i, mlink->name);
+                       SQL_STEP(stmts[STMT_DELETE_PAGE]);
+                       sqlite3_reset(stmts[STMT_DELETE_PAGE]);
+               }
        }
+
+       if (0 == nodb)
+               SQL_EXEC("END TRANSACTION");
 }
 
 /*
@@ -1893,19 +1943,15 @@ dbopen(int real)
              " \"arch\" TEXT NOT NULL,\n"
              " \"name\" TEXT NOT NULL,\n"
              " \"pageid\" INTEGER NOT NULL REFERENCES mpages(id) "
-               "ON DELETE CASCADE,\n"
-             " \"id\" INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL\n"
+               "ON DELETE CASCADE\n"
              ");\n"
              "\n"
              "CREATE TABLE \"keys\" (\n"
              " \"bits\" INTEGER NOT NULL,\n"
              " \"key\" TEXT NOT NULL,\n"
              " \"pageid\" INTEGER NOT NULL REFERENCES mpages(id) "
-               "ON DELETE CASCADE,\n"
-             " \"id\" INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL\n"
-             ");\n"
-             "\n"
-             "CREATE INDEX \"key_index\" ON keys (key);\n";
+               "ON DELETE CASCADE\n"
+             ");\n";
 
        if (SQLITE_OK != sqlite3_exec(db, sql, NULL, NULL, NULL)) {
                exitcode = (int)MANDOCLEVEL_SYSERR;
@@ -1915,7 +1961,9 @@ dbopen(int real)
 
 prepare_statements:
        SQL_EXEC("PRAGMA foreign_keys = ON");
-       sql = "DELETE FROM mpages where file=?";
+       sql = "DELETE FROM mpages WHERE id IN "
+               "(SELECT pageid FROM mlinks WHERE "
+               "sec=? AND arch=? AND name=?)";
        sqlite3_prepare_v2(db, sql, -1, &stmts[STMT_DELETE_PAGE], NULL);
        sql = "INSERT INTO mpages "
                "(form) VALUES (?)";