#!/usr/bin/env php
<?php

/*
 +-----------------------------------------------------------------------+
 | This file is part of the Roundcube Webmail client                     |
 |                                                                       |
 | Copyright (C) The Roundcube Dev Team                                  |
 |                                                                       |
 | Licensed under the GNU General Public License version 3 or            |
 | any later version with exceptions for skins & plugins.                |
 | See the README file for a full license statement.                     |
 +-----------------------------------------------------------------------+
 | Author: Thomas Bruederli <thomas@roundcube.net>                       |
 +-----------------------------------------------------------------------+
*/

define('INSTALL_PATH', realpath(__DIR__ . '/..') . '/');
ini_set('memory_limit', -1);

require_once INSTALL_PATH . 'program/include/clisetup.php';

function print_usage()
{
    echo "Usage: msgimport.sh -h imap-host -u user-name -m mailbox -f message-file\n";
    echo "--host   IMAP host\n";
    echo "--user   IMAP user name\n";
    echo "--mbox   Target mailbox\n";
    echo "--file   Message file to upload\n";
}

// get arguments
$opts = ['h' => 'host', 'u' => 'user', 'p' => 'pass', 'm' => 'mbox', 'f' => 'file'];
$args = rcube_utils::get_opt($opts) + ['host' => 'localhost', 'mbox' => 'INBOX'];

if (!isset($_SERVER['argv'][1]) || $_SERVER['argv'][1] == 'help') {
    print_usage();
    exit;
} elseif (empty($args['host']) || empty($args['file'])) {
    echo "Missing required parameters.\n";
    print_usage();
    exit;
} elseif (!is_file($args['file'])) {
    rcube::raise_error('Cannot read message file.', false, true);
}

// prompt for username if not set
if (empty($args['user'])) {
    // fwrite(STDOUT, "Please enter your name\n");
    echo 'IMAP user: ';
    $args['user'] = trim(fgets(\STDIN));
}

// prompt for password
if (empty($args['pass'])) {
    $args['pass'] = rcube_utils::prompt_silent('Password: ');
}

// parse $host URL
$a_host = parse_url($args['host']);
if (!empty($a_host['host'])) {
    $host = $a_host['host'];
    $imap_ssl = (isset($a_host['scheme']) && in_array($a_host['scheme'], ['ssl', 'imaps', 'tls'])) ? true : false;
    $imap_port = $a_host['port'] ?? ($imap_ssl ? 993 : 143);
} else {
    $host = $args['host'];
    $imap_port = 143;
    $imap_ssl = false;
}

// instantiate IMAP class
$IMAP = new rcube_imap(null);

// try to connect to IMAP server
if ($IMAP->connect($host, $args['user'], $args['pass'], $imap_port, $imap_ssl)) {
    echo "IMAP login successful.\n";
    echo "Uploading messages...\n";

    $count = 0;
    $message = $lastline = '';

    $fp = fopen($args['file'], 'r');
    while (($line = fgets($fp)) !== false) {
        if (preg_match('/^From\s+-/', $line) && $lastline == '') {
            if (!empty($message)) {
                if ($IMAP->save_message($args['mbox'], rtrim($message))) {
                    $count++;
                } else {
                    rcube::raise_error("Failed to save message to {$args['mbox']}", false, true);
                }
                $message = '';
            }

            continue;
        }

        $message .= $line;
        $lastline = rtrim($line);
    }

    if (!empty($message) && $IMAP->save_message($args['mbox'], rtrim($message))) {
        $count++;
    }

    // upload message from file
    if ($count) {
        echo "{$count} messages successfully added to {$args['mbox']}.\n";
    } else {
        echo "Adding messages failed!\n";
    }
} else {
    rcube::raise_error('IMAP login failed.', false, true);
}
