bin/msgimport.sh

   1 #!/usr/bin/php
   2 <?php
   3
   4 define('INSTALL_PATH', realpath(dirname(__FILE__) . '/..') . '/' );
   5 ini_set('memory_limit', -1);
   6
   7 require_once INSTALL_PATH.'program/include/iniset.php';
   8
   9 /**
  10  * Parse commandline arguments into a hash array
  11  */
  12 function get_opt($aliases=array())
  13 {
  14         $args = array();
  15         for ($i=1; $i<count($_SERVER['argv']); $i++)
  16         {
  17                 $arg = $_SERVER['argv'][$i];
  18                 if (substr($arg, 0, 2) == '--')
  19                 {
  20                         $sp = strpos($arg, '=');
  21                         $key = substr($arg, 2, $sp - 2);
  22                         $value = substr($arg, $sp+1);
  23                 }
  24                 else if ($arg{0} == '-')
  25                 {
  26                         $key = substr($arg, 1);
  27                         $value = $_SERVER['argv'][++$i];
  28                 }
  29                 else
  30                         continue;
  31
  32                 $args[$key] = preg_replace(array('/^["\']/', '/["\']$/'), '', $value);
  33
  34                 if ($alias = $aliases[$key])
  35                         $args[$alias] = $args[$key];
  36         }
  37
  38         return $args;
  39 }
  40
  41 function print_usage()
  42 {
  43         print "Usage:  msgimport -h imap-host -u user-name -m mailbox -f message-file\n";
  44         print "--host   IMAP host\n";
  45         print "--user   IMAP user name\n";
  46         print "--mbox   Target mailbox\n";
  47         print "--file   Message file to upload\n";
  48 }
  49
  50
  51 // get arguments
  52 $args = get_opt(array('h' => 'host', 'u' => 'user', 'p' => 'pass', 'm' => 'mbox', 'f' => 'file')) + array('host' => 'localhost', 'mbox' => 'INBOX');
  53
  54 if ($_SERVER['argv'][1] == 'help')
  55 {
  56         print_usage();
  57         exit;
  58 }
  59 else if (!($args['host'] && $args['file']))
  60 {
  61         print "Missing required parameters.\n";
  62         print_usage();
  63         exit;
  64 }
  65 else if (!is_file($args['file']))
  66 {
  67         print "Cannot read message file\n";
  68         exit;
  69 }
  70
  71 // prompt for username if not set
  72 if (empty($args['user']))
  73 {
  74         //fwrite(STDOUT, "Please enter your name\n");
  75         echo "IMAP user: ";
  76         $args['user'] = trim(fgets(STDIN));
  77 }
  78
  79 // prompt for password
  80 if (empty($args['pass']))
  81 {
  82         echo "Password: ";
  83         $args['pass'] = trim(fgets(STDIN));
  84
  85         // clear password input
  86         echo chr(8)."\rPassword: ".str_repeat("*", strlen($args['pass']))."\n";
  87 }
  88
  89 // parse $host URL
  90 $a_host = parse_url($args['host']);
  91 if ($a_host['host'])
  92 {
  93         $host = $a_host['host'];
  94         $imap_ssl = (isset($a_host['scheme']) && in_array($a_host['scheme'], array('ssl','imaps','tls'))) ? TRUE : FALSE;
  95         $imap_port = isset($a_host['port']) ? $a_host['port'] : ($imap_ssl ? 993 : 143);
  96 }
  97 else
  98 {
  99         $host = $args['host'];
 100         $imap_port = 143;
 101 }
 102
 103 // instantiate IMAP class
 104 $IMAP = new rcube_imap(null);
 105
 106 // try to connect to IMAP server
 107 if ($IMAP->connect($host, $args['user'], $args['pass'], $imap_port, $imap_ssl))
 108 {
 109         print "IMAP login successful.\n";
 110         print "Uploading messages...\n";
 111
 112         $count = 0;
 113         $message = $lastline = '';
 114
 115         $fp = fopen($args['file'], 'r');
 116         while (($line = fgets($fp)) !== false)
 117         {
 118                 if (preg_match('/^From\s+/', $line) && $lastline == '')
 119                 {
 120                         if (!empty($message))
 121                         {
 122                                 if ($IMAP->save_message($args['mbox'], rtrim($message)))
 123                                         $count++;
 124                                 else
 125                                         die("Failed to save message to {$args['mbox']}\n");
 126                                 $message = '';
 127                         }
 128                         continue;
 129                 }
 130
 131                 $message .= $line;
 132                 $lastline = rtrim($line);
 133         }
 134
 135         if (!empty($message) && $IMAP->save_message($args['mbox'], rtrim($message)))
 136                 $count++;
 137
 138         // upload message from file
 139         if ($count)
 140                 print "$count messages successfully added to {$args['mbox']}.\n";
 141         else
 142                 print "Adding messages failed!\n";
 143 }
 144 else
 145 {
 146         print "IMAP login failed.\n";
 147 }
 148
 149 ?>