#!/usr/bin/php5
<?php
#ini_set('memory_limit', '64M');

/*
    Gajim log export tool
    Copyleft (C) 2007 BohwaZ - http://dev.kd2.org/

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

// Application defaults ////////////////////////////////////////////////////////////////////

// Gajim path
$source_path = $_SERVER['HOME'].'/.gajim';

// Destination path (leave empty, user *have* to specify it)
$dest_path = '';

// Output format (text or irssi) only irssi available
$output_format = 'irssi';

// User nick for logs
$user_nick = 'You';

// From gajim source code, not well documented
define('GAJIM_KIND_STATUS', 0);
define('GAJIM_KIND_GCSTATUS', 1);
define('GAJIM_KIND_GC_MSG', 2);
define('GAJIM_KIND_SINGLE_MSG_RECV', 3);
define('GAJIM_KIND_CHAT_MSG_RECV', 4);
define('GAJIM_KIND_SINGLE_MSG_SENT', 5);
define('GAJIM_KIND_CHAT_MSG_SENT', 6);
define('GAJIM_KIND_STATUS', 7);


// Common functions and objects ///////////////////////////////////////////////////////////

function parse_arguments()
{
    $args = array();
    foreach ($_SERVER['argv'] as $arg)
    {
        $arg = trim($arg);
        if (preg_match('/^--([a-z-]+)$/', $arg, $match))
        {
            if (!empty($current_key) && !isset($args[$current_key]))
                $args[$current_key] = true;

            $current_key = $match[1];
            continue;
        }
        if (empty($current_key))
            continue;

        $args[$current_key] = $arg;
    }

    return $args;
}

function cmd_error($msg)
{
    echo $msg;
    echo "\n";
    die();
}

function irssi_time($time)
{
    return strtolower(date('D M d H:i:s Y', $time));
}

function log_export_irssi($jid, $result)
{
    $jid = preg_replace('/^([^@%]*)([@%].*)$/', '\\1', $jid);

    $start = 0;
    $end = 0;
    $out = '';

    $fp = fopen($GLOBALS['dest_path'].'/'.$jid.'.log', 'w');

    while($line = $result->fetch(PDO::FETCH_ASSOC))
    {
        if (date('Ymd', $line['time']) > $start)
        {
            if (!empty($end))
                fputs($fp, "--- Log closed ".irssi_time($end)."\n");
            fputs($fp, "--- Log opened ".irssi_time($line['time'])."\n");
            $start = date('Ymd', $line['time']);
        }

        fputs($fp, date('H:i', $line['time'])." ");

        switch ($line['kind'])
        {
            case GAJIM_KIND_SINGLE_MSG_RECV:
            case GAJIM_KIND_CHAT_MSG_RECV:
                fputs($fp, "<".$jid.">");
                break;
            case GAJIM_KIND_SINGLE_MSG_SENT:
            case GAJIM_KIND_CHAT_MSG_SENT:
                fputs($fp, "<".$GLOBALS['user_nick'].">");
                break;
            default:
                if (empty($line['message']))
                    continue(2);
                fputs($fp, " * Gajim message:");
                break;
        }

        fputs($fp, " ".utf8_decode($line['message'])."\n");

        $end = $line['time'];
    }

    if (!empty($end))
        fputs($fp, "--- Log closed ".irssi_time($end)."\n");

    fclose($fp);
    unset($out);
}

// Begin of application itself ////////////////////////////////////////////////////////////////////////

// Fetching command line arguments
$args = parse_arguments();

if (isset($args['help']))
{
    echo
        "Gajim log export tool\nCopyleft (C) 2007 BohwaZ - http://dev.kd2.org/ - GNU/GPL\n\n".
        $_SERVER['argv'][0]." --destination DESTINATION_PATH [options]\n\n".
        " --destination     MANDATORY. Destination path, where to put generated log files\n".
        " --source-path     Source path, where are gajim things (eg. /home/user/.gajim)\n".
        //" --output-format   Defines output format. 'irssi' or 'text'. Default is irssi.\n".
        " --user-nick       Nick of yourself when you're speaking in conversations.\n".
        "\n";

    die();
}

if (!empty($args['source-path']))
    $source_path = $args['source-path'];

if (!file_exists($source_path) OR !is_dir($source_path))
    cmd_error("Path ".$path." don't seems to exist or is not a directory.");

if (!empty($args['destination']))
    $dest_path = $args['destination'];

if (empty($dest_path))
    cmd_error("Destination path is mandatory. Check help with --help for options and arguments.");

//if (!empty($args['output-format']) AND preg_match('/^(irssi|text)$/i', $args['output-format']))
//    $output_format = strtolower($args['output_format']);

if (!empty($args['user-nick']))
    $user_nick = $args['user-nick'];

// SQLite Connection via PDO
try {
    $db = new PDO('sqlite:'.$source_path.'/logs.db');
}
catch (PDOException $dbex) {
    cmd_error("Can't connect to logs.db file: ".$dbex->getMessage());
}

try {
    $results = $db->query('SELECT * FROM jids;')->fetchAll();
}
catch (PDOException $dbex) {
    cmd_error("Error in fetching jid list: ".$dbex->getMessage());
}

if (empty($results))
    cmd_error("Can't continue: you have no JID !");

$jids = array();

foreach ($results as $line) {
    $jids[$line['jid_id']] = $line['jid'];
}

foreach ($jids as $id=>$jid)
{
    echo "Exporting $jid ...  ";

    try {
        $results = $db->query('SELECT * FROM logs WHERE jid_id="'.(int) $id.'" ORDER BY time ASC;');
    }
    catch (PDOException $dbex) {
        cmd_error("Error: ".$dbex->getMessage());
    }

    if (empty($results))
        echo "Nothing to export.\n";
    else
    {
        if ($output_format == 'irssi')
            log_export_irssi($jid, $results);

        echo "Done.\n";
    }
    unset($results);
}

?>