Older versions of GNU gcc do not allow 'NULL' as sentinel.
[asterisk/asterisk.git] / apps / app_directory.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2005, Digium, Inc.
5  *
6  * Mark Spencer <markster@digium.com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*! \file
20  *
21  * \brief Provide a directory of extensions
22  *
23  * \author Mark Spencer <markster@digium.com>
24  *
25  * \ingroup applications
26  */
27
28 /*** MODULEINFO
29         <depend>app_voicemail</depend>
30  ***/
31 #include "asterisk.h"
32
33 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
34
35 #include <ctype.h>
36
37 #include "asterisk/paths.h" /* use ast_config_AST_SPOOL_DIR */
38 #include "asterisk/file.h"
39 #include "asterisk/pbx.h"
40 #include "asterisk/module.h"
41 #include "asterisk/say.h"
42 #include "asterisk/app.h"
43 #include "asterisk/utils.h"
44
45 static char *app = "Directory";
46
47 static char *synopsis = "Provide directory of voicemail extensions";
48 static char *descrip =
49 "  Directory(vm-context[,dial-context[,options]]): This application will present\n"
50 "the calling channel with a directory of extensions from which they can search\n"
51 "by name. The list of names and corresponding extensions is retrieved from the\n"
52 "voicemail configuration file, voicemail.conf.\n"
53 "  This application will immediately exit if one of the following DTMF digits are\n"
54 "received and the extension to jump to exists:\n"
55 "    0 - Jump to the 'o' extension, if it exists.\n"
56 "    * - Jump to the 'a' extension, if it exists.\n\n"
57 "  Parameters:\n"
58 "    vm-context   - This is the context within voicemail.conf to use for the\n"
59 "                   Directory.\n"
60 "    dial-context - This is the dialplan context to use when looking for an\n"
61 "                   extension that the user has selected, or when jumping to the\n"
62 "                   'o' or 'a' extension.\n\n"
63 "  Options:\n"
64 "    e           In addition to the name, also read the extension number to the\n"
65 "              caller before presenting dialing options.\n"
66 "    f[(<n>)]    Allow the caller to enter the first name of a user in the\n"
67 "              directory instead of using the last name.  If specified, the\n"
68 "              optional number argument will be used for the number of\n"
69 "              characters the user should enter.\n"
70 "    l[(<n>)]    Allow the caller to enter the last name of a user in the\n"
71 "              directory.  This is the default.  If specified, the\n"
72 "              optional number argument will be used for the number of\n"
73 "              characters the user should enter.\n"
74 "    b[(<n>)]    Allow the caller to enter either the first or the last name\n"
75 "              of a user in the directory.  If specified, the optional number\n"
76 "              argument will be used for the number of characters the user\n"
77 "              should enter.\n"
78 "    m           Instead of reading each name sequentially and asking for\n"
79 "              confirmation, create a menu of up to 8 names.\n"
80 "    p(<n>)      Pause for n milliseconds after the digits are typed.  This is\n"
81 "              helpful for people with cellphones, who are not holding the\n"
82 "              receiver to their ear while entering DTMF.\n"
83 "\n"
84 "    Only one of the f, l, or b options may be specified.  If more than one is\n"
85 "    specified, then Directory will act as if 'b' was specified.  The number\n"
86 "    of characters for the user to type defaults to 3.\n";
87
88 /* For simplicity, I'm keeping the format compatible with the voicemail config,
89    but i'm open to suggestions for isolating it */
90
91 #define VOICEMAIL_CONFIG "voicemail.conf"
92
93 enum {
94         OPT_LISTBYFIRSTNAME = (1 << 0),
95         OPT_SAYEXTENSION =    (1 << 1),
96         OPT_FROMVOICEMAIL =   (1 << 2),
97         OPT_SELECTFROMMENU =  (1 << 3),
98         OPT_LISTBYLASTNAME =  (1 << 4),
99         OPT_LISTBYEITHER =    OPT_LISTBYFIRSTNAME | OPT_LISTBYLASTNAME,
100         OPT_PAUSE =           (1 << 5),
101 } directory_option_flags;
102
103 enum {
104         OPT_ARG_FIRSTNAME =   0,
105         OPT_ARG_LASTNAME =    1,
106         OPT_ARG_EITHER =      2,
107         OPT_ARG_PAUSE =       3,
108         /* This *must* be the last value in this enum! */
109         OPT_ARG_ARRAY_SIZE =  4,
110 };
111
112 struct directory_item {
113         char exten[AST_MAX_EXTENSION + 1];
114         char name[AST_MAX_EXTENSION + 1];
115         char key[50]; /* Text to order items. Either lastname+firstname or firstname+lastname */
116
117         AST_LIST_ENTRY(directory_item) entry;
118 };
119
120 AST_APP_OPTIONS(directory_app_options, {
121         AST_APP_OPTION_ARG('f', OPT_LISTBYFIRSTNAME, OPT_ARG_FIRSTNAME),
122         AST_APP_OPTION_ARG('l', OPT_LISTBYLASTNAME, OPT_ARG_LASTNAME),
123         AST_APP_OPTION_ARG('b', OPT_LISTBYEITHER, OPT_ARG_EITHER),
124         AST_APP_OPTION_ARG('p', OPT_PAUSE, OPT_ARG_PAUSE),
125         AST_APP_OPTION('e', OPT_SAYEXTENSION),
126         AST_APP_OPTION('v', OPT_FROMVOICEMAIL),
127         AST_APP_OPTION('m', OPT_SELECTFROMMENU),
128 });
129
130 static int compare(const char *text, const char *template)
131 {
132         char digit;
133
134         while (*template) {
135                 digit = toupper(*text++);
136                 switch (digit) {
137                 case 0:
138                         return -1;
139                 case '1':
140                         digit = '1';
141                         break;
142                 case '2':
143                 case 'A':
144                 case 'B':
145                 case 'C':
146                         digit = '2';
147                         break;
148                 case '3':
149                 case 'D':
150                 case 'E':
151                 case 'F':
152                         digit = '3';
153                         break;
154                 case '4':
155                 case 'G':
156                 case 'H':
157                 case 'I':
158                         digit = '4';
159                         break;
160                 case '5':
161                 case 'J':
162                 case 'K':
163                 case 'L':
164                         digit = '5';
165                         break;
166                 case '6':
167                 case 'M':
168                 case 'N':
169                 case 'O':
170                         digit = '6';
171                         break;
172                 case '7':
173                 case 'P':
174                 case 'Q':
175                 case 'R':
176                 case 'S':
177                         digit = '7';
178                         break;
179                 case '8':
180                 case 'T':
181                 case 'U':
182                 case 'V':
183                         digit = '8';
184                         break;
185                 case '9':
186                 case 'W':
187                 case 'X':
188                 case 'Y':
189                 case 'Z':
190                         digit = '9';
191                         break;
192
193                 default:
194                         if (digit > ' ')
195                                 return -1;
196                         continue;
197                 }
198
199                 if (*template++ != digit)
200                         return -1;
201         }
202
203         return 0;
204 }
205
206 /* play name of mailbox owner.
207  * returns:  -1 for bad or missing extension
208  *           '1' for selected entry from directory
209  *           '*' for skipped entry from directory
210  */
211 static int play_mailbox_owner(struct ast_channel *chan, const char *context,
212         const char *ext, const char *name, struct ast_flags *flags)
213 {
214         int res = 0;
215         if ((res = ast_app_sayname(chan, ext, context)) >= 0) {
216                 ast_stopstream(chan);
217                 /* If Option 'e' was specified, also read the extension number with the name */
218                 if (ast_test_flag(flags, OPT_SAYEXTENSION)) {
219                         ast_stream_and_wait(chan, "vm-extension", AST_DIGIT_ANY);
220                         res = ast_say_character_str(chan, ext, AST_DIGIT_ANY, chan->language);
221                 }
222         } else {
223                 res = ast_say_character_str(chan, S_OR(name, ext), AST_DIGIT_ANY, chan->language);
224                 if (!ast_strlen_zero(name) && ast_test_flag(flags, OPT_SAYEXTENSION)) {
225                         ast_stream_and_wait(chan, "vm-extension", AST_DIGIT_ANY);
226                         res = ast_say_character_str(chan, ext, AST_DIGIT_ANY, chan->language);
227                 }
228         }
229
230         return res;
231 }
232
233 static int select_entry(struct ast_channel *chan, const char *context, const char *dialcontext, const struct directory_item *item, struct ast_flags *flags)
234 {
235         ast_debug(1, "Selecting '%s' - %s@%s\n", item->name, item->exten, dialcontext);
236
237         if (ast_test_flag(flags, OPT_FROMVOICEMAIL)) {
238                 /* We still want to set the exten though */
239                 ast_copy_string(chan->exten, item->exten, sizeof(chan->exten));
240         } else if (ast_goto_if_exists(chan, dialcontext, item->exten, 1)) {
241                 ast_log(LOG_WARNING,
242                         "Can't find extension '%s' in context '%s'.  "
243                         "Did you pass the wrong context to Directory?\n",
244                         item->exten, dialcontext);
245                 return -1;
246         }
247
248         return 0;
249 }
250
251 static int select_item_seq(struct ast_channel *chan, struct directory_item **items, int count, const char *context, const char *dialcontext, struct ast_flags *flags)
252 {
253         struct directory_item *item, **ptr;
254         int i, res, loop;
255
256         for (ptr = items, i = 0; i < count; i++, ptr++) {
257                 item = *ptr;
258
259                 for (loop = 3 ; loop > 0; loop--) {
260                         res = play_mailbox_owner(chan, context, item->exten, item->name, flags);
261
262                         if (!res)
263                                 res = ast_stream_and_wait(chan, "dir-instr", AST_DIGIT_ANY);
264                         if (!res)
265                                 res = ast_waitfordigit(chan, 3000);
266                         ast_stopstream(chan);
267         
268                         if (res == '1') { /* Name selected */
269                                 return select_entry(chan, context, dialcontext, item, flags) ? -1 : 1;
270                         } else if (res == '*') {
271                                 /* Skip to next match in list */
272                                 break;
273                         }
274
275                         if (res < 0)
276                                 return -1;
277
278                         res = 0;
279                 }
280         }
281
282         /* Nothing was selected */
283         return 0;
284 }
285
286 static int select_item_menu(struct ast_channel *chan, struct directory_item **items, int count, const char *context, const char *dialcontext, struct ast_flags *flags)
287 {
288         struct directory_item **block, *item;
289         int i, limit, res = 0;
290         char buf[9];
291
292         for (block = items; count; block += limit, count -= limit) {
293                 limit = count;
294                 if (limit > 8)
295                         limit = 8;
296
297                 for (i = 0; i < limit && !res; i++) {
298                         item = block[i];
299
300                         snprintf(buf, sizeof(buf), "digits/%d", i + 1);
301                         /* Press <num> for <name>, [ extension <ext> ] */
302                         res = ast_streamfile(chan, "dir-multi1", chan->language);
303                         if (!res)
304                                 res = ast_waitstream(chan, AST_DIGIT_ANY);
305                         if (!res)
306                                 res = ast_streamfile(chan, buf, chan->language);
307                         if (!res)
308                                 res = ast_waitstream(chan, AST_DIGIT_ANY);
309                         if (!res)
310                                 res = ast_streamfile(chan, "dir-multi2", chan->language);
311                         if (!res)
312                                 res = ast_waitstream(chan, AST_DIGIT_ANY);
313                         if (!res)
314                                 res = play_mailbox_owner(chan, context, item->exten, item->name, flags);
315                         if (!res)
316                                 res = ast_waitstream(chan, AST_DIGIT_ANY);
317                         if (!res)
318                                 res = ast_waitfordigit(chan, 800);
319                 }
320
321                 /* Press "9" for more names. */
322                 if (!res && count > limit) {
323                         res = ast_streamfile(chan, "dir-multi9", chan->language);
324                         if (!res)
325                                 res = ast_waitstream(chan, AST_DIGIT_ANY);
326                 }
327
328                 if (!res) {
329                         res = ast_waitfordigit(chan, 3000);
330                 }
331
332                 if (res && res > '0' && res < '1' + limit) {
333                         return select_entry(chan, context, dialcontext, block[res - '1'], flags) ? -1 : 1;
334                 }
335
336                 if (res < 0)
337                         return -1;
338
339                 res = 0;
340         }
341
342         /* Nothing was selected */
343         return 0;
344 }
345
346 static struct ast_config *realtime_directory(char *context)
347 {
348         struct ast_config *cfg;
349         struct ast_config *rtdata;
350         struct ast_category *cat;
351         struct ast_variable *var;
352         char *mailbox;
353         const char *fullname;
354         const char *hidefromdir;
355         char tmp[100];
356         struct ast_flags config_flags = { 0 };
357
358         /* Load flat file config. */
359         cfg = ast_config_load(VOICEMAIL_CONFIG, config_flags);
360
361         if (!cfg) {
362                 /* Loading config failed. */
363                 ast_log(LOG_WARNING, "Loading config failed.\n");
364                 return NULL;
365         }
366
367         /* Get realtime entries, categorized by their mailbox number
368            and present in the requested context */
369         rtdata = ast_load_realtime_multientry("voicemail", "mailbox LIKE", "%", "context", context, SENTINEL);
370
371         /* if there are no results, just return the entries from the config file */
372         if (!rtdata)
373                 return cfg;
374
375         /* Does the context exist within the config file? If not, make one */
376         cat = ast_category_get(cfg, context);
377         if (!cat) {
378                 cat = ast_category_new(context, "", 99999);
379                 if (!cat) {
380                         ast_log(LOG_WARNING, "Out of memory\n");
381                         ast_config_destroy(cfg);
382                         return NULL;
383                 }
384                 ast_category_append(cfg, cat);
385         }
386
387         mailbox = NULL;
388         while ( (mailbox = ast_category_browse(rtdata, mailbox)) ) {
389                 fullname = ast_variable_retrieve(rtdata, mailbox, "fullname");
390                 if (ast_true((hidefromdir = ast_variable_retrieve(rtdata, mailbox, "hidefromdir")))) {
391                         /* Skip hidden */
392                         continue;
393                 }
394                 snprintf(tmp, sizeof(tmp), "no-password,%s", S_OR(fullname, ""));
395                 var = ast_variable_new(mailbox, tmp, "");
396                 if (var)
397                         ast_variable_append(cat, var);
398                 else
399                         ast_log(LOG_WARNING, "Out of memory adding mailbox '%s'\n", mailbox);
400         }
401         ast_config_destroy(rtdata);
402
403         return cfg;
404 }
405
406 static int check_match(struct directory_item **result, const char *item_fullname, const char *item_ext, const char *pattern_ext, int use_first_name)
407 {
408         struct directory_item *item;
409         const char *key = NULL;
410         int namelen;
411
412
413         /* Set key to last name or first name depending on search mode */
414         if (!use_first_name)
415                 key = strchr(item_fullname, ' ');
416
417         if (key)
418                 key++;
419         else
420                 key = item_fullname;
421
422         if (compare(key, pattern_ext))
423                 return 0;
424
425         /* Match */
426         item = ast_calloc(1, sizeof(*item));
427         if (!item)
428                 return -1;
429         ast_copy_string(item->name, item_fullname, sizeof(item->name));
430         ast_copy_string(item->exten, item_ext, sizeof(item->exten));
431
432         ast_copy_string(item->key, key, sizeof(item->key));
433         if (key != item_fullname) {
434                 /* Key is the last name. Append first name to key in order to sort Last,First */
435                 namelen = key - item_fullname - 1;
436                 if (namelen > sizeof(item->key) - strlen(item->key) - 1)
437                         namelen = sizeof(item->key) - strlen(item->key) - 1;
438                 strncat(item->key, item_fullname, namelen);
439         }
440
441         *result = item;
442         return 1;
443 }
444
445 typedef AST_LIST_HEAD_NOLOCK(, directory_item) itemlist;
446
447 static int search_directory(const char *context, struct ast_config *vmcfg, struct ast_config *ucfg, const char *ext, struct ast_flags flags, itemlist *alist)
448 {
449         struct ast_variable *v;
450         char buf[AST_MAX_EXTENSION + 1], *pos, *bufptr, *cat;
451         struct directory_item *item;
452         int res;
453
454         ast_debug(2, "Pattern: %s\n", ext);
455
456         for (v = ast_variable_browse(vmcfg, context); v; v = v->next) {
457
458                 /* Ignore hidden */
459                 if (strcasestr(v->value, "hidefromdir=yes"))
460                         continue;
461
462                 ast_copy_string(buf, v->value, sizeof(buf));
463                 bufptr = buf;
464
465                 /* password,Full Name,email,pager,options */
466                 strsep(&bufptr, ",");
467                 pos = strsep(&bufptr, ",");
468
469                 res = 0;
470                 if (ast_test_flag(&flags, OPT_LISTBYLASTNAME)) {
471                         res = check_match(&item, pos, v->name, ext, 0 /* use_first_name */);
472                 }
473                 if (!res && ast_test_flag(&flags, OPT_LISTBYFIRSTNAME)) {
474                         res = check_match(&item, pos, v->name, ext, 1 /* use_first_name */);
475                 }
476
477                 if (!res)
478                         continue;
479                 else if (res < 0)
480                         return -1;
481
482                 AST_LIST_INSERT_TAIL(alist, item, entry);
483         }
484
485         if (ucfg) {
486                 for (cat = ast_category_browse(ucfg, NULL); cat ; cat = ast_category_browse(ucfg, cat)) {
487                         const char *pos;
488                         if (!strcasecmp(cat, "general"))
489                                 continue;
490                         if (!ast_true(ast_config_option(ucfg, cat, "hasdirectory")))
491                                 continue;
492
493                         /* Find all candidate extensions */
494                         pos = ast_variable_retrieve(ucfg, cat, "fullname");
495                         if (!pos)
496                                 continue;
497
498                         res = 0;
499                         if (ast_test_flag(&flags, OPT_LISTBYLASTNAME)) {
500                                 res = check_match(&item, pos, cat, ext, 0 /* use_first_name */);
501                         }
502                         if (!res && ast_test_flag(&flags, OPT_LISTBYFIRSTNAME)) {
503                                 res = check_match(&item, pos, cat, ext, 1 /* use_first_name */);
504                         }
505
506                         if (!res)
507                                 continue;
508                         else if (res < 0)
509                                 return -1;
510
511                         AST_LIST_INSERT_TAIL(alist, item, entry);
512                 }
513         }
514         return 0;
515 }
516
517 static void sort_items(struct directory_item **sorted, int count)
518 {
519         int reordered, i;
520         struct directory_item **ptr, *tmp;
521
522         if (count < 2)
523                 return;
524
525         /* Bubble-sort items by the key */
526         do {
527                 reordered = 0;
528                 for (ptr = sorted, i = 0; i < count - 1; i++, ptr++) {
529                         if (strcasecmp(ptr[0]->key, ptr[1]->key) > 0) {
530                                 tmp = ptr[0];
531                                 ptr[0] = ptr[1];
532                                 ptr[1] = tmp;
533                                 reordered++;
534                         }
535                 }
536         } while (reordered);
537 }
538
539 static int goto_exten(struct ast_channel *chan, const char *dialcontext, char *ext)
540 {
541         if (!ast_goto_if_exists(chan, dialcontext, ext, 1) ||
542                 (!ast_strlen_zero(chan->macrocontext) &&
543                 !ast_goto_if_exists(chan, chan->macrocontext, ext, 1))) {
544                 return 0;
545         } else {
546                 ast_log(LOG_WARNING, "Can't find extension '%s' in current context.  "
547                         "Not Exiting the Directory!\n", ext);
548                 return -1;
549         }
550 }
551
552 static int do_directory(struct ast_channel *chan, struct ast_config *vmcfg, struct ast_config *ucfg, char *context, char *dialcontext, char digit, int digits, struct ast_flags *flags)
553 {
554         /* Read in the first three digits..  "digit" is the first digit, already read */
555         int res = 0;
556         itemlist alist = AST_LIST_HEAD_NOLOCK_INIT_VALUE;
557         struct directory_item *item, **ptr, **sorted = NULL;
558         int count, i;
559         char ext[10] = "";
560
561         if (ast_strlen_zero(context)) {
562                 ast_log(LOG_WARNING,
563                         "Directory must be called with an argument "
564                         "(context in which to interpret extensions)\n");
565                 return -1;
566         }
567
568         if (digit == '0' && !goto_exten(chan, dialcontext, "o")) {
569                 return 0;
570         }
571
572         if (digit == '*' && !goto_exten(chan, dialcontext, "a")) {
573                 return 0;
574         }
575
576         ext[0] = digit;
577         if (ast_readstring(chan, ext + 1, digits - 1, 3000, 3000, "#") < 0)
578                 return -1;
579
580         res = search_directory(context, vmcfg, ucfg, ext, *flags, &alist);
581         if (res)
582                 goto exit;
583
584         /* Count items in the list */
585         count = 0;
586         AST_LIST_TRAVERSE(&alist, item, entry) {
587                 count++;
588         }
589
590         if (count < 1) {
591                 res = ast_streamfile(chan, "dir-nomatch", chan->language);
592                 goto exit;
593         }
594
595
596         /* Create plain array of pointers to items (for sorting) */
597         sorted = ast_calloc(count, sizeof(*sorted));
598
599         ptr = sorted;
600         AST_LIST_TRAVERSE(&alist, item, entry) {
601                 *ptr++ = item;
602         }
603
604         /* Sort items */
605         sort_items(sorted, count);
606
607         if (option_debug) {
608                 ast_debug(2, "Listing matching entries:\n");
609                 for (ptr = sorted, i = 0; i < count; i++, ptr++) {
610                         ast_log(LOG_DEBUG, "%s: %s\n", ptr[0]->exten, ptr[0]->name);
611                 }
612         }
613
614         if (ast_test_flag(flags, OPT_SELECTFROMMENU)) {
615                 /* Offer multiple entries at the same time */
616                 res = select_item_menu(chan, sorted, count, context, dialcontext, flags);
617         } else {
618                 /* Offer entries one by one */
619                 res = select_item_seq(chan, sorted, count, context, dialcontext, flags);
620         }
621
622         if (!res) {
623                 res = ast_streamfile(chan, "dir-nomore", chan->language);
624         }
625
626 exit:
627         if (sorted)
628                 ast_free(sorted);
629
630         while ((item = AST_LIST_REMOVE_HEAD(&alist, entry)))
631                 ast_free(item);
632
633         return res;
634 }
635
636 static int directory_exec(struct ast_channel *chan, void *data)
637 {
638         int res = 0, digit = 3;
639         struct ast_config *cfg, *ucfg;
640         const char *dirintro;
641         char *parse, *opts[OPT_ARG_ARRAY_SIZE];
642         struct ast_flags flags = { 0 };
643         struct ast_flags config_flags = { 0 };
644         enum { FIRST, LAST, BOTH } which = LAST;
645         char digits[9] = "digits/3";
646         AST_DECLARE_APP_ARGS(args,
647                 AST_APP_ARG(vmcontext);
648                 AST_APP_ARG(dialcontext);
649                 AST_APP_ARG(options);
650         );
651
652         if (ast_strlen_zero(data)) {
653                 ast_log(LOG_WARNING, "Directory requires an argument (context[,dialcontext])\n");
654                 return -1;
655         }
656
657         parse = ast_strdupa(data);
658
659         AST_STANDARD_APP_ARGS(args, parse);
660
661         if (args.options && ast_app_parse_options(directory_app_options, &flags, opts, args.options))
662                 return -1;
663
664         if (ast_strlen_zero(args.dialcontext))
665                 args.dialcontext = args.vmcontext;
666
667         cfg = realtime_directory(args.vmcontext);
668         if (!cfg) {
669                 ast_log(LOG_ERROR, "Unable to read the configuration data!\n");
670                 return -1;
671         }
672
673         ucfg = ast_config_load("users.conf", config_flags);
674
675         dirintro = ast_variable_retrieve(cfg, args.vmcontext, "directoryintro");
676         if (ast_strlen_zero(dirintro))
677                 dirintro = ast_variable_retrieve(cfg, "general", "directoryintro");
678
679         if (ast_test_flag(&flags, OPT_LISTBYFIRSTNAME) && ast_test_flag(&flags, OPT_LISTBYLASTNAME)) {
680                 if (!ast_strlen_zero(opts[OPT_ARG_EITHER])) {
681                         digit = atoi(opts[OPT_ARG_EITHER]);
682                 }
683                 which = BOTH;
684         } else if (ast_test_flag(&flags, OPT_LISTBYFIRSTNAME)) {
685                 if (!ast_strlen_zero(opts[OPT_ARG_FIRSTNAME])) {
686                         digit = atoi(opts[OPT_ARG_FIRSTNAME]);
687                 }
688                 which = FIRST;
689         } else {
690                 if (!ast_strlen_zero(opts[OPT_ARG_LASTNAME])) {
691                         digit = atoi(opts[OPT_ARG_LASTNAME]);
692                 }
693                 which = LAST;
694         }
695
696         /* If no options specified, search by last name */
697         if (!ast_test_flag(&flags, OPT_LISTBYFIRSTNAME) && !ast_test_flag(&flags, OPT_LISTBYLASTNAME)) {
698                 ast_set_flag(&flags, OPT_LISTBYLASTNAME);
699                 which = LAST;
700         }
701
702         if (digit > 9) {
703                 digit = 9;
704         } else if (digit < 1) {
705                 digit = 3;
706         }
707         digits[7] = digit + '0';
708
709         if (chan->_state != AST_STATE_UP)
710                 res = ast_answer(chan);
711
712         for (;;) {
713                 if (!ast_strlen_zero(dirintro) && !res) {
714                         res = ast_stream_and_wait(chan, dirintro, AST_DIGIT_ANY);
715                 } else if (!res) {
716                         /* Stop playing sounds as soon as we have a digit. */
717                         res = ast_stream_and_wait(chan, "dir-welcome", AST_DIGIT_ANY);
718                         if (!res) {
719                                 res = ast_stream_and_wait(chan, "dir-pls-enter", AST_DIGIT_ANY);
720                         }
721                         if (!res) {
722                                 res = ast_stream_and_wait(chan, digits, AST_DIGIT_ANY);
723                         }
724                         if (!res) {
725                                 res = ast_stream_and_wait(chan, 
726                                         which == FIRST ? "dir-first" :
727                                         which == LAST ? "dir-last" :
728                                         "dir-firstlast", AST_DIGIT_ANY);
729                         }
730                         if (!res) {
731                                 ast_stream_and_wait(chan, "dir-usingkeypad", AST_DIGIT_ANY);
732                         }
733                 }
734                 ast_stopstream(chan);
735                 if (!res)
736                         res = ast_waitfordigit(chan, 5000);
737
738                 if (res <= 0)
739                         break;
740
741                 res = do_directory(chan, cfg, ucfg, args.vmcontext, args.dialcontext, res, digit, &flags);
742                 if (res)
743                         break;
744
745                 res = ast_waitstream(chan, AST_DIGIT_ANY);
746                 ast_stopstream(chan);
747
748                 if (res)
749                         break;
750         }
751
752         if (ucfg)
753                 ast_config_destroy(ucfg);
754         ast_config_destroy(cfg);
755
756         return res < 0 ? -1 : 0;
757 }
758
759 static int unload_module(void)
760 {
761         int res;
762         res = ast_unregister_application(app);
763         return res;
764 }
765
766 static int load_module(void)
767 {
768         return ast_register_application(app, directory_exec, synopsis, descrip);
769 }
770
771 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Extension Directory");