From fb7ec576ab3a353b6eb99614e9636a9096462807 Mon Sep 17 00:00:00 2001
From: alecpl <alec@alec.pl>
Date: Thu, 25 Mar 2010 12:29:02 -0400
Subject: [PATCH] - Added possibility to select all messages in a folder (#1484756)

---
 program/lib/imap.inc |  780 ++++++++++++++++++++---------------------------------------
 1 files changed, 268 insertions(+), 512 deletions(-)

diff --git a/program/lib/imap.inc b/program/lib/imap.inc
index 437ea59..fc55d73 100644
--- a/program/lib/imap.inc
+++ b/program/lib/imap.inc
@@ -173,18 +173,10 @@
 	var $forwarded = false;
 	var $junk = false;
 	var $flagged = false;
+	var $has_children = false;
+	var $depth = 0;
+	var $unread_children = 0;
 	var $others = array();
-}
-
-/**
- * @todo Change class vars to public/private
- */
-class iilThreadHeader
-{
-	var $id;
-	var $sbj;
-	var $irt;
-	var $mid;
 }
 
 function iil_xor($string, $string2) {
@@ -234,7 +226,7 @@
 	$line = '';
 
 	if (!$fp) {
-    		return $line;
+    		return NULL;
 	}
     
 	if (!$size) {
@@ -242,6 +234,10 @@
 	}
     
 	do {
+		if (feof($fp)) {
+			return $line ? $line : NULL;
+		}
+		
     		$buffer = fgets($fp, $size);
 
     		if ($buffer === false) {
@@ -264,6 +260,8 @@
 		$bytes = $a[2][0];
 		while (strlen($out) < $bytes) {
 			$line = iil_ReadBytes($fp, $bytes); 
+			if ($line === NULL)
+				break;
 			$out .= $line;
 		}
 
@@ -276,7 +274,8 @@
 	global $my_prefs;
 	$data = '';
 	$len  = 0;
-	do {
+	while ($len < $bytes && !feof($fp))
+	{
 		$d = fread($fp, $bytes-$len);
 		if (!empty($my_prefs['debug_mode']))
 			write_log('imap', 'S: '. $d);
@@ -286,7 +285,7 @@
     		        break; //nothing was read -> exit to avoid apache lockups
     		}
     		$len = $data_len;
-	} while ($len < $bytes);
+	};
 	
 	return $data;
 }
@@ -551,26 +550,19 @@
 	$iil_error = '';
 	$iil_errornum = 0;
 
-	// set some imap options
+	// set options
 	if (is_array($options)) {
-		foreach($options as $optkey => $optval) {
-			if ($optkey == 'imap') {
-				$auth_method = strtoupper($optval);
-			} else if ($optkey == 'rootdir') {
-    				$my_prefs['rootdir'] = $optval;
-			} else if ($optkey == 'delimiter') {
-    				$my_prefs['delimiter'] = $optval;
-			} else if ($optkey == 'debug_mode') {
-    				$my_prefs['debug_mode'] = $optval;
-			}
-		}
-	}
-
-	if (empty($auth_method))
+        $my_prefs = $options;
+    }
+    // set auth method
+    if (!empty($my_prefs['auth_method'])) {
+            $auth_method = strtoupper($my_prefs['auth_method']);
+	} else {
     		$auth_method = 'CHECK';
-		
+    }
+
 	$message = "INITIAL: $auth_method\n";
-		
+
 	$result = false;
 	
 	// initialize connection
@@ -580,13 +572,13 @@
 	$conn->selected    = '';
 	$conn->user        = $user;
 	$conn->host        = $host;
-	
+
 	if ($my_prefs['sort_field'] == 'INTERNALDATE') {
 		$IMAP_USE_INTERNAL_DATE = true;
 	} else if ($my_prefs['sort_field'] == 'DATE') {
     		$IMAP_USE_INTERNAL_DATE = false;
 	}
-	
+
 	//check input
 	if (empty($host)) {
 		$iil_error = "Empty host";
@@ -640,7 +632,7 @@
 		$conn->capability = explode(' ', strtoupper($matches[1]));
 	}
 
-	$conn->message .= $line . "\n";
+	$conn->message .= $line;
 
 	// TLS connection
 	if ($ICL_SSL == 'tls' && iil_C_GetCapability($conn, 'STARTTLS')) {
@@ -665,46 +657,50 @@
         	}
 	}
 
