From 12f47656be36f4b2c68ecf79d69543ab1343248c Mon Sep 17 00:00:00 2001
From: Wolfgang Sourdeau <wsourdeau@inverse.ca>
Date: Thu, 4 Aug 2011 17:03:29 +0000
Subject: [PATCH] Monotone-Parent: 260a782f71d54b70c19078f5cbc97b7317db579a
 Monotone-Revision: a18a109c58aa27267200fdcfd8fda6939aa301ff

Monotone-Author: wsourdeau@inverse.ca
Monotone-Date: 2011-08-04T17:03:29
Monotone-Branch: ca.inverse.sogo
---
 OpenChange/MAPIStoreDraftsMessage.m | 25 +++++++++++++++++++++++++
 OpenChange/MAPIStoreMailMessage.h   |  5 +++++
 2 files changed, 30 insertions(+)

diff --git a/OpenChange/MAPIStoreDraftsMessage.m b/OpenChange/MAPIStoreDraftsMessage.m
index ec199c0da..282abb780 100644
--- a/OpenChange/MAPIStoreDraftsMessage.m
+++ b/OpenChange/MAPIStoreDraftsMessage.m
@@ -33,6 +33,7 @@
 
 #import "MAPIStoreContext.h"
 #import "MAPIStoreTypes.h"
+#import "NSObject+MAPIStore.h"
 #import "NSString+MAPIStore.h"
 
 #import "MAPIStoreDraftsMessage.h"
@@ -147,6 +148,22 @@ typedef void (*getMessageData_inMemCtx_) (MAPIStoreMessage *, SEL,
     [super getMessageData: dataPtr inMemCtx: memCtx];
 }
 
+- (int) getPrIconIndex: (void **) data
+              inMemCtx: (TALLOC_CTX *) memCtx
+{
+  int rc;
+
+  if ([sogoObject isKindOfClass: SOGoDraftObjectK])
+    {
+      *data = MAPILongValue (memCtx, 0xffffffff);
+      rc = MAPISTORE_SUCCESS;
+    }
+  else
+    rc = [super getPrIconIndex: data inMemCtx: memCtx];
+
+  return rc;
+}
+
 - (int) getPrImportance: (void **) data
                inMemCtx: (TALLOC_CTX *) memCtx
 {
@@ -197,6 +214,14 @@ typedef void (*getMessageData_inMemCtx_) (MAPIStoreMessage *, SEL,
   return rc;
 }
 
+- (int) getPrFlagStatus: (void **) data
+               inMemCtx: (TALLOC_CTX *) memCtx
+{
+  return ([sogoObject isKindOfClass: SOGoDraftObjectK]
+          ? [self getLongZero: data inMemCtx: memCtx]
+          : [super getPrFlagStatus: data inMemCtx: memCtx]);
+}
+
 - (void) _saveAttachment: (NSString *) attachmentKey
 {
   NSDictionary *properties, *metadata;
diff --git a/OpenChange/MAPIStoreMailMessage.h b/OpenChange/MAPIStoreMailMessage.h
index 430c0f80d..a8c56a530 100644
--- a/OpenChange/MAPIStoreMailMessage.h
+++ b/OpenChange/MAPIStoreMailMessage.h
@@ -48,6 +48,11 @@
 
 - (NSString *) subject;
 
+- (int) getPrIconIndex: (void **) data
+              inMemCtx: (TALLOC_CTX *) memCtx;
+- (int) getPrFlagStatus: (void **) data
+               inMemCtx: (TALLOC_CTX *) memCtx;
+
 - (int) getPrMessageFlags: (void **) data
                  inMemCtx: (TALLOC_CTX *) memCtx;
 - (int) getPrImportance: (void **) data