X-Git-Url: https://git.notmuchmail.org/git?p=notmuch;a=blobdiff_plain;f=bindings%2Fruby%2Fquery.c;h=8a2b4d3d48350e6b37ea9de73d86927ddb80e301;hp=ef9e1a077b28ddbbb31e33fde16fe1e584b40e3d;hb=HEAD;hpb=ed38940323ba80d0136688405cbd923fa088d706 diff --git a/bindings/ruby/query.c b/bindings/ruby/query.c index ef9e1a07..077def02 100644 --- a/bindings/ruby/query.c +++ b/bindings/ruby/query.c @@ -1,6 +1,6 @@ /* The Ruby interface to the notmuch mail library * - * Copyright © 2010 Ali Polatel + * Copyright © 2010, 2011, 2012 Ali Polatel * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by @@ -13,7 +13,7 @@ * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License - * along with this program. If not, see http://www.gnu.org/licenses/ . + * along with this program. If not, see https://www.gnu.org/licenses/ . * * Author: Ali Polatel */ @@ -21,19 +21,14 @@ #include "defs.h" /* - * call-seq: QUERY.destroy => nil + * call-seq: QUERY.destroy! => nil * * Destroys the query, freeing all resources allocated for it. */ VALUE -notmuch_rb_query_destroy(VALUE self) +notmuch_rb_query_destroy (VALUE self) { - notmuch_query_t *query; - - Data_Get_Notmuch_Query(self, query); - - notmuch_query_destroy(query); - DATA_PTR(self) = NULL; + notmuch_rb_object_destroy (self, ¬much_rb_query_type); return Qnil; } @@ -44,13 +39,13 @@ notmuch_rb_query_destroy(VALUE self) * Get sort type of the +QUERY+ */ VALUE -notmuch_rb_query_get_sort(VALUE self) +notmuch_rb_query_get_sort (VALUE self) { notmuch_query_t *query; - Data_Get_Notmuch_Query(self, query); + Data_Get_Notmuch_Query (self, query); - return FIX2INT(notmuch_query_get_sort(query)); + return INT2FIX (notmuch_query_get_sort (query)); } /* @@ -59,16 +54,16 @@ notmuch_rb_query_get_sort(VALUE self) * Set sort type of the +QUERY+ */ VALUE -notmuch_rb_query_set_sort(VALUE self, VALUE sortv) +notmuch_rb_query_set_sort (VALUE self, VALUE sortv) { notmuch_query_t *query; - Data_Get_Notmuch_Query(self, query); + Data_Get_Notmuch_Query (self, query); - if (!FIXNUM_P(sortv)) - rb_raise(rb_eTypeError, "Not a Fixnum"); + if (!FIXNUM_P (sortv)) + rb_raise (rb_eTypeError, "Not a Fixnum"); - notmuch_query_set_sort(query, FIX2UINT(sortv)); + notmuch_query_set_sort (query, FIX2UINT (sortv)); return Qnil; } @@ -79,13 +74,51 @@ notmuch_rb_query_set_sort(VALUE self, VALUE sortv) * Get query string of the +QUERY+ */ VALUE -notmuch_rb_query_get_string(VALUE self) +notmuch_rb_query_get_string (VALUE self) +{ + notmuch_query_t *query; + + Data_Get_Notmuch_Query (self, query); + + return rb_str_new2 (notmuch_query_get_query_string (query)); +} + +/* + * call-seq: QUERY.add_tag_exclude(tag) => nil + * + * Add a tag that will be excluded from the query results by default. + */ +VALUE +notmuch_rb_query_add_tag_exclude (VALUE self, VALUE tagv) +{ + notmuch_query_t *query; + const char *tag; + + Data_Get_Notmuch_Query (self, query); + tag = RSTRING_PTR(tagv); + + notmuch_query_add_tag_exclude(query, tag); + return Qnil; +} + +/* + * call-seq: QUERY.omit_excluded=(fixnum) => nil + * + * Specify whether to omit excluded results or simply flag them. + * By default, this is set to +Notmuch::EXCLUDE_TRUE+. + */ +VALUE +notmuch_rb_query_set_omit_excluded (VALUE self, VALUE omitv) { notmuch_query_t *query; + notmuch_exclude_t value; - Data_Get_Notmuch_Query(self, query); + Data_Get_Notmuch_Query (self, query); - return rb_str_new2(notmuch_query_get_query_string(query)); + value = FIXNUM_P (omitv) ? FIX2UINT (omitv) : RTEST(omitv); + notmuch_query_set_omit_excluded (query, value); + + return Qnil; } /* @@ -94,18 +127,19 @@ notmuch_rb_query_get_string(VALUE self) * Search for threads */ VALUE -notmuch_rb_query_search_threads(VALUE self) +notmuch_rb_query_search_threads (VALUE self) { notmuch_query_t *query; notmuch_threads_t *threads; + notmuch_status_t status; - Data_Get_Notmuch_Query(self, query); + Data_Get_Notmuch_Query (self, query); - threads = notmuch_query_search_threads(query); - if (!threads) - rb_raise(notmuch_rb_eMemoryError, "Out of memory"); + status = notmuch_query_search_threads (query, &threads); + if (status) + notmuch_rb_status_raise (status); - return Data_Wrap_Struct(notmuch_rb_cThreads, NULL, NULL, threads); + return Data_Wrap_Notmuch_Object (notmuch_rb_cThreads, ¬much_rb_threads_type, threads); } /* @@ -114,16 +148,59 @@ notmuch_rb_query_search_threads(VALUE self) * Search for messages */ VALUE -notmuch_rb_query_search_messages(VALUE self) +notmuch_rb_query_search_messages (VALUE self) { notmuch_query_t *query; notmuch_messages_t *messages; + notmuch_status_t status; + + Data_Get_Notmuch_Query (self, query); + + status = notmuch_query_search_messages (query, &messages); + if (status) + notmuch_rb_status_raise (status); + + return Data_Wrap_Notmuch_Object (notmuch_rb_cMessages, ¬much_rb_messages_type, messages); +} + +/* + * call-seq: QUERY.count_messages => Fixnum + * + * Return an estimate of the number of messages matching a search + */ +VALUE +notmuch_rb_query_count_messages (VALUE self) +{ + notmuch_query_t *query; + notmuch_status_t status; + unsigned int count; + + Data_Get_Notmuch_Query (self, query); + + status = notmuch_query_count_messages (query, &count); + if (status) + notmuch_rb_status_raise (status); + + return UINT2NUM(count); +} + +/* + * call-seq: QUERY.count_threads => Fixnum + * + * Return an estimate of the number of threads matching a search + */ +VALUE +notmuch_rb_query_count_threads (VALUE self) +{ + notmuch_query_t *query; + notmuch_status_t status; + unsigned int count; - Data_Get_Notmuch_Query(self, query); + Data_Get_Notmuch_Query (self, query); - messages = notmuch_query_search_messages(query); - if (!messages) - rb_raise(notmuch_rb_eMemoryError, "Out of memory"); + status = notmuch_query_count_threads (query, &count); + if (status) + notmuch_rb_status_raise (status); - return Data_Wrap_Struct(notmuch_rb_cMessages, NULL, NULL, messages); + return UINT2NUM(count); }