oc: monkeypatched NGImap4Connection class from sope

To improve fetchUids method so we can retrieve uids in batches.
Otherwise we will face an error 'too long request' kind

OpenChange/GNUmakefile

@@ -122,7 +122,9 @@ $(SOGOBACKEND)_OBJC_FILES += \
 	\
 	RTFHandler.m \
 	\
-	Codepages.m
+	Codepages.m \
+	\
+	NGImap4Connection+Monkeypatching.m
 
 
 $(SOGOBACKEND)_RESOURCE_FILES += \

OpenChange/NGImap4Connection+Monkeypatching.h

@@ -0,0 +1,37 @@
+/* NGImap4Connection+Monkeypatching.h - this file is part of SOGo
+ *
+ * Copyright (C) 2014 Jesús García Sáez
+ *
+ * This file is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3, or (at your option)
+ * any later version.
+ *
+ * This file is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#ifndef __OpenChange_NGImap4Connection_Monkeypatching_H__
+#define __OpenChange_NGImap4Connection_Monkeypatching_H__
+
+#import <Foundation/NSArray.h>
+#import <NGImap4/NGImap4Client.h>
+#import <NGImap4/NGImap4Connection.h>
+
+
+@interface NGImap4Connection (Monkeypatching)
+
+- (NSArray *)fetchUIDs:(NSArray *)_uids inURL:(NSURL *)_url
+                                        parts:(NSArray *)_parts;
+
+@end
+
+
+#endif // __OpenChange_NGImap4Connection_Monkeypatching_H__

OpenChange/NGImap4Connection+Monkeypatching.m

@@ -0,0 +1,108 @@
+/* NGImap4Connection+Monkeypatching.m - this file is part of SOGo
+ *
+ * Copyright (C) 2014 Jesús García Sáez
+ *
+ * This file is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3, or (at your option)
+ * any later version.
+ *
+ * This file is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#import "NGImap4Connection+Monkeypatching.h"
+
+#import <Foundation/NSObject.h>
+#import <Foundation/NSDictionary.h>
+
+#import <NGExtensions/NSObject+Logs.h>
+
+
+@implementation NGImap4Connection (Monkeypatching)
+
+- (NSArray *)fetchUIDs:(NSArray *)_uids inURL:(NSURL *)_url
+                                        parts:(NSArray *)_parts
+{
+  // currently returns a dict?!
+  /*
+    Allowed fetch keys:
+      UID
+      BODY.PEEK[<section>]<<partial>>
+      BODY            [this is the bodystructure, supported]
+      BODYSTRUCTURE   [not supported yet!]
+      ENVELOPE        [this is a parsed header, but does not include type]
+      FLAGS
+      INTERNALDATE
+      RFC822
+      RFC822.HEADER
+      RFC822.SIZE
+      RFC822.TEXT
+  */
+  NSMutableDictionary *result = nil;
+  NSUInteger i, total, step = 1000;
+
+  if (_uids == nil || [_uids count] == 0)
+    return nil;
+
+  /* select folder */
+
+  if (![self selectFolder:_url])
+    return nil;
+
+  /* fetch parts */
+
+  total = [_uids count];
+  for (i = 0; i < total; i += step) {
+    NSRange range;
+    NSArray *partial_uids;
+    NSDictionary *partial_result;
+
+    range = NSMakeRange(i, (i + step) > total ? (total - i) : step);
+    partial_uids = [_uids subarrayWithRange: range];
+
+    /* We will only fetch "step" uids each time */
+    partial_result = [[self client] fetchUids:partial_uids parts:_parts];
+
+    if (![[partial_result valueForKey:@"result"] boolValue]) {
+      [self errorWithFormat: @"could not fetch %d uids for url: %@", [_uids count], _url];
+      return nil;
+    }
+
+    if (!result) {
+      /* First iteration, first result */
+      result = [[partial_result mutableCopy] autorelease];
+      /* RawResponse has already been processed, ignore it */
+      [result removeObjectForKey: @"RawResponse"];
+      continue;
+    }
+
+    /* Merge partial_result into previous result */
+    for (id key in [partial_result keyEnumerator]) {
+      id obj, current_obj;
+
+      current_obj = [result objectForKey: key];
+      if (!current_obj) continue;
+
+      obj = [partial_result objectForKey: key];
+      if ([obj isKindOfClass: [NSArray class]]) {
+        NSArray *data, *current_data, *new_data;
+        data = obj;
+        current_data = current_obj;
+        new_data = [current_data arrayByAddingObjectsFromArray: data];
+        [result setObject: new_data forKey: key];
+      }
+    }
+  }
+
+  return (id)result;
+}
+
+@end