From: Marco Bettini <marco.bettini@open-xchange.com>
Date: Wed, 8 Nov 2023 14:20:49 +0000 (+0000)
Subject: imapc: Propagate INPROGRESS notifications from backends
X-Git-Tag: 2.4.0~2410
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=3c26e25b3117aa87d3c1683d51c0f7f57b73b305;p=thirdparty%2Fdovecot%2Fcore.git

imapc: Propagate INPROGRESS notifications from backends
---

diff --git a/src/lib-storage/index/imapc/imapc-storage.c b/src/lib-storage/index/imapc/imapc-storage.c
index 2f09c975d3..03ee58a7a9 100644
--- a/src/lib-storage/index/imapc/imapc-storage.c
+++ b/src/lib-storage/index/imapc/imapc-storage.c
@@ -60,6 +60,8 @@ static void imapc_untagged_status(const struct imapc_untagged_reply *reply,
 				  struct imapc_storage_client *client);
 static void imapc_untagged_namespace(const struct imapc_untagged_reply *reply,
 				     struct imapc_storage_client *client);
+static void imapc_untagged_inprogress(const struct imapc_untagged_reply *reply,
+				      struct imapc_storage_client *client);
 static int imapc_mailbox_run_status(struct mailbox *box,
 				    enum mailbox_status_items items,
 				    struct mailbox_status *status_r);
@@ -460,7 +462,8 @@ imapc_storage_create(struct mail_storage *_storage,
 					       imapc_untagged_status);
 	imapc_storage_client_register_untagged(storage->client, "NAMESPACE",
 					       imapc_untagged_namespace);
-
+	imapc_storage_client_register_untagged(storage->client, "OK",
+					       imapc_untagged_inprogress);
 	return 0;
 }
 
@@ -996,6 +999,69 @@ static void imapc_untagged_namespace(const struct imapc_untagged_reply *reply,
 	}
 }
 
+static void
+imapc_parse_inprogress_start_time(struct imapc_storage_client *client,
+				  const char *tag,
+				  struct mail_storage_progress_details *detail_r)
+{
+	struct imapc_command *cmd =
+		imapc_client_find_command_by_tag(client->client, tag);
+	if (cmd == NULL)
+		return;
+
+	struct timeval start_time = imapc_command_get_start_time(cmd);
+	if (start_time.tv_sec == 0)
+		return;
+
+	detail_r->start_time = start_time;
+	detail_r->now = ioloop_timeval;
+}
+
+static void imapc_parse_inprogress(const struct imapc_untagged_reply *reply,
+				   struct imapc_storage_client *client,
+				   struct mail_storage_progress_details *detail_r)
+{
+	unsigned int count;
+	const struct imap_arg *code_args;
+
+	i_zero(detail_r);
+	if (!imap_arg_get_list_full(&reply->args[1], &code_args, &count) || count != 3)
+		return;
+
+	const char *value;
+	if (!imap_arg_get_atom(&code_args[1], &value) ||
+	    str_to_uint32(value, &detail_r->processed) < 0)
+	    	detail_r->processed = 0;
+
+	if (detail_r->processed == 0 ||
+	    !imap_arg_get_atom(&code_args[2], &value) ||
+	    str_to_uint32(value, &detail_r->total) < 0)
+		detail_r->total = 0;
+
+	const char *tag;
+	if (detail_r->processed != 0 && imap_arg_get_astring(&code_args[0], &tag))
+		imapc_parse_inprogress_start_time(client, tag, detail_r);
+
+	detail_r->verb = "Processed";
+}
+
+static void imapc_untagged_inprogress(const struct imapc_untagged_reply *reply,
+				      struct imapc_storage_client *client)
+{
+	struct mail_storage *storage = &client->_storage->storage;
+	if (storage->callbacks.notify_progress == NULL ||
+	    reply->resp_text_key == NULL ||
+	    strcasecmp(reply->resp_text_key, "INPROGRESS") != 0)
+		return;
+
+	struct imapc_mailbox *mbox = reply->untagged_box_context;
+	struct mailbox *box = mbox == NULL ? NULL : &mbox->box;
+
+	struct mail_storage_progress_details dtl;
+	imapc_parse_inprogress(reply, client, &dtl);
+	storage->callbacks.notify_progress(box, &dtl, storage->callback_context);
+}
+
 static void imapc_mailbox_get_selected_status(struct imapc_mailbox *mbox,
 					      enum mailbox_status_items items,
 					      struct mailbox_status *status_r)