sort.c

Go to the documentation of this file.

 
#include "config.h"
#include <stdbool.h>
#include <stdlib.h>
#include "mutt/lib.h"
#include "address/lib.h"
#include "config/lib.h"
#include "email/lib.h"
#include "core/lib.h"
#include "alias/lib.h"
#include "sort.h"
#include "nntp/lib.h"
#include "globals.h"
#include "mutt_logging.h"
#include "mutt_thread.h"
#include "mview.h"
#include "mx.h"
#include "score.h"
 
struct EmailCompare
{
  enum MailboxType type; 
  short sort;            
  short sort_aux;        
};
 
static int compare_email_shim(const void *a, const void *b, void *sdata)
{
  const struct Email *ea = *(struct Email const *const *) a;
  const struct Email *eb = *(struct Email const *const *) b;
  const struct EmailCompare *cmp = sdata;
  return mutt_compare_emails(ea, eb, cmp->type, cmp->sort, cmp->sort_aux);
}
 
static int compare_score(const struct Email *a, const struct Email *b, bool reverse)
{
  int result = mutt_numeric_cmp(b->score, a->score); /* note that this is reverse */
  return reverse ? -result : result;
}
 
static int compare_size(const struct Email *a, const struct Email *b, bool reverse)
{
  int result = mutt_numeric_cmp(a->body->length, b->body->length);
  return reverse ? -result : result;
}
 
static int compare_date_sent(const struct Email *a, const struct Email *b, bool reverse)
{
  int result = mutt_numeric_cmp(a->date_sent, b->date_sent);
  return reverse ? -result : result;
}
 
static int compare_subject(const struct Email *a, const struct Email *b, bool reverse)
{
  int rc;
 
  if (!a->env->real_subj)
  {
    if (!b->env->real_subj)
      rc = compare_date_sent(a, b, false);
    else
      rc = -1;
  }
  else if (!b->env->real_subj)
  {
    rc = 1;
  }
  else
  {
    rc = mutt_istr_cmp(a->env->real_subj, b->env->real_subj);
  }
  return reverse ? -rc : rc;
}
 
const char *mutt_get_name(const struct Address *a)
{
  struct Address *ali = NULL;
 
  if (a)
  {
    const bool c_reverse_alias = cs_subset_bool(NeoMutt->sub, "reverse_alias");
    if (c_reverse_alias && (ali = alias_reverse_lookup(a)) && ali->personal)
      return buf_string(ali->personal);
    if (a->personal)
      return buf_string(a->personal);
    if (a->mailbox)
      return mutt_addr_for_display(a);
  }
  /* don't return NULL to avoid segfault when printing/comparing */
  return "";
}
 
static int compare_to(const struct Email *a, const struct Email *b, bool reverse)
{
  char fa[128] = { 0 };
 
  mutt_str_copy(fa, mutt_get_name(TAILQ_FIRST(&a->env->to)), sizeof(fa));
  const char *fb = mutt_get_name(TAILQ_FIRST(&b->env->to));
  int result = mutt_istrn_cmp(fa, fb, sizeof(fa));
  return reverse ? -result : result;
}
 
static int compare_from(const struct Email *a, const struct Email *b, bool reverse)
{
  char fa[128] = { 0 };
 
  mutt_str_copy(fa, mutt_get_name(TAILQ_FIRST(&a->env->from)), sizeof(fa));
  const char *fb = mutt_get_name(TAILQ_FIRST(&b->env->from));
  int result = mutt_istrn_cmp(fa, fb, sizeof(fa));
  return reverse ? -result : result;
}
 
static int compare_date_received(const struct Email *a, const struct Email *b, bool reverse)
{
  int result = mutt_numeric_cmp(a->received, b->received);
  return reverse ? -result : result;
}
 
static int compare_order(const struct Email *a, const struct Email *b, bool reverse)
{
  int result = mutt_numeric_cmp(a->index, b->index);
  return reverse ? -result : result;
}
 
static int compare_spam(const struct Email *a, const struct Email *b, bool reverse)
{
  char *aptr = NULL, *bptr = NULL;
  int ahas, bhas;
  int result = 0;
  double difference;
 
  /* Firstly, require spam attributes for both msgs */
  /* to compare. Determine which msgs have one.     */
  ahas = a->env && !buf_is_empty(&a->env->spam);
  bhas = b->env && !buf_is_empty(&b->env->spam);
 
  /* If one msg has spam attr but other does not, sort the one with first. */
  if (ahas && !bhas)
    return reverse ? -1 : 1;
  if (!ahas && bhas)
    return reverse ? 1 : -1;
 
  /* Else, if neither has a spam attr, presume equality. Fall back on aux. */
  if (!ahas && !bhas)
    return 0;
 
  /* Both have spam attrs. */
 
  /* preliminary numeric examination */
  difference = (strtod(a->env->spam.data, &aptr) - strtod(b->env->spam.data, &bptr));
 
  /* map double into comparison (-1, 0, or 1) */
  result = ((difference < 0.0) ? -1 : (difference > 0.0) ? 1 : 0);
 
  /* If either aptr or bptr is equal to data, there is no numeric    */
  /* value for that spam attribute. In this case, compare lexically. */
  if ((aptr == a->env->spam.data) || (bptr == b->env->spam.data))
  {
    result = mutt_str_cmp(aptr, bptr);
    return reverse ? -result : result;
  }
 
  /* Otherwise, we have numeric value for both attrs. If these values */
  /* are equal, then we first fall back upon string comparison, then  */
  /* upon auxiliary sort.                                             */
  if (result == 0)
    result = mutt_str_cmp(aptr, bptr);
  return reverse ? -result : result;
}
 
