regex.c

Go to the documentation of this file.

 
#include "config.h"
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include "mutt/lib.h"
#include "config/lib.h"
#include "core/lib.h"
#include "index/lib.h"
#include "pattern/lib.h"
#include "attr.h"
#include "color.h"
#include "commands.h"
#include "debug.h"
#include "module_data.h"
#include "notify2.h"
#include "regex4.h"

void regex_colors_init(void)
{
  struct ColorModuleData *mod_data = neomutt_get_module_data(NeoMutt, MODULE_ID_COLOR);
  color_debug(LL_DEBUG5, "init AttachList, BodyList, etc\n");
  STAILQ_INIT(&mod_data->attach_list);
  STAILQ_INIT(&mod_data->body_list);
  STAILQ_INIT(&mod_data->header_list);
  STAILQ_INIT(&mod_data->index_author_list);
  STAILQ_INIT(&mod_data->index_collapsed_list);
  STAILQ_INIT(&mod_data->index_date_list);
  STAILQ_INIT(&mod_data->index_label_list);
  STAILQ_INIT(&mod_data->index_number_list);
  STAILQ_INIT(&mod_data->index_size_list);
  STAILQ_INIT(&mod_data->index_tags_list);
  STAILQ_INIT(&mod_data->index_flags_list);
  STAILQ_INIT(&mod_data->index_list);
  STAILQ_INIT(&mod_data->index_subject_list);
  STAILQ_INIT(&mod_data->index_tag_list);
  STAILQ_INIT(&mod_data->status_list);
}

void regex_colors_reset(void)
{
  struct ColorModuleData *mod_data = neomutt_get_module_data(NeoMutt, MODULE_ID_COLOR);
  color_debug(LL_DEBUG5, "reset regex\n");
  regex_color_list_clear(&mod_data->attach_list);
  regex_color_list_clear(&mod_data->body_list);
  regex_color_list_clear(&mod_data->header_list);
  regex_color_list_clear(&mod_data->index_list);
  regex_color_list_clear(&mod_data->index_author_list);
  regex_color_list_clear(&mod_data->index_collapsed_list);
  regex_color_list_clear(&mod_data->index_date_list);
  regex_color_list_clear(&mod_data->index_label_list);
  regex_color_list_clear(&mod_data->index_number_list);
  regex_color_list_clear(&mod_data->index_size_list);
  regex_color_list_clear(&mod_data->index_tags_list);
  regex_color_list_clear(&mod_data->index_flags_list);
  regex_color_list_clear(&mod_data->index_subject_list);
  regex_color_list_clear(&mod_data->index_tag_list);
  regex_color_list_clear(&mod_data->status_list);
}

void regex_colors_cleanup(void)
{
  regex_colors_reset();
}

void regex_color_clear(struct RegexColor *rcol)
{
  if (!rcol)
    return;
 
  rcol->match = 0;
  rcol->stop_matching = false;
 
  attr_color_clear(&rcol->attr_color);
  FREE(&rcol->pattern);
  regfree(&rcol->regex);
  mutt_pattern_free(&rcol->color_pattern);
}

void regex_color_free(struct RegexColor **ptr)
{
  if (!ptr || !*ptr)
    return;
 
  struct RegexColor *rcol = *ptr;
  regex_color_clear(rcol);
 
  FREE(ptr);
}

struct RegexColor *regex_color_new(void)
{
  return MUTT_MEM_CALLOC(1, struct RegexColor);
}

struct RegexColorList *regex_color_list_new(void)
{
  struct RegexColorList *rcl = MUTT_MEM_CALLOC(1, struct RegexColorList);
 
  STAILQ_INIT(rcl);
 
  struct RegexColor *rcol = regex_color_new();
  STAILQ_INSERT_TAIL(rcl, rcol, entries);
 
  return rcl;
}

void regex_color_list_clear(struct RegexColorList *rcl)
{
  if (!rcl)
    return;
 
  struct RegexColor *np = NULL, *tmp = NULL;
  STAILQ_FOREACH_SAFE(np, rcl, entries, tmp)
  {
    STAILQ_REMOVE(rcl, np, RegexColor, entries);
    regex_color_free(&np);
  }
}

