lib-ui/enter.c

   1 /*
   2  * Copyright notice from original mutt:
   3  * Copyright (C) 1996-2000 Michael R. Elkins <me@mutt.org>
   4  * Copyright (C) 2000 Edmund Grimley Evans <edmundo@rano.org>
   5  *
   6  * This file is part of mutt-ng, see http://www.muttng.org/.
   7  * It's licensed under the GNU General Public License,
   8  * please see the file GPL in the top level source directory.
   9  */
  10
  11 #include <lib-lib/lib-lib.h>
  12
  13 #include "curses.h"
  14 #include "enter.h"
  15 #include "menu.h"
  16
  17 #include "mutt.h"
  18 #include "alias.h"
  19 #include "keymap.h"
  20 #include "history.h"
  21 #include "buffy.h"
  22
  23 /* redraw flags for mutt_enter_string() */
  24 enum {
  25   M_REDRAW_INIT = 1,            /* go to end of line and redraw */
  26   M_REDRAW_LINE                 /* redraw entire line */
  27 };
  28
  29 static int my_wcwidth (wchar_t wc)
  30 {
  31   int n = wcwidth (wc);
  32
  33   if (iswprint(wc) && n > 0)
  34     return n;
  35   if (!(wc & ~0x7f))
  36     return 2;
  37   if (!(wc & ~0xffff))
  38     return 6;
  39   return 10;
  40 }
  41
  42 /* combining mark / non-spacing character */
  43 #define COMB_CHAR(wc) (iswprint(wc) && !wcwidth(wc))
  44
  45 static int my_wcswidth (const wchar_t * s, size_t n)
  46 {
  47   int w = 0;
  48
  49   while (n--)
  50     w += my_wcwidth (*s++);
  51   return w;
  52 }
  53
  54 static int my_addwch (wchar_t wc)
  55 {
  56   int n = wcwidth (wc);
  57
  58   if (iswprint(wc) && n > 0)
  59     return mutt_addwch (wc);
  60   if (!(wc & ~0x7f))
  61     return printw ("^%c", ((int) wc + 0x40) & 0x7f);
  62   if (!(wc & ~0xffff))
  63     return printw ("\\u%04x", (int) wc);
  64   return printw ("\\u%08x", (int) wc);
  65 }
  66
  67 static size_t width_ceiling (const wchar_t * s, size_t n, int w1)
  68 {
  69   const wchar_t *s0 = s;
  70   int w = 0;
  71
  72   for (; n; s++, n--)
  73     if ((w += my_wcwidth (*s)) > w1)
  74       break;
  75   return s - s0;
  76 }
  77
  78 static void my_wcstombs (char *dest, ssize_t dlen, const wchar_t * src,
  79                          size_t slen)
  80 {
  81   mbstate_t st;
  82   size_t k;
  83
  84   /* First convert directly into the destination buffer */
  85   p_clear(&st, 1);
  86   for (; slen && dlen >= MB_LEN_MAX; dest += k, dlen -= k, src++, slen--)
  87     if ((k = wcrtomb (dest, *src, &st)) == (size_t) (-1))
  88       break;
  89
  90   /* If this works, we can stop now */
  91   if (dlen >= MB_LEN_MAX) {
  92     wcrtomb (dest, 0, &st);
  93     return;
  94   }
  95
  96   /* Otherwise convert any remaining data into a local buffer */
  97   {
  98     char buf[3 * MB_LEN_MAX];
  99     char *p = buf;
 100
 101     for (; slen && p - buf < dlen; p += k, src++, slen--)
 102       if ((k = wcrtomb (p, *src, &st)) == (size_t) (-1))
 103         break;
 104     p += wcrtomb (p, 0, &st);
 105
 106     /* If it fits into the destination buffer, we can stop now */
 107     if (p - buf <= dlen) {
 108       memcpy (dest, buf, p - buf);
 109       return;
 110     }
 111
 112     /* Otherwise we truncate the string in an ugly fashion */
 113     memcpy (dest, buf, dlen);
 114     dest[dlen - 1] = '\0';      /* assume original dlen > 0 */
 115   }
 116 }
 117
 118 static size_t my_mbstowcs (wchar_t ** pwbuf, size_t * pwbuflen, size_t i, const char *buf)
 119 {
 120   wchar_t wc;
 121   mbstate_t st;
 122   size_t k;
 123   wchar_t *wbuf;
 124   size_t wbuflen;
 125
 126   wbuf = *pwbuf, wbuflen = *pwbuflen;
 127   p_clear(&st, 1);
 128   for (; (k = mbrtowc (&wc, buf, MB_LEN_MAX, &st)) &&
 129        k != (size_t) (-1) && k != (size_t) (-2); buf += k) {
 130     if (i >= wbuflen) {
 131       wbuflen = i + 20;
 132       p_realloc(&wbuf, wbuflen);
 133     }
 134     wbuf[i++] = wc;
 135   }
 136   *pwbuf = wbuf, *pwbuflen = wbuflen;
 137   return i;
 138 }
 139
 140 /*
 141  * Replace part of the wchar_t buffer, from FROM to CURPOS, by BUF.
 142  */
 143
 144 static void replace_part (ENTER_STATE *state, size_t from, const char *buf)
 145 {
 146   /* Save the suffix */
 147   size_t savelen = state->lastchar - state->curpos;
 148   wchar_t *savebuf = p_new(wchar_t, savelen);
 149
 150   memcpy (savebuf, state->wbuf + state->curpos, savelen * sizeof (wchar_t));
 151
 152   /* Convert to wide characters */
 153   state->curpos = my_mbstowcs(&state->wbuf, &state->wbuflen, from, buf);
 154
 155   /* Make space for suffix */
 156   if (state->curpos + savelen > state->wbuflen) {
 157     state->wbuflen = state->curpos + savelen;
 158     p_realloc(&state->wbuf, state->wbuflen);
 159   }
 160
 161   /* Restore suffix */
 162   memcpy (state->wbuf + state->curpos, savebuf, savelen * sizeof (wchar_t));
 163   state->lastchar = state->curpos + savelen;
 164
 165   p_delete(&savebuf);
 166 }
 167
 168 /*
 169  * Returns:
 170  *        1 need to redraw the screen and call me again
 171  *        0 if input was given
 172  *         -1 if abort.
 173  */
 174
 175 int mutt_enter_string (char *buf, size_t buflen, int y, int x, int flags)
 176 {
 177   int rv;
 178   ENTER_STATE *es = mutt_new_enter_state ();
 179
 180   rv = _mutt_enter_string(buf, buflen, y, x, flags, 0, NULL, NULL, es);
 181   mutt_free_enter_state (&es);
 182   return rv;
 183 }
 184
 185 int _mutt_enter_string (char *buf, size_t buflen, int y, int x,
 186                         int flags, int multiple, char ***files, int *numfiles,
 187                         ENTER_STATE * state)
 188 {
 189   int width = COLS - x - 1;
 190   int redraw;
 191   int pass = (flags & M_PASS);
 192   int first = 1;
 193   int ch, w, r;
 194   size_t i;
 195   wchar_t *tempbuf = 0;
 196   size_t templen = 0;
 197   history_class_t hclass;
 198   wchar_t wc;
 199   mbstate_t mbstate;
 200
 201   int rv = 0;
 202
 203   p_clear(&mbstate, 1);
 204
 205   if (state->wbuf) {
 206     /* Coming back after return 1 */
 207     redraw = M_REDRAW_LINE;
 208   }
 209   else {
 210     /* Initialise wbuf from buf */
 211     state->wbuflen = 0;
 212     state->lastchar = my_mbstowcs (&state->wbuf, &state->wbuflen, 0, buf);
 213     redraw = M_REDRAW_INIT;
 214   }
 215
 216   if (flags & (M_FILE | M_EFILE))
 217     hclass = HC_FILE;
 218   else if (flags & M_CMD)
 219     hclass = HC_CMD;
 220   else if (flags & M_ALIAS)
 221     hclass = HC_ALIAS;
 222   else if (flags & M_COMMAND)
 223     hclass = HC_COMMAND;
 224   else if (flags & M_PATTERN)
 225     hclass = HC_PATTERN;
 226   else
 227     hclass = HC_OTHER;
 228
 229   for (;;) {
 230     if (redraw && !pass) {
 231       if (redraw == M_REDRAW_INIT) {
 232         /* Go to end of line */
 233         state->curpos = state->lastchar;
 234         state->begin =
 235           width_ceiling (state->wbuf, state->lastchar,
 236                          my_wcswidth (state->wbuf,
 237                                       state->lastchar) - width + 1);
 238       }
 239       if (state->curpos < state->begin ||
 240           my_wcswidth (state->wbuf + state->begin,
 241                        state->curpos - state->begin) >= width)
 242         state->begin =
 243           width_ceiling (state->wbuf, state->lastchar,
 244                          my_wcswidth (state->wbuf,
 245                                       state->curpos) - width / 2);
 246       move (y, x);
 247       w = 0;
 248       for (i = state->begin; i < state->lastchar; i++) {
 249         w += my_wcwidth (state->wbuf[i]);
 250         if (w > width)
 251           break;
 252         my_addwch (state->wbuf[i]);
 253       }
 254       clrtoeol ();
 255       move (y,
 256             x + my_wcswidth (state->wbuf + state->begin,
 257                              state->curpos - state->begin));
 258     }
 259     mutt_refresh ();
 260
 261     if ((ch = km_dokey (MENU_EDITOR)) == -1) {
 262       rv = -1;
 263       goto bye;
 264     }
 265
 266     if (ch != OP_NULL) {
 267       first = 0;
 268       if (ch != OP_EDITOR_COMPLETE && ch != OP_EDITOR_COMPLETE_QUERY)
 269         state->tabs = 0;
 270       redraw = M_REDRAW_LINE;
 271       switch (ch) {
 272       case OP_EDITOR_HISTORY_UP:
 273         state->curpos = state->lastchar;
 274         replace_part(state, 0, mutt_history_prev(hclass));
 275         redraw = M_REDRAW_INIT;
 276         break;
 277
 278       case OP_EDITOR_HISTORY_DOWN:
 279         state->curpos = state->lastchar;
 280         replace_part(state, 0, mutt_history_next(hclass));
 281         redraw = M_REDRAW_INIT;
 282         break;
 283
 284       case OP_EDITOR_BACKSPACE:
 285         if (state->curpos == 0)
 286           BEEP ();
 287         else {
 288           i = state->curpos;
 289           while (i && COMB_CHAR (state->wbuf[i - 1]))
 290             --i;
 291           if (i)
 292             --i;
 293           memmove (state->wbuf + i, state->wbuf + state->curpos,
 294                    (state->lastchar - state->curpos) * sizeof (wchar_t));
 295           state->lastchar -= state->curpos - i;
 296           state->curpos = i;
 297         }
 298         break;
 299
 300       case OP_EDITOR_BOL:
 301         state->curpos = 0;
 302         break;
 303
 304       case OP_EDITOR_EOL:
 305         redraw = M_REDRAW_INIT;
 306         break;
 307
 308       case OP_EDITOR_KILL_LINE:
 309         state->curpos = state->lastchar = 0;
 310         break;
 311
 312       case OP_EDITOR_KILL_EOL:
 313         state->lastchar = state->curpos;
 314         break;
 315
 316       case OP_EDITOR_BACKWARD_CHAR:
 317         if (state->curpos == 0)
 318           BEEP ();
 319         else {
 320           while (state->curpos && COMB_CHAR (state->wbuf[state->curpos - 1]))
 321             state->curpos--;
 322           if (state->curpos)
 323             state->curpos--;
 324         }
 325         break;
 326
 327       case OP_EDITOR_FORWARD_CHAR:
 328         if (state->curpos == state->lastchar)
 329           BEEP ();
 330         else {
 331           ++state->curpos;
 332           while (state->curpos < state->lastchar
 333                  && COMB_CHAR (state->wbuf[state->curpos]))
 334             ++state->curpos;
 335         }
 336         break;
 337
 338       case OP_EDITOR_BACKWARD_WORD:
 339         if (state->curpos == 0)
 340           BEEP ();
 341         else {
 342           while (state->curpos && iswspace (state->wbuf[state->curpos - 1]))
 343             --state->curpos;
 344           while (state->curpos && !iswspace (state->wbuf[state->curpos - 1]))
 345             --state->curpos;
 346         }
 347         break;
 348
 349       case OP_EDITOR_FORWARD_WORD:
 350         if (state->curpos == state->lastchar)
 351           BEEP ();
 352         else {
 353           while (state->curpos < state->lastchar
 354                  && iswspace (state->wbuf[state->curpos]))
 355             ++state->curpos;
 356           while (state->curpos < state->lastchar
 357                  && !iswspace (state->wbuf[state->curpos]))
 358             ++state->curpos;
 359         }
 360         break;
 361
 362       case OP_EDITOR_CAPITALIZE_WORD:
 363       case OP_EDITOR_UPCASE_WORD:
 364       case OP_EDITOR_DOWNCASE_WORD:
 365         if (state->curpos == state->lastchar) {
 366           BEEP ();
 367           break;
 368         }
 369         while (state->curpos && !iswspace (state->wbuf[state->curpos]))
 370           --state->curpos;
 371         while (state->curpos < state->lastchar
 372                && iswspace (state->wbuf[state->curpos]))
 373           ++state->curpos;
 374         while (state->curpos < state->lastchar
 375                && !iswspace (state->wbuf[state->curpos])) {
 376           if (ch == OP_EDITOR_DOWNCASE_WORD)
 377             state->wbuf[state->curpos] =
 378               towlower (state->wbuf[state->curpos]);
 379           else {
 380             state->wbuf[state->curpos] =
 381               towupper (state->wbuf[state->curpos]);
 382             if (ch == OP_EDITOR_CAPITALIZE_WORD)
 383               ch = OP_EDITOR_DOWNCASE_WORD;
 384           }
 385           state->curpos++;
 386         }
 387         break;
 388
 389       case OP_EDITOR_DELETE_CHAR:
 390         if (state->curpos == state->lastchar)
 391           BEEP ();
 392         else {
 393           i = state->curpos;
 394           while (i < state->lastchar && COMB_CHAR (state->wbuf[i]))
 395             ++i;
 396           if (i < state->lastchar)
 397             ++i;
 398           while (i < state->lastchar && COMB_CHAR (state->wbuf[i]))
 399             ++i;
 400           memmove (state->wbuf + state->curpos, state->wbuf + i,
 401                    (state->lastchar - i) * sizeof (wchar_t));
 402           state->lastchar -= i - state->curpos;
 403         }
 404         break;
 405
 406       case OP_EDITOR_KILL_WORD:
 407         /* delete to begining of word */
 408         if (state->curpos != 0) {
 409           i = state->curpos;
 410           while (i && iswspace (state->wbuf[i - 1]))
 411             --i;
 412           if (i) {
 413             if (iswalnum (state->wbuf[i - 1])) {
 414               for (--i; i && iswalnum (state->wbuf[i - 1]); i--);
 415             }
 416             else
 417               --i;
 418           }
 419           memmove (state->wbuf + i, state->wbuf + state->curpos,
 420                    (state->lastchar - state->curpos) * sizeof (wchar_t));
 421           state->lastchar += i - state->curpos;
 422           state->curpos = i;
 423         }
 424         break;
 425
 426       case OP_EDITOR_KILL_EOW:
 427         /* delete to end of word */
 428         for (i = state->curpos;
 429              i < state->lastchar && iswspace (state->wbuf[i]); i++);
 430         for (; i < state->lastchar && !iswspace (state->wbuf[i]); i++);
 431         memmove (state->wbuf + state->curpos, state->wbuf + i,
 432                  (state->lastchar - i) * sizeof (wchar_t));
 433         state->lastchar += state->curpos - i;
 434         break;
 435
 436       case OP_EDITOR_BUFFY_CYCLE:
 437         if (flags & M_EFILE) {
 438           first = 1;            /* clear input if user types a real key later */
 439           my_wcstombs (buf, buflen, state->wbuf, state->curpos);
 440           buffy_next (buf, buflen);
 441           state->curpos = state->lastchar =
 442             my_mbstowcs (&state->wbuf, &state->wbuflen, 0, buf);
 443           break;
 444         }
 445         else if (!(flags & M_FILE))
 446           goto self_insert;
 447         /* fall through to completion routine (M_FILE) */
 448
 449       case OP_EDITOR_COMPLETE:
 450       case OP_EDITOR_COMPLETE_QUERY:
 451         state->tabs++;
 452         if (flags & M_CMD) {
 453           for (i = state->curpos; i && state->wbuf[i - 1] != ' '; i--);
 454           my_wcstombs (buf, buflen, state->wbuf + i, state->curpos - i);
 455           if (tempbuf && templen == state->lastchar - i &&
 456               !memcmp (tempbuf, state->wbuf + i,
 457                        (state->lastchar - i) * sizeof (wchar_t))) {
 458             mutt_select_file (buf, buflen,
 459                               (flags & M_EFILE) ? M_SEL_FOLDER : 0);
 460             set_option (OPTNEEDREDRAW);
 461             if (*buf)
 462               replace_part (state, i, buf);
 463             rv = 1;
 464             goto bye;
 465           }
 466           if (!mutt_complete (buf, buflen)) {
 467             templen = state->lastchar - i;
 468             p_realloc(&tempbuf, templen);
 469           }
 470           else
 471             BEEP ();
 472
 473           replace_part (state, i, buf);
 474         }
 475         else if (flags & M_ALIAS && ch == OP_EDITOR_COMPLETE) {
 476           /* invoke the alias-menu to get more addresses */
 477           for (i = state->curpos; i && state->wbuf[i - 1] != ',' &&
 478                state->wbuf[i - 1] != ':'; i--);
 479           for (; i < state->lastchar && state->wbuf[i] == ' '; i++);
 480           my_wcstombs (buf, buflen, state->wbuf + i, state->curpos - i);
 481           r = mutt_alias_complete (buf, buflen);
 482           replace_part (state, i, buf);
 483           if (!r) {
 484             rv = 1;
 485             goto bye;
 486           }
 487           break;
 488         } else if (flags & M_ALIAS && ch == OP_EDITOR_COMPLETE_QUERY) {
 489           /* invoke the query-menu to get more addresses */
 490           if ((i = state->curpos)) {
 491             for (; i && state->wbuf[i - 1] != ','; i--);
 492             for (; i < state->curpos && state->wbuf[i] == ' '; i++);
 493           }
 494           my_wcstombs (buf, buflen, state->wbuf + i, state->curpos - i);
 495           mutt_query_complete (buf, buflen);
 496           replace_part (state, i, buf);
 497           rv = 1;
 498           goto bye;
 499         } else if (flags & M_COMMAND) {
 500           my_wcstombs (buf, buflen, state->wbuf, state->curpos);
 501           i = m_strlen(buf);
 502           if (i && buf[i - 1] == '=' &&
 503               mutt_var_value_complete (buf, buflen, i))
 504             state->tabs = 0;
 505           else if (!mutt_command_complete (buf, buflen, i, state->tabs))
 506             BEEP ();
 507           replace_part (state, 0, buf);
 508         }
 509         else if (flags & (M_FILE | M_EFILE)) {
 510           my_wcstombs (buf, buflen, state->wbuf, state->curpos);
 511
 512           /* see if the path has changed from the last time */
 513           if ((!tempbuf && !state->lastchar)
 514               || (tempbuf && templen == state->lastchar
 515                   && !memcmp (tempbuf, state->wbuf,
 516                               state->lastchar * sizeof (wchar_t)))) {
 517             _mutt_select_file (buf, buflen,
 518                                ((flags & M_EFILE) ? M_SEL_FOLDER : 0) |
 519                                (multiple ? M_SEL_MULTI : 0), files, numfiles);
 520             set_option (OPTNEEDREDRAW);
 521             if (*buf) {
 522               mutt_pretty_mailbox (buf);
 523               if (!pass)
 524                 mutt_history_add (hclass, buf);
 525               rv = 0;
 526               goto bye;
 527             }
 528
 529             /* file selection cancelled */
 530             rv = 1;
 531             goto bye;
 532           }
 533
 534           if (!mutt_complete (buf, buflen)) {
 535             templen = state->lastchar;
 536             p_realloc(&tempbuf, templen);
 537             memcpy (tempbuf, state->wbuf, templen * sizeof (wchar_t));
 538           }
 539           else
 540             BEEP ();            /* let the user know that nothing matched */
 541           replace_part (state, 0, buf);
 542         }
 543         else
 544           goto self_insert;
 545         break;
 546
 547       case OP_EDITOR_QUOTE_CHAR:
 548         {
 549           event_t event;
 550
 551           /*ADDCH (LastKey); */
 552           event = mutt_getch ();
 553           if (event.ch != -1) {
 554             LastKey = event.ch;
 555             goto self_insert;
 556           }
 557         }
 558
 559       case OP_EDITOR_TRANSPOSE_CHARS:
 560         if (state->lastchar < 2)
 561           BEEP ();
 562         else {
 563           wchar_t t;
 564
 565           if (state->curpos == 0)
 566             state->curpos = 2;
 567           else if (state->curpos < state->lastchar)
 568             ++state->curpos;
 569
 570           t = state->wbuf[state->curpos - 2];
 571           state->wbuf[state->curpos - 2] = state->wbuf[state->curpos - 1];
 572           state->wbuf[state->curpos - 1] = t;
 573         }
 574         break;
 575
 576       default:
 577         BEEP ();
 578       }
 579     }
 580     else {
 581
 582     self_insert:
 583
 584       state->tabs = 0;
 585       /* use the raw keypress */
 586       ch = LastKey;
 587
 588       if ((ch == '#') && (flags & M_LASTFOLDER)) {
 589         rv = 2;
 590         my_wcstombs (buf, buflen, state->wbuf, state->lastchar);
 591         goto bye;
 592       }
 593
 594 #ifdef KEY_ENTER
 595       /* treat ENTER the same as RETURN */
 596       if (ch == KEY_ENTER)
 597         ch = '\r';
 598 #endif
 599
 600       /* quietly ignore all other function keys */
 601       if (ch & ~0xff)
 602         continue;
 603
 604       /* gather the octets into a wide character */
 605       {
 606         char c;
 607         size_t k;
 608
 609         c = ch;
 610         k = mbrtowc (&wc, &c, 1, &mbstate);
 611         if (k == (size_t) (-2))
 612           continue;
 613         else if (k && k != 1) {
 614           p_clear(&mbstate, 1);
 615           continue;
 616         }
 617       }
 618
 619       if (first && (flags & M_CLEAR)) {
 620         first = 0;
 621         if (iswprint(wc))      /* why? */
 622           state->curpos = state->lastchar = 0;
 623       }
 624
 625       if (wc == '\r' || wc == '\n') {
 626         /* Convert from wide characters */
 627         my_wcstombs (buf, buflen, state->wbuf, state->lastchar);
 628         if (!pass)
 629           mutt_history_add (hclass, buf);
 630
 631         if (multiple) {
 632           char **tfiles;
 633
 634           *numfiles = 1;
 635           tfiles = p_new(char *, *numfiles);
 636           mutt_expand_path (buf, buflen);
 637           tfiles[0] = m_strdup(buf);
 638           *files = tfiles;
 639         }
 640         rv = 0;
 641         goto bye;
 642       }
 643       else if (wc && (wc < ' ' || iswprint(wc))) {     /* why? */
 644         if (state->lastchar >= state->wbuflen) {
 645           state->wbuflen = state->lastchar + 20;
 646           p_realloc(&state->wbuf, state->wbuflen);
 647         }
 648         memmove (state->wbuf + state->curpos + 1, state->wbuf + state->curpos,
 649                  (state->lastchar - state->curpos) * sizeof (wchar_t));
 650         state->wbuf[state->curpos++] = wc;
 651         state->lastchar++;
 652       }
 653       else {
 654         mutt_flushinp ();
 655         BEEP ();
 656       }
 657     }
 658   }
 659
 660 bye:
 661
 662   p_delete(&tempbuf);
 663   return rv;
 664 }
 665
 666 void mutt_free_enter_state (ENTER_STATE ** esp)
 667 {
 668   if (!esp)
 669     return;
 670
 671   p_delete(&(*esp)->wbuf);
 672   p_delete(esp);
 673 }
 674
 675 /*
 676  * TODO:
 677  * very narrow screen might crash it
 678  * sort out the input side
 679  * unprintable chars
 680  */