Essentially replace each call to notmuch_count_* with an if block. --- notmuch-count.c | 14 ++++++++++++-- notmuch-reply.c | 8 +++++++- notmuch-search.c | 14 ++++++++++++-- notmuch-show.c | 8 +++++++- 4 files changed, 38 insertions(+), 6 deletions(-)
diff --git a/notmuch-count.c b/notmuch-count.c index 6058f7c..cd6b0c9 100644 --- a/notmuch-count.c +++ b/notmuch-count.c @@ -71,6 +71,8 @@ print_count (notmuch_database_t *notmuch, const char *query_str, { notmuch_query_t *query; size_t i; + unsigned count; + notmuch_status_t status; query = notmuch_query_create (notmuch, query_str); if (query == NULL) { @@ -83,10 +85,18 @@ print_count (notmuch_database_t *notmuch, const char *query_str, switch (output) { case OUTPUT_MESSAGES: - printf ("%u\n", notmuch_query_count_messages (query)); + if ((status = notmuch_query_count_messages_st (query, &count))) { + fprintf (stderr, "Error: %s\n", notmuch_status_to_string (status)); + return 1; + } + printf ("%u\n", count); break; case OUTPUT_THREADS: - printf ("%u\n", notmuch_query_count_threads (query)); + if ((status = notmuch_query_count_threads_st (query, &count))) { + fprintf (stderr, "Error: %s\n", notmuch_status_to_string (status)); + return 1; + } + printf ("%u\n", count); break; case OUTPUT_FILES: printf ("%u\n", count_files (query)); diff --git a/notmuch-reply.c b/notmuch-reply.c index 7c1c809..0b3e5fd 100644 --- a/notmuch-reply.c +++ b/notmuch-reply.c @@ -650,8 +650,14 @@ notmuch_reply_format_sprinter(void *ctx, notmuch_messages_t *messages; notmuch_message_t *message; mime_node_t *node; + unsigned count; + notmuch_status_t status; - if (notmuch_query_count_messages (query) != 1) { + if ((status = notmuch_query_count_messages_st (query, &count))) { + fprintf (stderr, "Error: %s.\n", notmuch_status_to_string (status)); + return 1; + } + if (count != 1) { fprintf (stderr, "Error: search term did not match precisely one message.\n"); return 1; } diff --git a/notmuch-search.c b/notmuch-search.c index a591d45..ba631c0 100644 --- a/notmuch-search.c +++ b/notmuch-search.c @@ -113,7 +113,14 @@ do_search_threads (search_context_t *ctx) int i; if (ctx->offset < 0) { - ctx->offset += notmuch_query_count_threads (ctx->query); + unsigned count; + notmuch_status_t status; + if ((status = notmuch_query_count_threads_st (ctx->query, &count))) { + fprintf (stderr, "Error: %s\n", notmuch_status_to_string (status)); + return 1; + } + + ctx->offset += count; if (ctx->offset < 0) ctx->offset = 0; } @@ -414,7 +421,10 @@ do_search_messages (search_context_t *ctx) int i; if (ctx->offset < 0) { - ctx->offset += notmuch_query_count_messages (ctx->query); + unsigned count; + if (notmuch_query_count_messages_st (ctx->query, &count)) + return 1; + ctx->offset += count; if (ctx->offset < 0) ctx->offset = 0; } diff --git a/notmuch-show.c b/notmuch-show.c index d416fbd..749b1af 100644 --- a/notmuch-show.c +++ b/notmuch-show.c @@ -982,8 +982,14 @@ do_show_single (void *ctx, { notmuch_messages_t *messages; notmuch_message_t *message; + unsigned count; + notmuch_status_t status; + if ((status = notmuch_query_count_messages_st (query, &count))) { + fprintf (stderr, "Error: %s\n", notmuch_status_to_string (status)); + return 1; + } - if (notmuch_query_count_messages (query) != 1) { + if (count != 1) { fprintf (stderr, "Error: search term did not match precisely one message.\n"); return 1; } -- 2.1.4