format.c

Go to the documentation of this file.

 
#include "config.h"
#include <errno.h>
#include <limits.h>
#include <stdbool.h>
#include <string.h>
#include <wchar.h>
#include "mutt/lib.h"
#include "format.h"
#include "mutt_thread.h"
#ifdef HAVE_ISWBLANK
#include <wctype.h>
#endif
 
void buf_justify(struct Buffer *buf, enum FormatJustify justify, size_t max_cols, char pad_char)
{
  if (!buf || (pad_char == '\0'))
    return;
 
  size_t len = buf_len(buf);
  if (len >= max_cols)
    return;
 
  buf_alloc(buf, len + max_cols);
 
  max_cols -= len;
 
  switch (justify)
  {
    case JUSTIFY_LEFT:
    {
      memset(buf->dptr, pad_char, max_cols);
      break;
    }
 
    case JUSTIFY_CENTER:
    {
      if (max_cols > 1)
      {
        memmove(buf->data + (max_cols / 2), buf->data, len);
        memset(buf->data, pad_char, max_cols / 2);
      }
      memset(buf->data + len + (max_cols / 2), pad_char, (max_cols + 1) / 2);
      break;
    }
 
    case JUSTIFY_RIGHT:
    {
      memmove(buf->data + max_cols, buf->data, len);
      memset(buf->data, pad_char, max_cols);
      break;
    }
  }
 
  buf->dptr += max_cols;
  buf->dptr[0] = '\0';
}
 
int format_string(struct Buffer *buf, int min_cols, int max_cols, enum FormatJustify justify,
                  char pad_char, const char *str, size_t n, bool arboreal)
{
  wchar_t wc = 0;
  int w = 0;
  size_t k = 0;
  char scratch[MB_LEN_MAX] = { 0 };
  mbstate_t mbstate1 = { 0 };
  mbstate_t mbstate2 = { 0 };
  bool escaped = false;
  int used_cols = 0;
 
  for (; (n > 0) && (k = mbrtowc(&wc, str, n, &mbstate1)); str += k, n -= k)
  {
    if ((k == ICONV_ILLEGAL_SEQ) || (k == ICONV_BUF_TOO_SMALL))
    {
      if ((k == ICONV_ILLEGAL_SEQ) && (errno == EILSEQ))
        memset(&mbstate1, 0, sizeof(mbstate1));
 
      k = (k == ICONV_ILLEGAL_SEQ) ? 1 : n;
      wc = ReplacementChar;
    }
 
    // How many screen cells will the character require?
    if (escaped)
    {
      escaped = false;
      w = 0;
    }
    else if (arboreal && (wc == MUTT_SPECIAL_INDEX))
    {
      escaped = true;
      w = 0;
    }
    else if (arboreal && (wc < MUTT_TREE_MAX))
    {
      w = 1; /* hack */
    }
    else
    {
#ifdef HAVE_ISWBLANK
      if (iswblank(wc))
        wc = ' ';
      else
#endif
          if (!IsWPrint(wc))
        wc = '?';
      w = wcwidth(wc);
    }
 
    // It'll require _some_ space
    if (w >= 0)
    {
      if (w > max_cols)
        continue;
      size_t k2 = wcrtomb(scratch, wc, &mbstate2);
      if (k2 == ICONV_ILLEGAL_SEQ)
        continue; // LCOV_EXCL_LINE
 
      used_cols += w;
 
      min_cols -= w;
      max_cols -= w;
      buf_addstr_n(buf, scratch, k2);
    }
  }
 
  // How much space is left?
  w = min_cols;
  if (w > 0)
  {
    used_cols += w;
  }
 
  if (w > 0)
    buf_justify(buf, justify, buf_len(buf) + w, pad_char);
 
  return used_cols;
}