complete.c

Go to the documentation of this file.

 
#include "config.h"
#include <notmuch.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include "private.h"
#include "mutt/lib.h"
#include "config/lib.h"
#include "core/lib.h"
#include "gui/lib.h"
#include "lib.h"
#include "complete/lib.h"
#include "editor/lib.h"
#include "index/lib.h"

static void nm_complete_data_init(struct CompletionData *cd, const char *pt)
{
  completion_data_free_match_strings(cd);
  mutt_str_copy(cd->user_typed, pt, sizeof(cd->user_typed));
  memset(cd->match_list, 0, cd->match_list_len * sizeof(const char *));
  memset(cd->completed, 0, sizeof(cd->completed));
  cd->num_matched = 0;
  cd->free_match_strings = true;
}

static int complete_all_nm_tags_fresh(struct CompletionData *cd)
{
  int rc = -1;
  struct Mailbox *m_cur = get_current_mailbox();
  notmuch_database_t *db = NULL;
  notmuch_tags_t *tags = NULL;
  const char *db_filename = nm_db_get_filename(m_cur);
 
#if !LIBNOTMUCH_CHECK_VERSION(5, 4, 0)
  if (!db_filename)
    db_filename = cs_subset_string(NeoMutt->sub, "folder");
#endif
 
  if (!(db = nm_db_do_open(db_filename, false, false)) ||
      !(tags = notmuch_database_get_all_tags(db)))
  {
    goto done;
  }
 
  while (notmuch_tags_valid(tags))
  {
    const char *tag = notmuch_tags_get(tags);
 
    /* Keep a private copy because notmuch owns the iterator storage. */
    if (*tag && candidate(cd, cd->user_typed, tag, cd->completed, sizeof(cd->completed)))
      cd->match_list[cd->num_matched - 1] = mutt_str_dup(tag);
 
    notmuch_tags_move_to_next(tags);
  }
 
  rc = 0;
 
done:
  if (tags)
    notmuch_tags_destroy(tags);
  if (db)
    nm_db_free(db);
 
  return rc;
}

int complete_all_nm_tags(struct CompletionData *cd, const char *pt)
{
  int rc = -1;
 
  nm_complete_data_init(cd, pt);
  rc = complete_all_nm_tags_fresh(cd);
  if (rc == 0)
  {
    matches_ensure_morespace(cd, cd->num_matched);
    cd->match_list[cd->num_matched++] = mutt_str_dup(cd->user_typed);
  }
 
  return rc;
}

bool mutt_nm_query_complete(struct CompletionData *cd, struct Buffer *buf, int numtabs)
{
  char *pt = buf->data;
  int spaces;
 
  SKIPWS(pt);
  spaces = pt - buf->data;
 
  pt = (char *) buf_rfind(buf, "tag:");
  if (pt)
  {
    pt += 4;
    if (numtabs == 1)
    {
      /* First TAB. Collect all the matches */
      complete_all_nm_tags(cd, pt);
 
      /* All matches are stored. Longest non-ambiguous string is ""
       * i.e. don't change 'buf'. Fake successful return this time.  */
      if (cd->user_typed[0] == '\0')
        return true;
    }
 
    if ((cd->completed[0] == '\0') && (cd->user_typed[0] != '\0'))
      return false;
 
    /* cd->num_matched will _always_ be at least 1 since the initial
     * user-typed string is always stored */
    if ((numtabs == 1) && (cd->num_matched == 2))
    {
      snprintf(cd->completed, sizeof(cd->completed), "%s", NONULL(cd->match_list[0]));
    }
    else if ((numtabs > 1) && (cd->num_matched > 2))
    {
      /* cycle through all the matches */
      snprintf(cd->completed, sizeof(cd->completed), "%s",
               NONULL(cd->match_list[(numtabs - 2) % cd->num_matched]));
    }
 
    /* return the completed query */
    strncpy(pt, cd->completed, buf->data + buf->dsize - pt - spaces);
  }
  else
  {
    return false;
  }
 
  return true;
}

bool mutt_nm_tag_complete(struct CompletionData *cd, struct Buffer *buf, int numtabs)
{
  if (!buf)
    return false;
 
  char *pt = buf->data;
 
  /* Only examine the last token */
  char *last_space = strrchr(buf->data, ' ');
  if (last_space)
    pt = (last_space + 1);
 
  /* Skip the +/- */
  if ((pt[0] == '+') || (pt[0] == '-'))
    pt++;
 
  if (numtabs == 1)
  {
    /* First TAB. Collect all the matches */
    complete_all_nm_tags(cd, pt);
 
    /* All matches are stored. Longest non-ambiguous string is ""
     * i.e. don't change 'buf'. Fake successful return this time.  */
    if (cd->user_typed[0] == '\0')
      return true;
  }
 
  if ((cd->completed[0] == '\0') && (cd->user_typed[0] != '\0'))
    return false;
 
  /* cd->num_matched will _always_ be at least 1 since the initial
   * user-typed string is always stored */
  if ((numtabs == 1) && (cd->num_matched == 2))
  {
    snprintf(cd->completed, sizeof(cd->completed), "%s", NONULL(cd->match_list[0]));
  }
  else if ((numtabs > 1) && (cd->num_matched > 2))
  {
    /* cycle through all the matches */
    snprintf(cd->completed, sizeof(cd->completed), "%s",
             NONULL(cd->match_list[(numtabs - 2) % cd->num_matched]));
  }
 
  /* return the completed query */
  strncpy(pt, cd->completed, buf->data + buf->dsize - pt);
 
  return true;
}

enum FunctionRetval complete_nm_query(struct EnterWindowData *wdata, int op)
{
  if (!wdata || ((op != OP_EDITOR_COMPLETE) && (op != OP_EDITOR_COMPLETE_QUERY)))
    return FR_NO_ACTION;
 
  int rc = FR_SUCCESS;
  buf_mb_wcstombs(wdata->buffer, wdata->state->wbuf, wdata->state->curpos);
  if (!mutt_nm_query_complete(wdata->cd, wdata->buffer, wdata->tabs))
    rc = FR_ERROR;
 
  replace_part(wdata->state, 0, buf_string(wdata->buffer));
  return rc;
}

enum FunctionRetval complete_nm_tag(struct EnterWindowData *wdata, int op)
{
  if (!wdata || ((op != OP_EDITOR_COMPLETE) && (op != OP_EDITOR_COMPLETE_QUERY)))
    return FR_NO_ACTION;
 
  int rc = FR_SUCCESS;
  buf_mb_wcstombs(wdata->buffer, wdata->state->wbuf, wdata->state->curpos);
  if (!mutt_nm_tag_complete(wdata->cd, wdata->buffer, wdata->tabs))
    rc = FR_ERROR;
 
  replace_part(wdata->state, 0, buf_string(wdata->buffer));
  return rc;
}

const struct CompleteOps CompleteNmQueryOps = {
  .complete = complete_nm_query,
};

const struct CompleteOps CompleteNmTagOps = {
  .complete = complete_nm_tag,
};