WS changes for function calls.
[idzebra-moved-to-github.git] / index / zebraidx.c
1 /* $Id: zebraidx.c,v 1.9 2007-10-29 09:25:41 adam Exp $
2    Copyright (C) 1995-2007
3    Index Data ApS
4
5 This file is part of the Zebra server.
6
7 Zebra 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 Zebra 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
18 along with this program; if not, write to the Free Software
19 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
20
21 */
22
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <assert.h>
27 #ifdef WIN32
28 #include <io.h>
29 #endif
30 #if HAVE_UNISTD_H
31 #include <unistd.h>
32 #endif
33
34 #if HAVE_ICU
35 #include <yaz/icu_I18N.h>
36 #endif
37 #include <yaz/log.h>
38 #include <yaz/options.h>
39 #include <idzebra/version.h>
40 #include <idzebra/api.h>
41
42 char *prog;
43
44 static void filter_cb(void *cd, const char *name)
45 {
46     puts(name);
47 }
48
49 static void show_filters(ZebraService zs)
50 {
51     zebra_filter_info(zs, 0, filter_cb);
52 }
53
54 int main(int argc, char **argv)
55 {
56     int ret;
57     int cmd = 0;
58     char *arg;
59     char *configName = 0;
60     int nsections = 0;
61     int enable_commit = 1;
62     char *database = 0;
63     Res res = res_open(0, 0);
64     Res default_res = res_open(0, 0);
65     
66     int trans_started = 0;
67 #ifndef WIN32
68     char nbuf[100];
69 #endif
70     ZebraService zs = 0;
71     ZebraHandle zh = 0;
72
73 #ifdef WIN32
74 #else
75     sprintf(nbuf, "%.40s(%ld)", *argv, (long) getpid());
76     yaz_log_init_prefix(nbuf);
77 #endif
78     prog = *argv;
79     if (argc < 2)
80     {
81         fprintf(stderr, "%s [options] command <dir> ...\n"
82         "Commands:\n"
83         " update <dir>  Update index with files below <dir>.\n"
84         "               If <dir> is empty filenames are read from stdin.\n"
85         " delete <dir>  Delete index with files below <dir>.\n"
86         " create <db>   Create database <db>\n"
87         " drop <db>     Drop database <db>\n"
88         " commit        Commit changes\n"
89         " clean         Clean shadow files\n"
90         "Options:\n"
91         " -t <type>     Index files as <type> (grs or text).\n"
92         " -c <config>   Read configuration file <config>.\n"
93         " -g <group>    Index files according to group settings.\n"
94         " -d <database> Records belong to Z39.50 database <database>.\n"
95         " -m <mbytes>   Use <mbytes> before flushing keys to disk.\n"
96         " -n            Don't use shadow system.\n"
97         " -s            Show analysis on stdout, but do no work.\n"
98         " -v <level>    Set logging to <level>.\n"
99         " -l <file>     Write log to <file>.\n"
100         " -L            Don't follow symbolic links.\n"
101         " -f <n>        Display information for the first <n> records.\n"
102         " -V            Show version.\n", *argv
103                  );
104         exit(1);
105     }
106     res_set(default_res, "profilePath", DEFAULT_PROFILE_PATH);
107     res_set(default_res, "modulePath", DEFAULT_MODULE_PATH);
108     while ((ret = options("sVt:c:g:d:m:v:nf:l:L", argv, argc, &arg)) != -2)
109     {
110         if (ret == 0)
111         {
112             if(cmd == 0) /* command */
113             {
114                 if (!zs)
115                 {
116                     const char *config = configName ? configName : "zebra.cfg";
117                     zs = zebra_start_res(config, default_res, res);
118                     if (!zs)
119                     {
120                         yaz_log(YLOG_FATAL, "Cannot read config %s", config);
121                         exit(1);
122                     }   
123                     zh = zebra_open(zs, 0);
124                     zebra_shadow_enable(zh, enable_commit);
125                 }
126
127                 if (database &&
128                     zebra_select_database(zh, database) == ZEBRA_FAIL)
129                 {
130                     yaz_log(YLOG_FATAL, "Could not select database %s "
131                             "errCode=%d",
132                             database, zebra_errCode(zh) );
133                     exit(1);
134                 }
135                 if (!strcmp(arg, "update"))
136                     cmd = 'u';
137                 else if (!strcmp(arg, "update1"))
138                     cmd = 'U';
139                 else if (!strcmp(arg, "update2"))
140                     cmd = 'm';
141                 else if (!strcmp(arg, "dump"))
142                     cmd = 's';
143                 else if (!strcmp(arg, "del") || !strcmp(arg, "delete"))
144                     cmd = 'd';
145                 else if (!strcmp(arg, "init"))
146                 {
147                     zebra_init(zh);
148                 }
149                 else if (!strcmp(arg, "drop"))
150                 {
151                     cmd = 'D';
152                 }
153                 else if (!strcmp(arg, "create"))
154                 {
155                     cmd = 'C';
156                 }
157                 else if (!strcmp(arg, "commit"))
158                 {
159                     zebra_commit(zh);
160                 }
161                 else if (!strcmp(arg, "clean"))
162                 {
163                     zebra_clean(zh);
164                 }
165                 else if (!strcmp(arg, "stat") || !strcmp(arg, "status"))
166                 {
167                     zebra_register_statistics(zh, 0);
168                 }
169                 else if (!strcmp(arg, "dumpdict"))
170                 {
171                     zebra_register_statistics(zh, 1);
172                 }
173                 else if (!strcmp(arg, "compact"))
174                 {
175                     zebra_compact(zh);
176                 }
177                 else if (!strcmp(arg, "filters"))
178                 {
179                     show_filters(zs);
180                 }
181                 else
182                 {
183                     yaz_log(YLOG_FATAL, "unknown command: %s", arg);
184                     exit(1);
185                 }
186             }
187             else
188             {
189                 ZEBRA_RES res = ZEBRA_OK;
190                 if (!trans_started)
191                 {
192                     trans_started=1;
193                     if (zebra_begin_trans(zh, 1) != ZEBRA_OK)
194                         exit(1);
195                 }
196                 switch (cmd)
197                 {
198                 case 'u':
199                     res = zebra_repository_update(zh, arg);
200                     break;
201                 case 'd':
202                     res = zebra_repository_delete(zh, arg);
203                     break;
204                 case 's':
205                     res = zebra_repository_show(zh, arg);
206                     nsections = 0;
207                     break;
208                 case 'C':
209                     res = zebra_create_database(zh, arg);
210                     break;
211                 case 'D':
212                     res = zebra_drop_database(zh, arg);
213                     break;
214                 default:
215                     nsections = 0;
216                 }
217                 if (res != ZEBRA_OK)
218                 {
219                     const char *add = zebra_errAdd(zh);
220                     yaz_log(YLOG_FATAL, "Operation failed: %s %s",
221                             zebra_errString(zh), add ? add : "");
222                     
223                     if (trans_started)
224                         if (zebra_end_trans(zh) != ZEBRA_OK)
225                             yaz_log(YLOG_WARN, "zebra_end_trans failed");
226
227
228                     zebra_close(zh);
229                     zebra_stop(zs);
230                     exit(1);
231                 }
232                 log_event_end(NULL, NULL);
233             }
234         }
235         else if (ret == 'V')
236         {
237             char version_str[20];
238             char sys_str[80];
239             zebra_get_version(version_str, sys_str);
240
241             printf("Zebra %s\n", version_str);
242             printf("(C) 1994-2007, Index Data ApS\n");
243             printf("Zebra is free software, covered by the GNU General Public License, and you are\n");
244             printf("welcome to change it and/or distribute copies of it under certain conditions.\n");
245             printf("Configured as: %s\n", sys_str);
246             if (strcmp(version_str, ZEBRAVER))
247                 printf("zebraidx compiled version %s\n", ZEBRAVER);
248 #if HAVE_ICU
249             printf("Using ICU\n");
250 #endif
251         }
252         else if (ret == 'v')
253             yaz_log_init_level(yaz_log_mask_str(arg));
254         else if (ret == 'l')
255             yaz_log_init_file(arg);
256         else if (ret == 'm')
257             res_set(res, "memMax", arg);
258         else if (ret == 'd')
259             database = arg;
260         else if (ret == 's')
261             res_set(res, "openRW", "0");
262         else if (ret == 'g')
263             res_set(res, "group", arg);
264         else if (ret == 'f')
265             res_set(res, "fileVerboseLimit", arg);
266         else if (ret == 'c')
267             configName = arg;
268         else if (ret == 't')
269             res_set(res, "recordType", arg);
270         else if (ret == 'n')
271             enable_commit = 0;
272         else if (ret == 'L')
273             res_set(res, "followLinks", "0");
274         else
275             yaz_log(YLOG_WARN, "unknown option '-%s'", arg);
276     } /* while arg */
277
278     if (trans_started)
279         if (zebra_end_trans(zh) != ZEBRA_OK)
280             yaz_log(YLOG_WARN, "zebra_end_trans failed");
281
282     zebra_close(zh);
283     zebra_stop(zs);
284
285     res_close(res);
286     res_close(default_res);
287     exit(0);
288     return 0;
289 }
290
291 /*
292  * Local variables:
293  * c-basic-offset: 4
294  * indent-tabs-mode: nil
295  * End:
296  * vim: shiftwidth=4 tabstop=8 expandtab
297  */
298