msg_set.c

Go to the documentation of this file.

 
#include "config.h"
#include <stdio.h>
#include "private.h"
#include "mutt/lib.h"
#include "config/lib.h"
#include "msg_set.h"
 
int ImapMaxCmdlen = 8192;
 
int imap_sort_uid(const void *a, const void *b, void *sdata)
{
  unsigned int ua = *(unsigned int *) a;
  unsigned int ub = *(unsigned int *) b;
 
  return mutt_numeric_cmp(ua, ub);
}
 
int imap_make_msg_set(struct UidArray *uida, struct Buffer *buf, int *pos)
{
  if (!uida || !buf || !pos)
    return 0;
 
  const size_t array_size = ARRAY_SIZE(uida);
  if ((array_size == 0) || (*pos >= array_size))
    return 0;
 
  int count = 1; // Number of UIDs added to the set
  size_t i = *pos;
  unsigned int start = *ARRAY_GET(uida, i);
  unsigned int prev = start;
 
  for (i++; (i < array_size) && (buf_len(buf) < ImapMaxCmdlen); i++, count++)
  {
    unsigned int uid = *ARRAY_GET(uida, i);
 
    // Keep adding to current set
    if (uid == (prev + 1))
    {
      prev = uid;
      continue;
    }
 
    // End the current set
    if (start == prev)
      buf_add_printf(buf, "%u,", start);
    else
      buf_add_printf(buf, "%u:%u,", start, prev);
 
    // Start a new set
    start = uid;
    prev = uid;
  }
 
  if (start == prev)
    buf_add_printf(buf, "%u", start);
  else
    buf_add_printf(buf, "%u:%u", start, prev);
 
  *pos = i;
 
  return count;
}
 
int imap_exec_msg_set(struct ImapAccountData *adata, const char *pre,
                      const char *post, struct UidArray *uida)
{
  struct Buffer *cmd = buf_pool_get();
 
  int count = 0;
  int pos = 0;
  int rc = 0;
 
  do
  {
    buf_reset(cmd);
    buf_add_printf(cmd, "%s ", pre);
    rc = imap_make_msg_set(uida, cmd, &pos);
    if (rc > 0)
    {
      buf_add_printf(cmd, " %s", post);
      if (imap_exec(adata, buf_string(cmd), IMAP_CMD_QUEUE) != IMAP_EXEC_SUCCESS)
      {
        rc = -1;
        goto out;
      }
      count += rc;
    }
  } while (rc > 0);
 
  rc = count;
 
out:
  buf_pool_release(&cmd);
  return rc;
}