2 * Copyright (C) 1995-2005, Index Data ApS
3 * See the file LICENSE for details.
5 * $Id: zoomsh.c,v 1.38 2005-11-01 15:07:29 adam Exp $
15 #include <yaz/comstack.h>
17 #if HAVE_READLINE_READLINE_H
18 #include <readline/readline.h>
20 #if HAVE_READLINE_HISTORY_H
21 #include <readline/history.h>
24 #include <yaz/xmalloc.h>
33 static int next_token (const char **cpp, const char **t_start)
36 const char *cp = *cpp;
43 while (*cp && *cp != '"')
54 while (*cp && *cp != ' ' && *cp != '\r' && *cp != '\n')
63 return len; /* return -1 if no token was read .. */
66 static int next_token_copy (const char **cpp, char *buf_out, int buf_max)
69 int len = next_token (cpp, &start);
77 memcpy (buf_out, start, len);
82 static int is_command (const char *cmd_str, const char *this_str, int this_len)
84 int cmd_len = strlen(cmd_str);
85 if (cmd_len != this_len)
87 if (memcmp (cmd_str, this_str, cmd_len))
92 static void cmd_set (ZOOM_connection *c, ZOOM_resultset *r,
96 char key[40], val[80];
98 if (next_token_copy (args, key, sizeof(key)) < 0)
100 printf ("missing argument for set\n");
103 if (next_token_copy (args, val, sizeof(val)) < 0)
104 ZOOM_options_set(options, key, 0);
106 ZOOM_options_set(options, key, val);
109 static void cmd_get (ZOOM_connection *c, ZOOM_resultset *r,
110 ZOOM_options options,
114 if (next_token_copy (args, key, sizeof(key)) < 0)
116 printf ("missing argument for get\n");
120 const char *val = ZOOM_options_get(options, key);
121 printf ("%s = %s\n", key, val ? val : "<null>");
125 static void cmd_rget(ZOOM_connection *c, ZOOM_resultset *r,
126 ZOOM_options options,
130 if (next_token_copy (args, key, sizeof(key)) < 0)
132 printf ("missing argument for get\n");
137 for (i = 0; i<MAX_CON; i++)
142 const char *val = ZOOM_resultset_option_get(r[i], key);
143 printf ("%s = %s\n", key, val ? val : "<null>");
148 static void cmd_close (ZOOM_connection *c, ZOOM_resultset *r,
149 ZOOM_options options,
154 next_token_copy (args, host, sizeof(host));
155 for (i = 0; i<MAX_CON; i++)
160 if ((h = ZOOM_connection_option_get(c[i], "host"))
161 && !strcmp (h, host))
163 ZOOM_connection_destroy (c[i]);
166 else if (*host == '\0')
168 ZOOM_connection_destroy (c[i]);
174 static void display_records (ZOOM_connection c,
176 int start, int count)
179 for (i = 0; i<count; i++)
182 ZOOM_record rec = ZOOM_resultset_record (r, pos);
183 const char *db = ZOOM_record_get (rec, "database", 0);
185 const char *render = ZOOM_record_get (rec, "render", &len);
186 const char *opac_render = ZOOM_record_get (rec, "opac", &opac_len);
187 const char *syntax = ZOOM_record_get (rec, "syntax", 0);
188 /* if rec is non-null, we got a record for display */
192 (void) oid_name_to_dotstring(CLASS_RECSYN, syntax, oidbuf);
193 printf ("%d %s %s (%s)\n",
194 pos+1, (db ? db : "unknown"), syntax, oidbuf);
196 fwrite (render, 1, len, stdout);
199 fwrite (opac_render, 1, opac_len, stdout);
205 static void cmd_show (ZOOM_connection *c, ZOOM_resultset *r,
206 ZOOM_options options,
210 char start_str[10], count_str[10];
212 if (next_token_copy (args, start_str, sizeof(start_str)) >= 0)
213 ZOOM_options_set (options, "start", start_str);
215 if (next_token_copy (args, count_str, sizeof(count_str)) >= 0)
216 ZOOM_options_set (options, "count", count_str);
218 for (i = 0; i<MAX_CON; i++)
219 ZOOM_resultset_records (r[i], 0, atoi(start_str), atoi(count_str));
220 while (ZOOM_event (MAX_CON, c))
223 for (i = 0; i<MAX_CON; i++)
226 const char *errmsg, *addinfo, *dset;
227 /* display errors if any */
230 if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset)))
231 printf ("%s error: %s (%s:%d) %s\n",
232 ZOOM_connection_option_get(c[i], "host"), errmsg,
233 dset, error, addinfo);
236 /* OK, no major errors. Display records... */
237 int start = ZOOM_options_get_int (options, "start", 0);
238 int count = ZOOM_options_get_int (options, "count", 0);
239 display_records (c[i], r[i], start, count);
242 ZOOM_options_set (options, "count", "0");
243 ZOOM_options_set (options, "start", "0");
246 static void cmd_ext (ZOOM_connection *c, ZOOM_resultset *r,
247 ZOOM_options options,
250 ZOOM_package p[MAX_CON];
251 char ext_type_str[10];
255 if (next_token_copy (args, ext_type_str, sizeof(ext_type_str)) < 0)
258 for (i = 0; i<MAX_CON; i++)
262 p[i] = ZOOM_connection_package (c[i], 0);
263 ZOOM_package_send(p[i], ext_type_str);
269 while (ZOOM_event (MAX_CON, c))
272 for (i = 0; i<MAX_CON; i++)
275 const char *errmsg, *addinfo, *dset;
276 /* display errors if any */
279 if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset)))
280 printf ("%s error: %s (%s:%d) %s\n",
281 ZOOM_connection_option_get(c[i], "host"), errmsg,
282 dset, error, addinfo);
287 v = ZOOM_package_option_get (p[i], "targetReference");
289 printf("targetReference: %s\n", v);
290 v = ZOOM_package_option_get (p[i], "xmlUpdateDoc");
292 printf("xmlUpdateDoc: %s\n", v);
294 ZOOM_package_destroy (p[i]);
298 static void cmd_debug (ZOOM_connection *c, ZOOM_resultset *r,
299 ZOOM_options options,
302 yaz_log_init_level(YLOG_ALL);
305 static void cmd_search (ZOOM_connection *c, ZOOM_resultset *r,
306 ZOOM_options options,
310 const char *query_str = *args;
313 s = ZOOM_query_create ();
314 while (*query_str == ' ')
316 if (memcmp(query_str, "cql:", 4) == 0)
318 ZOOM_query_cql (s, query_str + 4);
320 else if (ZOOM_query_prefix (s, query_str))
322 printf ("Bad PQF: %s\n", query_str);
325 for (i = 0; i<MAX_CON; i++)
329 ZOOM_resultset_destroy (r[i]);
333 r[i] = ZOOM_connection_search (c[i], s);
336 while (ZOOM_event (MAX_CON, c))
339 for (i = 0; i<MAX_CON; i++)
342 const char *errmsg, *addinfo, *dset;
343 /* display errors if any */
346 if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset)))
347 printf ("%s error: %s (%s:%d) %s\n",
348 ZOOM_connection_option_get(c[i], "host"), errmsg,
349 dset, error, addinfo);
352 /* OK, no major errors. Look at the result count */
353 int start = ZOOM_options_get_int (options, "start", 0);
354 int count = ZOOM_options_get_int (options, "count", 0);
356 printf ("%s: %d hits\n", ZOOM_connection_option_get(c[i], "host"),
357 ZOOM_resultset_size(r[i]));
359 display_records (c[i], r[i], start, count);
362 ZOOM_query_destroy (s);
365 static void cmd_scan (ZOOM_connection *c, ZOOM_resultset *r,
366 ZOOM_options options,
369 const char *start_term = *args;
371 ZOOM_scanset s[MAX_CON];
373 while (*start_term == ' ')
376 for (i = 0; i<MAX_CON; i++)
379 s[i] = ZOOM_connection_scan(c[i], start_term);
383 while (ZOOM_event(MAX_CON, c))
385 for (i = 0; i<MAX_CON; i++)
388 size_t p, sz = ZOOM_scanset_size(s[i]);
389 for (p = 0; p < sz; p++)
393 const char *term = ZOOM_scanset_display_term(s[i], p,
395 fwrite(term, 1, len, stdout);
396 printf (" %d\n", occ);
398 ZOOM_scanset_destroy(s[i]);
403 static void cmd_sort (ZOOM_connection *c, ZOOM_resultset *r,
404 ZOOM_options options,
407 const char *sort_spec = *args;
410 while (*sort_spec == ' ')
413 for (i = 0; i<MAX_CON; i++)
416 ZOOM_resultset_sort(r[i], "yaz", sort_spec);
418 while (ZOOM_event(MAX_CON, c))
422 static void cmd_help (ZOOM_connection *c, ZOOM_resultset *r,
423 ZOOM_options options,
426 printf ("connect <zurl>\n");
427 printf ("search <pqf>\n");
428 printf ("show [<start> [<count>]\n");
429 printf ("scan <term>\n");
431 printf ("close <zurl>\n");
432 printf ("ext <type>\n");
433 printf ("set <option> [<value>]\n");
434 printf ("get <option>\n");
436 printf ("options:\n");
439 printf (" databaseName\n");
440 printf (" preferredRecordSyntax\n");
442 printf (" elementSetName\n");
443 printf (" maximumRecordSize\n");
444 printf (" preferredRecordSize\n");
446 printf (" piggyback\n");
449 printf (" password\n");
450 printf (" implementationName\n");
451 printf (" charset\n");
455 static void cmd_connect (ZOOM_connection *c, ZOOM_resultset *r,
456 ZOOM_options options,
460 const char *errmsg, *addinfo, *dset;
463 if (next_token_copy (args, host, sizeof(host)) < 0)
465 printf ("missing host after connect\n");
468 for (j = -1, i = 0; i<MAX_CON; i++)
471 if (c[i] && (h = ZOOM_connection_option_get(c[i], "host")) &&
474 ZOOM_connection_destroy (c[i]);
477 else if (c[i] == 0 && j == -1)
480 if (i == MAX_CON) /* no match .. */
484 printf ("no more connection available\n");
487 i = j; /* OK, use this one is available */
489 c[i] = ZOOM_connection_create (options);
490 ZOOM_connection_connect (c[i], host, 0);
492 if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset)))
493 printf ("%s error: %s (%s:%d) %s\n",
494 ZOOM_connection_option_get(c[i], "host"), errmsg,
495 dset, error, addinfo);
498 static int cmd_parse (ZOOM_connection *c, ZOOM_resultset *r,
499 ZOOM_options options,
505 cmd_len = next_token (buf, &cmd_str);
508 if (is_command ("quit", cmd_str, cmd_len))
510 else if (is_command ("set", cmd_str, cmd_len))
511 cmd_set (c, r, options, buf);
512 else if (is_command ("get", cmd_str, cmd_len))
513 cmd_get (c, r, options, buf);
514 else if (is_command ("rget", cmd_str, cmd_len))
515 cmd_rget (c, r, options, buf);
516 else if (is_command ("connect", cmd_str, cmd_len))
517 cmd_connect (c, r, options, buf);
518 else if (is_command ("open", cmd_str, cmd_len))
519 cmd_connect (c, r, options, buf);
520 else if (is_command ("search", cmd_str, cmd_len))
521 cmd_search (c, r, options, buf);
522 else if (is_command ("find", cmd_str, cmd_len))
523 cmd_search (c, r, options, buf);
524 else if (is_command ("show", cmd_str, cmd_len))
525 cmd_show (c, r, options, buf);
526 else if (is_command ("close", cmd_str, cmd_len))
527 cmd_close (c, r, options, buf);
528 else if (is_command ("help", cmd_str, cmd_len))
529 cmd_help(c, r, options, buf);
530 else if (is_command ("ext", cmd_str, cmd_len))
531 cmd_ext(c, r, options, buf);
532 else if (is_command ("debug", cmd_str, cmd_len))
533 cmd_debug(c, r, options, buf);
534 else if (is_command ("scan", cmd_str, cmd_len))
535 cmd_scan(c, r, options, buf);
536 else if (is_command ("sort", cmd_str, cmd_len))
537 cmd_sort(c, r, options, buf);
539 printf ("unknown command %.*s\n", cmd_len, cmd_str);
543 void shell(ZOOM_connection *c, ZOOM_resultset *r,
544 ZOOM_options options)
550 const char *bp = buf;
551 #if HAVE_READLINE_READLINE_H
553 line_in=readline("ZOOM>");
556 #if HAVE_READLINE_HISTORY_H
558 add_history(line_in);
560 if(strlen(line_in) > 999) {
561 printf("Input line too long\n");
567 printf ("ZOOM>"); fflush (stdout);
568 if (!fgets (buf, 999, stdin))
571 if ((cp = strchr(buf, '\n')))
573 if (!cmd_parse (c, r, options, &bp))
578 static void zoomsh(int argc, char **argv)
580 ZOOM_options options = ZOOM_options_create();
582 ZOOM_connection z39_con[MAX_CON];
583 ZOOM_resultset z39_res[MAX_CON];
585 for (i = 0; i<MAX_CON; i++)
591 for (i = 0; i<MAX_CON; i++)
595 for (i = 1; i<argc; i++)
597 const char *bp = argv[i];
598 res = cmd_parse(z39_con, z39_res, options, &bp);
599 if (res == 0) /* received quit */
602 if (res) /* do cmdline shell only if not quitting */
603 shell(z39_con, z39_res, options);
604 ZOOM_options_destroy(options);
606 for (i = 0; i<MAX_CON; i++)
608 ZOOM_connection_destroy(z39_con[i]);
609 ZOOM_resultset_destroy(z39_res[i]);
613 int main(int argc, char **argv)
615 const char *maskstr = 0;
616 if (argc > 2 && !strcmp(argv[1], "-v"))
622 else if (argc > 1 && !strncmp(argv[1], "-v", 2))
630 int mask = yaz_log_mask_str(maskstr);
631 yaz_log_init_level(mask);
641 * indent-tabs-mode: nil
643 * vim: shiftwidth=4 tabstop=8 expandtab