alecpl
2008-08-29 910d07e3002a9077500e09abea968fc7f2eaeb91
commit | author | age
687b22 1 #!/usr/bin/php
0712a3 2 <?php
T 3
47124c 4 define('INSTALL_PATH', preg_replace('/bin\/$/', '', getcwd()) . '/');
6cc6e7 5 ini_set('memory_limit', -1);
0712a3 6
47124c 7 require_once INSTALL_PATH.'program/include/iniset.php';
0712a3 8
T 9 /**
10  * Parse commandline arguments into a hash array
11  */
687b22 12 function get_opt($aliases=array())
0712a3 13 {
T 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
6d0998 41 function print_usage()
0712a3 42 {
687b22 43     print "Usage:  msgimport -h imap-host -u user-name -m mailbox -f message-file\n";
b48bd0 44     print "--host   IMAP host\n";
T 45     print "--user   IMAP user name\n";
687b22 46     print "--mbox   Target mailbox\n";
b48bd0 47     print "--file   Message file to upload\n";
6d0998 48 }
T 49
50
51 // get arguments
687b22 52 $args = get_opt(array('h' => 'host', 'u' => 'user', 'p' => 'pass', 'm' => 'mbox', 'f' => 'file')) + array('host' => 'localhost', 'mbox' => 'INBOX');
6d0998 53
T 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();
0712a3 63     exit;
T 64 }
65 else if (!is_file($args['file']))
66 {
67     print "Cannot read message file\n";
6d0998 68     exit;
0712a3 69 }
T 70
6d0998 71 // prompt for username if not set
T 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 echo "Password: ";
81 $args['pass'] = trim(fgets(STDIN));
0712a3 82
b48bd0 83 // clear password input
T 84 echo chr(8)."\rPassword: ".str_repeat("*", strlen($args['pass']))."\n";
85
0712a3 86 // parse $host URL
T 87 $a_host = parse_url($args['host']);
88 if ($a_host['host'])
89 {
90     $host = $a_host['host'];
91     $imap_ssl = (isset($a_host['scheme']) && in_array($a_host['scheme'], array('ssl','imaps','tls'))) ? TRUE : FALSE;
92     $imap_port = isset($a_host['port']) ? $a_host['port'] : ($imap_ssl ? 993 : 143);
93 }
94 else
95 {
96     $host = $args['host'];
97     $imap_port = 143;
98 }
99
100 // instantiate IMAP class
101 $IMAP = new rcube_imap(null);
102
103 // try to connect to IMAP server
104 if ($IMAP->connect($host, $args['user'], $args['pass'], $imap_port, $imap_ssl))
105 {
106     print "IMAP login successful.\n";
687b22 107     print "Uploading messages...\n";
0712a3 108     
687b22 109     $count = 0;
T 110     $message = $lastline = '';
111     
112     $fp = fopen($args['file'], 'r');
113     while (($line = fgets($fp)) !== false)
114     {
115         if (preg_match('/^From\s+/', $line) && $lastline == '')
116         {
117             if (!empty($message))
118             {
119                 if ($IMAP->save_message($args['mbox'], rtrim($message)))
120                     $count++;
121                 else
122                     die("Failed to save message to $mailbox\n");
123                 $message = '';
124             }
125             continue;
126         }
127
128         $message .= $line;
129         $lastline = rtrim($line);
130     }
131
132     if (!empty($message) && $IMAP->save_message($args['mbox'], rtrim($message)))
133         $count++;
134
0712a3 135     // upload message from file
687b22 136     if ($count)
T 137         print "$count messages successfully added to $mailbox.\n";
0712a3 138     else
687b22 139         print "Adding messages failed!\n";
0712a3 140 }
T 141 else
142 {
143     print "IMAP login failed.\n";
144 }
145
146 ?>