subjectrx.c

Go to the documentation of this file.

 
#include "config.h"
#include <stddef.h>
#include "mutt/lib.h"
#include "email/lib.h"
#include "core/lib.h"
#include "gui/lib.h"
#include "subjectrx.h"
#include "parse/lib.h"
#include "module_data.h"

void subjectrx_init(struct NeoMutt *n, struct IndexModuleData *md)
{
  STAILQ_INIT(&md->subject_rx_list);
 
  md->subject_rx_notify = notify_new();
  notify_set_parent(md->subject_rx_notify, n->notify);
}

void subjectrx_cleanup(struct IndexModuleData *md)
{
  notify_free(&md->subject_rx_notify);
 
  mutt_replacelist_free(&md->subject_rx_list);
}

static enum CommandResult parse_unreplace_list(const struct Command *cmd,
                                               struct Buffer *line,
                                               struct ReplaceList *list, struct Buffer *err)
{
  if (!MoreArgs(line))
  {
    buf_printf(err, _("%s: too few arguments"), cmd->name);
    return MUTT_CMD_WARNING;
  }
 
  struct Buffer *token = buf_pool_get();
 
  /* First token is a regex. */
  parse_extract_token(token, line, TOKEN_NO_FLAGS);
 
  /* "*" is a special case. */
  if (mutt_str_equal(buf_string(token), "*"))
  {
    mutt_replacelist_free(list);
    buf_pool_release(&token);
    return MUTT_CMD_SUCCESS;
  }
 
  mutt_replacelist_remove(list, buf_string(token));
  buf_pool_release(&token);
  return MUTT_CMD_SUCCESS;
}

static enum CommandResult parse_replace_list(const struct Command *cmd, struct Buffer *line,
                                             struct ReplaceList *list, struct Buffer *err)
{
  struct Buffer *templ = buf_pool_get();
  struct Buffer *regex = buf_pool_get();
  int rc = MUTT_CMD_WARNING;
 
  if (!MoreArgs(line))
  {
    buf_printf(err, _("%s: too few arguments"), cmd->name);
    goto done;
  }
 
  /* First token is a regex. */
  parse_extract_token(regex, line, TOKEN_NO_FLAGS);
 
  if (!MoreArgs(line))
  {
    buf_printf(err, _("%s: too few arguments"), cmd->name);
    goto done;
  }
 
  /* Second token is a replacement template */
  parse_extract_token(templ, line, TOKEN_NO_FLAGS);
 
  if (mutt_replacelist_add(list, buf_string(regex), buf_string(templ), err) != 0)
  {
    rc = MUTT_CMD_ERROR;
    goto done;
  }
 
  rc = MUTT_CMD_SUCCESS;
 
done:
  buf_pool_release(&regex);
  buf_pool_release(&templ);
  return rc;
}

bool subjectrx_apply_mods(struct Envelope *env)
{
  if (!env || !env->subject || (*env->subject == '\0'))
    return false;
 
  if (env->disp_subj)
    return true;
 
  struct IndexModuleData *md = neomutt_get_module_data(NeoMutt, MODULE_ID_INDEX);
  ASSERT(md);
 
  if (STAILQ_EMPTY(&md->subject_rx_list))
    return false;
 
  env->disp_subj = mutt_replacelist_apply(&md->subject_rx_list, env->subject);
  return true;
}

void subjectrx_clear_mods(struct MailboxView *mv)
{
  if (!mv || !mv->mailbox)
    return;
 
  struct Mailbox *m = mv->mailbox;
 
  for (int i = 0; i < m->msg_count; i++)
  {
    struct Email *e = m->emails[i];
    if (!e || !e->env)
      continue;
    FREE(&e->env->disp_subj);
  }
}

enum CommandResult parse_subjectrx_list(const struct Command *cmd, struct Buffer *line,
                                        const struct ParseContext *pc,
                                        struct ParseError *pe)
{
  struct Buffer *err = pe->message;
 
  if (!MoreArgs(line))
  {
    buf_printf(err, _("%s: too few arguments"), cmd->name);
    return MUTT_CMD_WARNING;
  }
 
  struct IndexModuleData *md = neomutt_get_module_data(NeoMutt, MODULE_ID_INDEX);
  ASSERT(md);
 
  enum CommandResult rc;
 
  rc = parse_replace_list(cmd, line, &md->subject_rx_list, err);
  if (rc == MUTT_CMD_SUCCESS)
  {
    mutt_debug(LL_NOTIFY, "NT_SUBJECTRX_ADD: %s\n", cmd->name);
    notify_send(md->subject_rx_notify, NT_SUBJECTRX, NT_SUBJECTRX_ADD, NULL);
  }
  return rc;
}

enum CommandResult parse_unsubjectrx_list(const struct Command *cmd, struct Buffer *line,
                                          const struct ParseContext *pc,
                                          struct ParseError *pe)
{
  struct Buffer *err = pe->message;
 
  if (!MoreArgs(line))
  {
    buf_printf(err, _("%s: too few arguments"), cmd->name);
    return MUTT_CMD_WARNING;
  }
 
  struct IndexModuleData *md = neomutt_get_module_data(NeoMutt, MODULE_ID_INDEX);
  ASSERT(md);
 
  enum CommandResult rc;
 
  rc = parse_unreplace_list(cmd, line, &md->subject_rx_list, err);
  if (rc == MUTT_CMD_SUCCESS)
  {
    mutt_debug(LL_NOTIFY, "NT_SUBJECTRX_DELETE: %s\n", cmd->name);
    notify_send(md->subject_rx_notify, NT_SUBJECTRX, NT_SUBJECTRX_DELETE, NULL);
  }
  return rc;
}