lib-ui/query.c

   1 /*
   2  * Copyright notice from original mutt:
   3  * Copyright (C) 1996-2000 Michael R. Elkins <me@mutt.org>
   4  *
   5  * This file is part of mutt-ng, see http://www.muttng.org/.
   6  * It's licensed under the GNU General Public License,
   7  * please see the file GPL in the top level source directory.
   8  */
   9
  10 #include <lib-ui/lib-ui.h>
  11
  12 #include <lib-sys/unix.h>
  13
  14 #include <lib-ui/menu.h>
  15
  16 #include "mutt.h"
  17 #include "alias.h"
  18 #include "mutt_idna.h"
  19 #include "sort.h"
  20
  21 typedef struct query {
  22   address_t *addr;
  23   char *name;
  24   char *other;
  25   struct query *next;
  26 } QUERY;
  27
  28 typedef struct entry {
  29   int tagged;
  30   QUERY *data;
  31 } ENTRY;
  32
  33 /* Variables for outsizing output format */
  34 static int FirstColumn;
  35 static int SecondColumn;
  36
  37 static void query_menu (char *buf, ssize_t buflen, QUERY * results,
  38                         int retbuf);
  39
  40 static address_t *result_to_addr (QUERY * r)
  41 {
  42   static address_t *tmp;
  43
  44   tmp = address_list_dup (r->addr);
  45
  46   if (!tmp->next && !tmp->personal)
  47     tmp->personal = m_strdup(r->name);
  48
  49   mutt_addrlist_to_idna (tmp, NULL);
  50   return tmp;
  51 }
  52
  53 static QUERY *run_query (char *s, int quiet)
  54 {
  55   FILE *fp;
  56   QUERY *first = NULL;
  57   QUERY *cur = NULL;
  58   char cmd[_POSIX_PATH_MAX];
  59   char *buf = NULL;
  60   ssize_t buflen;
  61   int dummy = 0;
  62   char msg[STRING];
  63   char *p;
  64   pid_t thepid;
  65   int l;
  66
  67
  68   m_quotefile_fmt(cmd, sizeof (cmd), QueryCmd, s);
  69
  70   if ((thepid = mutt_create_filter (cmd, NULL, &fp, NULL)) < 0) {
  71     return 0;
  72   }
  73   if (!quiet)
  74     mutt_message _("Waiting for response...");
  75
  76   fgets (msg, sizeof (msg), fp);
  77   if ((p = strrchr (msg, '\n')))
  78     *p = '\0';
  79   while ((buf = mutt_read_line (buf, &buflen, fp, &dummy)) != NULL) {
  80     if ((p = strtok (buf, "\t\n"))) {
  81       if (first == NULL) {
  82         FirstColumn = 0;
  83         SecondColumn = 0;
  84         first = p_new(QUERY, 1);
  85         cur = first;
  86       }
  87       else {
  88         cur->next = p_new(QUERY, 1);
  89         cur = cur->next;
  90       }
  91
  92       l = m_strwidth(p);
  93       if (l > SecondColumn)
  94         SecondColumn = l;
  95
  96       cur->addr = rfc822_parse_adrlist (cur->addr, p);
  97       p = strtok (NULL, "\t\n");
  98       if (p) {
  99         l = m_strwidth(p);
 100         if (l > FirstColumn)
 101           FirstColumn = l;
 102         cur->name = m_strdup(p);
 103         p = strtok (NULL, "\t\n");
 104         if (p) {
 105           cur->other = m_strdup(p);
 106         }
 107       }
 108     }
 109   }
 110   p_delete(&buf);
 111   m_fclose(&fp);
 112   if (mutt_wait_filter (thepid)) {
 113     if (!quiet)
 114       mutt_error ("%s", msg);
 115   }
 116   else {
 117     if (!quiet)
 118       mutt_message ("%s", msg);
 119   }
 120
 121   return first;
 122 }
 123
 124 static int query_search (MUTTMENU * m, regex_t * re, int n)
 125 {
 126   ENTRY *table = (ENTRY *) m->data;
 127
 128   if (table[n].data->name && !regexec (re, table[n].data->name, 0, NULL, 0))
 129     return 0;
 130   if (table[n].data->other && !regexec (re, table[n].data->other, 0, NULL, 0))
 131     return 0;
 132   if (table[n].data->addr) {
 133     if (table[n].data->addr->personal &&
 134         !regexec (re, table[n].data->addr->personal, 0, NULL, 0))
 135       return 0;
 136     if (table[n].data->addr->mailbox &&
 137         !regexec (re, table[n].data->addr->mailbox, 0, NULL, 0))
 138       return 0;
 139   }
 140
 141   return REG_NOMATCH;
 142 }
 143
 144 /* This is the callback routine from mutt_menuLoop() which is used to generate
 145  * a menu entry for the requested item number.
 146  */
 147 #define QUERY_MIN_COLUMN_LENGHT 20      /* Must be < 70/2 */
 148 static void query_entry (char *s, ssize_t slen, MUTTMENU * m, int num)
 149 {
 150   ENTRY *table = (ENTRY *) m->data;
 151   char buf2[STRING], buf[STRING] = "";
 152
 153   /* need a query format ... hard coded constants are not good */
 154   while (FirstColumn + SecondColumn > 70) {
 155     FirstColumn = FirstColumn * 3 / 4;
 156     SecondColumn = SecondColumn * 3 / 4;
 157     if (FirstColumn < QUERY_MIN_COLUMN_LENGHT)
 158       FirstColumn = QUERY_MIN_COLUMN_LENGHT;
 159     if (SecondColumn < QUERY_MIN_COLUMN_LENGHT)
 160       SecondColumn = QUERY_MIN_COLUMN_LENGHT;
 161   }
 162
 163   rfc822_addrcat(buf, sizeof (buf), table[num].data->addr, 1);
 164
 165   mutt_format_string (buf2, sizeof (buf2),
 166                       FirstColumn + 2, FirstColumn + 2,
 167                       0, ' ', table[num].data->name,
 168                       m_strlen(table[num].data->name), 0);
 169
 170   snprintf (s, slen, " %c %3d %s %-*.*s %s",
 171             table[num].tagged ? '*' : ' ',
 172             num + 1,
 173             buf2,
 174             SecondColumn + 2,
 175             SecondColumn + 2, buf, NONULL (table[num].data->other));
 176 }
 177
 178 static int query_tag (MUTTMENU * menu, int n, int m)
 179 {
 180   ENTRY *cur = &((ENTRY *) menu->data)[n];
 181   int ot = cur->tagged;
 182
 183   cur->tagged = m >= 0 ? m : !cur->tagged;
 184   return cur->tagged - ot;
 185 }
 186
 187 int mutt_query_complete (char *buf, ssize_t buflen)
 188 {
 189   QUERY *results = NULL;
 190   address_t *tmpa;
 191
 192   if (!QueryCmd) {
 193     mutt_error _("Query command not defined.");
 194
 195     return 0;
 196   }
 197
 198   results = run_query (buf, 1);
 199   if (results) {
 200     /* only one response? */
 201     if (results->next == NULL) {
 202       tmpa = result_to_addr (results);
 203       mutt_addrlist_to_local (tmpa);
 204       buf[0] = '\0';
 205       rfc822_addrcat(buf, buflen, tmpa, 0);
 206       address_list_wipe(&tmpa);
 207       mutt_clear_error ();
 208       return (0);
 209     }
 210     /* multiple results, choose from query menu */
 211     query_menu (buf, buflen, results, 1);
 212   }
 213   return (0);
 214 }
 215
 216 void mutt_query_menu (char *buf, ssize_t buflen)
 217 {
 218   if (!QueryCmd) {
 219     mutt_error _("Query command not defined.");
 220
 221     return;
 222   }
 223
 224   if (buf == NULL) {
 225     char buffer[STRING] = "";
 226
 227     query_menu (buffer, sizeof (buffer), NULL, 0);
 228   }
 229   else {
 230     query_menu (buf, buflen, NULL, 1);
 231   }
 232 }
 233
 234 static void query_menu (char *buf, ssize_t buflen, QUERY * results, int retbuf)
 235 {
 236   MUTTMENU *menu;
 237   HEADER *msg = NULL;
 238   ENTRY *QueryTable = NULL;
 239   QUERY *queryp = NULL;
 240   int i, done = 0;
 241   int op;
 242   char title[STRING];
 243
 244   snprintf (title, sizeof (title), _("Query")); /* FIXME */
 245
 246   menu = mutt_new_menu ();
 247   menu->make_entry = query_entry;
 248   menu->search = query_search;
 249   menu->tag = query_tag;
 250   menu->menu = MENU_QUERY;
 251   menu->title = title;
 252
 253   if (results == NULL) {
 254     /* Prompt for Query */
 255     if (mutt_get_field (_("Query: "), buf, buflen, 0) == 0 && buf[0]) {
 256       results = run_query (buf, 0);
 257     }
 258   }
 259
 260   if (results) {
 261     snprintf (title, sizeof (title), _("Query '%s'"), buf);
 262
 263     /* count the number of results */
 264     for (queryp = results; queryp; queryp = queryp->next)
 265       menu->max++;
 266
 267     menu->data = QueryTable = p_new(ENTRY, menu->max);
 268
 269     for (i = 0, queryp = results; queryp; queryp = queryp->next, i++)
 270       QueryTable[i].data = queryp;
 271
 272     while (!done) {
 273       switch ((op = mutt_menuLoop (menu))) {
 274       case OP_QUERY_APPEND:
 275       case OP_QUERY:
 276         if (mutt_get_field (_("Query: "), buf, buflen, 0) == 0 && buf[0]) {
 277           QUERY *newresults = NULL;
 278
 279           newresults = run_query (buf, 0);
 280
 281           menu->redraw = REDRAW_FULL;
 282           if (newresults) {
 283             snprintf (title, sizeof (title), _("Query '%s'"), buf);
 284
 285             if (op == OP_QUERY) {
 286               queryp = results;
 287               while (queryp) {
 288                 address_list_wipe(&queryp->addr);
 289                 p_delete(&queryp->name);
 290                 p_delete(&queryp->other);
 291                 results = queryp->next;
 292                 p_delete(&queryp);
 293                 queryp = results;
 294               }
 295               results = newresults;
 296               p_delete(&QueryTable);
 297             }
 298             else {
 299               /* append */
 300               for (queryp = results; queryp->next; queryp = queryp->next);
 301
 302               queryp->next = newresults;
 303             }
 304
 305
 306             menu->current = 0;
 307             mutt_menuDestroy (&menu);
 308             menu = mutt_new_menu ();
 309             menu->make_entry = query_entry;
 310             menu->search = query_search;
 311             menu->tag = query_tag;
 312             menu->menu = MENU_QUERY;
 313             menu->title = title;
 314
 315             /* count the number of results */
 316             for (queryp = results; queryp; queryp = queryp->next)
 317               menu->max++;
 318
 319             if (op == OP_QUERY) {
 320               menu->data = QueryTable = p_new(ENTRY, menu->max);
 321
 322               for (i = 0, queryp = results; queryp;
 323                    queryp = queryp->next, i++)
 324                 QueryTable[i].data = queryp;
 325             }
 326             else {
 327               int doclear = 0;
 328
 329               /* append */
 330               p_realloc(&QueryTable, menu->max);
 331
 332               menu->data = QueryTable;
 333
 334               for (i = 0, queryp = results; queryp;
 335                    queryp = queryp->next, i++) {
 336                 /* once we hit new entries, clear/init the tag */
 337                 if (queryp == newresults)
 338                   doclear = 1;
 339
 340                 QueryTable[i].data = queryp;
 341                 if (doclear)
 342                   QueryTable[i].tagged = 0;
 343               }
 344             }
 345           }
 346         }
 347         break;
 348
 349       case OP_CREATE_ALIAS:
 350         if (menu->tagprefix) {
 351           address_t *naddr = NULL;
 352
 353           for (i = 0; i < menu->max; i++)
 354             if (QueryTable[i].tagged) {
 355               address_list_append(&naddr, result_to_addr(QueryTable[i].data));
 356             }
 357
 358           mutt_create_alias (NULL, naddr);
 359         }
 360         else {
 361           address_t *a = result_to_addr (QueryTable[menu->current].data);
 362
 363           mutt_create_alias (NULL, a);
 364           address_list_wipe(&a);
 365         }
 366         break;
 367
 368       case OP_GENERIC_SELECT_ENTRY:
 369         if (retbuf) {
 370           done = 2;
 371           break;
 372         }
 373         /* fall through to OP_MAIL */
 374
 375       case OP_MAIL:
 376         msg = header_new();
 377         msg->env = envelope_new();
 378         if (!menu->tagprefix) {
 379           msg->env->to = result_to_addr (QueryTable[menu->current].data);
 380         }
 381         else {
 382           for (i = 0; i < menu->max; i++)
 383             if (QueryTable[i].tagged) {
 384               address_list_append(&msg->env->to, result_to_addr(QueryTable[i].data));
 385             }
 386         }
 387         ci_send_message (0, msg, NULL, Context, NULL);
 388         menu->redraw = REDRAW_FULL;
 389         break;
 390
 391       case OP_EXIT:
 392         done = 1;
 393         break;
 394       }
 395     }
 396
 397     /* if we need to return the selected entries */
 398     if (retbuf && (done == 2)) {
 399       int tagged = 0;
 400       ssize_t curpos = 0;
 401
 402       p_clear(buf, buflen);
 403
 404       /* check for tagged entries */
 405       for (i = 0; i < menu->max; i++) {
 406         if (QueryTable[i].tagged) {
 407           if (curpos == 0) {
 408             address_t *tmpa = result_to_addr (QueryTable[i].data);
 409
 410             mutt_addrlist_to_local (tmpa);
 411             tagged = 1;
 412             rfc822_addrcat(buf, buflen, tmpa, 0);
 413             curpos = m_strlen(buf);
 414             address_list_wipe(&tmpa);
 415           }
 416           else if (curpos + 2 < buflen) {
 417             address_t *tmpa = result_to_addr (QueryTable[i].data);
 418
 419             mutt_addrlist_to_local (tmpa);
 420             m_strcat(buf, buflen, ", ");
 421             rfc822_addrcat(buf + curpos + 1, buflen - curpos - 1, tmpa, 0);
 422             curpos = m_strlen(buf);
 423             address_list_wipe(&tmpa);
 424           }
 425         }
 426       }
 427       /* then enter current message */
 428       if (!tagged) {
 429         address_t *tmpa = result_to_addr (QueryTable[menu->current].data);
 430
 431         mutt_addrlist_to_local (tmpa);
 432         rfc822_addrcat(buf, buflen, tmpa, 0);
 433         address_list_wipe(&tmpa);
 434       }
 435
 436     }
 437
 438     queryp = results;
 439     while (queryp) {
 440       address_list_wipe(&queryp->addr);
 441       p_delete(&queryp->name);
 442       p_delete(&queryp->other);
 443       results = queryp->next;
 444       p_delete(&queryp);
 445       queryp = results;
 446     }
 447     p_delete(&QueryTable);
 448
 449     /* tell whoever called me to redraw the screen when I return */
 450     set_option (OPTNEEDREDRAW);
 451   }
 452
 453   mutt_menuDestroy (&menu);
 454 }