/*
   Copyright 2009 Last.fm Ltd. 

   This file is part of liblastfm.

   liblastfm 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 3 of the License, or
   (at your option) any later version.

   liblastfm 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 liblastfm.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <QRegExp>
#include <QStringList>

#include "RadioStation.h"
#include "../core/XmlQuery.h"


const float k_defaultRep(0.5);
const float k_defaultMainstr(0.5);
const bool k_defaultDisco(false);

lastfm::RadioStation
lastfm::RadioStation::library( const lastfm::User& user )
{
    QList<lastfm::User> users;
    users << user;
    return library( users );
}

lastfm::RadioStation
lastfm::RadioStation::library( QList<lastfm::User>& users )
{
    RadioStation s( libraryStr( users ) );
    if( users.count() == 1 ) 
        s.setTitle( QObject::tr( "%1%2s Library Radio").arg( lastfm::ws::Username, QChar(0x2019) ));

    else {
        QString title;
        for( QList<lastfm::User>::const_iterator i = users.begin(); i != users.end(); i++ ) {
            if( i == users.end() - 1 )
                title += " and " + *i;
            else
                title += ", " + *i;
        }

        s.setTitle( title );
    }

    return s;
}


lastfm::RadioStation
lastfm::RadioStation::recommendations( const lastfm::User& user )
{
    RadioStation s( recommendationsStr( user ) );
    
    s.setTitle( QObject::tr( "%1%2s Recommended Radio").arg( lastfm::ws::Username, QChar(0x2019) ));

    return s;
}

lastfm::RadioStation
lastfm::RadioStation::friends( const lastfm::User& user )
{
    RadioStation s( friendsStr( user ) );

    s.setTitle( QObject::tr( "%1%2s Friends Radio").arg( lastfm::ws::Username, QChar(0x2019) ));

    return s; 
}

lastfm::RadioStation
lastfm::RadioStation::neighbourhood( const lastfm::User& user )
{
    RadioStation s( neighbourhoodStr( user ) );
    s.setTitle( QObject::tr( "%1%2s Neighbours%2 Radio").arg( lastfm::ws::Username, QChar(0x2019) ));
    return s;
}


lastfm::RadioStation
lastfm::RadioStation::tag( const lastfm::Tag& tag )
{
    QList<lastfm::Tag> tags;
    tags << tag;
    return lastfm::RadioStation::tag( tags );
}


lastfm::RadioStation
lastfm::RadioStation::tag( QList<lastfm::Tag>& tag )
{
    return RadioStation( tagStr( tag ) );
}


lastfm::RadioStation
lastfm::RadioStation::similar( const lastfm::Artist& artist )
{
    QList<lastfm::Artist> artists;
    artists << artist;
    return similar( artists );
}


lastfm::RadioStation
lastfm::RadioStation::similar( QList<lastfm::Artist>& artists )
{
    return RadioStation( similarStr( artists ) );
}


lastfm::RadioStation
lastfm::RadioStation::mix( const lastfm::User& user )
{
    RadioStation s( mixStr( user ) );
    s.setTitle( QObject::tr( "%1%2s Mix Radio").arg( lastfm::ws::Username, QChar(0x2019) ) );
    return s;
}


QString
lastfm::RadioStation::url() const
{
    return m_url.toString() + (m_tagFilter.isEmpty() ? "" : "/tag/" + m_tagFilter);
}


void
lastfm::RadioStation::setTitle( const QString& s )
{
    // Stop the radio station getting renamed when the web services don't know what it's called
    if ( !m_title.isEmpty() && s.compare( "a radio station", Qt::CaseInsensitive ) == 0 )
        return;

    QString title = s.trimmed();

    if ( title.compare( QObject::tr("%1%2s Library Radio").arg( lastfm::ws::Username, QChar(0x2019) ), Qt::CaseInsensitive ) == 0 )
        title = QObject::tr("My Library Radio");
    else if ( title.compare( QObject::tr("%1%2s Mix Radio").arg( lastfm::ws::Username, QChar(0x2019) ), Qt::CaseInsensitive ) == 0  )
        title = QObject::tr("My Mix Radio");
    else if ( title.compare( QObject::tr("%1%2s Recommended Radio").arg( lastfm::ws::Username, QChar(0x2019) ), Qt::CaseInsensitive ) == 0  )
        title = QObject::tr("My Recommended Radio");
    else if ( title.compare( QObject::tr("%1%2s Friends%2 Radio").arg( lastfm::ws::Username, QChar(0x2019) ), Qt::CaseInsensitive ) == 0  )
        title = QObject::tr("My Friends%1 Radio").arg( QChar( 0x2019 ) );
    else if ( title.compare( QObject::tr("%1%2s Friends Radio").arg( lastfm::ws::Username, QChar(0x2019) ), Qt::CaseInsensitive ) == 0  )
        title = QObject::tr("My Friends%1 Radio").arg( QChar( 0x2019 ) );
    else if ( title.compare( QObject::tr("%1%2s Neighbours%2 Radio").arg( lastfm::ws::Username, QChar(0x2019) ), Qt::CaseInsensitive ) == 0  )
        title = QObject::tr("My Neighbours%1 Radio").arg( QChar( 0x2019 ) );
    else if ( title.compare( QObject::tr("%1%2s Neighbours Radio").arg( lastfm::ws::Username ), Qt::CaseInsensitive ) == 0  )
        title = QObject::tr("My Neighbours%1 Radio").arg( QChar( 0x2019 ) );

    m_title = title;
}


QString
lastfm::RadioStation::title() const
{
    return m_title; // + (m_tagFilter.isEmpty() ? "" : ": " + m_tagFilter);
}


void
lastfm::RadioStation::setTagFilter( const QString& tag )
{
    m_tagFilter = tag;
}


QNetworkReply*
lastfm::RadioStation::getSampleArtists( int limit ) const
{
    QMap<QString, QString> map;
    map["method"] = "radio.getSampleArtists";
    map["station"] = m_url.toString();
    map["limit"] = QString::number( limit );
    return ws::get( map );
}


QNetworkReply*
lastfm::RadioStation::getTagSuggestions( int limit ) const
{
    QMap<QString, QString> map;
    map["method"] = "radio.getTagSuggestions";
    map["station"] = m_url.toString();
    map["limit"] = QString::number( limit );
    return ws::get( map );
}


//static 
QList<lastfm::RadioStation> 
lastfm::RadioStation::list( QNetworkReply* r )
{
    QList<lastfm::RadioStation> result;
    try {
        foreach (XmlQuery xq, XmlQuery( r->readAll() ).children("station")) {
            lastfm::RadioStation rs( QUrl::fromPercentEncoding( xq["url"].text().toUtf8() ) );
            rs.setTitle(xq["name"].text());
            result.append(rs);
        }
    }
    catch (ws::ParseError& e)
    {
        qWarning() << e.what();
    }
    return result;
}


bool
lastfm::RadioStation::operator==( const RadioStation& that ) const
{
    return this->m_url == that.m_url && this->m_tagFilter == that.m_tagFilter;
}


void
lastfm::RadioStation::setString( const QString& string )
{
    // If it's a tag filtered station then extract that part
    QString tempString = string;

    if ( !tempString.startsWith("lastfm://tag/") )
    {
        int index = tempString.indexOf("/tag/");

        if ( index != -1 )
        {
            m_tagFilter = tempString.mid( index + 5, tempString.count() - (index + 5) );
            tempString = tempString.mid( 0, index );
        }
    }

    m_url = tempString;
}


void
lastfm::RadioStation::setRep(float rep)
{
    m_rep = rep;
}


void
lastfm::RadioStation::setMainstr(float mainstr)
{
    m_mainstr = mainstr;
}


void
lastfm::RadioStation::setDisco(bool disco)
{
    m_disco = disco;
}


float lastfm::RadioStation::rep() const
{
    return m_rep;
}


float lastfm::RadioStation::mainstr() const
{
    return m_mainstr;
}


bool lastfm::RadioStation::disco() const
{
    return m_disco;
}


QString lastfm::RadioStation::libraryStr( QList<lastfm::User>& users )
{
    qSort(users.begin(), users.end());

    QString url = (users.count() > 1) ? "lastfm://users/" : "lastfm://user/";

    url.append( users[0].name() );

    for ( int i = 1 ; i < users.count() ; ++i )
        url.append( "," + users[i].name() );

    url.append("/personal");

    return url;
}


QString lastfm::RadioStation::tagStr( QList<lastfm::Tag>& tags )
{
    qSort(tags.begin(), tags.end());

    QString url = (tags.count() > 1) ? "lastfm://tag/" : "lastfm://globaltags/";

    url.append( tags[0].name() );

    for ( int i = 1 ; i < tags.count() ; ++i )
        url.append( "*" + tags[i].name() );

    return url;
}


QString lastfm::RadioStation::similarStr( QList<lastfm::Artist>& artists )
{
    qSort(artists.begin(), artists.end());

    QString url = (artists.count() > 1) ? "lastfm://artistnames/" : "lastfm://artist/";

    url.append( artists[0].name() );

    for ( int i = 1 ; i < artists.count() ; ++i )
        url.append( "," + artists[i].name() );

    if (artists.count() == 1)
        url.append( "/similarartists" );

    return url;
}