From 24201dc1f48770d20ffaa44fabe1ef571f979da9 Mon Sep 17 00:00:00 2001 From: alecpl <alec@alec.pl> Date: Sat, 18 Jun 2011 07:28:43 -0400 Subject: [PATCH] - Fix set_row_attribs() for backward compat. --- bin/msgexport.sh | 126 +++++++++++++++++++---------------------- 1 files changed, 59 insertions(+), 67 deletions(-) diff --git a/bin/msgexport.sh b/bin/msgexport.sh index 890d48a..0aa33b5 100755 --- a/bin/msgexport.sh +++ b/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"); } -?> \ No newline at end of file +?> -- Gitblit v1.9.1