mail-search-register-human.c revision 5c2d695acf9f95ae0dcdda89c4d2391ceda4d672
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen/* Copyright (c) 2010 Dovecot authors, see the included COPYING file */
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#include "lib.h"
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#include "ioloop.h"
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#include "array.h"
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#include "settings-parser.h"
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#include "imap-date.h"
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#include "mail-search-register.h"
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen#include "mail-search-parser.h"
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#include "mail-search-build.h"
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#include <time.h>
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#include <ctype.h>
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstruct mail_search_register *mail_search_register_human;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstatic struct mail_search_arg *
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainenhuman_search_or(struct mail_search_build_context *ctx)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen{
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen struct mail_search_arg *sarg;
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen /* this changes the parent arg to be an OR block instead of AND block */
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen ctx->parent->type = SEARCH_OR;
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen if (mail_search_build_key(ctx, ctx->parent, &sarg) < 0)
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen return NULL;
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen return sarg;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen}
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstatic struct mail_search_arg *
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenarg_new_human_date(struct mail_search_build_context *ctx,
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen enum mail_search_arg_type type,
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen enum mail_search_date_type date_type)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen{
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen struct mail_search_arg *sarg;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen const char *value, *error;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen struct tm tm;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen unsigned int secs;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen int tz;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen sarg = mail_search_build_new(ctx, type);
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen if (mail_search_parse_string(ctx->parser, &value) < 0)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen return NULL;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen /* a) yyyy-mm-dd
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen b) imap date-time
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen c) interval (e.g. n days) */
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen if (i_isdigit(value[0]) && i_isdigit(value[1]) &&
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen i_isdigit(value[2]) && i_isdigit(value[3]) && value[4] == '-' &&
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen i_isdigit(value[5]) && i_isdigit(value[6]) && value[7] == '-' &&
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen i_isdigit(value[8]) && i_isdigit(value[9]) && value[10] == '\0') {
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen memset(&tm, 0, sizeof(tm));
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen tm.tm_year = (value[0]-'0') * 1000 + (value[1]-'0') * 100 +
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen (value[2]-'0') * 10 + (value[3]-'0');
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen tm.tm_mon = (value[5]-'0') * 10 + (value[6]-'0') - 1;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen tm.tm_mday = (value[8]-'0') * 10 + (value[9]-'0');
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen sarg->value.time = mktime(&tm);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen } else if (imap_parse_datetime(value, &sarg->value.time, &tz)) {
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen /* imap date */
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen } else if (settings_get_time(value, &secs, &error) == 0) {
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen sarg->value.time = ioloop_time - secs;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen } else {
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen sarg->value.time = (time_t)-1;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen }
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen sarg->value.search_flags = MAIL_SEARCH_ARG_FLAG_USE_TZ;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen if (sarg->value.time == (time_t)-1) {
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen ctx->_error = p_strconcat(ctx->pool,
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen "Invalid search date parameter: ", value, NULL);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen return NULL;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen }
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen sarg->value.date_type = date_type;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen return sarg;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen}
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen#define CALLBACK_DATE(_func, _type, _date_type) \
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstatic struct mail_search_arg *\
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainenhuman_search_##_func(struct mail_search_build_context *ctx) \
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen{ \
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen return arg_new_human_date(ctx, _type, _date_type); \
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen}
4fc52b7b25c3d3f42348903e0154840f8761f306Timo SirainenCALLBACK_DATE(before, SEARCH_BEFORE, MAIL_SEARCH_DATE_TYPE_RECEIVED);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo SirainenCALLBACK_DATE(on, SEARCH_ON, MAIL_SEARCH_DATE_TYPE_RECEIVED);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo SirainenCALLBACK_DATE(since, SEARCH_SINCE, MAIL_SEARCH_DATE_TYPE_RECEIVED);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo SirainenCALLBACK_DATE(sentbefore, SEARCH_BEFORE, MAIL_SEARCH_DATE_TYPE_SENT);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo SirainenCALLBACK_DATE(senton, SEARCH_ON, MAIL_SEARCH_DATE_TYPE_SENT);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo SirainenCALLBACK_DATE(sentsince, SEARCH_SINCE, MAIL_SEARCH_DATE_TYPE_SENT);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo SirainenCALLBACK_DATE(savedbefore, SEARCH_BEFORE, MAIL_SEARCH_DATE_TYPE_SAVED);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo SirainenCALLBACK_DATE(savedon, SEARCH_ON, MAIL_SEARCH_DATE_TYPE_SAVED);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo SirainenCALLBACK_DATE(savedsince, SEARCH_SINCE, MAIL_SEARCH_DATE_TYPE_SAVED);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstatic struct mail_search_arg *
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenarg_new_human_size(struct mail_search_build_context *ctx,
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen enum mail_search_arg_type type)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen{
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen struct mail_search_arg *sarg;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen const char *value, *error;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen sarg = mail_search_build_new(ctx, type);
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen if (mail_search_parse_string(ctx->parser, &value) < 0)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen return NULL;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen if (settings_get_size(value, &sarg->value.size, &error) < 0) {
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen ctx->_error = p_strdup(ctx->pool, error);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen return NULL;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen }
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen return sarg;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen}
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstatic struct mail_search_arg *
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainenhuman_search_larger(struct mail_search_build_context *ctx)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen{
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen return arg_new_human_size(ctx, SEARCH_LARGER);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen}
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstatic struct mail_search_arg *
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainenhuman_search_smaller(struct mail_search_build_context *ctx)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen{
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen return arg_new_human_size(ctx, SEARCH_SMALLER);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen}
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstatic struct mail_search_arg *
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainenhuman_search_guid(struct mail_search_build_context *ctx)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen{
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen return mail_search_build_str(ctx, SEARCH_GUID);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen}
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstatic struct mail_search_arg *
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainenhuman_search_mailbox(struct mail_search_build_context *ctx)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen{
5c2d695acf9f95ae0dcdda89c4d2391ceda4d672Timo Sirainen return mail_search_build_str(ctx, SEARCH_MAILBOX);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen}
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstatic const struct mail_search_register_arg human_register_args[] = {
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "OR", human_search_or },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen /* dates */
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "BEFORE", human_search_before },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "ON", human_search_on },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "SINCE", human_search_since },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "SENTBEFORE", human_search_sentbefore },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "SENTON", human_search_senton },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "SENTSINCE", human_search_sentsince },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "SAVEDBEFORE", human_search_savedbefore },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "SAVEDON", human_search_savedon },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "SAVEDSINCE", human_search_savedsince },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "X-SAVEDBEFORE", human_search_savedbefore },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "X-SAVEDON", human_search_savedon },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "X-SAVEDSINCE", human_search_savedsince },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen /* sizes */
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "LARGER", human_search_larger },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "SMALLER", human_search_smaller },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen /* Other Dovecot extensions: */
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "GUID", human_search_guid },
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen { "MAILBOX", human_search_mailbox }
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen};
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenstruct mail_search_register *
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainenmail_search_register_init_human(struct mail_search_register *imap_register)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen{
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen struct mail_search_register *reg;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen mail_search_register_fallback_t *fallback;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen ARRAY_DEFINE(copy_args, const struct mail_search_register_arg);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen const struct mail_search_register_arg *human_args, *imap_args;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen unsigned int i, j, human_count, imap_count;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen int ret;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen reg = mail_search_register_init();
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen mail_search_register_add(reg, human_register_args,
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen N_ELEMENTS(human_register_args));
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen /* find and register args in imap that don't exist in human */
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen imap_args = mail_search_register_get(imap_register, &imap_count);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen human_args = mail_search_register_get(reg, &human_count);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen t_array_init(&copy_args, imap_count);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen for (i = j = 0; i < imap_count && j < human_count; ) {
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen ret = strcmp(imap_args[i].key, human_args[j].key);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen if (ret < 0) {
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen array_append(&copy_args, &imap_args[i], 1);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen i++;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen } else if (ret > 0) {
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen j++;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen } else {
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen i++; j++;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen }
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen }
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen for (; i < imap_count; i++)
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen array_append(&copy_args, &imap_args[i], 1);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen imap_args = array_get(&copy_args, &imap_count);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen mail_search_register_add(reg, imap_args, imap_count);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen if (mail_search_register_get_fallback(imap_register, &fallback))
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen mail_search_register_fallback(reg, fallback);
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen return reg;
4fc52b7b25c3d3f42348903e0154840f8761f306Timo Sirainen}