modindex.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3 #include <getopt.h>
   4 #include <errno.h>
   5 #include <string.h>
   6
   7 #include "logging.h"
   8 #include "index.h"
   9
  10 static char *getline_wrapped(FILE *file, unsigned int *linenum)
  11 {
  12         int size = 256;
  13         int i = 0;
  14         char *buf = NOFAIL(malloc(size));
  15         for(;;) {
  16                 int ch = getc_unlocked(file);
  17
  18                 switch(ch) {
  19                         case EOF:
  20                                 if (i == 0) {
  21                                         free(buf);
  22                                         return NULL;
  23                                 }
  24                                 /* else fall through */
  25
  26                         case '\n':
  27                                 if (linenum)
  28                                         (*linenum)++;
  29                                 if (i == size)
  30                                         buf = NOFAIL(realloc(buf, size + 1));
  31                                 buf[i] = '\0';
  32                                 return buf;
  33
  34                         case '\\':
  35                                 ch = getc_unlocked(file);
  36
  37                                 if (ch == '\n') {
  38                                         if (linenum)
  39                                                 (*linenum)++;
  40                                         continue;
  41                                 }
  42                                 /* else fall through */
  43
  44                         default:
  45                                 buf[i++] = ch;
  46
  47                                 if (i == size) {
  48                                         size *= 2;
  49                                         buf = NOFAIL(realloc(buf, size));
  50                                 }
  51                 }
  52         }
  53 }
  54
  55 static void write_index(const char *filename)
  56 {
  57         struct index_node *index;
  58         char *line, *pos;
  59         FILE *cfile;
  60         unsigned int linenum = 0;
  61
  62         cfile = fopen(filename, "w");
  63         if (!cfile)
  64                 fatal("Could not open %s for writing: %s\n",
  65                       filename, strerror(errno));
  66
  67         index = index_create();
  68
  69         while((line = getline_wrapped(stdin, &linenum))) {
  70                 pos = strchr(line, ' ');
  71                 *pos++ = '\0';
  72                 index_insert(index, line, pos, linenum);
  73                 free(line);
  74         }
  75
  76         index_write(index, cfile);
  77         index_destroy(index);
  78         fclose(cfile);
  79 }
  80
  81 static struct index_file *open_index(const char *filename)
  82 {
  83         struct index_file *index;
  84
  85         index = index_file_open(filename);
  86         if (!index) {
  87                 if (errno == EINVAL)
  88                         fatal("%s has wrong magic or version number", filename);
  89
  90                 fatal("Could not open %s for reading: %s\n",
  91                       filename, strerror(errno));
  92         }
  93
  94         return index;
  95 }
  96
  97 static void dump_index(const char *filename)
  98 {
  99         struct index_file *index = open_index(filename);
 100
 101         index_dump(index, stdout, "");
 102
 103         index_file_close(index);
 104 }
 105
 106 static void search_index(const char *filename, char *key)
 107 {
 108         struct index_file *index = open_index(filename);
 109         char *value;
 110
 111         value = index_search(index, key);
 112         if (value)
 113                 printf("Found value:\n%s\n", value);
 114         else
 115                 printf("Not found.\n");
 116
 117         free(value);
 118         index_file_close(index);
 119 }
 120
 121 static void searchwild_index(const char *filename, char *key)
 122 {
 123         struct index_file *index = open_index(filename);
 124         struct index_value *values, *v;
 125
 126         values = index_searchwild(index, key);
 127         if (values)
 128                 printf("Found value(s):\n");
 129         else
 130                 printf("Not found.\n");
 131
 132         for (v = values; v; v = v->next)
 133                 printf("%s\n", v->value);
 134
 135         index_values_free(values);
 136         index_file_close(index);
 137 }
 138
 139 static void print_usage(const char *progname)
 140 {
 141         fprintf(stderr,
 142                 "Usage: %s [MODE] [FILE] ...\n"
 143                 " -o, --output <outfile>\n"
 144                 " -d, --dump <infile>\n"
 145                 " -s, --search <infile> <key>\n"
 146                 " -w, --searchwild <infile> <key>\n"
 147                 ,progname);
 148         exit(1);
 149 }
 150
 151 static struct option options[] = {
 152         { "output", 0, NULL, 'o' },
 153         { "dump", 0, NULL, 'd' },
 154         { "search", 0, NULL, 's' },
 155         { "searchwild", 0, NULL, 'w' },
 156 };
 157
 158 int main(int argc, char *argv[])
 159 {
 160         char opt;
 161         char mode = 0;
 162         char *filename = NULL;
 163         char *key = NULL;
 164
 165         while ((opt = getopt_long(argc, argv, "odsw", options, NULL))
 166                        != -1) {
 167                 switch (opt) {
 168                         case 'o':
 169                                 mode = 'o';
 170                                 break;
 171                         case 'd':
 172                                 mode = 'd';
 173                                 break;
 174                         case 's':
 175                                 mode = 's';
 176                                 break;
 177                         case 'w':
 178                                 mode = 'w';
 179                                 break;
 180                         default:
 181                                 print_usage(argv[0]);
 182                 }
 183         }
 184         if (!mode)
 185                 print_usage(argv[0]);
 186
 187         if (optind >= argc)
 188                 print_usage(argv[0]);
 189         filename = argv[optind];
 190
 191         if (mode == 's' || mode == 'w') {
 192                 if (optind+1 >= argc)
 193                         print_usage(argv[0]);
 194                 key = argv[optind+1];
 195         }
 196
 197         switch(mode) {
 198                 case 'o':
 199                         write_index(filename);
 200                         break;
 201                 case 'd':
 202                         dump_index(filename);
 203                         break;
 204                 case 's':
 205                         search_index(filename, key);
 206                         break;
 207                 case 'w':
 208                         searchwild_index(filename, key);
 209                         break;
 210         }
 211
 212         return 0;
 213 }