X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=zoom%2Fzoomsh.c;h=26aab7b54f840f4bfc48c341be56fccf2c860eb9;hp=0d2047117e95ff3c021bfba6c489c510ede32608;hb=80184e0ebd6d82ad97874372236c2ad956e01d72;hpb=eceac7b7869288887b1b14212208e73ab54116db diff --git a/zoom/zoomsh.c b/zoom/zoomsh.c index 0d20471..26aab7b 100644 --- a/zoom/zoomsh.c +++ b/zoom/zoomsh.c @@ -1,16 +1,20 @@ /* This file is part of the YAZ toolkit. - * Copyright (C) 1995-2009 Index Data + * Copyright (C) 1995-2012 Index Data * See the file LICENSE for details. */ - /** \file zoomsh.c \brief ZOOM C command line tool (shell) */ +#if HAVE_CONFIG_H +#include +#endif #include #include #include -#include +#include +#include +#include #if HAVE_READLINE_READLINE_H #include @@ -24,7 +28,24 @@ #define MAX_CON 100 -static int next_token(const char **cpp, const char **t_start) +static void process_events(ZOOM_connection *c) +{ + int i; + + yaz_log(YLOG_DEBUG, "process_events"); + while ((i = ZOOM_event(MAX_CON, c)) != 0) + { + int peek = ZOOM_connection_peek_event(c[i-1]); + int event = ZOOM_connection_last_event(c[i-1]); + yaz_log(YLOG_DEBUG, "no = %d peek = %d event = %d %s", i-1, + peek, + event, + ZOOM_get_event_str(event)); + } +} + +static int next_token_chars(const char **cpp, const char **t_start, + const char *tok_chars) { int len = 0; const char *cp = *cpp; @@ -45,7 +66,7 @@ static int next_token(const char **cpp, const char **t_start) else { *t_start = cp; - while (*cp && *cp != ' ' && *cp != '\r' && *cp != '\n') + while (*cp && !strchr(tok_chars, *cp)) { cp++; len++; @@ -57,20 +78,23 @@ static int next_token(const char **cpp, const char **t_start) return len; /* return -1 if no token was read .. */ } -static int next_token_copy(const char **cpp, char *buf_out, int buf_max) +static int next_token(const char **cpp, const char **t_start) { + return next_token_chars(cpp, t_start, "\r\n "); +} + + +static WRBUF next_token_new_wrbuf(const char **cpp) +{ + WRBUF w = 0; const char *start; int len = next_token(cpp, &start); if (len < 0) - { - *buf_out = 0; - return len; - } - if (len >= buf_max) - len = buf_max-1; - memcpy(buf_out, start, len); - buf_out[len] = '\0'; - return len; + return 0; + w = wrbuf_alloc(); + if (len > 0) + wrbuf_write(w, start, len); + return w; } static int is_command(const char *cmd_str, const char *this_str, int this_len) @@ -83,47 +107,56 @@ static int is_command(const char *cmd_str, const char *this_str, int this_len) return 1; } -static void cmd_set(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_set(ZOOM_connection *c, ZOOM_resultset *r, + ZOOM_options options, + const char **args) { - char key[40], val[80]; + WRBUF key; + const char *val_buf; + int val_len; - if (next_token_copy(args, key, sizeof(key)) < 0) + if (!(key = next_token_new_wrbuf(args))) { printf("missing argument for set\n"); - return ; + return 1; } - if (next_token_copy(args, val, sizeof(val)) < 0) - ZOOM_options_set(options, key, 0); + val_len = next_token_chars(args, &val_buf, ""); + if (val_len != -1) + ZOOM_options_setl(options, wrbuf_cstr(key), val_buf, val_len); else - ZOOM_options_set(options, key, val); + ZOOM_options_set(options, wrbuf_cstr(key), 0); + wrbuf_destroy(key); + return 0; } -static void cmd_get(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_get(ZOOM_connection *c, ZOOM_resultset *r, + ZOOM_options options, + const char **args) { - char key[40]; - if (next_token_copy(args, key, sizeof(key)) < 0) + WRBUF key; + if (!(key = next_token_new_wrbuf(args))) { printf("missing argument for get\n"); + return 1; } else { - const char *val = ZOOM_options_get(options, key); - printf("%s = %s\n", key, val ? val : ""); + const char *val = ZOOM_options_get(options, wrbuf_cstr(key)); + printf("%s = %s\n", wrbuf_cstr(key), val ? val : ""); + wrbuf_destroy(key); } + return 0; } -static void cmd_rget(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, +static int cmd_rget(ZOOM_connection *c, ZOOM_resultset *r, + ZOOM_options options, const char **args) { - char key[40]; - if (next_token_copy(args, key, sizeof(key)) < 0) + WRBUF key; + if (!(key = next_token_new_wrbuf(args))) { printf("missing argument for get\n"); + return 1; } else { @@ -134,36 +167,41 @@ static void cmd_rget(ZOOM_connection *c, ZOOM_resultset *r, if (!r[i]) continue; - val = ZOOM_resultset_option_get(r[i], key); - printf("%s = %s\n", key, val ? val : ""); + val = ZOOM_resultset_option_get(r[i], wrbuf_cstr(key)); + printf("%s = %s\n", wrbuf_cstr(key), val ? val : ""); } + wrbuf_destroy(key); } + return 0; } -static void cmd_close(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_close(ZOOM_connection *c, ZOOM_resultset *r, + ZOOM_options options, + const char **args) { - char host[60]; + WRBUF host; int i; - next_token_copy(args, host, sizeof(host)); + host = next_token_new_wrbuf(args); for (i = 0; i 0) - start = atoi(start_str); - - if (next_token_copy(args, count_str, sizeof(count_str)) > 0) - count = atoi(count_str); - - if (next_token_copy(args, render_str, sizeof(render_str)) > 0) - type = render_str; + WRBUF tmp; + + if ((tmp = next_token_new_wrbuf(args))) + { + start = atoi(wrbuf_cstr(tmp)); + wrbuf_destroy(tmp); + } + + if ((tmp = next_token_new_wrbuf(args))) + { + count = atoi(wrbuf_cstr(tmp)); + wrbuf_destroy(tmp); + } + render_str = next_token_new_wrbuf(args); } + if (render_str) + type = wrbuf_cstr(render_str); for (i = 0; i < MAX_CON; i++) ZOOM_resultset_records(r[i], 0, start, count); - while (ZOOM_event(MAX_CON, c)) - ; + process_events(c); for (i = 0; i < MAX_CON; i++) { @@ -254,42 +292,133 @@ static void cmd_show(ZOOM_connection *c, ZOOM_resultset *r, if (!c[i]) continue; if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset))) + { printf("%s error: %s (%s:%d) %s\n", ZOOM_connection_option_get(c[i], "host"), errmsg, dset, error, addinfo); + ret = 1; + } else if (r[i]) { /* OK, no major errors. Display records... */ display_records(c[i], r[i], start, count, type); } } + if (render_str) + wrbuf_destroy(render_str); + return ret; } -static void cmd_ext(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static void display_facets(ZOOM_facet_field *facets, int count) { + int index; + printf("Facets: \n"); + for (index = 0; index < count; index++) { + int term_index; + const char *facet_name = ZOOM_facet_field_name(facets[index]); + printf(" %s: \n", facet_name); + for (term_index = 0; term_index < ZOOM_facet_field_term_count(facets[index]); term_index++) { + int freq = 0; + const char *term = ZOOM_facet_field_get_term(facets[index], term_index, &freq); + printf(" %s(%d) \n", term, freq); + } + } +} + +static int cmd_facets(ZOOM_connection *c, ZOOM_resultset *r, + ZOOM_options options, + const char **args) { - ZOOM_package p[MAX_CON]; - char ext_type_str[10]; - int i; + int ret = 0; + + process_events(c); + + for (i = 0; i < MAX_CON; i++) + { + int error; + const char *errmsg, *addinfo, *dset; + /* display errors if any */ + if (!c[i]) + continue; + if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset))) + { + printf("%s error: %s (%s:%d) %s\n", + ZOOM_connection_option_get(c[i], "host"), errmsg, + dset, error, addinfo); + ret = 1; + } + else if (r[i]) + { + int num_facets = ZOOM_resultset_facets_size(r[i]); + if (num_facets) { + ZOOM_facet_field *facets = ZOOM_resultset_facets(r[i]); + display_facets(facets, num_facets); + } + } + } + return ret; +} + +static int cmd_suggestions(ZOOM_connection *c, ZOOM_resultset *r, ZOOM_options options, const char **args) +{ + int i; + int ret = 0; + + process_events(c); + + for (i = 0; i < MAX_CON; i++) + { + int error; + const char *errmsg, *addinfo, *dset; + /* display errors if any */ + if (!c[i]) + continue; + if ((error = ZOOM_connection_error_x(c[i], &errmsg, &addinfo, &dset))) + { + printf("%s error: %s (%s:%d) %s\n", + ZOOM_connection_option_get(c[i], "host"), errmsg, + dset, error, addinfo); + ret = 1; + } + else if (r[i]) + { + const char *suggestions = ZOOM_resultset_option_get(r[i], "suggestions"); + if (suggestions) { + printf("Suggestions: \n%s\n", suggestions); + } + } + } + return ret; +} + - if (next_token_copy(args, ext_type_str, sizeof(ext_type_str)) < 0) - return; +static int cmd_ext(ZOOM_connection *c, ZOOM_resultset *r, + ZOOM_options options, + const char **args) +{ + ZOOM_package p[MAX_CON]; + int i; + int ret = 0; + WRBUF ext_type_str = next_token_new_wrbuf(args); + if (!ext_type_str) + { + printf("es: missing type " + "(itemorder, create, drop, commit, update, xmlupdate)\n"); + return 1; + } for (i = 0; i\n"); printf("search \n"); printf("show [ [ [\n"); printf("quit\n"); printf("close \n"); @@ -496,26 +667,31 @@ static void cmd_help(ZOOM_connection *c, ZOOM_resultset *r, printf(" charset\n"); printf(" lang\n"); printf(" timeout\n"); + printf(" facets\n"); + printf(" extraArgs\n"); + printf(" suggestions\n"); + return 0; } -static void cmd_connect(ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_connect(ZOOM_connection *c, ZOOM_resultset *r, + ZOOM_options options, + const char **args) { + int ret = 0; int error; const char *errmsg, *addinfo, *dset; - char host[60]; int j, i; - if (next_token_copy(args, host, sizeof(host)) < 0) + WRBUF host = next_token_new_wrbuf(args); + if (!host) { printf("missing host after connect\n"); - return ; + return 1; } for (j = -1, i = 0; i"); + line_in = readline("ZOOM>"); if (!line_in) + { + res = -1; break; + } #if HAVE_READLINE_HISTORY_H if (*line_in) add_history(line_in); #endif - if(strlen(line_in) > 999) { + if (strlen(line_in) > sizeof(buf)-1) + { printf("Input line too long\n"); + res = 1; break; - }; + } strcpy(buf,line_in); free(line_in); #else printf("ZOOM>"); fflush(stdout); - if (!fgets(buf, 999, stdin)) + if (!fgets(buf, sizeof(buf)-1, stdin)) + { + res = -1; break; + } #endif if ((cp = strchr(buf, '\n'))) *cp = '\0'; - if (!cmd_parse(c, r, options, &bp)) + res = cmd_parse(c, r, options, &bp); + if (res == -1) break; + if (!exit_on_error && res > 0) + res = 0; } + return res; } -static void zoomsh(int argc, char **argv) +static int zoomsh(int argc, char **argv) { - ZOOM_options options = ZOOM_options_create(); - int i, res; + ZOOM_options zoom_options = ZOOM_options_create(); + int i, res = 0; /* -1: EOF; 0 = OK, > 0 ERROR */ + int exit_on_error = 0; ZOOM_connection z39_con[MAX_CON]; ZOOM_resultset z39_res[MAX_CON]; @@ -633,51 +846,51 @@ static void zoomsh(int argc, char **argv) z39_con[i] = 0; z39_res[i] = 0; } - - for (i = 0; i 0) + res = 0; /* hide error */ + break; + case YAZ_OPTIONS_EOF: + res = shell(z39_con, z39_res, zoom_options, exit_on_error); + break; + case 'e': + exit_on_error = 1; break; + case 'v': + mask = yaz_log_mask_str(arg); + yaz_log_init_level(mask); + break; + default: + fprintf(stderr, "zoomsh: [-e] [-v] [commands]\n"); + res = 1; + } } - if (res) /* do cmdline shell only if not quitting */ - shell(z39_con, z39_res, options); - ZOOM_options_destroy(options); for (i = 0; i 2 && !strcmp(argv[1], "-v")) - { - maskstr = argv[2]; - argv += 2; - argc -= 2; - } - else if (argc > 1 && !strncmp(argv[1], "-v", 2)) - { - maskstr = argv[1]+2; - argv++; - argc--; - } - if (maskstr) - { - int mask = yaz_log_mask_str(maskstr); - yaz_log_init_level(mask); - } - zoomsh(argc, argv); - exit(0); + int ret = zoomsh(argc, argv); + exit(ret); } /* * Local variables: