From 37d06c6f211fe0c61ed96a255f36bfb1b8a329b3 Mon Sep 17 00:00:00 2001
From: Francis Lachapelle <flachapelle@inverse.ca>
Date: Fri, 22 Oct 2021 15:20:49 -0400
Subject: [PATCH] fix(mail(js)): update list of labels when adding one to a
 message

---
 UI/Templates/MailerUI/UIxMailFolderTemplate.wox    |  2 +-
 UI/WebServerResources/js/Mailer/Mailbox.service.js |  9 +++++----
 UI/WebServerResources/js/Mailer/Message.service.js | 13 ++++++++++++-
 3 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/UI/Templates/MailerUI/UIxMailFolderTemplate.wox b/UI/Templates/MailerUI/UIxMailFolderTemplate.wox
index e04d473a0..f31453cc4 100644
--- a/UI/Templates/MailerUI/UIxMailFolderTemplate.wox
+++ b/UI/Templates/MailerUI/UIxMailFolderTemplate.wox
@@ -94,7 +94,7 @@
                   sg-false-value="0"> <var:string label:value="Show flagged messages only"/></sg-checkmark>
             </md-menu-item>
             <md-menu-divider> <!-- divider --></md-menu-divider>
-            <md-menu-item ng-repeat="label in ::mailbox.selectedFolder.$labels track by label.imapName">
+            <md-menu-item ng-repeat="label in mailbox.selectedFolder.$labels track by label.imapName">
               <sg-checkmark
                   ng-change="mailbox.selectedFolder.$filter(mailbox.service.$query)"
                   ng-model="mailbox.selectedFolder.$filteredLabels[label.imapName]"
diff --git a/UI/WebServerResources/js/Mailer/Mailbox.service.js b/UI/WebServerResources/js/Mailer/Mailbox.service.js
index c29cecfea..ba2572b95 100644
--- a/UI/WebServerResources/js/Mailer/Mailbox.service.js
+++ b/UI/WebServerResources/js/Mailer/Mailbox.service.js
@@ -706,13 +706,14 @@
    * @desc Fetch the list of labels associated to the mailbox. Use the cached value if available.
    * @returns a promise of the HTTP operation
    */
-  Mailbox.prototype.getLabels = function() {
+  Mailbox.prototype.getLabels = function(options) {
     var _this = this;
 
-    if (this.$labels)
-      return this.$labels;
+    if (this.$labels && !(options && options.reload))
+      return Mailbox.$q.when(this.$labels);
 
-    this.$filteredLabels = {};
+    if (angular.isUndefined(this.$filteredLabels))
+      this.$filteredLabels = {};
     return Mailbox.$$resource.fetch(this.id, 'labels').then(function(data) {
       _this.$labels = data;
       return _this.$labels;
diff --git a/UI/WebServerResources/js/Mailer/Message.service.js b/UI/WebServerResources/js/Mailer/Message.service.js
index 9c7bd6746..fc3dc1c24 100644
--- a/UI/WebServerResources/js/Mailer/Message.service.js
+++ b/UI/WebServerResources/js/Mailer/Message.service.js
@@ -496,7 +496,18 @@
    * @returns a promise of the HTTP operation
    */
   Message.prototype.addTag = function(tag) {
-    return this.$addOrRemoveTag('add', tag);
+    var _this = this,
+        _tag = tag.replace(/^_\$/, '$');
+    return this.$mailbox.getLabels().then(function(labels) {
+      var reload = !_.find(labels, function(label) {
+        return label.imapName == _tag;
+      });
+      return _this.$addOrRemoveTag('add', tag).then(function() {
+        if (reload)
+          // Update the list of labels for the mailbox
+          _this.$mailbox.getLabels({reload: true});
+      });
+    });
   };
 
   /**