New upstream version.

[roundcube.git] / bin / msgexport.sh

diff --git a/bin/msgexport.sh b/bin/msgexport.sh
index 890d48a062b94809e8a0cb1b2e2dae9223d93510..0aa33b5a85482f7710bc3732f76e276fa73f804f 100755 (executable)
--- 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)
@@ -59,6 +27,45 @@ function progress_update($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);
+}
+
 
 // get arguments
 $args = get_opt(array('h' => 'host', 'u' => 'user', 'p' => 'pass', 'm' => 'mbox', 'f' => 'file')) + array('host' => 'localhost', 'mbox' => 'INBOX');
@@ -83,8 +90,7 @@ if (empty($args['user']))
 }
 
 // prompt for password
-vputs("Password: ");
-$args['pass'] = trim(fgets(STDIN));
+$args['pass'] = prompt_silent("Password: ");
 
 
 // parse $host URL
@@ -109,39 +115,25 @@ if ($IMAP->connect($host, $args['user'], $args['pass'], $imap_port, $imap_ssl))
 {
        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
+?>