contrib/groff-1.19/src/utils/lkbib/lkbib.cpp

   1 // -*- C++ -*-
   2 /* Copyright (C) 1989-1992, 2000, 2001 Free Software Foundation, Inc.
   3      Written by James Clark (jjc@jclark.com)
   4
   5 This file is part of groff.
   6
   7 groff is free software; you can redistribute it and/or modify it under
   8 the terms of the GNU General Public License as published by the Free
   9 Software Foundation; either version 2, or (at your option) any later
  10 version.
  11
  12 groff is distributed in the hope that it will be useful, but WITHOUT ANY
  13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  15 for more details.
  16
  17 You should have received a copy of the GNU General Public License along
  18 with groff; see the file COPYING.  If not, write to the Free Software
  19 Foundation, 51 Franklin St - Fifth Floor, Boston, MA 02110-1301, USA. */
  20
  21 #include "lib.h"
  22
  23 #include <stdlib.h>
  24 #include <errno.h>
  25 #include <assert.h>
  26
  27 #include "errarg.h"
  28 #include "error.h"
  29
  30 #include "defs.h"
  31 #include "refid.h"
  32 #include "search.h"
  33
  34 extern "C" const char *Version_string;
  35
  36 static void usage(FILE *stream)
  37 {
  38   fprintf(stream, "usage: %s [-nv] [-p database] [-i XYZ] [-t N] keys ...\n",
  39           program_name);
  40 }
  41
  42 int main(int argc, char **argv)
  43 {
  44   program_name = argv[0];
  45   static char stderr_buf[BUFSIZ];
  46   setbuf(stderr, stderr_buf);
  47   int search_default = 1;
  48   search_list list;
  49   int opt;
  50   static const struct option long_options[] = {
  51     { "help", no_argument, 0, CHAR_MAX + 1 },
  52     { "version", no_argument, 0, 'v' },
  53     { NULL, 0, 0, 0 }
  54   };
  55   while ((opt = getopt_long(argc, argv, "nvVi:t:p:", long_options, NULL))
  56          != EOF)
  57     switch (opt) {
  58     case 'V':
  59       verify_flag = 1;
  60       break;
  61     case 'n':
  62       search_default = 0;
  63       break;
  64     case 'i':
  65       linear_ignore_fields = optarg;
  66       break;
  67     case 't':
  68       {
  69         char *ptr;
  70         long n = strtol(optarg, &ptr, 10);
  71         if (n == 0 && ptr == optarg) {
  72           error("bad integer `%1' in `t' option", optarg);
  73           break;
  74         }
  75         if (n < 1)
  76           n = 1;
  77         linear_truncate_len = int(n);
  78         break;
  79       }
  80     case 'v':
  81       {
  82         printf("GNU lkbib (groff) version %s\n", Version_string);
  83         exit(0);
  84         break;
  85       }
  86     case 'p':
  87       list.add_file(optarg);
  88       break;
  89     case CHAR_MAX + 1: // --help
  90       usage(stdout);
  91       exit(0);
  92       break;
  93     case '?':
  94       usage(stderr);
  95       exit(1);
  96       break;
  97     default:
  98       assert(0);
  99     }
 100   if (optind >= argc) {
 101     usage(stderr);
 102     exit(1);
 103   }
 104   char *filename = getenv("REFER");
 105   if (filename)
 106     list.add_file(filename);
 107   else if (search_default)
 108     list.add_file(DEFAULT_INDEX, 1);
 109   if (list.nfiles() == 0)
 110     fatal("no databases");
 111   int total_len = 0;
 112   int i;
 113   for (i = optind; i < argc; i++)
 114     total_len += strlen(argv[i]);
 115   total_len += argc - optind - 1 + 1; // for spaces and '\0'
 116   char *buffer = new char[total_len];
 117   char *ptr = buffer;
 118   for (i = optind; i < argc; i++) {
 119     if (i > optind)
 120       *ptr++ = ' ';
 121     strcpy(ptr, argv[i]);
 122     ptr = strchr(ptr, '\0');
 123   }
 124   search_list_iterator iter(&list, buffer);
 125   const char *start;
 126   int len;
 127   int count;
 128   for (count = 0; iter.next(&start, &len); count++) {
 129     if (fwrite(start, 1, len, stdout) != (size_t)len)
 130       fatal("write error on stdout: %1", strerror(errno));
 131     // Can happen for last reference in file.
 132     if (start[len - 1] != '\n')
 133       putchar('\n');
 134     putchar('\n');
 135   }
 136   return !count;
 137 }