<?
App::import('Sanitize');
class VaultShell extends Shell {
	/* vault.php - To be run in CLI via Cron. Fetches the mail from a certain mailbox, searches for a CMC Technologies Enquiry Numbeer
	 * then loads the data into the 'emails' table and the attachments in the 'email_attachments' table.
	 * Emails are discared or moved to a subfolder once processed
	 */
	

	var $uses = array('Enquiry', 'Email', 'EmailAttachment');
	
	function main() {
		/****************************************************** 
		 * Config Variables 
		 * *****************************************************/
		$testing = 0; //Whether to actually move the emails. 1=test, 0=production
		/* Setup Connection to the IMAP server */
		$username = 'vault';
		$password = 'xjdYOsmJWc37'; /* The password for the account to be checked */
		$email_dir = '/var/www/cakephp/app/emails/working';
		$temp_filename = 'temp.eml';
		
		$mbox = imap_open("{saturn:143/novalidate-cert}INBOX", $username, $password) or die("can't connect: " . imap_last_error());
		$MC = imap_check($mbox);
		$number_of_messages = $MC->Nmsgs;
		
		/* Loop through the messages and sort them into ones to be processed or discarded */
		for ($i=1; $i < $number_of_messages; $i++) {
			$this_header = imap_headerinfo($mbox, $i);
			$message = $this->getMessage($mbox, $i, $this_header);
			$enquiry = $this->checkIfValidEnquiry($message['subject'], $testing);
			if($enquiry) {			
			 //Process it and store the message and its attachments.
				$this->Email->create();
				$this->data['Email']['enquiry_id'] = $enquiry['Enquiry']['id'];
				$this->data['Email']['to'] = $message['to'];
				$this->data['Email']['cc'] = $message['cc'];
				$this->data['Email']['from'] = $message['from'];
				$this->data['Email']['date'] = $message['date'];
				$this->data['Email']['subject'] = $message['subject'];
				$this->data['Email']['body'] = "";
				$this->data['Email']['plainbody'] = "";
				$structure = imap_fetchstructure($mbox, $i);
				if (empty($structure->parts)) { /* A single part message. No attachments and is plain text */
					$this->data['Email']['body'] = imap_body($mbox, $i);
				}
				else {
				$attachments = $this->fetchBodyAttachments($mbox, $i, $temp_filename, $email_dir);
					foreach ($attachments as $attachment) {
						if($attachment['type'] == 'text/html') { //Assuming All HTML attachments are the body of the email
							
							$filecontents = file_get_contents($email_dir.'/'.$attachment['name']);
							$size = filesize($email_dir.'/'.$attachment['name']);
							$this->data['Email']['body'] .= $filecontents;
							
							
						}
						if($attachment['type'] == 'text/plain') { //Found plain text
							$filecontents = file_get_contents($email_dir.'/'.$attachment['name']);
							$size = filesize($email_dir.'/'.$attachment['name']);
							$this->data['Email']['plainbody'] .= $filecontents;
							
					
						}		
					}
					
				}
			
				//Sanitize::clean($this->data);				
				
				if($this->Email->save($this->data)) {
					$email_id = $this->Email->id;
					if(isset($attachments)) {
						foreach ($attachments as $attachment) {
								if( ($attachment['type'] != 'text/html') && ($attachment['type'] != 'multipart/mixed') && 
									($attachment['type'] != 'multipart/alternative') ) {
									$this->EmailAttachment->create();
									$this->data['EmailAttachment']['email_id'] = $email_id;
									$this->data['EmailAttachment']['name'] = $attachment['name'];
									$this->data['EmailAttachment']['type'] = $attachment['type'];
									$this->data['EmailAttachment']['size'] = filesize($email_dir.'/'.$attachment['name']);
									$this->data['EmailAttachment']['data'] = file_get_contents($email_dir.'/'.$attachment['name']);
									if ($this->EmailAttachment->save($this->data)) {
										echo "Saved file successfully to database\n";
									}
									else {
										echo "Something went wrong saving the file to the DB\n";
									}
								}
							}
						}
						echo "Email stored in the DB under enquiry ".$enquiry['Enquiry']['title']." Will be moved to the stored folder\n";
					
						if($testing == 0) { //Testing Mode. Don't actually move these emails unless we're in production.
							imap_mail_move($mbox, $i, 'INBOX/Stored'); //Move it to the stored folder.
						}
						
						
						if(isset($attachments)) {
							$this->clearEmailAttachmentDirs($email_dir, $temp_filename, $attachments);
						}
					
					
				}
				else {
					echo 'Unable to save the Email\n';
				}
				

			} else {
				/* Can't find a valid-looking CMC Enquiry Number. Move the message to the discarded folder 
				 * I may change this to simply delete the emails. This will do for now, but it's doubling up on the storage for useless files.
				 * */
				if($testing == 0) {
					
					if(imap_mail_move($mbox, $i, 'INBOX/Discarded')) {
						echo "Message Number $i contains no valid Enquiry number. Has been discarded\n";
					}
					else {
						echo "Message Number $i contains no valid Enquiry number. But could not be discarded\n";
					}
				} 
			}
	
	}
		/* Finished working with the IMAP server. Make the changes and close the connection */
		imap_expunge($mbox);
		imap_close($mbox); 
	
}


/*
* Fetches the body and attachments from a MIME encoded email. Uses ripmime to do the decoding
 * 
 */ 	
	
function fetchBodyAttachments($mailbox, $msg_number, $filename, $email_dir) {

  $email_file = $email_dir.'/'.$filename;
  imap_savebody($mailbox, $email_file, $msg_number);
  $command = "ripmime -i $email_file -d $email_dir -v --verbose-contenttype --paranoid --prefix $msg_number";
  $output = array();
  exec($command, $output);
  
  /* Check the $output array and find the filenames of the attachments */
  $attachments = array();
  
  for($i=0, $j=0; $i< count($output); $i++, $j++) { 	
  	$words = explode(' ', $output[$i]);
  	$type = explode('=', $words[1]);
  	$name = explode('=', $words[2]);	
  	$attachments[$j]['type'] = $type[1];
  	$attachments[$j]['name'] = $name[1];
    echo "in message number $msg_number: found attachment ".$attachments[$j]['name'].' '.$attachments[$j]['type']."\n"; 
  }
return $attachments;
}


/* clearEmailyAttachmentDirs($email_dir, $filename, $attachment_dir, $attachments)
 * 
 * Deletes $email_dir/$filename.eml and the attachments specified by $attachments
 * */ 
function clearEmailAttachmentDirs($email_dir, $filename, $attachments) {
		unlink("$email_dir/$filename");
		foreach ($attachments as $attachment) {
			unlink($email_dir.'/'.$attachment['name']);
		}
}


function getMessage($mbox, $msgnumber, $headers) {

	$subject = $headers->subject;
	$date = $headers->date;
	$recipients = $this->getRecipients($headers);
	$message['subject'] = $subject;
	$message['date'] = $date;
	$message['to'] = $recipients['to'];
	$message['from'] = $recipients['from'];
	$message['cc'] = $recipients['cc'];
	return $message;

}
/* get Recipients from the headers of an email  */
function getRecipients($headers) {

	$recipients = array();	
	$recipients['to'] = "";
	$recipients['from'] = "";
	$recipients['cc'] = "";	
	
	if(isset($headers->to)) {
		$to = $headers->to;
		if(count($to) > 0) {
		foreach ($to as $id => $object) {
			$recipients['to'] .= "$object->mailbox@$object->host ";	
			}
		}
	}
	if(isset($headers->cc)) {
		$cc = $headers->cc;
		if(count($cc) > 0) {
		foreach ($cc as $id => $object) {
				$recipients['cc'] .= "$object->mailbox@$object->host ";
			}
		
		}
	}
	
	if(isset($headers->from)) {
		$from = $headers->from;
		if(count($from) > 0) {
		foreach ($from as $id => $object) {
				$recipients['from'] .= "$object->mailbox@$object->host";
			}
		}
	}

return $recipients;
}
	 

function getBody($mbox, $msgnumber) {
	$structure = imap_fetchstructure($mbox, $msgnumber);
	if (!empty($structure->parts)) {
    	$body = $this->getParts($structure->parts, $mbox, $msgnumber);
	}
     else {
    $body = imap_body($mbox, $msgnumber); /* Or get the plaintext */
	}
	return $body;
}

function checkIfValidEnquiry($subject, $testing) {
	
	$output = array();
	
	
	preg_match("/CMC\d+([NVQWSO]|ACT|NT)E\d+-\d+/", $subject, $output);
	
	if(isset($output[0])) { //Found a valid-looking Enquiry Number
		$fetched_enquirynumber = $output[0];
		$enquiry = $this->Enquiry->findByTitle($fetched_enquirynumber);
		if($enquiry) {
			return $enquiry;
		}
	}
	else {
			
	return FALSE;
	}
	
}

}	
	
	?>