-#!/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)
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);
+}
+
// get arguments
$args = get_opt(array('h' => 'host', 'u' => 'user', 'p' => 'pass', 'm' => 'mbox', 'f' => 'file')) + array('host' => 'localhost', 'mbox' => 'INBOX');
}
// prompt for password
-vputs("Password: ");
-$args['pass'] = trim(fgets(STDIN));
+$args['pass'] = prompt_silent("Password: ");
// parse $host URL
{
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'])
- {
- if (!($out = fopen($args['file'], 'w')))
- {
- vputs("Cannot write to output file\n");
- exit;
- }
- }
- else
- $out = STDOUT;
-
- for ($count = $IMAP->messagecount(), $i=1; $i <= $count; $i++)
+ $filename = null;
+ $mailboxes = $args['mbox'] == '*' ? $IMAP->list_mailboxes(null) : array($args['mbox']);
+
+ foreach ($mailboxes as $mbox)
{
- $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);
+ 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);
}
- vputs("\ncomplete.\n");
}
else
{
vputs("IMAP login failed.\n");
}
-?>
\ No newline at end of file
+?>