my_header.c

Go to the documentation of this file.

 
#include "config.h"
#include <string.h>
#include "mutt/lib.h"
#include "email/lib.h"
#include "core/lib.h"
#include "my_header.h"
#include "parse/lib.h"
#include "module_data.h"

enum CommandResult parse_my_header(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;
 
  parse_extract_token(token, line, TOKEN_SPACE | TOKEN_QUOTE);
  char *p = strpbrk(buf_string(token), ": \t");
  if (!p || (*p != ':'))
  {
    buf_strcpy(err, _("invalid header field"));
    goto done;
  }
 
  struct SendModuleData *md = neomutt_get_module_data(NeoMutt, MODULE_ID_SEND);
  ASSERT(md);
 
  struct EventHeader ev_h = { token->data };
  struct ListNode *node = header_find(&md->user_header, buf_string(token));
 
  if (node)
  {
    header_update(node, buf_string(token));
    mutt_debug(LL_NOTIFY, "NT_HEADER_CHANGE: %s\n", buf_string(token));
    notify_send(NeoMutt->notify, NT_HEADER, NT_HEADER_CHANGE, &ev_h);
  }
  else
  {
    header_add(&md->user_header, buf_string(token));
    mutt_debug(LL_NOTIFY, "NT_HEADER_ADD: %s\n", buf_string(token));
    notify_send(NeoMutt->notify, NT_HEADER, NT_HEADER_ADD, &ev_h);
  }
 
  rc = MUTT_CMD_SUCCESS;
 
done:
  buf_pool_release(&token);
  return rc;
}

enum CommandResult parse_unmy_header(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 ListNode *np = NULL, *tmp = NULL;
  size_t l;
 
  struct SendModuleData *md = neomutt_get_module_data(NeoMutt, MODULE_ID_SEND);
  ASSERT(md);
 
  do
  {
    parse_extract_token(token, line, TOKEN_NO_FLAGS);
    if (mutt_str_equal("*", buf_string(token)))
    {
      /* Clear all headers, send a notification for each header */
      STAILQ_FOREACH(np, &md->user_header, entries)
      {
        mutt_debug(LL_NOTIFY, "NT_HEADER_DELETE: %s\n", np->data);
        struct EventHeader ev_h = { np->data };
        notify_send(NeoMutt->notify, NT_HEADER, NT_HEADER_DELETE, &ev_h);
      }
      mutt_list_free(&md->user_header);
      continue;
    }
 
    l = mutt_str_len(buf_string(token));
    if (buf_at(token, l - 1) == ':')
      l--;
 
    STAILQ_FOREACH_SAFE(np, &md->user_header, entries, tmp)
    {
      if (mutt_istrn_equal(buf_string(token), np->data, l) && (np->data[l] == ':'))
      {
        mutt_debug(LL_NOTIFY, "NT_HEADER_DELETE: %s\n", np->data);
        struct EventHeader ev_h = { np->data };
        notify_send(NeoMutt->notify, NT_HEADER, NT_HEADER_DELETE, &ev_h);
 
        header_free(&md->user_header, np);
      }
    }
  } while (MoreArgs(line));
  buf_pool_release(&token);
  return MUTT_CMD_SUCCESS;
}