move more files.
[apps/madmutt.git] / lib-ui / query.c
diff --git a/lib-ui/query.c b/lib-ui/query.c
new file mode 100644 (file)
index 0000000..2a60afa
--- /dev/null
@@ -0,0 +1,481 @@
+/*
+ * Copyright notice from original mutt:
+ * Copyright (C) 1996-2000 Michael R. Elkins <me@mutt.org>
+ *
+ * This file is part of mutt-ng, see http://www.muttng.org/.
+ * It's licensed under the GNU General Public License,
+ * please see the file GPL in the top level source directory.
+ */
+
+#if HAVE_CONFIG_H
+# include "config.h"
+#endif
+
+#include <string.h>
+#include <stdlib.h>
+#include <ctype.h>
+
+#include <lib-lib/mem.h>
+#include <lib-lib/str.h>
+#include <lib-lib/macros.h>
+#include <lib-lib/file.h>
+#include <lib-lib/mapping.h>
+
+#include <lib-ui/menu.h>
+
+#include "mutt.h"
+#include "alias.h"
+#include "mutt_idna.h"
+#include "sort.h"
+
+typedef struct query {
+  address_t *addr;
+  char *name;
+  char *other;
+  struct query *next;
+} QUERY;
+
+typedef struct entry {
+  int tagged;
+  QUERY *data;
+} ENTRY;
+
+static struct mapping_t QueryHelp[] = {
+  {N_("Exit"), OP_EXIT},
+  {N_("Mail"), OP_MAIL},
+  {N_("New Query"), OP_QUERY},
+  {N_("Make Alias"), OP_CREATE_ALIAS},
+  {N_("Search"), OP_SEARCH},
+  {N_("Help"), OP_HELP},
+  {NULL, OP_NULL}
+};
+
+/* Variables for outsizing output format */
+static int FirstColumn;
+static int SecondColumn;
+
+static void query_menu (char *buf, ssize_t buflen, QUERY * results,
+                        int retbuf);
+
+static address_t *result_to_addr (QUERY * r)
+{
+  static address_t *tmp;
+
+  tmp = address_list_dup (r->addr);
+
+  if (!tmp->next && !tmp->personal)
+    tmp->personal = m_strdup(r->name);
+
+  mutt_addrlist_to_idna (tmp, NULL);
+  return tmp;
+}
+
+static QUERY *run_query (char *s, int quiet)
+{
+  FILE *fp;
+  QUERY *first = NULL;
+  QUERY *cur = NULL;
+  char cmd[_POSIX_PATH_MAX];
+  char *buf = NULL;
+  ssize_t buflen;
+  int dummy = 0;
+  char msg[STRING];
+  char *p;
+  pid_t thepid;
+  int l;
+
+
+  mutt_expand_file_fmt (cmd, sizeof (cmd), QueryCmd, s);
+
+  if ((thepid = mutt_create_filter (cmd, NULL, &fp, NULL)) < 0) {
+    return 0;
+  }
+  if (!quiet)
+    mutt_message _("Waiting for response...");
+
+  fgets (msg, sizeof (msg), fp);
+  if ((p = strrchr (msg, '\n')))
+    *p = '\0';
+  while ((buf = mutt_read_line (buf, &buflen, fp, &dummy)) != NULL) {
+    if ((p = strtok (buf, "\t\n"))) {
+      if (first == NULL) {
+        FirstColumn = 0;
+        SecondColumn = 0;
+        first = p_new(QUERY, 1);
+        cur = first;
+      }
+      else {
+        cur->next = p_new(QUERY, 1);
+        cur = cur->next;
+      }
+
+      l = mutt_strwidth (p);
+      if (l > SecondColumn)
+        SecondColumn = l;
+
+      cur->addr = rfc822_parse_adrlist (cur->addr, p);
+      p = strtok (NULL, "\t\n");
+      if (p) {
+        l = mutt_strwidth (p);
+        if (l > FirstColumn)
+          FirstColumn = l;
+        cur->name = m_strdup(p);
+        p = strtok (NULL, "\t\n");
+        if (p) {
+          cur->other = m_strdup(p);
+        }
+      }
+    }
+  }
+  p_delete(&buf);
+  fclose (fp);
+  if (mutt_wait_filter (thepid)) {
+    if (!quiet)
+      mutt_error ("%s", msg);
+  }
+  else {
+    if (!quiet)
+      mutt_message ("%s", msg);
+  }
+
+  return first;
+}
+
+static int query_search (MUTTMENU * m, regex_t * re, int n)
+{
+  ENTRY *table = (ENTRY *) m->data;
+
+  if (table[n].data->name && !regexec (re, table[n].data->name, 0, NULL, 0))
+    return 0;
+  if (table[n].data->other && !regexec (re, table[n].data->other, 0, NULL, 0))
+    return 0;
+  if (table[n].data->addr) {
+    if (table[n].data->addr->personal &&
+        !regexec (re, table[n].data->addr->personal, 0, NULL, 0))
+      return 0;
+    if (table[n].data->addr->mailbox &&
+        !regexec (re, table[n].data->addr->mailbox, 0, NULL, 0))
+      return 0;
+  }
+
+  return REG_NOMATCH;
+}
+
+/* This is the callback routine from mutt_menuLoop() which is used to generate
+ * a menu entry for the requested item number.
+ */
+#define QUERY_MIN_COLUMN_LENGHT 20      /* Must be < 70/2 */
+static void query_entry (char *s, ssize_t slen, MUTTMENU * m, int num)
+{
+  ENTRY *table = (ENTRY *) m->data;
+  char buf2[SHORT_STRING], buf[SHORT_STRING] = "";
+
+  /* need a query format ... hard coded constants are not good */
+  while (FirstColumn + SecondColumn > 70) {
+    FirstColumn = FirstColumn * 3 / 4;
+    SecondColumn = SecondColumn * 3 / 4;
+    if (FirstColumn < QUERY_MIN_COLUMN_LENGHT)
+      FirstColumn = QUERY_MIN_COLUMN_LENGHT;
+    if (SecondColumn < QUERY_MIN_COLUMN_LENGHT)
+      SecondColumn = QUERY_MIN_COLUMN_LENGHT;
+  }
+
+  rfc822_write_address (buf, sizeof (buf), table[num].data->addr, 1);
+
+  mutt_format_string (buf2, sizeof (buf2),
+                      FirstColumn + 2, FirstColumn + 2,
+                      0, ' ', table[num].data->name,
+                      m_strlen(table[num].data->name), 0);
+
+  snprintf (s, slen, " %c %3d %s %-*.*s %s",
+            table[num].tagged ? '*' : ' ',
+            num + 1,
+            buf2,
+            SecondColumn + 2,
+            SecondColumn + 2, buf, NONULL (table[num].data->other));
+}
+
+static int query_tag (MUTTMENU * menu, int n, int m)
+{
+  ENTRY *cur = &((ENTRY *) menu->data)[n];
+  int ot = cur->tagged;
+
+  cur->tagged = m >= 0 ? m : !cur->tagged;
+  return cur->tagged - ot;
+}
+
+int mutt_query_complete (char *buf, ssize_t buflen)
+{
+  QUERY *results = NULL;
+  address_t *tmpa;
+
+  if (!QueryCmd) {
+    mutt_error _("Query command not defined.");
+
+    return 0;
+  }
+
+  results = run_query (buf, 1);
+  if (results) {
+    /* only one response? */
+    if (results->next == NULL) {
+      tmpa = result_to_addr (results);
+      mutt_addrlist_to_local (tmpa);
+      buf[0] = '\0';
+      rfc822_write_address (buf, buflen, tmpa, 0);
+      address_list_wipe(&tmpa);
+      mutt_clear_error ();
+      return (0);
+    }
+    /* multiple results, choose from query menu */
+    query_menu (buf, buflen, results, 1);
+  }
+  return (0);
+}
+
+void mutt_query_menu (char *buf, ssize_t buflen)
+{
+  if (!QueryCmd) {
+    mutt_error _("Query command not defined.");
+
+    return;
+  }
+
+  if (buf == NULL) {
+    char buffer[STRING] = "";
+
+    query_menu (buffer, sizeof (buffer), NULL, 0);
+  }
+  else {
+    query_menu (buf, buflen, NULL, 1);
+  }
+}
+
+static void query_menu (char *buf, ssize_t buflen, QUERY * results, int retbuf)
+{
+  MUTTMENU *menu;
+  HEADER *msg = NULL;
+  ENTRY *QueryTable = NULL;
+  QUERY *queryp = NULL;
+  int i, done = 0;
+  int op;
+  char helpstr[SHORT_STRING];
+  char title[STRING];
+
+  snprintf (title, sizeof (title), _("Query")); /* FIXME */
+
+  menu = mutt_new_menu ();
+  menu->make_entry = query_entry;
+  menu->search = query_search;
+  menu->tag = query_tag;
+  menu->menu = MENU_QUERY;
+  menu->title = title;
+  menu->help =
+    mutt_compile_help (helpstr, sizeof (helpstr), MENU_QUERY, QueryHelp);
+
+  if (results == NULL) {
+    /* Prompt for Query */
+    if (mutt_get_field (_("Query: "), buf, buflen, 0) == 0 && buf[0]) {
+      results = run_query (buf, 0);
+    }
+  }
+
+  if (results) {
+    snprintf (title, sizeof (title), _("Query '%s'"), buf);
+
+    /* count the number of results */
+    for (queryp = results; queryp; queryp = queryp->next)
+      menu->max++;
+
+    menu->data = QueryTable = p_new(ENTRY, menu->max);
+
+    for (i = 0, queryp = results; queryp; queryp = queryp->next, i++)
+      QueryTable[i].data = queryp;
+
+    while (!done) {
+      switch ((op = mutt_menuLoop (menu))) {
+      case OP_QUERY_APPEND:
+      case OP_QUERY:
+        if (mutt_get_field (_("Query: "), buf, buflen, 0) == 0 && buf[0]) {
+          QUERY *newresults = NULL;
+
+          newresults = run_query (buf, 0);
+
+          menu->redraw = REDRAW_FULL;
+          if (newresults) {
+            snprintf (title, sizeof (title), _("Query '%s'"), buf);
+
+            if (op == OP_QUERY) {
+              queryp = results;
+              while (queryp) {
+                address_list_wipe(&queryp->addr);
+                p_delete(&queryp->name);
+                p_delete(&queryp->other);
+                results = queryp->next;
+                p_delete(&queryp);
+                queryp = results;
+              }
+              results = newresults;
+              p_delete(&QueryTable);
+            }
+            else {
+              /* append */
+              for (queryp = results; queryp->next; queryp = queryp->next);
+
+              queryp->next = newresults;
+            }
+
+
+            menu->current = 0;
+            mutt_menuDestroy (&menu);
+            menu = mutt_new_menu ();
+            menu->make_entry = query_entry;
+            menu->search = query_search;
+            menu->tag = query_tag;
+            menu->menu = MENU_QUERY;
+            menu->title = title;
+            menu->help =
+              mutt_compile_help (helpstr, sizeof (helpstr), MENU_QUERY,
+                                 QueryHelp);
+
+            /* count the number of results */
+            for (queryp = results; queryp; queryp = queryp->next)
+              menu->max++;
+
+            if (op == OP_QUERY) {
+              menu->data = QueryTable = p_new(ENTRY, menu->max);
+
+              for (i = 0, queryp = results; queryp;
+                   queryp = queryp->next, i++)
+                QueryTable[i].data = queryp;
+            }
+            else {
+              int clear = 0;
+
+              /* append */
+              p_realloc(&QueryTable, menu->max);
+
+              menu->data = QueryTable;
+
+              for (i = 0, queryp = results; queryp;
+                   queryp = queryp->next, i++) {
+                /* once we hit new entries, clear/init the tag */
+                if (queryp == newresults)
+                  clear = 1;
+
+                QueryTable[i].data = queryp;
+                if (clear)
+                  QueryTable[i].tagged = 0;
+              }
+            }
+          }
+        }
+        break;
+
+      case OP_CREATE_ALIAS:
+        if (menu->tagprefix) {
+          address_t *naddr = NULL;
+
+          for (i = 0; i < menu->max; i++)
+            if (QueryTable[i].tagged) {
+              address_list_append(&naddr, result_to_addr(QueryTable[i].data));
+            }
+
+          mutt_create_alias (NULL, naddr);
+        }
+        else {
+          address_t *a = result_to_addr (QueryTable[menu->current].data);
+
+          mutt_create_alias (NULL, a);
+          address_list_wipe(&a);
+        }
+        break;
+
+      case OP_GENERIC_SELECT_ENTRY:
+        if (retbuf) {
+          done = 2;
+          break;
+        }
+        /* fall through to OP_MAIL */
+
+      case OP_MAIL:
+        msg = header_new();
+        msg->env = envelope_new();
+        if (!menu->tagprefix) {
+          msg->env->to = result_to_addr (QueryTable[menu->current].data);
+        }
+        else {
+          for (i = 0; i < menu->max; i++)
+            if (QueryTable[i].tagged) {
+              address_list_append(&msg->env->to, result_to_addr(QueryTable[i].data));
+            }
+        }
+        ci_send_message (0, msg, NULL, Context, NULL);
+        menu->redraw = REDRAW_FULL;
+        break;
+
+      case OP_EXIT:
+        done = 1;
+        break;
+      }
+    }
+
+    /* if we need to return the selected entries */
+    if (retbuf && (done == 2)) {
+      int tagged = 0;
+      ssize_t curpos = 0;
+
+      p_clear(buf, buflen);
+
+      /* check for tagged entries */
+      for (i = 0; i < menu->max; i++) {
+        if (QueryTable[i].tagged) {
+          if (curpos == 0) {
+            address_t *tmpa = result_to_addr (QueryTable[i].data);
+
+            mutt_addrlist_to_local (tmpa);
+            tagged = 1;
+            rfc822_write_address (buf, buflen, tmpa, 0);
+            curpos = m_strlen(buf);
+            address_list_wipe(&tmpa);
+          }
+          else if (curpos + 2 < buflen) {
+            address_t *tmpa = result_to_addr (QueryTable[i].data);
+
+            mutt_addrlist_to_local (tmpa);
+            strcat (buf, ", "); /* __STRCAT_CHECKED__ */
+            rfc822_write_address ((char *) buf + curpos + 1,
+                                  buflen - curpos - 1, tmpa, 0);
+            curpos = m_strlen(buf);
+            address_list_wipe(&tmpa);
+          }
+        }
+      }
+      /* then enter current message */
+      if (!tagged) {
+        address_t *tmpa = result_to_addr (QueryTable[menu->current].data);
+
+        mutt_addrlist_to_local (tmpa);
+        rfc822_write_address (buf, buflen, tmpa, 0);
+        address_list_wipe(&tmpa);
+      }
+
+    }
+
+    queryp = results;
+    while (queryp) {
+      address_list_wipe(&queryp->addr);
+      p_delete(&queryp->name);
+      p_delete(&queryp->other);
+      results = queryp->next;
+      p_delete(&queryp);
+      queryp = results;
+    }
+    p_delete(&QueryTable);
+
+    /* tell whoever called me to redraw the screen when I return */
+    set_option (OPTNEEDREDRAW);
+  }
+
+  mutt_menuDestroy (&menu);
+}