X-Git-Url: https://git.notmuchmail.org/git?a=blobdiff_plain;f=lib%2Fparse-sexp.cc;h=2555605853f3b0803ca8b463ae4288da85244aae;hb=200e164dc7acc3f19c6fe39164c472985e1ba384;hp=f031d7904a4bf8246979c819882736b9346f36ed;hpb=a2785c3919c56a370a7f860f4eeb93846f4a1e63;p=notmuch diff --git a/lib/parse-sexp.cc b/lib/parse-sexp.cc index f031d790..25556058 100644 --- a/lib/parse-sexp.cc +++ b/lib/parse-sexp.cc @@ -7,20 +7,83 @@ /* _sexp is used for file scope symbols to avoid clashing with * definitions from sexp.h */ +typedef enum { + SEXP_FLAG_NONE = 0, + SEXP_FLAG_FIELD = 1 << 0, +} _sexp_flag_t; + +typedef struct { + const char *name; + Xapian::Query::op xapian_op; + Xapian::Query initial; + _sexp_flag_t flags; +} _sexp_prefix_t; + +static _sexp_prefix_t prefixes[] = +{ + { "and", Xapian::Query::OP_AND, Xapian::Query::MatchAll, + SEXP_FLAG_NONE }, + { "not", Xapian::Query::OP_AND_NOT, Xapian::Query::MatchAll, + SEXP_FLAG_NONE }, + { "or", Xapian::Query::OP_OR, Xapian::Query::MatchNothing, + SEXP_FLAG_NONE }, + { "subject", Xapian::Query::OP_AND, Xapian::Query::MatchAll, + SEXP_FLAG_FIELD }, + { } +}; + +static notmuch_status_t _sexp_to_xapian_query (notmuch_database_t *notmuch, + const _sexp_prefix_t *parent, + const sexp_t *sx, + Xapian::Query &output); + +static notmuch_status_t +_sexp_combine_query (notmuch_database_t *notmuch, + const _sexp_prefix_t *parent, + Xapian::Query::op operation, + Xapian::Query left, + const sexp_t *sx, + Xapian::Query &output) +{ + Xapian::Query subquery; + + notmuch_status_t status; + + /* if we run out elements, return accumulator */ + + if (! sx) { + output = left; + return NOTMUCH_STATUS_SUCCESS; + } + + status = _sexp_to_xapian_query (notmuch, parent, sx, subquery); + if (status) + return status; + + return _sexp_combine_query (notmuch, + parent, + operation, + Xapian::Query (operation, left, subquery), + sx->next, output); +} + /* Here we expect the s-expression to be a proper list, with first * element defining and operation, or as a special case the empty * list */ static notmuch_status_t -_sexp_to_xapian_query (notmuch_database_t *notmuch, const sexp_t *sx, +_sexp_to_xapian_query (notmuch_database_t *notmuch, const _sexp_prefix_t *parent, const sexp_t *sx, Xapian::Query &output) { if (sx->ty == SEXP_VALUE) { std::string term = Xapian::Unicode::tolower (sx->val); Xapian::Stem stem = *(notmuch->stemmer); + std::string term_prefix = parent ? _find_prefix (parent->name) : ""; if (sx->aty == SEXP_BASIC) - term = "Z" + stem (term); + term = "Z" + term_prefix + stem (term); + else + term = term_prefix + term; output = Xapian::Query (term); return NOTMUCH_STATUS_SUCCESS; @@ -32,11 +95,29 @@ _sexp_to_xapian_query (notmuch_database_t *notmuch, const sexp_t *sx, return NOTMUCH_STATUS_SUCCESS; } - if (sx->list->ty == SEXP_VALUE) - _notmuch_database_log (notmuch, "unknown prefix '%s'\n", sx->list->val); - else + if (sx->list->ty == SEXP_LIST) { _notmuch_database_log (notmuch, "unexpected list in field/operation position\n", sx->list->val); + return NOTMUCH_STATUS_BAD_QUERY_SYNTAX; + } + + for (_sexp_prefix_t *prefix = prefixes; prefix && prefix->name; prefix++) { + if (strcmp (prefix->name, sx->list->val) == 0) { + if (prefix->flags & SEXP_FLAG_FIELD) { + if (parent) { + _notmuch_database_log (notmuch, "nested field: '%s' inside '%s'\n", + prefix->name, parent->name); + return NOTMUCH_STATUS_BAD_QUERY_SYNTAX; + } + parent = prefix; + } + + return _sexp_combine_query (notmuch, parent, prefix->xapian_op, prefix->initial, + sx->list->next, output); + } + } + + _notmuch_database_log (notmuch, "unknown prefix '%s'\n", sx->list->val); return NOTMUCH_STATUS_BAD_QUERY_SYNTAX; } @@ -54,6 +135,6 @@ _notmuch_sexp_string_to_xapian_query (notmuch_database_t *notmuch, const char *q return NOTMUCH_STATUS_BAD_QUERY_SYNTAX; } - return _sexp_to_xapian_query (notmuch, sx, output); + return _sexp_to_xapian_query (notmuch, NULL, sx, output); } #endif