struct RegexColorList *regex_colors_get_list(enum ColorId cid)
{
  struct ColorModuleData *mod_data = neomutt_get_module_data(NeoMutt, MODULE_ID_COLOR);
  switch (cid)
  {
    case MT_COLOR_ATTACH_HEADERS:
      return &mod_data->attach_list;
    case MT_COLOR_BODY:
      return &mod_data->body_list;
    case MT_COLOR_HEADER:
      return &mod_data->header_list;
    case MT_COLOR_INDEX:
      return &mod_data->index_list;
    case MT_COLOR_INDEX_AUTHOR:
      return &mod_data->index_author_list;
    case MT_COLOR_INDEX_COLLAPSED:
      return &mod_data->index_collapsed_list;
    case MT_COLOR_INDEX_DATE:
      return &mod_data->index_date_list;
    case MT_COLOR_INDEX_FLAGS:
      return &mod_data->index_flags_list;
    case MT_COLOR_INDEX_LABEL:
      return &mod_data->index_label_list;
    case MT_COLOR_INDEX_NUMBER:
      return &mod_data->index_number_list;
    case MT_COLOR_INDEX_SIZE:
      return &mod_data->index_size_list;
    case MT_COLOR_INDEX_SUBJECT:
      return &mod_data->index_subject_list;
    case MT_COLOR_INDEX_TAG:
      return &mod_data->index_tag_list;
    case MT_COLOR_INDEX_TAGS:
      return &mod_data->index_tags_list;
    case MT_COLOR_STATUS:
      return &mod_data->status_list;
    default:
      return NULL;
  }
}

static enum CommandResult add_pattern(struct RegexColorList *rcl, const char *s,
                                      struct AttrColor *ac_val,
                                      struct Buffer *err, bool is_index, int match)
{
  struct RegexColor *rcol = NULL;
 
  STAILQ_FOREACH(rcol, rcl, entries)
  {
    if (mutt_str_equal(s, rcol->pattern))
      break;
  }
 
  if (rcol) // found a matching regex
  {
    struct AttrColor *ac = &rcol->attr_color;
    attr_color_overwrite(ac, ac_val);
  }
  else
  {
    rcol = regex_color_new();
    if (is_index)
    {
      struct Buffer *buf = buf_pool_get();
      buf_strcpy(buf, s);
      const char *const c_simple_search = cs_subset_string(NeoMutt->sub, "simple_search");
      mutt_check_simple(buf, NONULL(c_simple_search));
      struct MailboxView *mv_cur = get_current_mailbox_view();
      rcol->color_pattern = mutt_pattern_comp(mv_cur, buf_string(buf), MUTT_PC_FULL_MSG, err);
      buf_pool_release(&buf);
      if (!rcol->color_pattern)
      {
        regex_color_free(&rcol);
        return MUTT_CMD_ERROR;
      }
    }
    else
    {
      // Smart case matching
      uint16_t flags = mutt_mb_is_lower(s) ? REG_ICASE : 0;
 
      const int r = REG_COMP(&rcol->regex, s, flags);
      if (r != 0)
      {
        regerror(r, &rcol->regex, err->data, err->dsize);
        buf_fix_dptr(err);
        regex_color_free(&rcol);
        return MUTT_CMD_ERROR;
      }
    }
    rcol->pattern = mutt_str_dup(s);
    rcol->match = match;
 
    struct AttrColor *ac = &rcol->attr_color;
 
    attr_color_overwrite(ac, ac_val);
 
    STAILQ_INSERT_TAIL(rcl, rcol, entries);
  }
 
  if (is_index)
  {
    /* force re-caching of index colors */
    struct ColorModuleData *mod_data = neomutt_get_module_data(NeoMutt, MODULE_ID_COLOR);
    struct EventColor ev_c = { MT_COLOR_INDEX, NULL };
    notify_send(mod_data->colors_notify, NT_COLOR, NT_COLOR_SET, &ev_c);
  }
 
  return MUTT_CMD_SUCCESS;
}

bool regex_colors_parse_color_list(enum ColorId cid, const char *pat,
                                   struct AttrColor *ac, int *rc, struct Buffer *err)
 
{
  if (cid == MT_COLOR_STATUS)
    return false;
 