static int compare_label(const struct Email *a, const struct Email *b, bool reverse)
{
  int ahas, bhas, result = 0;
 
  /* As with compare_spam, not all messages will have the x-label
   * property.  Blank X-Labels are treated as null in the index
   * display, so we'll consider them as null for sort, too.       */
  ahas = a->env && a->env->x_label && *(a->env->x_label);
  bhas = b->env && b->env->x_label && *(b->env->x_label);
 
  /* First we bias toward a message with a label, if the other does not. */
  if (ahas && !bhas)
    return reverse ? 1 : -1;
  if (!ahas && bhas)
    return reverse ? -1 : 1;
 
  /* If neither has a label, use aux sort. */
  if (!ahas && !bhas)
    return 0;
 
  /* If both have a label, we just do a lexical compare. */
  result = mutt_istr_cmp(a->env->x_label, b->env->x_label);
  return reverse ? -result : result;
}
 
static sort_mail_t get_sort_func(enum SortType method, enum MailboxType type)
{
  switch (method)
  {
    case SORT_DATE:
      return compare_date_sent;
    case SORT_FROM:
      return compare_from;
    case SORT_LABEL:
      return compare_label;
    case SORT_ORDER:
      if (type == MUTT_NNTP)
        return nntp_compare_order;
      else
        return compare_order;
    case SORT_RECEIVED:
      return compare_date_received;
    case SORT_SCORE:
      return compare_score;
    case SORT_SIZE:
      return compare_size;
    case SORT_SPAM:
      return compare_spam;
    case SORT_SUBJECT:
      return compare_subject;
    case SORT_TO:
      return compare_to;
    default:
      mutt_error(_("Could not find sorting function [report this bug]"));
      return NULL;
  }
  /* not reached */
}
 
int mutt_compare_emails(const struct Email *a, const struct Email *b,
                        enum MailboxType type, short sort, short sort_aux)
{
  sort_mail_t func = get_sort_func(sort & SORT_MASK, type);
  int rc = func(a, b, (sort & SORT_REVERSE) != 0);
  if (rc == 0)
  {
    func = get_sort_func(sort_aux & SORT_MASK, type);
    rc = func(a, b, (sort_aux & SORT_REVERSE) != 0);
  }
  if (rc == 0)
  {
    /* Fallback of last resort to preserve stable order; will only
     * return 0 if a and b have the same index, which is probably a
     * bug in the code. */
    func = compare_order;
    rc = func(a, b, false);
  }
  return rc;
}
 
void mutt_sort_headers(struct MailboxView *mv, bool init)
{
  if (!mv)
    return;
 
  struct Mailbox *m = mv->mailbox;
  if (!m || !m->emails[0])
    return;
 
  OptNeedResort = false;
 
  if (m->msg_count == 0)
  {
    /* this function gets called by mutt_sync_mailbox(), which may have just
     * deleted all the messages.  the virtual message numbers are not updated
     * in that routine, so we must make sure to zero the vcount member.  */
    m->vcount = 0;
    mutt_clear_threads(mv->threads);
    mv->vsize = 0;
    return; /* nothing to do! */
  }
 
  if (m->verbose)
    mutt_message(_("Sorting mailbox..."));
 
  const bool c_score = cs_subset_bool(NeoMutt->sub, "score");
  if (OptNeedRescore && c_score)
  {
    for (int i = 0; i < m->msg_count; i++)
    {
      struct Email *e = m->emails[i];
      if (!e)
        break;
      mutt_score_message(m, e, true);
    }
  }
  OptNeedRescore = false;
 
  if (OptResortInit)
  {
    OptResortInit = false;
    init = true;
  }
 
  if (init)
    mutt_clear_threads(mv->threads);
 
  const bool threaded = mutt_using_threads();
  if (threaded)
  {
    mutt_sort_threads(mv->threads, init);
  }
  else
  {
    struct EmailCompare cmp = { 0 };
    cmp.type = mx_type(m);
    cmp.sort = cs_subset_sort(NeoMutt->sub, "sort");
    cmp.sort_aux = cs_subset_sort(NeoMutt->sub, "sort_aux");
    mutt_qsort_r((void *) m->emails, m->msg_count, sizeof(struct Email *),
                 compare_email_shim, &cmp);
  }
 
  /* adjust the virtual message numbers */
  m->vcount = 0;
  for (int i = 0; i < m->msg_count; i++)
  {
    struct Email *e_cur = m->emails[i];
    if (!e_cur)
      break;
 
    if ((e_cur->vnum != -1) || (e_cur->collapsed && e_cur->visible))
    {
      e_cur->vnum = m->vcount;
      m->v2r[m->vcount] = i;
      m->vcount++;
    }
    e_cur->msgno = i;
  }
 
  /* re-collapse threads marked as collapsed */
  if (threaded)
  {
    mutt_thread_collapse_collapsed(mv->threads);
    mv->vsize = mutt_set_vnum(m);
  }
 
  if (m->verbose)
    mutt_clear_error();
}
 
void mutt_sort_order(struct Mailbox *m)
{
  if (!m)
    return;
 
  struct EmailCompare cmp = { 0 };
  cmp.type = mx_type(m);
  cmp.sort = SORT_ORDER;
  mutt_qsort_r((void *) m->emails, m->msg_count, sizeof(struct Email *),
               compare_email_shim, &cmp);
}