1#!/usr/bin/env php
2<?php
3/*
4 +-----------------------------------------------------------------------+
5 | This file is part of the Roundcube Webmail client                     |
6 |                                                                       |
7 | Copyright (C) The Roundcube Dev Team                                  |
8 |                                                                       |
9 | Licensed under the GNU General Public License version 3 or            |
10 | any later version with exceptions for skins & plugins.                |
11 | See the README file for a full license statement.                     |
12 +-----------------------------------------------------------------------+
13 | Author: Thomas Bruederli <thomas@roundcube.net>                       |
14 +-----------------------------------------------------------------------+
15*/
16
17define('INSTALL_PATH', realpath(__DIR__ . '/..') . '/' );
18ini_set('memory_limit', -1);
19
20require_once INSTALL_PATH.'program/include/clisetup.php';
21
22function print_usage()
23{
24    print "Usage: msgimport.sh -h imap-host -u user-name -m mailbox -f message-file\n";
25    print "--host   IMAP host\n";
26    print "--user   IMAP user name\n";
27    print "--mbox   Target mailbox\n";
28    print "--file   Message file to upload\n";
29}
30
31// get arguments
32$opts = ['h' => 'host', 'u' => 'user', 'p' => 'pass', 'm' => 'mbox', 'f' => 'file'];
33$args = rcube_utils::get_opt($opts) + ['host' => 'localhost', 'mbox' => 'INBOX'];
34
35if (!isset($_SERVER['argv'][1]) || $_SERVER['argv'][1] == 'help') {
36    print_usage();
37    exit;
38}
39else if (empty($args['host']) || empty($args['file'])) {
40    print "Missing required parameters.\n";
41    print_usage();
42    exit;
43}
44else if (!is_file($args['file'])) {
45    rcube::raise_error("Cannot read message file.", false, true);
46}
47
48// prompt for username if not set
49if (empty($args['user'])) {
50    //fwrite(STDOUT, "Please enter your name\n");
51    echo "IMAP user: ";
52    $args['user'] = trim(fgets(STDIN));
53}
54
55// prompt for password
56if (empty($args['pass'])) {
57    $args['pass'] = rcube_utils::prompt_silent("Password: ");
58}
59
60// parse $host URL
61$a_host = parse_url($args['host']);
62if (!empty($a_host['host'])) {
63    $host      = $a_host['host'];
64    $imap_ssl  = (isset($a_host['scheme']) && in_array($a_host['scheme'], ['ssl','imaps','tls'])) ? TRUE : FALSE;
65    $imap_port = isset($a_host['port']) ? $a_host['port'] : ($imap_ssl ? 993 : 143);
66}
67else {
68    $host      = $args['host'];
69    $imap_port = 143;
70    $imap_ssl  = false;
71}
72
73// instantiate IMAP class
74$IMAP = new rcube_imap(null);
75
76// try to connect to IMAP server
77if ($IMAP->connect($host, $args['user'], $args['pass'], $imap_port, $imap_ssl)) {
78    print "IMAP login successful.\n";
79    print "Uploading messages...\n";
80
81    $count   = 0;
82    $message = $lastline = '';
83
84    $fp = fopen($args['file'], 'r');
85    while (($line = fgets($fp)) !== false) {
86        if (preg_match('/^From\s+-/', $line) && $lastline == '') {
87            if (!empty($message)) {
88                if ($IMAP->save_message($args['mbox'], rtrim($message))) {
89                    $count++;
90                }
91                else {
92                    rcube::raise_error("Failed to save message to {$args['mbox']}", false, true);
93                }
94                $message = '';
95            }
96            continue;
97        }
98
99        $message .= $line;
100        $lastline = rtrim($line);
101    }
102
103    if (!empty($message) && $IMAP->save_message($args['mbox'], rtrim($message))) {
104        $count++;
105    }
106
107    // upload message from file
108    if ($count) {
109        print "$count messages successfully added to {$args['mbox']}.\n";
110    }
111    else {
112        print "Adding messages failed!\n";
113    }
114}
115else {
116    rcube::raise_error("IMAP login failed.", false, true);
117}
118