X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=zoom%2Fzoomsh.c;h=37f5452223f3572d20e9da37e3eafdae9ef17493;hp=5f64c040da815a9d442ad3ea77669a1119f0ef5d;hb=11ed1645174d8ca8d8dcf5ef2bbe91e70a4a3e00;hpb=c12f2351ad72e5e4788f30684de0d98e5bc0dc1e diff --git a/zoom/zoomsh.c b/zoom/zoomsh.c index 5f64c04..37f5452 100644 --- a/zoom/zoomsh.c +++ b/zoom/zoomsh.c @@ -1,36 +1,86 @@ -/* - * $Id: zoomsh.c,v 1.21 2003-07-09 23:00:21 mike Exp $ - * - * ZOOM-C Shell +/* This file is part of the YAZ toolkit. + * Copyright (C) Index Data + * See the file LICENSE for details. */ +/** \file zoomsh.c + \brief ZOOM C command line tool (shell) +*/ +#if HAVE_CONFIG_H +#include +#endif + +#if HAVE_UNISTD_H +#include +#endif #include #include #include -#include +#include +#include +#include +#include #if HAVE_READLINE_READLINE_H -#include +#include #endif #if HAVE_READLINE_HISTORY_H #include #endif -#include - #include -#include #include -#include + +struct zoom_sh { + WRBUF strategy; + WRBUF criteria; + ZOOM_options options; + struct zoom_db *list; +}; + +struct zoom_db { + ZOOM_connection con; + ZOOM_resultset res; + struct zoom_db *next; +}; #define MAX_CON 100 -static int next_token (const char **cpp, const char **t_start) +static void process_events(struct zoom_sh *sh) +{ + ZOOM_connection *c; + int i, number; + struct zoom_db *db; + + for (number = 0, db = sh->list; db; db = db->next) + if (db->con) + number++; + c = xmalloc(sizeof(*c) * number); + + for (i = 0, db = sh->list; db; db = db->next) + if (db->con) + c[i++] = db->con; + + yaz_log(YLOG_DEBUG, "process_events"); + while ((i = ZOOM_event(number, 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)); + } + xfree(c); +} + +static int next_token_chars(const char **cpp, const char **t_start, + const char *tok_chars) { int len = 0; const char *cp = *cpp; while (*cp == ' ') - cp++; + cp++; if (*cp == '"') { cp++; @@ -46,7 +96,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++; @@ -58,517 +108,915 @@ 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); + 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) +static int is_command(const char *cmd_str, const char *this_str, int this_len) { int cmd_len = strlen(cmd_str); if (cmd_len != this_len) - return 0; - if (memcmp (cmd_str, this_str, cmd_len)) - return 0; + return 0; + if (memcmp(cmd_str, this_str, cmd_len)) + return 0; return 1; } -static void cmd_set (ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_set(struct zoom_sh *sh, 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 ; + printf("missing argument for set\n"); + 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(sh->options, wrbuf_cstr(key), val_buf, val_len); else - ZOOM_options_set(options, key, val); + ZOOM_options_set(sh->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(struct zoom_sh *sh, 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"); + 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(sh->options, 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_shell(struct zoom_sh *sh, const char **args) { - char host[60]; - int i; - next_token_copy (args, host, sizeof(host)); - for (i = 0; ilist; + for (; db; db = db->next) + { + if (db->res) + { + const char *val = + ZOOM_resultset_option_get(db->res, wrbuf_cstr(key)); + printf("%s = %s\n", wrbuf_cstr(key), val ? val : ""); + } + } + wrbuf_destroy(key); } + return 0; } -static const char *oid_name_to_dotstring(const char *name) { - struct oident ent; - int oid[OID_SIZE]; - static char oidbuf[100]; /* ### bad interface */ - int i; +static int cmd_close(struct zoom_sh *sh, const char **args) +{ + struct zoom_db **dbp; + WRBUF host; + host = next_token_new_wrbuf(args); + + for (dbp = &sh->list; *dbp; ) + { + const char *h; + struct zoom_db *db = *dbp; + if (!db->con || !host || + ((h = ZOOM_connection_option_get(db->con, "host")) + && !strcmp(h, wrbuf_cstr(host)))) + { + *dbp = (*dbp)->next; + ZOOM_connection_destroy(db->con); + ZOOM_resultset_destroy(db->res); + xfree(db); + } + else + dbp = &(*dbp)->next; + } + if (host) + wrbuf_destroy(host); + return 0; +} + +static void display_records(ZOOM_connection c, + ZOOM_resultset r, + size_t start, size_t count, const char *type) +{ + size_t i; + for (i = 0; i < count; i++) + { + size_t pos = i + start; + ZOOM_record rec = ZOOM_resultset_record(r, pos); + const char *db = ZOOM_record_get(rec, "database", 0); - /* Translate syntax to oid_val */ - oid_value value = oid_getvalbyname(name); + if (ZOOM_record_error(rec, 0, 0, 0)) + { + const char *msg; + const char *addinfo; + const char *diagset; + int error = ZOOM_record_error(rec, &msg, &addinfo, &diagset); + + printf("%lld %s: %s (%s:%d) %s\n", (long long) pos, + (db ? db : "unknown"), + msg, diagset, error, addinfo ? addinfo : "none"); + } + else + { + int len; + const char *render = ZOOM_record_get(rec, type, &len); + const char *syntax = ZOOM_record_get(rec, "syntax", 0); + const char *schema = ZOOM_record_get(rec, "schema", 0); + /* if rec is non-null, we got a record for display */ + if (rec) + { + printf("%lld database=%s syntax=%s schema=%s\n", + (long long) pos, (db ? db : "unknown"), syntax, + schema ? schema : "unknown"); + if (render) + { + if (fwrite(render, 1, len, stdout) != (size_t) len) + { + printf("write to stdout failed\n"); + } + } + printf("\n"); + } + } + } +} - /* Build it into an oident */ - ent.proto = PROTO_Z3950; - ent.oclass = CLASS_RECSYN; - ent.value = value; +static int cmd_show(struct zoom_sh *sh, const char **args) +{ + size_t start = 0, count = 1; + const char *type = "render"; + WRBUF render_str = 0; + int ret = 0; + struct zoom_db *db; - /* Translate to an array of int */ - (void) oid_ent_to_oid(&ent, oid); + { + WRBUF tmp; - /* Write the array of int into a dotted string (phew!) */ - oidbuf[0] = '\0'; - for (i = 0; oid[i] != -1; i++) { - char tmpbuf[20]; - sprintf(tmpbuf, "%d", oid[i]); - if (i > 0) strcat(oidbuf, "."); - strcat(oidbuf, tmpbuf); + 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); - return oidbuf; + for (db = sh->list; db; db = db->next) + if (db->res) + ZOOM_resultset_records(db->res, 0, start, count); + process_events(sh); + + for (db = sh->list; db; db = db->next) + { + int error; + const char *errmsg, *addinfo, *dset; + /* display errors if any */ + if (!db->con) + continue; + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, &dset))) + { + printf("%s error: %s (%s:%d) %s\n", + ZOOM_connection_option_get(db->con, "host"), errmsg, + dset, error, addinfo); + ret = 1; + } + else if (db->res) + { + /* OK, no major errors. Display records... */ + display_records(db->con, db->res, start, count, type); + } + } + if (render_str) + wrbuf_destroy(render_str); + return ret; } -static void display_records (ZOOM_connection c, - ZOOM_resultset r, - int start, int count) +static void display_facets(ZOOM_facet_field *facets, int count) { int i; - for (i = 0; i= 0) - ZOOM_options_set (options, "start", start_str); - - if (next_token_copy (args, count_str, sizeof(count_str)) >= 0) - ZOOM_options_set (options, "count", count_str); - - for (i = 0; ilist; db; db = db->next) + { + int error; + const char *errmsg, *addinfo, *dset; + /* display errors if any */ + if (!db->con) + continue; + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) + { + printf("%s error: %s (%s:%d) %s\n", + ZOOM_connection_option_get(db->con, "host"), errmsg, + dset, error, addinfo); + ret = 1; + } + else if (db->res) + { + int num_facets = ZOOM_resultset_facets_size(db->res); + if (num_facets) { + ZOOM_facet_field *facets = ZOOM_resultset_facets(db->res); + display_facets(facets, num_facets); + } + } + } + return ret; } -static void cmd_ext (ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_suggestions(struct zoom_sh *sh, const char **args) { - ZOOM_package p[MAX_CON]; - - int i; - - for (i = 0; ilist; db; db = db->next) { - if (c[i]) + int error; + const char *errmsg, *addinfo, *dset; + /* display errors if any */ + if (!db->con) + continue; + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) { - p[i] = ZOOM_connection_package (c[i], 0); - ZOOM_package_send(p[i], "itemorder"); + printf("%s error: %s (%s:%d) %s\n", + ZOOM_connection_option_get(db->con, "host"), errmsg, + dset, error, addinfo); + ret = 1; } - else - p[i] = 0; + else if (db->res) + { + const char *suggestions = + ZOOM_resultset_option_get(db->res, "suggestions"); + if (suggestions) + printf("Suggestions: \n%s\n", suggestions); + } + } + return ret; +} + +static int cmd_ext(struct zoom_sh *sh, const char **args) +{ + int ret = 0; + ZOOM_package *p = 0; + struct zoom_db *db; + int i, number; + 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 (number = 0, db = sh->list; db; db = db->next) + if (db->con) + number++; - while (ZOOM_event (MAX_CON, c)) - ; + p = xmalloc(sizeof(*p) * number); - for (i = 0; ilist; db; db = db->next) + if (db->con) + { + p[i] = ZOOM_connection_package(db->con, 0); + ZOOM_package_send(p[i], ext_type_str ? wrbuf_cstr(ext_type_str):0); + i++; + } + + process_events(sh); + + for (i = 0, db = sh->list; db; db = db->next) { - int error; - const char *errmsg, *addinfo, *dset; - /* display errors if any */ - if (!p[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); - else if (p[i]) - { - printf ("ok\n"); - } - ZOOM_package_destroy (p[i]); + int error; + const char *errmsg, *addinfo, *dset; + /* display errors if any */ + if (!db->con) + continue; + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) + { + printf("%s error: %s (%s:%d) %s\n", + ZOOM_connection_option_get(db->con, "host"), errmsg, + dset, error, addinfo); + ret = 1; + } + else if (p[i]) + { + const char *v; + printf("ok\n"); + v = ZOOM_package_option_get(p[i], "targetReference"); + if (v) + printf("targetReference: %s\n", v); + v = ZOOM_package_option_get(p[i], "xmlUpdateDoc"); + if (v) + printf("xmlUpdateDoc: %s\n", v); + } + ZOOM_package_destroy(p[i]); + i++; } + if (ext_type_str) + wrbuf_destroy(ext_type_str); + xfree(p); + return ret; } -static void cmd_debug (ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_debug(struct zoom_sh *sh, const char **args) { - yaz_log_init_level(LOG_ALL); + yaz_log_init_level(YLOG_ALL); + return 0; } -static void cmd_search (ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static void display_search_result(struct zoom_db *db) +{ + const char *v; + int num; + + v = ZOOM_resultset_option_get(db->res, "searchresult.size"); + if (v && (num = atoi(v))) + { + int i; + printf("SearchResult-1:"); + for (i = 0; i < num; i++) + { + const char *v; + char str[60]; + + if (i) + printf(","); + + sprintf(str, "searchresult.%d.id", i); + v = ZOOM_resultset_option_get(db->res, str); + if (v) + printf(" id=%s", v); + + sprintf(str, "searchresult.%d.subquery.term", i); + v = ZOOM_resultset_option_get(db->res, str); + if (v) + printf(" term=%s", v); + + sprintf(str, "searchresult.%d.count", i); + v = ZOOM_resultset_option_get(db->res, str); + if (v) + printf(" cnt=%s", v); + } + printf("\n"); + } +} + +static int cmd_search(struct zoom_sh *sh, const char **args) { ZOOM_query s; const char *query_str = *args; - int i; - - s = ZOOM_query_create (); + int ret = 0; + struct zoom_db *db; + + s = ZOOM_query_create(); while (*query_str == ' ') query_str++; if (memcmp(query_str, "cql:", 4) == 0) { - ZOOM_query_cql (s, query_str + 4); + ZOOM_query_cql(s, query_str + 4); } - else if (ZOOM_query_prefix (s, query_str)) + else if (ZOOM_query_prefix(s, query_str)) { - printf ("Bad PQF: %s\n", query_str); - return; + printf("Bad PQF: %s\n", query_str); + ZOOM_query_destroy(s); + return 1; } - for (i = 0; istrategy && wrbuf_len(sh->strategy) && wrbuf_len(sh->criteria)) { - if (c[i]) - { - ZOOM_resultset_destroy (r[i]); - r[i] = 0; - } - if (c[i]) - r[i] = ZOOM_connection_search (c[i], s); + int r = ZOOM_query_sortby2(s, wrbuf_cstr(sh->strategy), + wrbuf_cstr(sh->criteria)); + if (r) + { + if (r == -1) + printf("Bad sortby strategy: %s\n", wrbuf_cstr(sh->strategy)); + else + printf("Bad sortby criteria: %s\n", wrbuf_cstr(sh->criteria)); + ZOOM_query_destroy(s); + return 1; + } + printf("sortby added\n"); + } + for (db = sh->list; db; db = db->next) + { + if (db->con) + { + ZOOM_resultset_destroy(db->res); + db->res = ZOOM_connection_search(db->con, s); + } } + ZOOM_query_destroy(s); - while (ZOOM_event (MAX_CON, c)) - ; + process_events(sh); - for (i = 0; ilist; db; db = db->next) { - 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); - else if (r[i]) - { - /* OK, no major errors. Look at the result count */ - int start = ZOOM_options_get_int (options, "start", 0); - int count = ZOOM_options_get_int (options, "count", 0); + int error; + const char *errmsg, *addinfo, *dset; + /* display errors if any */ + if (!db->con) + continue; + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) + { + printf("%s error: %s (%s:%d) %s\n", + ZOOM_connection_option_get(db->con, "host"), errmsg, + dset, error, addinfo); + ret = 1; + } + else if (db->res) + { + /* OK, no major errors. Look at the result count */ + int start = ZOOM_options_get_int(sh->options, "start", 0); + int count = ZOOM_options_get_int(sh->options, "count", 0); + int facet_num; + + printf("%s: %lld hits\n", ZOOM_connection_option_get(db->con, + "host"), + (long long int) ZOOM_resultset_size(db->res)); - printf ("%s: %d hits\n", ZOOM_connection_option_get(c[i], "host"), - ZOOM_resultset_size(r[i])); - /* and display */ - display_records (c[i], r[i], start, count); - } + facet_num = ZOOM_resultset_facets_size(db->res); + if (facet_num) + { + ZOOM_facet_field *facets = ZOOM_resultset_facets(db->res); + int facet_idx; + for (facet_idx = 0; facet_idx < facet_num; facet_idx++) + { + const char *name = ZOOM_facet_field_name(facets[facet_idx]); + size_t term_idx; + size_t term_num = ZOOM_facet_field_term_count(facets[facet_idx]); + printf("facet: %s\n", name); + for (term_idx = 0; term_idx < term_num; term_idx++ ) + { + int freq; + const char *term = + ZOOM_facet_field_get_term(facets[facet_idx], term_idx, &freq); + printf("term: %s %d\n", term, freq); + } + } + } + display_search_result(db); + /* and display */ + display_records(db->con, db->res, start, count, "render"); + } } - ZOOM_query_destroy (s); + return ret; } -static void cmd_scan (ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_scan(struct zoom_sh *sh, const char **args) { - const char *start_term = *args; - int i; - ZOOM_scanset s[MAX_CON]; - - while (*start_term == ' ') - start_term++; + const char *query_str = *args; + ZOOM_query query = ZOOM_query_create(); + int i, number; + int ret = 0; + ZOOM_scanset *s = 0; + struct zoom_db *db; + + while (*query_str == ' ') + query_str++; - for (i = 0; ilist; db; db = db->next) + if (db->con) + number++; + + s = xmalloc(sizeof(*s) * number); + + for (i = 0, db = sh->list; db; db = db->next) + if (db->con) + s[i++] = ZOOM_connection_scan1(db->con, query); + + ZOOM_query_destroy(query); + + process_events(sh); + + for (i = 0, db = sh->list; db; db = db->next) + { + int error; + const char *errmsg, *addinfo, *dset; + /* display errors if any */ + if (!db->con) + continue; + if ((error = ZOOM_connection_error_x(db->con, &errmsg, &addinfo, + &dset))) + { + printf("%s error: %s (%s:%d) %s\n", + ZOOM_connection_option_get(db->con, "host"), errmsg, + dset, error, addinfo); + ret = 1; + } + if (s[i]) + { size_t p, sz = ZOOM_scanset_size(s[i]); for (p = 0; p < sz; p++) { - int occ = 0; - int len = 0; - const char *term = ZOOM_scanset_term(s[i], p, &occ, &len); - fwrite(term, 1, len, stdout); - printf (" %d\n", occ); - } + size_t occ = 0; + size_t len = 0; + const char *term = ZOOM_scanset_display_term(s[i], p, + &occ, &len); + printf("%.*s %lld\n", (int) len, term, (long long int) occ); + } ZOOM_scanset_destroy(s[i]); } + i++; + } + xfree(s); + return ret; +} + +static int cmd_sortby(struct zoom_sh *sh, const char **args) +{ + WRBUF strategy; + const char *criteria; + if (!(strategy = next_token_new_wrbuf(args))) + { + printf("missing argument argument: strategy and criteria\n"); + return 1; } + criteria = *args; + while (*criteria == ' ') + criteria++; + wrbuf_destroy(sh->strategy); + sh->strategy = strategy; + + wrbuf_rewind(sh->criteria); + wrbuf_puts(sh->criteria, criteria); + return 0; +} + +static int cmd_sort(struct zoom_sh *sh, const char **args) +{ + const char *sort_spec = *args; + int ret = 0; + struct zoom_db *db; + + while (*sort_spec == ' ') + sort_spec++; + + for (db = sh->list; db; db = db->next) + if (db->res) + ZOOM_resultset_sort(db->res, "yaz", sort_spec); + process_events(sh); + return ret; } -static void cmd_help (ZOOM_connection *c, ZOOM_resultset *r, - ZOOM_options options, - const char **args) +static int cmd_help(struct zoom_sh *sh, const char **args) { - printf ("connect \n"); - printf ("search \n"); - printf ("show [ []\n"); - printf ("scan \n"); - printf ("quit\n"); - printf ("close \n"); - printf ("set