/*
 * This file Copyright (C) Mnemosyne LLC
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2
 * as published by the Free Software Foundation.
 *
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 *
 * $Id$
 */

#include <iostream>

#include <QPainter>
#include <QPixmap>
#include <QTextDocument>

#include <libtransmission/transmission.h>
#include <libtransmission/utils.h>

#include "favicon.h"
#include "formatter.h"
#include "torrent.h"
#include "tracker-delegate.h"
#include "tracker-model.h"

/***
****
***/

namespace
{
    const int mySpacing = 6;
    const QSize myMargin( 10, 6 );
}

QSize
TrackerDelegate :: margin( const QStyle& style ) const
{
    Q_UNUSED( style );

    return myMargin;
}

/***
****
***/

QSize
TrackerDelegate :: sizeHint( const QStyleOptionViewItem& option, const TrackerInfo& info ) const
{
    Q_UNUSED( option );

    QPixmap favicon = info.st.getFavicon( );

    const QString text = TrackerDelegate :: getText( info );
    QTextDocument textDoc;
    textDoc.setHtml( text );
    const QSize textSize = textDoc.size().toSize();

    return QSize( myMargin.width() + favicon.width() + mySpacing + textSize.width() + myMargin.width(),
                  myMargin.height() + qMax<int>( favicon.height(), textSize.height() ) + myMargin.height() );
}

QSize
TrackerDelegate :: sizeHint( const QStyleOptionViewItem  & option,
                             const QModelIndex           & index ) const
{
    const TrackerInfo trackerInfo = index.data( TrackerModel::TrackerRole ).value<TrackerInfo>();
    return sizeHint( option, trackerInfo );
}

void
TrackerDelegate :: paint( QPainter                    * painter,
                          const QStyleOptionViewItem  & option,
                          const QModelIndex           & index) const
{
    const TrackerInfo trackerInfo = index.data( TrackerModel::TrackerRole ).value<TrackerInfo>();
    painter->save( );
    painter->setClipRect( option.rect );
    drawBackground( painter, option, index );
    drawTracker( painter, option, trackerInfo );
    drawFocus(painter, option, option.rect );
    painter->restore( );
}

void
TrackerDelegate :: drawTracker( QPainter                    * painter,
                                const QStyleOptionViewItem  & option,
                                const TrackerInfo           & inf ) const
{
    painter->save( );

    QPixmap icon = inf.st.getFavicon( );
    QRect iconArea( option.rect.x() + myMargin.width(),
                    option.rect.y() + myMargin.height(),
                    icon.width(),
                    icon.height() );
    painter->drawPixmap( iconArea.x(), iconArea.y()+4, icon );

    const int textWidth = option.rect.width() - myMargin.width()*2 - mySpacing - icon.width();
    const int textX = myMargin.width() + icon.width() + mySpacing;
    const QString text = getText( inf );
    QTextDocument textDoc;
    textDoc.setHtml( text );
    const QRect textRect( textX, iconArea.y(), textWidth, option.rect.height() - myMargin.height()*2 );
    painter->translate( textRect.topLeft( ) );
    textDoc.drawContents( painter, textRect.translated( -textRect.topLeft( ) ) );

    painter->restore( );
}

void
TrackerDelegate :: setShowMore( bool b )
{
    myShowMore = b;
}

namespace
{
    QString timeToStringRounded( int seconds )
    {
        if( seconds > 60 ) seconds -= ( seconds % 60 );
        return Formatter::timeToString ( seconds );
    }
}

