spam.c

Go to the documentation of this file.

 
#include "config.h"
#include <stdio.h>
#include "mutt/lib.h"
#include "core/lib.h"
#include "spam.h"
#include "parse/lib.h"
#include "module_data.h"

enum CommandResult parse_nospam(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 Buffer *token = buf_pool_get();
  enum CommandResult rc = MUTT_CMD_WARNING;
 
  // Extract the first token, a regex or "*"
  parse_extract_token(token, line, TOKEN_NO_FLAGS);
 
  if (MoreArgs(line))
  {
    buf_printf(err, _("%s: too many arguments"), cmd->name);
    goto done;
  }
 
  struct EmailModuleData *md = neomutt_get_module_data(NeoMutt, MODULE_ID_EMAIL);
  ASSERT(md);
 
  // "*" is special - clear both spam and nospam lists
  if (mutt_str_equal(buf_string(token), "*"))
  {
    mutt_replacelist_free(&md->spam);
    mutt_regexlist_free(&md->no_spam);
    rc = MUTT_CMD_SUCCESS;
    goto done;
  }
 
  // If it's on the spam list, just remove it
  if (mutt_replacelist_remove(&md->spam, buf_string(token)) != 0)
  {
    rc = MUTT_CMD_SUCCESS;
    goto done;
  }
 
  // Otherwise, add it to the nospam list
  if (mutt_regexlist_add(&md->no_spam, buf_string(token), REG_ICASE, err) != 0)
  {
    rc = MUTT_CMD_ERROR;
    goto done;
  }
 
  rc = MUTT_CMD_SUCCESS;
 
done:
  buf_pool_release(&token);
  return rc;
}

enum CommandResult parse_spam(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 Buffer *token = buf_pool_get();
  struct Buffer *templ = NULL;
  enum CommandResult rc = MUTT_CMD_ERROR;
 
  // Extract the first token, a regex
  parse_extract_token(token, line, TOKEN_NO_FLAGS);
 
  struct EmailModuleData *md = neomutt_get_module_data(NeoMutt, MODULE_ID_EMAIL);
  ASSERT(md);
 
  // If there's a second parameter, it's a template for the spam tag
  if (MoreArgs(line))
  {
    templ = buf_pool_get();
    parse_extract_token(templ, line, TOKEN_NO_FLAGS);
 
    // Add to the spam list
    if (mutt_replacelist_add(&md->spam, buf_string(token), buf_string(templ), err) != 0)
      goto done;
  }
  else
  {
    // If not, try to remove from the nospam list
    mutt_regexlist_remove(&md->no_spam, buf_string(token));
  }
 
  rc = MUTT_CMD_SUCCESS;
 
done:
  buf_pool_release(&templ);
  buf_pool_release(&token);
  return rc;
}