selection.c

Go to the documentation of this file.

 
#include "config.h"
#include <stdbool.h>
#include <stddef.h>
#include "mutt/lib.h"
#include "email/lib.h"
#include "menu/lib.h"
#include "attach.h"
#include "recvattach.h"

static bool aa_contains_attach(struct AttachPtrArray *aa, struct AttachPtr *ap)
{
  if (!aa || !ap)
    return false;
 
  struct AttachPtr **app = NULL;
  ARRAY_FOREACH(app, aa)
  {
    if (*app == ap)
      return true;
  }
 
  return false;
}

static void aa_add_attach(struct AttachPtrArray *aa, struct AttachPtr *ap)
{
  if (!aa_contains_attach(aa, ap))
    ARRAY_ADD(aa, ap);
}

static void aa_add_folded(struct AttachPtrArray *aa, struct AttachCtx *actx, int rindex)
{
  if (!aa || !actx || (rindex < 0) || (rindex >= actx->idxlen))
    return;
 
  struct AttachPtr *ap = actx->idx[rindex];
  aa_add_attach(aa, ap);
 
  if (!ap->collapsed)
    return;
 
  const int level = ap->level;
  for (int i = rindex + 1; (i < actx->idxlen) && (actx->idx[i]->level > level); i++)
  {
    aa_add_attach(aa, actx->idx[i]);
  }
}

static int aa_add_tagged(struct AttachPtrArray *aa, struct AttachCtx *actx)
{
  if (!aa || !actx)
    return -1;
 
  for (int i = 0; i < actx->idxlen; i++)
  {
    if (actx->idx[i]->body->tagged)
      aa_add_folded(aa, actx, i);
  }
 
  return ARRAY_SIZE(aa);
}

int aa_add_selection(struct AttachPtrArray *aa, struct AttachCtx *actx,
                     struct Menu *menu, bool use_tagged, int count)
{
  if (!aa || !actx || !menu)
    return -1;
 
  if (use_tagged)
    return aa_add_tagged(aa, actx);
 
  const int index = menu_get_index(menu);
  if ((index < 0) || (index >= actx->vcount))
    return -1;
 
  int n = (count > 1) ? count : 1;
  if ((index + n) > actx->vcount)
    n = actx->vcount - index;
 
  for (int i = 0; i < n; i++)
  {
    const int rindex = actx->v2r[index + i];
    if ((rindex < 0) || (rindex >= actx->idxlen))
      return -1;
 
    aa_add_folded(aa, actx, rindex);
  }
 
  return ARRAY_SIZE(aa);
}

int ba_add_selection(struct BodyArray *ba, struct AttachCtx *actx,
                     struct Menu *menu, bool use_tagged, int count)
{
  if (!ba)
    return -1;
 
  struct AttachPtrArray aa = ARRAY_HEAD_INITIALIZER;
  if (aa_add_selection(&aa, actx, menu, use_tagged, count) < 0)
  {
    ARRAY_FREE(&aa);
    return -1;
  }
 
  struct AttachPtr **app = NULL;
  ARRAY_FOREACH(app, &aa)
  {
    ARRAY_ADD(ba, (*app)->body);
  }
 
  const int rc = ARRAY_SIZE(ba);
  ARRAY_FREE(&aa);
  return rc;
}