+	$orig_method = $auth_method;
+
 	if ($auth_method == 'CHECK') {
-		//check for supported auth methods
+		// check for supported auth methods
 		if (iil_C_GetCapability($conn, 'AUTH=CRAM-MD5') || iil_C_GetCapability($conn, 'AUTH=CRAM_MD5')) {
 			$auth_method = 'AUTH';
 		}
 		else {
-			//default to plain text auth
+			// default to plain text auth
 			$auth_method = 'PLAIN';
 		}
 	}
 
 	if ($auth_method == 'AUTH') {
-		//do CRAM-MD5 authentication
+		// do CRAM-MD5 authentication
 		iil_PutLine($conn->fp, "a000 AUTHENTICATE CRAM-MD5");
 		$line = trim(iil_ReadLine($conn->fp, 1024));
 
 		if ($line[0] == '+') {
-			//got a challenge string, try CRAM-5
+			// got a challenge string, try CRAM-MD5
 			$result = iil_C_Authenticate($conn, $user, $password, substr($line,2));
         		
 			// stop if server sent BYE response
-			if($result == -3) {
+			if ($result == -3) {
 	            		$iil_error = $conn->error;
 	            		$iil_errornum = $conn->errorNum;
 				return false;
 			}
-			$conn->message .= "AUTH CRAM-MD5: $result\n";
-		} else {
-			$conn->message .= "AUTH CRAM-MD5: failed\n";
+		}
+		
+		if (!is_resource($result) && $orig_method == 'CHECK') {
 			$auth_method = 'PLAIN';
 		}
 	}
 		
-	if (!$result || $auth_method == 'PLAIN') {
-		//do plain text auth
+	if ($auth_method == 'PLAIN') {
+		// do plain text auth
 		$result = iil_C_Login($conn, $user, $password);
-		$conn->message .= "AUTH PLAIN: $result\n";
 	}
-		
-	if (!is_int($result)) {
+
+	if (is_resource($result)) {
+        if ($my_prefs['force_caps']) {
+			iil_C_ClearCapability($conn);
+        }
 		iil_C_Namespace($conn);
 		return $conn;
 	} else {
@@ -716,7 +712,8 @@
 
 function iil_Close(&$conn) {
 	if (iil_PutLine($conn->fp, "I LOGOUT")) {
-		fgets($conn->fp, 1024);
+		if (!feof($conn->fp))
+			fgets($conn->fp, 1024);
 		fclose($conn->fp);
 		$conn->fp = false;
 	}
@@ -737,32 +734,6 @@
 	}
 
 	$result[] = substr($string, $p);
-	return $result;
-}
-
-function iil_CheckForRecent($host, $user, $password, $mailbox) {
-	if (empty($mailbox)) {
-		$mailbox = 'INBOX';
-	}
-    
-	$conn = iil_Connect($host, $user, $password, 'plain');
-	$fp   = $conn->fp;
-	if ($fp) {
-		iil_PutLine($fp, "a002 EXAMINE \"".iil_Escape($mailbox)."\"");
-		do {
-			$line=chop(iil_ReadLine($fp, 300));
-			$a=explode(' ', $line);
-			if (($a[0] == '*') && (strcasecmp($a[2], 'RECENT') == 0)) {
-			    $result = (int) $a[1];
-        		}
-		} while (!iil_StartsWith($a[0], 'a002', true));
-
-		iil_PutLine($fp, "a003 LOGOUT");
-		fclose($fp);
-	} else {
-	    $result = -2;
-	}
-    
 	return $result;
 }
 
@@ -864,7 +835,7 @@
 	
 	$fields = array('ARRIVAL' => 1,'CC' => 1,'DATE' => 1,
         'FROM' => 1, 'SIZE' => 1, 'SUBJECT' => 1, 'TO' => 1);
-	
+
 	if (!$fields[$field]) {
 	    return false;
 	}
@@ -876,9 +847,12 @@
     
 	$is_uid = $is_uid ? 'UID ' : '';
 	
-	if (!empty($add)) {
+	// message IDs
+	if (is_array($add))
+		$add = iil_CompressMessageSet(join(',', $add));
+
+	if (!empty($add))
 	    $add = " $add";
-	}
 
 	$command  = 's ' . $is_uid . 'SORT (' . $field . ') ';
 	$command .= $encoding . ' ALL' . $add;
@@ -906,22 +880,29 @@
 	return preg_split('/\s+/', $data, -1, PREG_SPLIT_NO_EMPTY);
 }
 
-function iil_C_FetchHeaderIndex(&$conn, $mailbox, $message_set, $index_field='', $skip_deleted=true) {
+function iil_C_FetchHeaderIndex(&$conn, $mailbox, $message_set, $index_field='', $skip_deleted=true, $uidfetch=false) {
 
-	list($from_idx, $to_idx) = explode(':', $message_set);
-	if (empty($message_set) ||
-		(isset($to_idx) && $to_idx != '*' && (int)$from_idx > (int)$to_idx)) {
-		return false;
+	if (is_array($message_set)) {
+		if (!($message_set = iil_CompressMessageSet(join(',', $message_set))))
+			return false;
+	} else {
+		list($from_idx, $to_idx) = explode(':', $message_set);
+		if (empty($message_set) ||
+			(isset($to_idx) && $to_idx != '*' && (int)$from_idx > (int)$to_idx)) {
+			return false;
+		}
 	}
-
+	
 	$index_field = empty($index_field) ? 'DATE' : strtoupper($index_field);
 	
 	$fields_a['DATE']         = 1;
 	$fields_a['INTERNALDATE'] = 4;
+	$fields_a['ARRIVAL'] 	  = 4;
 	$fields_a['FROM']         = 1;
 	$fields_a['REPLY-TO']     = 1;
 	$fields_a['SENDER']       = 1;
 	$fields_a['TO']           = 1;
+	$fields_a['CC']           = 1;
 	$fields_a['SUBJECT']      = 1;
 	$fields_a['UID']          = 2;
 	$fields_a['SIZE']         = 2;
@@ -940,19 +921,22 @@
 	
 	// build FETCH command string
 	$key     = 'fhi0';
+	$cmd     = $uidfetch ? 'UID FETCH' : 'FETCH';
 	$deleted = $skip_deleted ? ' FLAGS' : '';
 
-	if ($mode == 1)
-		$request = " FETCH $message_set (BODY.PEEK[HEADER.FIELDS ($index_field)]$deleted)";
+	if ($mode == 1 && $index_field == 'DATE')
+		$request = " $cmd $message_set (INTERNALDATE BODY.PEEK[HEADER.FIELDS (DATE)]$deleted)";
+	else if ($mode == 1)
+		$request = " $cmd $message_set (BODY.PEEK[HEADER.FIELDS ($index_field)]$deleted)";
 	else if ($mode == 2) {
 		if ($index_field == 'SIZE')
-			$request = " FETCH $message_set (RFC822.SIZE$deleted)";
+			$request = " $cmd $message_set (RFC822.SIZE$deleted)";
 		else
-			$request = " FETCH $message_set ($index_field$deleted)";
+			$request = " $cmd $message_set ($index_field$deleted)";
 	} else if ($mode == 3)
-		$request = " FETCH $message_set (FLAGS)";
+		$request = " $cmd $message_set (FLAGS)";
 	else // 4
-		$request = " FETCH $message_set (INTERNALDATE$deleted)";
+		$request = " $cmd $message_set (INTERNALDATE$deleted)";
 
 	$request = $key . $request;
 
@@ -978,15 +962,23 @@
 				}
 			}
 
-			if ($mode == 1) {
-				if (preg_match('/BODY\[HEADER\.FIELDS \("?(DATE|FROM|REPLY-TO|SENDER|TO|SUBJECT)"?\)\] (.*)/', $line, $matches)) {
-					$value = preg_replace(array('/^"*[a-z]+:/i', '/\s+$/sm'), array('', ''), $matches[2]);
+			if ($mode == 1 && $index_field == 'DATE') {
+				if (preg_match('/BODY\[HEADER\.FIELDS \("*DATE"*\)\] (.*)/', $line, $matches)) {
+					$value = preg_replace(array('/^"*[a-z]+:/i'), '', $matches[1]);
 					$value = trim($value);
-					if ($index_field == 'DATE') {
-						$result[$id] = iil_StrToTime($value);
-					} else {
-						$result[$id] = $value;
-					}
+					$result[$id] = iil_StrToTime($value);
+				}
+				// non-existent/empty Date: header, use INTERNALDATE
+				if (empty($result[$id])) {
+					if (preg_match('/INTERNALDATE "([^"]+)"/', $line, $matches))
+						$result[$id] = iil_StrToTime($matches[1]);
+					else
+						$result[$id] = 0;
+				}
+			} else if ($mode == 1) {
+				if (preg_match('/BODY\[HEADER\.FIELDS \("?(FROM|REPLY-TO|SENDER|TO|SUBJECT)"?\)\] (.*)/', $line, $matches)) {
+					$value = preg_replace(array('/^"*[a-z]+:/i', '/\s+$/sm'), array('', ''), $matches[2]);
+					$result[$id] = trim($value);
 				} else {
 					$result[$id] = '';
 				}
@@ -1011,22 +1003,6 @@
 		}
 	} while (!iil_StartsWith($line, $key, true));
 
-/*
-	//check number of elements...
-	if (is_numeric($from_idx) && is_numeric($to_idx)) {
-		//count how many we should have
-		$should_have = $to_idx - $from_idx + 1;
-		
-		//if we have less, try and fill in the "gaps"
-		if (count($result) < $should_have) {
-			for ($i=$from_idx; $i<=$to_idx; $i++) {
-				if (!isset($result[$i])) {
-			    		$result[$i] = '';
-            			}
-        		}
-		}
-	}
-*/
 	return $result;	
 }
 
@@ -1090,317 +1066,15 @@
         return false;
 }
 
-function iil_C_FetchUIDs(&$conn,$mailbox) {
+function iil_C_FetchUIDs(&$conn, $mailbox, $message_set=null) {
 	global $clock;
-	
-	$num = iil_C_CountMessages($conn, $mailbox);
-	if ($num == 0) {
-	    return array();
-	}
-	$message_set = '1' . ($num>1?':' . $num:'');
-	
-	return iil_C_FetchHeaderIndex($conn, $mailbox, $message_set, 'UID');
-}
 
-function iil_SortThreadHeaders($headers, $index_a, $uids) {
-	asort($index_a);
-	$result = array();
-	foreach ($index_a as $mid=>$foobar) {
-		$uid = $uids[$mid];
-		$result[$uid] = $headers[$uid];
-	}
-	return $result;
-}
-
-function iil_C_FetchThreadHeaders(&$conn, $mailbox, $message_set) {
-	global $clock;
-	global $index_a;
+	if (is_array($message_set))
+		$message_set = join(',', $message_set);
+    else if (empty($message_set))
+		$message_set = '1:*';
 	
-	list($from_idx, $to_idx) = explode(':', $message_set);
-	if (empty($message_set) || (isset($to_idx)
-        && (int)$from_idx > (int)$to_idx)) {
-		return false;
-	}
-
-	$result = array();
-	$uids   = iil_C_FetchUIDs($conn, $mailbox);
-	$debug  = false;
-	
-	$message_set = iil_CompressMessageSet($message_set);
-    
-	/* if we're missing any, get them */
-	if ($message_set) {
-		/* FETCH date,from,subject headers */
-		$key        = 'fh';
-		$fp         = $conn->fp;
-		$request    = $key . " FETCH $message_set ";
-    		$request   .= "(BODY.PEEK[HEADER.FIELDS (SUBJECT MESSAGE-ID IN-REPLY-TO)])";
-		$mid_to_id  = array();
-		if (!iil_PutLine($fp, $request)) {
-		    return false;
-    		}
-		do {
-			$line = chop(iil_ReadLine($fp, 1024));
-			if ($debug) {
-			    echo $line . "\n";
-        		}
-			if (preg_match('/\{[0-9]+\}$/', $line)) {
-				$a 	 = explode(' ', $line);
-				$new = array();
-
-				$new_thhd = new iilThreadHeader;
-				$new_thhd->id = $a[1];
-				do {
-					$line = chop(iil_ReadLine($fp, 1024), "\r\n");
-					if (iil_StartsWithI($line, 'Message-ID:')
-                    				|| (iil_StartsWithI($line,'In-Reply-To:'))
-                    				|| (iil_StartsWithI($line,'SUBJECT:'))) {
-
-						$pos        = strpos($line, ':');
-						$field_name = substr($line, 0, $pos);
-						$field_val  = substr($line, $pos+1);
-
-						$new[strtoupper($field_name)] = trim($field_val);
-
-					} else if (preg_match('/^\s+/', $line)) {
-						$new[strtoupper($field_name)] .= trim($line);
-					}
-				} while ($line[0] != ')');
-                
-				$new_thhd->sbj = $new['SUBJECT'];
-				$new_thhd->mid = substr($new['MESSAGE-ID'], 1, -1);
-				$new_thhd->irt = substr($new['IN-REPLY-TO'], 1, -1);
-				
-				$result[$uids[$new_thhd->id]] = $new_thhd;
-			}
-		} while (!iil_StartsWith($line, 'fh'));
-	}
-	
-	/* sort headers */
-	if (is_array($index_a)) {
-		$result = iil_SortThreadHeaders($result, $index_a, $uids);	
-	}
-	
-	//echo 'iil_FetchThreadHeaders:'."\n";
-	//print_r($result);
-	
-	return $result;
-}
-
-function iil_C_BuildThreads2(&$conn, $mailbox, $message_set, &$clock) {
-	global $index_a;
-
-	list($from_idx, $to_idx) = explode(':', $message_set);
-	if (empty($message_set) || (isset($to_idx)
-    		&& (int)$from_idx > (int)$to_idx)) {
-		return false;
-	}
-    
-	$result    = array();
-	$roots     = array();
-	$root_mids = array();
-	$sub_mids  = array();
-	$strays    = array();
-	$messages  = array();
-	$fp        = $conn->fp;
-	$debug     = false;
-	
-	$sbj_filter_pat = '/[a-z]{2,3}(\[[0-9]*\])?:(\s*)/i';
-	
-	/*  Do "SELECT" command */
-	if (!iil_C_Select($conn, $mailbox)) {
-	    return false;
-	}
-    
-	/* FETCH date,from,subject headers */
-	$mid_to_id = array();
-	$messages  = array();
-	$headers   = iil_C_FetchThreadHeaders($conn, $mailbox, $message_set);
-	if ($clock) {
-	    $clock->register('fetched headers');
-	}
-    
-	if ($debug) {
-	    print_r($headers);
-	}
-    
-	/* go through header records */
-	foreach ($headers as $header) {
-		//$id = $header['i'];
-		//$new = array('id'=>$id, 'MESSAGE-ID'=>$header['m'], 
-		//			'IN-REPLY-TO'=>$header['r'], 'SUBJECT'=>$header['s']);
-		$id  = $header->id;
-		$new = array('id' => $id, 'MESSAGE-ID' => $header->mid, 
-        		'IN-REPLY-TO' => $header->irt, 'SUBJECT' => $header->sbj);
-
-		/* add to message-id -> mid lookup table */
-		$mid_to_id[$new['MESSAGE-ID']] = $id;
-		
-		/* if no subject, use message-id */
-		if (empty($new['SUBJECT'])) {
-		    $new['SUBJECT'] = $new['MESSAGE-ID'];
-		}
-        
-		/* if subject contains 'RE:' or has in-reply-to header, it's a reply */
-		$sbj_pre = '';
-		$has_re = false;
-		if (preg_match($sbj_filter_pat, $new['SUBJECT'])) {
-		    $has_re = true;
-		}
-    		if ($has_re || $new['IN-REPLY-TO']) {
-        	    $sbj_pre = 'RE:';
-		}
-        
-		/* strip out 're:', 'fw:' etc */
-		if ($has_re) {
-		    $sbj = preg_replace($sbj_filter_pat, '', $new['SUBJECT']);
-		} else {
-		    $sbj = $new['SUBJECT'];
-		}
-    		$new['SUBJECT'] = $sbj_pre.$sbj;
-		
-		
-		/* if subject not a known thread-root, add to list */
-		if ($debug) {
-		    echo $id . ' ' . $new['SUBJECT'] . "\t" . $new['MESSAGE-ID'] . "\n";
-		}
-    		$root_id = $roots[$sbj];
-		
-		if ($root_id && ($has_re || !$root_in_root[$root_id])) {
-			if ($debug) {
-			    echo "\tfound root: $root_id\n";
-			}
-        		$sub_mids[$new['MESSAGE-ID']] = $root_id;
-			$result[$root_id][]           = $id;
-		} else if (!isset($roots[$sbj]) || (!$has_re && $root_in_root[$root_id])) {
-			/* try to use In-Reply-To header to find root 
-				unless subject contains 'Re:' */
-			if ($has_re&&$new['IN-REPLY-TO']) {
-				if ($debug) {
-				    echo "\tlooking: ".$new['IN-REPLY-TO']."\n";
-				}
-				//reply to known message?
-				$temp = $sub_mids[$new['IN-REPLY-TO']];
-				
-				if ($temp) {
-					//found it, root:=parent's root
-					if ($debug) {
-					    echo "\tfound parent: ".$new['SUBJECT']."\n";
-					}
-                			$result[$temp][]              = $id;
-					$sub_mids[$new['MESSAGE-ID']] = $temp;
-					$sbj                          = '';
-				} else {
-					//if we can't find referenced parent, it's a "stray"
-					$strays[$id] = $new['IN-REPLY-TO'];
-				}
-			}
-			
-			//add subject as root
-			if ($sbj) {
-				if ($debug) {
-				    echo "\t added to root\n";
-				}
-            			$roots[$sbj]                  = $id;
-				$root_in_root[$id]            = !$has_re;
-				$sub_mids[$new['MESSAGE-ID']] = $id;
-				$result[$id]                  = array($id);
-			}
-			if ($debug) {
-			    echo $new['MESSAGE-ID'] . "\t" . $sbj . "\n";
-        		}
-		}
-	}
-	
-	//now that we've gone through all the messages,
-	//go back and try and link up the stray threads
-	if (count($strays) > 0) {
-		foreach ($strays as $id=>$irt) {
-			$root_id = $sub_mids[$irt];
-			if (!$root_id || $root_id==$id) {
-			    continue;
-			}
-        		$result[$root_id] = array_merge($result[$root_id],$result[$id]);
-			unset($result[$id]);
-		}
-	}
-	
-	if ($clock) {
-	    $clock->register('data prepped');
-	}
-    
-	if ($debug) {
-	    print_r($roots);
-	}
-
-	return $result;
-}
-
-function iil_SortThreads(&$tree, $index, $sort_order = 'ASC') {
-	if (!is_array($tree) || !is_array($index)) {
-	    return false;
-	}
-    
-	//create an id to position lookup table
-	$i = 0;
-	foreach ($index as $id=>$val) {
-		$i++;
-		$index[$id] = $i;
-	}
-	$max = $i+1;
-	
-	//for each tree, set array key to position
-	$itree = array();
-	foreach ($tree as $id=>$node) {
-		if (count($tree[$id])<=1) {
-			//for "threads" with only one message, key is position of that message
-			$n         = $index[$id];
-			$itree[$n] = array($n=>$id);
-		} else {
-			//for "threads" with multiple messages, 
-			$min   = $max;
-			$new_a = array();
-			foreach ($tree[$id] as $mid) {
-				$new_a[$index[$mid]] = $mid;		//create new sub-array mapping position to id
-				$pos                 = $index[$mid];
-				if ($pos&&$pos<$min) {
-				    $min = $index[$mid];	//find smallest position
-				}
-			}
-			$n = $min;	//smallest position of child is thread position
-			
-			//assign smallest position to root level key
-			//set children array to one created above
-			ksort($new_a);
-			$itree[$n] = $new_a;
-		}
-	}
-	
-	//sort by key, this basically sorts all threads
-	ksort($itree);
-	$i   = 0;
-	$out = array();
-	foreach ($itree as $k=>$node) {
-		$out[$i] = $itree[$k];
-		$i++;
-	}
-	
-	return $out;
-}
-
-function iil_IndexThreads(&$tree) {
-	/* creates array mapping mid to thread id */
-	
-	if (!is_array($tree)) {
-	    return false;
-	}
-    
-	$t_index = array();
-	foreach ($tree as $pos=>$kids) {
-		foreach ($kids as $kid) $t_index[$kid] = $pos;
-	}
-	
-	return $t_index;
+	return iil_C_FetchHeaderIndex($conn, $mailbox, $message_set, 'UID', false);
 }
 
 function iil_C_FetchHeaders(&$conn, $mailbox, $message_set, $uidfetch=false, $bodystr=false, $add='')
@@ -1416,6 +1090,9 @@
 		return false;
 	}
 
+	if (is_array($message_set))
+		$message_set = join(',', $message_set);
+
 	$message_set = iil_CompressMessageSet($message_set);
 
 	if ($add)
@@ -1430,7 +1107,8 @@
 	$request .= "BODY.PEEK[HEADER.FIELDS ";
 	$request .= "(DATE FROM TO SUBJECT REPLY-TO IN-REPLY-TO CC BCC ";
 	$request .= "CONTENT-TRANSFER-ENCODING CONTENT-TYPE MESSAGE-ID ";
-	$request .= "REFERENCES DISPOSITION-NOTIFICATION-TO X-PRIORITY".$add.")])";
+	$request .= "REFERENCES DISPOSITION-NOTIFICATION-TO X-PRIORITY ";
+	$request .= "X-DRAFT-INFO".$add.")])";
 
 	if (!iil_PutLine($fp, $request)) {
 		return false;
@@ -1592,13 +1270,10 @@
 						$result[$id]->encoding = $string;
 						break;
 					case 'content-type':
-						$ctype_parts = explode(";", $string);
+						$ctype_parts = preg_split('/[; ]/', $string);
 						$result[$id]->ctype = array_shift($ctype_parts);
-						foreach ($ctype_parts as $ctype_add) {
-							if (preg_match('/charset="?([a-z0-9\-\.\_]+)"?/i',
-								$ctype_add, $regs)) {
-								$result[$id]->charset = $regs[1];
-							}
+						if (preg_match('/charset\s*=\s*"?([a-z0-9\-\.\_]+)"?/i', $string, $regs)) {
+							$result[$id]->charset = $regs[1];
 						}
 						break;
 					case 'in-reply-to':
@@ -1745,10 +1420,10 @@
 
 		iil_PutLine($conn->fp, "exp1 $command");
 		do {
-			$line=chop(iil_ReadLine($conn->fp, 100));
+			$line = iil_ReadLine($conn->fp, 100);
 			if ($line[0] == '*') {
-            			$c++;
-        		}
+            	$c++;
+        	}
 		} while (!iil_StartsWith($line, 'exp1', true));
 		
 		if (iil_ParseResult($line) == 0) {
@@ -1767,29 +1442,27 @@
 	    return -1;
 	}
     
-	$fp    = $conn->fp;
 	$flags = $GLOBALS['IMAP_FLAGS'];
-        
-	$flag = strtoupper($flag);
-	$flag = $flags[$flag];
+	$flag = $flags[strtoupper($flag)];
     
-	if (iil_C_Select($conn, $mailbox)) {
-		$c = 0;
-		iil_PutLine($fp, "flg UID STORE $messages " . $mod . "FLAGS (" . $flag . ")");
-		do {
-			$line=chop(iil_ReadLine($fp, 100));
-			if ($line[0] == '*') {
-			    $c++;
-        		}
-		} while (!iil_StartsWith($line, 'flg', true));
-
-		if (iil_ParseResult($line) == 0) {
-			return $c;
-		}
-		$conn->error = $line;
-		return -1;
+	if (!iil_C_Select($conn, $mailbox)) {
+	    return -1;
 	}
-	$conn->error = 'Select failed';
+    
+    $c = 0;
+	iil_PutLine($conn->fp, "flg UID STORE $messages " . $mod . "FLAGS (" . $flag . ")");
+	do {
+		$line = iil_ReadLine($conn->fp, 1000);
+		if ($line[0] == '*') {
+		    $c++;
+        }
+	} while (!iil_StartsWith($line, 'flg', true));
+
+	if (iil_ParseResult($line) == 0) {
+		return $c;
+	}
+
+	$conn->error = $line;
 	return -1;
 }
 
@@ -1806,28 +1479,25 @@
 }
 
 function iil_C_Copy(&$conn, $messages, $from, $to) {
-	$fp = $conn->fp;
 
 	if (empty($from) || empty($to)) {
 	    return -1;
 	}
     
-	if (iil_C_Select($conn, $from)) {
-		$c=0;
-		
-		iil_PutLine($fp, "cpy1 UID COPY $messages \"".iil_Escape($to)."\"");
-		$line = iil_ReadReply($fp);
-		return iil_ParseResult($line);
-	} else {
-		return -1;
+	if (!iil_C_Select($conn, $from)) {
+        return -1;
 	}
+	
+    iil_PutLine($conn->fp, "cpy1 UID COPY $messages \"".iil_Escape($to)."\"");
+	$line = iil_ReadReply($conn->fp);
+	return iil_ParseResult($line);
 }
 
 function iil_C_CountUnseen(&$conn, $folder) {
-        $index = iil_C_Search($conn, $folder, 'ALL UNSEEN');
-        if (is_array($index))
-                return count($index);
-        return false;
+    $index = iil_C_Search($conn, $folder, 'ALL UNSEEN');
+    if (is_array($index))
+        return count($index);
+    return false;
 }
 
 function iil_C_UID2ID(&$conn, $folder, $uid) {
@@ -1860,33 +1530,119 @@
 	return $result;
 }
 
+// Don't be tempted to change $str to pass by reference to speed this up - it will slow it down by about
+// 7 times instead :-) See comments on http://uk2.php.net/references and this article:
+// http://derickrethans.nl/files/phparch-php-variables-article.pdf
+function iil_ParseThread($str, $begin, $end, $root, $parent, $depth, &$depthmap, &$haschildren) {
+	$node = array();
+	if ($str[$begin] != '(') {
+		$stop = $begin + strspn($str, "1234567890", $begin, $end - $begin);
+		$msg = substr($str, $begin, $stop - $begin);
+		if ($msg == 0)
+		    return $node;
+		if (is_null($root))
+			$root = $msg;
+		$depthmap[$msg] = $depth;
+		$haschildren[$msg] = false;
+		if (!is_null($parent))
+			$haschildren[$parent] = true;
+		if ($stop + 1 < $end)
+			$node[$msg] = iil_ParseThread($str, $stop + 1, $end, $root, $msg, $depth + 1, $depthmap, $haschildren);
+		else
+			$node[$msg] = array();
+	} else {
+		$off = $begin;
+		while ($off < $end) {
+			$start = $off;
+			$off++;
+			$n = 1;
+			while ($n > 0) {
+				$p = strpos($str, ')', $off);
+				if ($p === false) {
+					error_log('Mismatched brackets parsing IMAP THREAD response:');
+					error_log(substr($str, ($begin < 10) ? 0 : ($begin - 10), $end - $begin + 20));
+					error_log(str_repeat(' ', $off - (($begin < 10) ? 0 : ($begin - 10))));
+					return $node;
+				}
+				$p1 = strpos($str, '(', $off);
+				if ($p1 !== false && $p1 < $p) {
+					$off = $p1 + 1;
+					$n++;
+				} else {
+					$off = $p + 1;
+					$n--;
+				}
+			}
+			$node += iil_ParseThread($str, $start + 1, $off - 1, $root, $parent, $depth, $depthmap, $haschildren);
+		}
+	}
+	
+	return $node;
+}
+
+function iil_C_Thread(&$conn, $folder, $algorithm='REFERENCES', $criteria='',
+    $encoding='US-ASCII') {
+
+	if (!iil_C_Select($conn, $folder)) {
+		$conn->error = "Couldn't select $folder";
+		return false;
+	}
+
+	$encoding = $encoding ? trim($encoding) : 'US-ASCII';
+	$algorithm = $algorithm ? trim($algorithm) : 'REFERENCES';
+	$criteria = $criteria ? 'ALL '.trim($criteria) : 'ALL';
+		
+	if (!iil_PutLineC($conn->fp, "thrd1 THREAD $algorithm $encoding $criteria")) {
+		return false;
+	}
+	do {
+		$line = trim(iil_ReadLine($conn->fp, 10000));
+		if (preg_match('/^\* THREAD/', $line)) {
+			$str = trim(substr($line, 8));
+			$depthmap = array();
+			$haschildren = array();
+			$tree = iil_ParseThread($str, 0, strlen($str), null, null, 0, $depthmap, $haschildren);
+		}
+	} while (!iil_StartsWith($line, 'thrd1', true));
+
+	$result_code = iil_ParseResult($line);
+	if ($result_code == 0) {
+	    return array($tree, $depthmap, $haschildren);
+	}
+
+	$conn->error = 'iil_C_Thread: ' . $line . "\n";
+	return false;	
+}
+
 function iil_C_Search(&$conn, $folder, $criteria) {
 
-	if (iil_C_Select($conn, $folder)) {
-		$data = '';
-		
-		$query = 'srch1 SEARCH ' . chop($criteria);
-		if (!iil_PutLineC($conn->fp, $query)) {
-			return false;
-		}
-		do {
-			$line = trim(iil_ReadLine($conn->fp));
-			if (iil_StartsWith($line, '* SEARCH')) {
-				$data .= substr($line, 8);
-    			} else if (preg_match('/^[0-9 ]+$/', $line)) {
-				$data .= $line;
-			}
-		} while (!iil_StartsWith($line, 'srch1', true));
-
-		$result_code = iil_ParseResult($line);
-		if ($result_code == 0) {
-		    return preg_split('/\s+/', $data, -1, PREG_SPLIT_NO_EMPTY);
-		}
-		$conn->error = 'iil_C_Search: ' . $line . "\n";
-		return false;	
+	if (!iil_C_Select($conn, $folder)) {
+		$conn->error = "Couldn't select $folder";
+		return false;
 	}
-	$conn->error = "iil_C_Search: Couldn't select \"$folder\"\n";
-	return false;
+
+	$data = '';
+	$query = 'srch1 SEARCH ' . chop($criteria);
+
+	if (!iil_PutLineC($conn->fp, $query)) {
+		return false;
+	}
+	do {
+		$line = trim(iil_ReadLine($conn->fp));
+		if (iil_StartsWith($line, '* SEARCH')) {
+			$data .= substr($line, 8);
+		} else if (preg_match('/^[0-9 ]+$/', $line)) {
+			$data .= $line;
+		}
+	} while (!iil_StartsWith($line, 'srch1', true));
+
+	$result_code = iil_ParseResult($line);
+	if ($result_code == 0) {
+		return preg_split('/\s+/', $data, -1, PREG_SPLIT_NO_EMPTY);
+	}
+
+	$conn->error = 'iil_C_Search: ' . $line . "\n";
+	return false;	
 }
 
 function iil_C_Move(&$conn, $messages, $from, $to) {
@@ -1982,9 +1738,6 @@
 }
 
 function iil_C_ListMailboxes(&$conn, $ref, $mailbox) {
-	global $IGNORE_FOLDERS;
-	
-	$ignore = $IGNORE_FOLDERS[strtolower($conn->host)];
 		
 	$fp = $conn->fp;
 	
@@ -2013,12 +1766,7 @@
         		// split one line
 			$a = iil_ExplodeQuotedString(' ', $line);
         		// last string is folder name
-			$folder = preg_replace(array('/^"/', '/"$/'), '', iil_UnEscape($a[count($a)-1]));
-            
-        		if (empty($ignore) || (!empty($ignore)
-            			&& !preg_match('/'.preg_quote(ignore, '/').'/i', $folder))) {
-            			$folders[$i] = $folder;
-        		}
+			$folders[$i] = preg_replace(array('/^"/', '/"$/'), '', iil_UnEscape($a[count($a)-1]));
             
 		        // second from last is delimiter
         		$delim = trim($a[count($a)-2], '"');
@@ -2046,9 +1794,6 @@
 }
 
 function iil_C_ListSubscribed(&$conn, $ref, $mailbox) {
-	global $IGNORE_FOLDERS;
-	
-	$ignore = $IGNORE_FOLDERS[strtolower($conn->host)];
 	
 	$fp = $conn->fp;
 	if (empty($mailbox)) {
@@ -2080,9 +1825,9 @@
 			$a = iil_ExplodeQuotedString(' ', $line);
         		// last string is folder name
     			$folder = preg_replace(array('/^"/', '/"$/'), '', iil_UnEscape($a[count($a)-1]));
-        
-			if ((!in_array($folder, $folders)) && (empty($ignore)
-            			|| (!empty($ignore) && !preg_match('/'.preg_quote(ignore, '/').'/i', $folder)))) {
+
+			// @TODO: do we need this check???
+			if (!in_array($folder, $folders)) {
 			    $folders[$i] = $folder;
         		}
             
@@ -2131,7 +1876,7 @@
 	return (iil_ParseResult($line) == 0);
 }
 
-function iil_C_FetchMIMEHeaders(&$conn, $mailbox, $id, $parts) {
+function iil_C_FetchMIMEHeaders(&$conn, $mailbox, $id, $parts, $mime=true) {
 	
 	$fp     = $conn->fp;
 
@@ -2144,10 +1889,11 @@
 	$key = 'fmh0';
 	$peeks = '';
 	$idx = 0;
+        $type = $mime ? 'MIME' : 'HEADER';
 
 	// format request
 	foreach($parts as $part)
-		$peeks[] = "BODY.PEEK[$part.MIME]";
+		$peeks[] = "BODY.PEEK[$part.$type]";
 	
 	$request = "$key FETCH $id (" . implode(' ', $peeks) . ')';
 
@@ -2160,9 +1906,9 @@
         	$line = iil_ReadLine($fp, 1000);
         	$line = iil_MultLine($fp, $line);
 
-		if (preg_match('/BODY\[([0-9\.]+)\.MIME\]/', $line, $matches)) {
+		if (preg_match('/BODY\[([0-9\.]+)\.'.$type.'\]/', $line, $matches)) {
 			$idx = $matches[1];
-			$result[$idx] = preg_replace('/^(\* '.$id.' FETCH \()?\s*BODY\['.$idx.'\.MIME\]\s+/', '', $line);
+			$result[$idx] = preg_replace('/^(\* '.$id.' FETCH \()?\s*BODY\['.$idx.'\.'.$type.'\]\s+/', '', $line);
 			$result[$idx] = trim($result[$idx], '"');
 	    		$result[$idx] = rtrim($result[$idx], "\t\r\n\0\x0B");
     		}
@@ -2401,7 +2147,7 @@
 	return false;
 }
 
-function iil_C_AppendFromFile(&$conn, $folder, $path) {
+function iil_C_AppendFromFile(&$conn, $folder, $path, $headers=null, $separator="\n\n") {
 	if (!$folder) {
 	    return false;
 	}
@@ -2421,7 +2167,12 @@
 	if (!$len) {
 		return false;
 	}
-    
+
+        if ($headers) {
+                $headers = preg_replace('/[\r\n]+$/', '', $headers);
+                $len += strlen($headers) + strlen($separator);
+        }
+
 	//send APPEND command
 	$request    = 'a APPEND "' . iil_Escape($folder) . '" (\\Seen) {' . $len . '}';
 	if (iil_PutLine($fp, $request)) {
@@ -2433,16 +2184,21 @@
 			return false;
 		}
 
-		//send file
+                // send headers with body separator
+                if ($headers) {
+			iil_PutLine($fp, $headers . $separator, false);
+                }
+
+		// send file
 		while (!feof($in_fp)) {
-			$buffer      = fgets($in_fp, 4096);
+			$buffer = fgets($in_fp, 4096);
 			iil_PutLine($fp, $buffer, false);
 		}
 		fclose($in_fp);
 
 		iil_PutLine($fp, ''); // \r\n
 
-		//read response
+		// read response
 		do {
 			$line = iil_ReadLine($fp);
 		} while (!iil_StartsWith($line, 'a ', true));

--
Gitblit v1.9.1