- Performance improvements

alecpl

2011-11-24 e4a4ca19480690193e36e14ef6b9d592e73e9e9c

 bin/msgexport.sh

@@ -1,50 +1,18 @@
#!/usr/bin/php -qC 
#!/usr/bin/env php
<?php

define('INSTALL_PATH', preg_replace('/bin\/$/', '', getcwd()) . '/');
define('INSTALL_PATH', realpath(dirname(__FILE__) . '/..') . '/' );
ini_set('memory_limit', -1);

require_once INSTALL_PATH.'program/include/iniset.php';

/**
 * Parse commandline arguments into a hash array
 */
function get_opt($aliases=array())
{
   $args = array();
   for ($i=1; $i<count($_SERVER['argv']); $i++)
   {
      $arg = $_SERVER['argv'][$i];
      if (substr($arg, 0, 2) == '--')
      {
         $sp = strpos($arg, '=');
         $key = substr($arg, 2, $sp - 2);
         $value = substr($arg, $sp+1);
      }
      else if ($arg{0} == '-')
      {
         $key = substr($arg, 1);
         $value = $_SERVER['argv'][++$i];
      }
      else
         continue;

      $args[$key] = preg_replace(array('/^["\']/', '/["\']$/'), '', $value);
		
      if ($alias = $aliases[$key])
         $args[$alias] = $args[$key];
   }

   return $args;
}
require_once INSTALL_PATH.'program/include/clisetup.php';

function print_usage()
{
   print "Usage:  msgexport -h imap-host -u user-name -m mailbox name\n";
   print "--host   IMAP host\n";
   print "--user   IMAP user name\n";
   print "--mbox   Mailbox/folder name\n";
   print "--file   Mailbox/folder name\n";
   print "--mbox   Folder name, set to '*' for all\n";
   print "--file   Output file\n";
}

function vputs($str)
@@ -57,6 +25,45 @@
{
   $percent = round(100 * $pos / $max);
   vputs(sprintf("%3d%% [%-51s] %d/%d\033[K\r", $percent, @str_repeat('=', $percent / 2) . '>', $pos, $max));
}

function export_mailbox($mbox, $filename)
{
   global $IMAP;
	
   $IMAP->set_mailbox($mbox);
	
   vputs("Getting message list of {$mbox}...");
   vputs($IMAP->messagecount()." messages\n");
	
   if ($filename)
   {
      if (!($out = fopen($filename, 'w')))
      {
         vputs("Cannot write to output file\n");
         return;
      }
      vputs("Writing to $filename\n");
   }
   else
      $out = STDOUT;
	
   for ($count = $IMAP->messagecount(), $i=1; $i <= $count; $i++)
   {
      $headers = $IMAP->get_headers($i, null, false);
      $from = current($IMAP->decode_address_list($headers->from, 1, false));
		
      fwrite($out, sprintf("From %s %s UID %d\n", $from['mailto'], $headers->date, $headers->uid));
      fwrite($out, $IMAP->conn->fetchPartHeader($mbox, $i));
      fwrite($out, $IMAP->conn->handlePartBody($mbox, $i));
      fwrite($out, "\n\n\n");
		
      progress_update($i, $count);
   }
   vputs("\ncomplete.\n");
	
   if ($filename)
      fclose($out);
}


@@ -83,8 +90,7 @@
}

// prompt for password
vputs("Password: ");
$args['pass'] = trim(fgets(STDIN));
$args['pass'] = prompt_silent("Password: ");


// parse $host URL
@@ -109,39 +115,25 @@
{
   vputs("IMAP login successful.\n");
   
   $IMAP->set_mailbox($args['mbox']);
	
   vputs("Getting message list of {$args['mbox']}...");
   vputs($IMAP->messagecount()." messages\n");
	
   if ($args['file'])
   $filename = null;
   $mailboxes = $args['mbox'] == '*' ? $IMAP->list_mailboxes(null) : array($args['mbox']);

   foreach ($mailboxes as $mbox)
   {
      if (!($out = fopen($args['file'], 'w')))
      {
         vputs("Cannot write to output file\n");
         exit;
      }
      if ($args['file'])
         $filename = preg_replace('/\.[a-z0-9]{3,4}$/i', '', $args['file']) . asciiwords($mbox) . '.mbox';
      else if ($args['mbox'] == '*')
         $filename = asciiwords($mbox) . '.mbox';
			
      if ($args['mbox'] == '*' && in_array(strtolower($mbox), array('junk','spam','trash')))
         continue;

      export_mailbox($mbox, $filename);
   }
   else
      $out = STDOUT;
	
   for ($count = $IMAP->messagecount(), $i=1; $i <= $count; $i++)
   {
      $headers = $IMAP->get_headers($i, null, false);
      $from = current($IMAP->decode_address_list($headers->from, 1, false));
		
      fwrite($out, sprintf("From %s %s UID %d\n", $from['mailto'], $headers->date, $headers->uid));
      fwrite($out, iil_C_FetchPartHeader($IMAP->conn, $IMAP->mailbox, $i, null));
      fwrite($out, iil_C_HandlePartBody($IMAP->conn, $IMAP->mailbox, $i, null, 1));
      fwrite($out, "\n\n\n");
		
      progress_update($i, $count);
   }
   vputs("\ncomplete.\n");
}
else
{
   vputs("IMAP login failed.\n");
}

?>
?>