state.c

Go to the documentation of this file.

 
#include "config.h"
#include <inttypes.h>
#include <limits.h>
#include <stdarg.h> // IWYU pragma: keep
#include <stdio.h>
#include <wchar.h>
#include "state.h"
#include "date.h"
#include "random.h"
 
const char *state_attachment_marker(void)
{
  static char marker[256] = { 0 };
  if (!marker[0])
  {
    snprintf(marker, sizeof(marker), "\033]9;%" PRIu64 "\a", mutt_rand64());
  }
  return marker;
}
 
const char *state_protected_header_marker(void)
{
  static char marker[256] = { 0 };
  if (!marker[0])
  {
    snprintf(marker, sizeof(marker), "\033]8;%lld\a", (long long) mutt_date_now());
  }
  return marker;
}
 
void state_mark_attach(struct State *state)
{
  if (!state || !state->fp_out)
    return;
 
  if ((state->flags & (STATE_DISPLAY | STATE_PAGER)) == (STATE_DISPLAY | STATE_PAGER))
  {
    state_puts(state, state_attachment_marker());
  }
}
 
void state_mark_protected_header(struct State *state)
{
  if (!state || !state->fp_out)
    return;
 
  if ((state->flags & (STATE_DISPLAY | STATE_PAGER)) == (STATE_DISPLAY | STATE_PAGER))
  {
    state_puts(state, state_protected_header_marker());
  }
}
 
void state_attach_puts(struct State *state, const char *t)
{
  if (!state || !state->fp_out || !t)
    return;
 
  if (*t != '\n')
    state_mark_attach(state);
  while (*t)
  {
    state_putc(state, *t);
    if ((*t++ == '\n') && *t)
      if (*t != '\n')
        state_mark_attach(state);
  }
}
 
static int state_putwc(struct State *state, wchar_t wc)
{
  char mb[MB_LEN_MAX] = { 0 };
  int rc;
 
  rc = wcrtomb(mb, wc, NULL);
  if (rc < 0)
    return rc;
  if (fputs(mb, state->fp_out) == EOF)
    return -1;
  return 0;
}
 
int state_putws(struct State *state, const wchar_t *ws)
{
  const wchar_t *p = ws;
 
  while (p && (*p != L'\0'))
  {
    if (state_putwc(state, *p) < 0)
      return -1;
    p++;
  }
  return 0;
}
 
void state_prefix_putc(struct State *state, char c)
{
  if (state->flags & STATE_PENDINGPREFIX)
  {
    state_reset_prefix(state);
    if (state->prefix)
      state_puts(state, state->prefix);
  }
 
  state_putc(state, c);
 
  if (c == '\n')
    state_set_prefix(state);
}
 
int state_printf(struct State *state, const char *fmt, ...)
{
  int rc;
  va_list ap;
 
  va_start(ap, fmt);
  rc = vfprintf(state->fp_out, fmt, ap);
  va_end(ap);
 
  return rc;
}
 
void state_prefix_put(struct State *state, const char *buf, size_t buflen)
{
  if (state->prefix)
  {
    while (buflen--)
      state_prefix_putc(state, *buf++);
  }
  else
  {
    fwrite(buf, buflen, 1, state->fp_out);
  }
}