thomascube
2008-09-19 42e328a85f5880e3e767eebaaa88b55a2b49d2ff
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
9787ca 80 if (empty($args['pass']))
T 81 {
82     echo "Password: ";
83     $args['pass'] = trim(fgets(STDIN));
0712a3 84
9787ca 85     // clear password input
T 86     echo chr(8)."\rPassword: ".str_repeat("*", strlen($args['pass']))."\n";
87 }
b48bd0 88
0712a3 89 // parse $host URL
T 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";
687b22 110     print "Uploading messages...\n";
0712a3 111     
687b22 112     $count = 0;
T 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
9787ca 125                     die("Failed to save message to {$args['mbox']}\n");
687b22 126                 $message = '';
T 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
0712a3 138     // upload message from file
687b22 139     if ($count)
9787ca 140         print "$count messages successfully added to {$args['mbox']}.\n";
0712a3 141     else
687b22 142         print "Adding messages failed!\n";
0712a3 143 }
T 144 else
145 {
146     print "IMAP login failed.\n";
147 }
148
149 ?>