2 * Copyright (c) 1998-2004, Index Data.
3 * See the file LICENSE for details.
5 * $Id: yaz-my-client.cpp,v 1.26 2007-04-10 12:14:04 adam Exp $
10 #include <yaz/options.h>
11 #include <yaz/diagbib1.h>
12 #include <yaz/marcdisp.h>
13 #include <yazpp/ir-assoc.h>
14 #include <yazpp/pdu-assoc.h>
15 #include <yazpp/socket-manager.h>
18 #if HAVE_READLINE_READLINE_H
19 #include <readline/readline.h>
21 #if HAVE_READLINE_HISTORY_H
22 #include <readline/history.h>
26 using namespace yazpp_1;
28 class YAZ_EXPORT MyClient : public IR_Assoc {
30 int m_interactive_flag;
31 char m_thisCommand[1024];
32 char m_lastCommand[1024];
34 SocketManager *m_socketManager;
36 MyClient(IPDU_Observable *the_PDU_Observable,
37 SocketManager *the_SocketManager);
38 IPDU_Observer *sessionNotify(
39 IPDU_Observable *the_PDU_Observable, int fd);
40 int args(SocketManager *socketManager, int argc, char **argv);
41 int interactive(SocketManager *socketManager);
43 void recv_initResponse(Z_InitResponse *initResponse);
44 void recv_searchResponse(Z_SearchResponse *searchResponse);
45 void recv_presentResponse(Z_PresentResponse *presentResponse);
46 void recv_records (Z_Records *records);
47 void recv_diagrecs(Z_DiagRec **pp, int num);
48 void recv_namePlusRecord (Z_NamePlusRecord *zpr, int offset);
49 void recv_record(Z_DatabaseRecord *record, int offset,
50 const char *databaseName);
51 void recv_textRecord(int type, const char *buf, size_t len);
52 void recv_genericRecord(Z_GenericRecord *r);
53 void display_genericRecord(Z_GenericRecord *r, int level);
54 void display_variant(Z_Variant *v, int level);
58 char *get_cookie (Z_OtherInformation **oi);
59 int processCommand(const char *cmd);
60 const char *getCommand();
61 int cmd_open(char *host);
62 int cmd_connect(char *host);
63 int cmd_quit(char *args);
64 int cmd_close(char *args);
65 int cmd_find(char *args);
66 int cmd_show(char *args);
67 int cmd_cookie(char *args);
68 int cmd_init(char *args);
69 int cmd_format(char *args);
70 int cmd_proxy(char *args);
74 void MyClient::connectNotify()
76 printf ("Connection accepted by target\n");
80 void MyClient::timeoutNotify()
82 printf ("Connection timeout\n");
86 void MyClient::failNotify()
88 printf ("Connection closed by target\n");
92 IPDU_Observer *MyClient::sessionNotify(IPDU_Observable *the_PDU_Observable,
95 return new MyClient(the_PDU_Observable, m_socketManager);
98 MyClient::MyClient(IPDU_Observable *the_PDU_Observable,
99 SocketManager *the_socketManager) :
100 IR_Assoc (the_PDU_Observable)
103 m_interactive_flag = 1;
104 m_thisCommand[0] = '\0';
105 m_lastCommand[0] = '\0';
106 m_socketManager = the_socketManager;
109 void usage(char *prog)
111 fprintf (stderr, "%s: [-v log] [-c cookie] [-p proxy] [zurl]\n", prog);
115 char *MyClient::get_cookie(Z_OtherInformation **otherInfo)
118 Z_OtherInformationUnit *oi;
120 ent.proto = PROTO_Z3950;
121 ent.oclass = CLASS_USERINFO;
122 ent.value = (oid_value) VAL_COOKIE;
124 if (oid_ent_to_oid (&ent, oid) &&
125 (oi = update_otherInformation(otherInfo, 0, oid, 1, 1)) &&
126 oi->which == Z_OtherInfo_characterInfo)
127 return oi->information.characterInfo;
131 void MyClient::recv_initResponse(Z_InitResponse *initResponse)
133 printf ("Got InitResponse. Status ");
134 if (*initResponse->result)
138 const char *p = get_cookie (&initResponse->otherInfo);
141 printf ("cookie = %s\n", p);
149 void MyClient::recv_diagrecs(Z_DiagRec **pp, int num)
153 Z_DefaultDiagFormat *r;
155 printf("Diagnostic message(s) from database:\n");
156 for (i = 0; i<num; i++)
158 Z_DiagRec *p = pp[i];
159 if (p->which != Z_DiagRec_defaultFormat)
161 printf("Diagnostic record not in default format.\n");
165 r = p->u.defaultFormat;
166 if (!(ent = oid_getentbyoid(r->diagnosticSetId)) ||
167 ent->oclass != CLASS_DIAGSET || ent->value != VAL_BIB1)
168 printf("Missing or unknown diagset\n");
169 printf(" [%d] %s", *r->condition, diagbib1_str(*r->condition));
173 case Z_DefaultDiagFormat_v2Addinfo:
174 printf (" -- v2 addinfo '%s'\n", r->u.v2Addinfo);
176 case Z_DefaultDiagFormat_v3Addinfo:
177 printf (" -- v3 addinfo '%s'\n", r->u.v3Addinfo);
181 if (r->addinfo && *r->addinfo)
182 printf(" -- '%s'\n", r->addinfo);
189 void MyClient::recv_textRecord(int type, const char *buf, size_t len)
191 fwrite (buf, 1, len, stdout);
192 fputc ('\n', stdout);
195 void MyClient::display_variant(Z_Variant *v, int level)
199 for (i = 0; i < v->num_triples; i++)
201 printf("%*sclass=%d,type=%d", level * 4, "", *v->triples[i]->zclass,
202 *v->triples[i]->type);
203 if (v->triples[i]->which == Z_Triple_internationalString)
204 printf(",value=%s\n", v->triples[i]->value.internationalString);
210 void MyClient::display_genericRecord(Z_GenericRecord *r, int level)
216 for (i = 0; i < r->num_elements; i++)
220 printf("%*s", level * 4, "");
224 printf("%d,", *t->tagType);
227 if (t->tagValue->which == Z_StringOrNumeric_numeric)
228 printf("%d) ", *t->tagValue->u.numeric);
230 printf("%s) ", t->tagValue->u.string);
231 if (t->content->which == Z_ElementData_subtree)
234 display_genericRecord(t->content->u.subtree, level+1);
236 else if (t->content->which == Z_ElementData_string)
237 printf("%s\n", t->content->u.string);
238 else if (t->content->which == Z_ElementData_numeric)
239 printf("%d\n", *t->content->u.numeric);
240 else if (t->content->which == Z_ElementData_oid)
242 int *ip = t->content->u.oid;
245 if ((oent = oid_getentbyoid(t->content->u.oid)))
246 printf("OID: %s\n", oent->desc);
250 while (ip && *ip >= 0)
251 printf(" %d", *(ip++));
255 else if (t->content->which == Z_ElementData_noDataRequested)
256 printf("[No data requested]\n");
257 else if (t->content->which == Z_ElementData_elementEmpty)
258 printf("[Element empty]\n");
259 else if (t->content->which == Z_ElementData_elementNotThere)
260 printf("[Element not there]\n");
263 if (t->appliedVariant)
264 display_variant(t->appliedVariant, level+1);
265 if (t->metaData && t->metaData->supportedVariants)
269 printf("%*s---- variant list\n", (level+1)*4, "");
270 for (c = 0; c < t->metaData->num_supportedVariants; c++)
272 printf("%*svariant #%d\n", (level+1)*4, "", c);
273 display_variant(t->metaData->supportedVariants[c], level + 2);
279 void MyClient::recv_genericRecord(Z_GenericRecord *r)
281 display_genericRecord(r, 0);
284 void MyClient::recv_record(Z_DatabaseRecord *record, int offset,
285 const char *databaseName)
287 Z_External *r = (Z_External*) record;
288 oident *ent = oid_getentbyoid(r->direct_reference);
291 * Tell the user what we got.
293 if (r->direct_reference)
295 printf("Record type: ");
297 printf("%s\n", ent->desc);
299 if (r->which == Z_External_octet && record->u.octet_aligned->len)
301 yaz_marc_t mt = yaz_marc_create();
326 const char *result_buf;
328 yaz_marc_decode_buf(mt, (const char *)
329 record->u.octet_aligned->buf,
330 record->u.octet_aligned->len,
331 &result_buf, &result_size);
332 fwrite(result_buf, 1, result_size, stdout);
335 recv_textRecord((int) ent->value,
336 (const char *) record->u.octet_aligned->buf,
337 (size_t) record->u.octet_aligned->len);
339 yaz_marc_destroy(mt);
341 else if (ent && ent->value == VAL_SUTRS && r->which == Z_External_sutrs)
342 recv_textRecord((int) VAL_SUTRS, (const char *) r->u.sutrs->buf,
343 (size_t) r->u.sutrs->len);
344 else if (ent && ent->value == VAL_GRS1 && r->which == Z_External_grs1)
345 recv_genericRecord(r->u.grs1);
348 printf("Unknown record representation.\n");
349 if (!z_External(odr_print(), &r, 0, 0))
351 odr_perror(odr_print(), "Printing external");
352 odr_reset(odr_print());
357 void MyClient::recv_namePlusRecord (Z_NamePlusRecord *zpr, int offset)
359 if (zpr->databaseName)
360 printf("[%s]", zpr->databaseName);
361 if (zpr->which == Z_NamePlusRecord_surrogateDiagnostic)
362 recv_diagrecs(&zpr->u.surrogateDiagnostic, 1);
364 recv_record(zpr->u.databaseRecord, offset, zpr->databaseName);
367 void MyClient::recv_records (Z_Records *records)
370 Z_DiagRec dr, *dr_p = &dr;
375 switch (records->which)
377 case Z_Records_DBOSD:
378 for (i = 0; i < records->u.databaseOrSurDiagnostics->num_records; i++)
379 recv_namePlusRecord(records->u.databaseOrSurDiagnostics->
380 records[i], i + m_setOffset);
381 m_setOffset += records->u.databaseOrSurDiagnostics->num_records;
385 dr.which = Z_DiagRec_defaultFormat;
386 dr.u.defaultFormat = records->u.nonSurrogateDiagnostic;
387 recv_diagrecs (&dr_p, 1);
389 recv_diagrecs (&records->u.nonSurrogateDiagnostic, 1);
392 case Z_Records_multipleNSD:
393 recv_diagrecs (records->u.multipleNonSurDiagnostics->diagRecs,
394 records->u.multipleNonSurDiagnostics->num_diagRecs);
399 void MyClient::recv_searchResponse(Z_SearchResponse *searchResponse)
401 printf ("Got SearchResponse. Status ");
402 if (!*searchResponse->searchStatus)
409 printf ("Hits: %d\n", *searchResponse->resultCount);
411 recv_records (searchResponse->records);
414 void MyClient::recv_presentResponse(Z_PresentResponse *presentResponse)
416 printf ("Got PresentResponse\n");
417 recv_records (presentResponse->records);
423 while (m_socketManager->processEvent() > 0)
425 if (get_lastReceived())
432 #define C_PROMPT "Z>"
434 int MyClient::cmd_connect(char *host)
443 int MyClient::cmd_open(char *host)
454 int MyClient::cmd_init(char *args)
456 if (send_initRequest() >= 0)
463 int MyClient::cmd_quit(char *args)
468 int MyClient::cmd_close(char *args)
474 int MyClient::cmd_find(char *args)
478 if (query.set_rpn(args) <= 0)
480 printf ("Bad RPN query\n");
483 if (send_searchRequest(&query) >= 0)
486 printf ("Not connected\n");
490 int MyClient::cmd_show(char *args)
492 int start = m_setOffset, number = 1;
494 sscanf (args, "%d %d", &start, &number);
496 if (send_presentRequest(start, number) >= 0)
499 printf ("Not connected\n");
503 int MyClient::cmd_cookie(char *args)
505 set_cookie(*args ? args : 0);
509 int MyClient::cmd_format(char *args)
511 set_preferredRecordSyntax(args);
515 int MyClient::cmd_proxy(char *args)
521 int MyClient::processCommand(const char *commandLine)
523 char cmdStr[1024], cmdArgs[1024];
528 int (MyClient::*fun)(char *arg);
531 {"open", &MyClient::cmd_open, "<host>[':'<port>][/<database>]"},
532 {"connect", &MyClient::cmd_connect, "<host>[':'<port>][/<database>]"},
533 {"quit", &MyClient::cmd_quit, ""},
534 {"close", &MyClient::cmd_close, ""},
535 {"find", &MyClient::cmd_find, "<query>"},
536 {"show", &MyClient::cmd_show, "[<start> [<number>]]"},
537 {"cookie", &MyClient::cmd_cookie, "<cookie>"},
538 {"init", &MyClient::cmd_init, ""},
539 {"format", &MyClient::cmd_format, "<record-syntax>"},
540 {"proxy", &MyClient::cmd_proxy, "<host>:[':'<port>]"},
544 if (sscanf(commandLine, "%s %[^;]", cmdStr, cmdArgs) < 1)
547 for (i = 0; cmd[i].cmd; i++)
548 if (!strncmp(cmd[i].cmd, cmdStr, strlen(cmdStr)))
552 if (cmd[i].cmd) // Invoke command handler
553 res = (this->*cmd[i].fun)(cmdArgs);
554 else // Dump help screen
556 printf("Unknown command: %s.\n", cmdStr);
557 printf("Currently recognized commands:\n");
558 for (i = 0; cmd[i].cmd; i++)
559 printf(" %s %s\n", cmd[i].cmd, cmd[i].ad);
564 const char *MyClient::getCommand()
566 #if HAVE_READLINE_READLINE_H
567 // Read using GNU readline
569 line_in=readline(C_PROMPT);
572 #if HAVE_READLINE_HISTORY_H
574 add_history(line_in);
576 strncpy(m_thisCommand,line_in, 1023);
577 m_thisCommand[1023] = '\0';
580 // Read using fgets(3)
583 if (!fgets(m_thisCommand, 1023, stdin))
586 // Remove trailing whitespace
587 char *cp = m_thisCommand + strlen(m_thisCommand);
588 while (cp != m_thisCommand && strchr("\t \n", cp[-1]))
592 // Remove leading spaces...
593 while (*cp && strchr ("\t \n", *cp))
595 // Save command if non-empty
597 strcpy (m_lastCommand, cp);
598 return m_lastCommand;
601 int MyClient::interactive(SocketManager *socketManager)
604 if (!m_interactive_flag)
606 while ((cmd = getCommand()))
608 if (!processCommand(cmd))
614 int MyClient::args(SocketManager *socketManager, int argc, char **argv)
619 char *prog = argv[0];
622 while ((ret = options("c:p:v:q", argv, argc, &arg)) != -2)
646 yaz_log_init_level (yaz_log_mask_str(arg));
649 m_interactive_flag = 0;
668 int main(int argc, char **argv)
670 SocketManager mySocketManager;
671 PDU_Assoc *some = new PDU_Assoc(&mySocketManager);
673 MyClient z(some, &mySocketManager);
675 if (z.args(&mySocketManager, argc, argv))
677 if (z.interactive(&mySocketManager))
684 * indent-tabs-mode: nil
686 * vim: shiftwidth=4 tabstop=8 expandtab