X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=zoom%2Fzoomsh.c;h=81c8d7d3273d37bac6e24787023b09163ab75c78;hp=2c93b6e2127c374280a94c8a06358131be750af5;hb=cc94e0b0a1ff9b42754b57049434df7db6fb0e7c;hpb=7dbee1f341772642df14925d639f03aec2eb3355 diff --git a/zoom/zoomsh.c b/zoom/zoomsh.c index 2c93b6e..81c8d7d 100644 --- a/zoom/zoomsh.c +++ b/zoom/zoomsh.c @@ -1,19 +1,23 @@ /* This file is part of the YAZ toolkit. - * Copyright (C) 1995-2009 Index Data + * Copyright (C) 1995-2013 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 +#include #endif #if HAVE_READLINE_HISTORY_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,65 @@ 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_shell(ZOOM_connection *c, ZOOM_resultset *r, + ZOOM_options options, const char **args) +{ + int ret = system(*args); + if (ret) + printf("system command returned %d\n", ret); + return 0; +} + +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 { @@ -133,37 +175,42 @@ static void cmd_rget(ZOOM_connection *c, ZOOM_resultset *r, const char *val; 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 +301,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) +{ + 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; +} + + +static int cmd_ext(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; + WRBUF ext_type_str = next_token_new_wrbuf(args); - if (next_token_copy(args, ext_type_str, sizeof(ext_type_str)) < 0) - return; - + 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"); printf("ext \n"); printf("set