QString
TrackerDelegate :: getText( const TrackerInfo& inf ) const
{
    QString key;
    QString str;
    const time_t now( time( 0 ) );
    const QString err_markup_begin = "<span style=\"color:red\">";
    const QString err_markup_end = "</span>";
    const QString timeout_markup_begin = "<span style=\"color:#224466\">";
    const QString timeout_markup_end = "</span>";
    const QString success_markup_begin = "<span style=\"color:#008B00\">";
    const QString success_markup_end = "</span>";

    // hostname
    str += inf.st.isBackup ? "<i>" : "<b>";
    char * host = NULL;
    int port = 0;
    tr_urlParse( inf.st.announce.toUtf8().constData(), -1, NULL, &host, &port, NULL );
    str += QString( "%1:%2" ).arg( host ).arg( port );
    tr_free( host );
    if( !key.isEmpty( ) ) str += " - " + key;
    str += inf.st.isBackup ? "</i>" : "</b>";

    // announce & scrape info
    if( !inf.st.isBackup )
    {
        if( inf.st.hasAnnounced && inf.st.announceState != TR_TRACKER_INACTIVE )
        {
            const QString tstr( timeToStringRounded( now - inf.st.lastAnnounceTime ) );
            str += "<br/>\n";
            if( inf.st.lastAnnounceSucceeded )
            {
                str += tr( "Got a list of %1%2 peers%3 %4 ago" )
                           .arg( success_markup_begin )
                           .arg( inf.st.lastAnnouncePeerCount )
                           .arg( success_markup_end )
                           .arg( tstr );
            }
            else if( inf.st.lastAnnounceTimedOut )
            {
                str += tr( "Peer list request %1timed out%2 %3 ago; will retry" )
                           .arg( timeout_markup_begin )
                           .arg( timeout_markup_end )
                           .arg( tstr );
            }
            else
            {
                str += tr( "Got an error %1\"%2\"%3 %4 ago" )
                           .arg( err_markup_begin )
                           .arg( inf.st.lastAnnounceResult )
                           .arg( err_markup_end )
                           .arg( tstr );
            }
        }

        switch( inf.st.announceState )
        {
            case TR_TRACKER_INACTIVE:
                str += "<br/>\n";
                str += tr( "No updates scheduled" );
                break;

            case TR_TRACKER_WAITING: {
                const QString tstr( timeToStringRounded( inf.st.nextAnnounceTime - now ) );
                str += "<br/>\n";
                str += tr( "Asking for more peers in %1" ).arg( tstr );
                break;
            }

            case TR_TRACKER_QUEUED:
                str += "<br/>\n";
                str += tr( "Queued to ask for more peers" );
                break;

            case TR_TRACKER_ACTIVE: {
                const QString tstr( timeToStringRounded( now - inf.st.lastAnnounceStartTime ) );
                str += "<br/>\n";
                str += tr( "Asking for more peers now... <small>%1</small>" ).arg( tstr );
                break;
            }
        }

        if( myShowMore )
        {
            if( inf.st.hasScraped )
            {
                str += "<br/>\n";
                const QString tstr( timeToStringRounded( now - inf.st.lastScrapeTime ) );
                if( inf.st.lastScrapeSucceeded )
                {
                    str += tr( "Tracker had %1%2 seeders%3 and %4%5 leechers%6 %7 ago" )
                               .arg( success_markup_begin )
                               .arg( inf.st.seederCount )
                               .arg( success_markup_end )
                               .arg( success_markup_begin )
                               .arg( inf.st.leecherCount )
                               .arg( success_markup_end )
                               .arg( tstr );
                }
                else
                {
                    str += tr( "Got a scrape error %1\"%2\"%3 %4 ago" )
                               .arg( err_markup_begin )
                               .arg( inf.st.lastScrapeResult )
                               .arg( err_markup_end )
                               .arg( tstr );
                }
            }

            switch( inf.st.scrapeState )
            {
                case TR_TRACKER_INACTIVE:
                    break;

                case TR_TRACKER_WAITING: {
                    str += "<br/>\n";
                    const QString tstr( timeToStringRounded( inf.st.nextScrapeTime - now ) );
                    str += tr( "Asking for peer counts in %1" ).arg( tstr );
                    break;
                }

                case TR_TRACKER_QUEUED: {
                    str += "<br/>\n";
                    str += tr( "Queued to ask for peer counts" );
                    break;
                }

                case TR_TRACKER_ACTIVE: {
                    str += "<br/>\n";
                    const QString tstr( timeToStringRounded( now - inf.st.lastScrapeStartTime ) );
                    str += tr( "Asking for peer counts now... <small>%1</small>" ).arg( tstr );
                    break;
                }
            }
        }
    }

    return str;
}