  struct RegexColorList *rcl = regex_colors_get_list(cid);
  if (!rcl)
    return false;
 
  bool is_index = false;
  switch (cid)
  {
    case MT_COLOR_ATTACH_HEADERS:
    case MT_COLOR_BODY:
      break;
    case MT_COLOR_HEADER:
      break;
    case MT_COLOR_INDEX:
    case MT_COLOR_INDEX_AUTHOR:
    case MT_COLOR_INDEX_COLLAPSED:
    case MT_COLOR_INDEX_DATE:
    case MT_COLOR_INDEX_FLAGS:
    case MT_COLOR_INDEX_LABEL:
    case MT_COLOR_INDEX_NUMBER:
    case MT_COLOR_INDEX_SIZE:
    case MT_COLOR_INDEX_SUBJECT:
    case MT_COLOR_INDEX_TAG:
    case MT_COLOR_INDEX_TAGS:
      is_index = true;
      break;
    default:
      return false;
  }
 
  *rc = add_pattern(rcl, pat, ac, err, is_index, 0);
 
  struct Buffer *buf = buf_pool_get();
  get_colorid_name(cid, buf);
  color_debug(LL_DEBUG5, "NT_COLOR_SET: %s\n", buf_string(buf));
  buf_pool_release(&buf);
 
  if (!is_index) // else it will be logged in add_pattern()
  {
    struct ColorModuleData *mod_data = neomutt_get_module_data(NeoMutt, MODULE_ID_COLOR);
    struct EventColor ev_c = { cid, NULL };
    notify_send(mod_data->colors_notify, NT_COLOR, NT_COLOR_SET, &ev_c);
  }
 
  return true;
}

int regex_colors_parse_status_list(enum ColorId cid, const char *pat,
                                   struct AttrColor *ac, int match, struct Buffer *err)
{
  if (cid != MT_COLOR_STATUS)
    return MUTT_CMD_ERROR;
 
  struct ColorModuleData *mod_data = neomutt_get_module_data(NeoMutt, MODULE_ID_COLOR);
  int rc = add_pattern(&mod_data->status_list, pat, ac, err, false, match);
  if (rc != MUTT_CMD_SUCCESS)
    return rc;
 
  struct Buffer *buf = buf_pool_get();
  get_colorid_name(cid, buf);
  color_debug(LL_DEBUG5, "NT_COLOR_SET: %s\n", buf_string(buf));
  buf_pool_release(&buf);
 
  struct EventColor ev_c = { cid, NULL };
  notify_send(mod_data->colors_notify, NT_COLOR, NT_COLOR_SET, &ev_c);
 
  return rc;
}

bool regex_colors_parse_uncolor(enum ColorId cid, const char *pat)
{
  struct RegexColorList *cl = regex_colors_get_list(cid);
  if (!cl)
    return false;
 
  struct ColorModuleData *mod_data = neomutt_get_module_data(NeoMutt, MODULE_ID_COLOR);
 
  if (!pat) // Reset all patterns
  {
    if (STAILQ_EMPTY(cl))
      return true;
 
    mutt_debug(LL_NOTIFY, "NT_COLOR_RESET: [ALL]\n");
    struct EventColor ev_c = { cid, NULL };
    notify_send(mod_data->colors_notify, NT_COLOR, NT_COLOR_RESET, &ev_c);
 
    regex_color_list_clear(cl);
    return true;
  }
 
  bool rc = false;
  struct RegexColor *np = NULL, *prev = NULL;
  prev = NULL;
  STAILQ_FOREACH(np, cl, entries)
  {
    if (mutt_str_equal(pat, np->pattern))
    {
      rc = true;
 
      color_debug(LL_DEBUG1, "Freeing pattern \"%s\" from XXX\n", pat);
      if (prev)
        STAILQ_REMOVE_AFTER(cl, prev, entries);
      else
        STAILQ_REMOVE_HEAD(cl, entries);
 
      mutt_debug(LL_NOTIFY, "NT_COLOR_RESET: XXX\n");
      struct EventColor ev_c = { cid, &np->attr_color };
      notify_send(mod_data->colors_notify, NT_COLOR, NT_COLOR_RESET, &ev_c);
 
      regex_color_free(&np);
      break;
    }
    prev = np;
  }
 
  return rc;
}