#!/usr/bin/python
# -*- coding: utf-8 -*-

# --------------------------------------------------------------------------- #
#                                                                             #
#    Plugin for iSida Jabber Bot                                              #
#    Copyright (C) diSabler <dsy@dsy.name>                                    #
#                                                                             #
#    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 3 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, see <http://www.gnu.org/licenses/>.    #
#                                                                             #
# --------------------------------------------------------------------------- #

global execute, lf_api, lfm_url

lfm_url = 'http://ws.audioscrobbler.com/2.0/'

def last_text_cnt(text,c):
    text = reduce_spaces_all(text).split(' ')
    try: cnt = int(text[1])
    except: cnt = GT('lastfm_max_limit')
    cnt += c
    text = text[0]
    return text,cnt

def last_check_ascii(type, jid, nick, text):
    for tmp in text:
        if tmp > '~' or tmp < ' ':
            send_msg(type, jid, nick, L('Error!','%s/%s'%(jid,nick)))
            return True
    return None

def last_time_short(tm):
    tm = time.localtime(tm)
    tnow = time.localtime()
    if tm[0] != tnow[0]: form = '%d.%m.%Y %H:%M'
    elif tm[1]!=tnow[1] or tm[2]!=tnow[2]: form = '%d.%m %H:%M'
    else: form = '%H:%M'
    return str(time.strftime(form,tm))

def last_date_now(body):
    if 'nowplaying=\"true\"' in body: return 'now'
    else:
        try: return last_time_short(int(get_subtag(get_tag_full(body,'date'),'uts')))
        except: return 'Unknown'

def lastonetrack(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    ms = lf_api('user.getrecenttracks',text, '<track')
    if len(ms): cnt = len(ms)
    else: cnt = 0
    if cnt >=2: msg = L('Last track %s: %s - %s %s','%s/%s'%(jid,nick)) % (text,get_tag(ms[1],'artist'),get_tag(ms[1],'name'),'['+last_date_now(ms[1])+']')
    else: msg = L('Unavailable!','%s/%s'%(jid,nick))
    send_msg(type, jid, nick, msg)

def lf_api(method, user, splitter):
    user = reduce_spaces_all(user.lower().encode('utf-8').replace('\\x','%')).replace(' ','%20')
    link = '%s?method=%s&user=%s&api_key=%s' % (lfm_url,method,user,GT('lfm_api'))
    return rss_replace(html_encode(load_page(link))).split(splitter)

def lasttracks(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    text,cnt = last_text_cnt(text,1)
    ms = lf_api('user.getrecenttracks',text, '<track')
    msg = L('Last tracks %s:','%s/%s'%(jid,nick)) % text
    for a in ms[1:cnt]: msg += '\n[%s] %s - %s' % (last_date_now(a),get_tag(a,'artist'),get_tag(a,'name'))
    send_msg(type, jid, nick, msg)

def lastfriends(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    ms = lf_api('user.getfriends',text, '<user')
    msg = L('Friends %s:','%s/%s'%(jid,nick)) % text
    for a in ms[1:]:
        rn = get_tag(a,'realname')
        if rn: msg += ' %s (%s),' % (get_tag(a,'name'),rn)
        else: msg += ' %s,' % get_tag(a,'name')
    msg = msg[:-1]
    send_msg(type, jid, nick, msg)

def lastloved(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    text,cnt = last_text_cnt(text,1)
    ms = lf_api('user.getlovedtracks',text, '<track')
    msg = L('Loved tracks %s:','%s/%s'%(jid,nick)) % text
    for a in ms[1:cnt]: msg += '\n[%s] %s - %s' % (last_date_now(a),get_tag(a.split('<artist')[1],'name'),get_tag(a,'name'))
    send_msg(type, jid, nick, msg)

def lastneighbours(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    text,cnt = last_text_cnt(text,1)
    ms = lf_api('user.getneighbours',text, '<user')
    msg = L('Neighbours %s:','%s/%s'%(jid,nick)) % text
    for a in ms[1:cnt]: msg += '\n%s - %s' % (get_tag(a,'match'),get_tag(a,'name'))
    send_msg(type, jid, nick, msg)

def lastplaylist(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    text,cnt = last_text_cnt(text,2)
    ms = lf_api('user.getplaylists',text, '<playlist')
    msg = L('Playlists %s:','%s/%s'%(jid,nick)) % text
    for a in ms[2:cnt]: msg += '\n[%s] %s (%s) - %s - %s' % (get_tag(a,'id'),get_tag(a,'title'),get_tag(a,'description'),get_tag(a,'size'),get_tag(a,'duration'))
    send_msg(type, jid, nick, msg)

def topalbums(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    text,cnt = last_text_cnt(text,1)
    ms = lf_api('user.gettopalbums',text, '<album')
    msg = L('Top albums %s:','%s/%s'%(jid,nick)) % text
    for a in ms[1:cnt]: msg += '\n[%s] %s - %s' % (get_tag(a,'playcount'),get_tag(a.split('<artist')[1],'name'),get_tag(a,'name'))
    send_msg(type, jid, nick, msg)

def topartists(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    text,cnt = last_text_cnt(text,1)
    ms = lf_api('user.gettopartists',text, '<artist')
    msg = L('Top artists %s:','%s/%s'%(jid,nick)) % text
    for a in ms[1:cnt]: msg += '\n[%s] %s' % (get_tag(a,'playcount'),get_tag(a,'name'))
    send_msg(type, jid, nick, msg)

def toptags(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    text,cnt = last_text_cnt(text,1)
    ms = lf_api('user.gettoptags',text, '<tag')
    msg = L('Top tags %s:','%s/%s'%(jid,nick)) % text
    for a in ms[1:cnt]: msg += '\n[%s] %s - %s' % (get_tag(a,'count'),get_tag(a,'name'),get_tag(a,'url'))
    send_msg(type, jid, nick, msg)

def toptracks(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    text,cnt = last_text_cnt(text,1)
    ms = lf_api('user.gettoptracks',text, '<track')
    if cnt > len(ms): cnt = len(ms)
    msg = L('Top tracks %s:','%s/%s'%(jid,nick)) % text
    for a in ms[1:cnt]:
        b = a.split('<artist')
        msg += '\n[%s] %s - %s' % (get_tag(a,'playcount'),get_tag(b[1],'name'),get_tag(a,'name'))
    send_msg(type, jid, nick, msg)

def tasteometer(type, jid, nick, text):
    if not text: text = nick
    if last_check_ascii(type, jid, nick, text): return
    text = reduce_spaces_all(text.lower().encode('utf-8').replace('\\x','%')).split(' ',1)
    try: (user1,user2) = text
    except:
        send_msg(type, jid, nick, L('Need two users','%s/%s'%(jid,nick)))
        return
    link = '%s?method=tasteometer.compare&type1=user&type2=user&value1=%s&value2=%s&api_key=%s' % (lfm_url,user1,user2,GT('lfm_api'))
    lfxml = html_encode(load_page(link))
    scor = get_tag(lfxml,'score')
    try: scor = float(scor)
    except: scor = 0
    if scor <= 0: msg = L('Tastes of %s and %s - soo different!','%s/%s'%(jid,nick)) % (user1,user2)
    else:
        msg,msg2 = L('Match tastes of %s and %s - %s','%s/%s'%(jid,nick)) % (user1,user2,str(int(scor*100))+'%'),''
        lfxml = lfxml.split('<artist')
        cnt = len(lfxml)
        for a in lfxml[2:cnt]: msg2 += '%s, ' % get_tag(a,'name')
        if len(msg2): msg += '\n%s' % L('Artists: %s','%s/%s'%(jid,nick)) % msg2[:-2]
    send_msg(type, jid, nick, msg)

def no_api(type, jid, nick): send_msg(type, jid, nick, L('Not found LastFM api','%s/%s'%(jid,nick)))

exec_yes = [(3, 'lasttracks', lasttracks, 2, 'Last scrobled tracks'),
        (3, 'last', lastonetrack, 2, 'Last scrobled track'),
        (3, 'lastfriends', lastfriends, 2, 'Last friends'),
        (3, 'lastloved', lastloved, 2, 'Last loved tracks'),
        (3, 'lastneighbours', lastneighbours, 2, 'Last neighbours'),
        (3, 'lastplaylist', lastplaylist, 2, 'Last playlist'),
        (3, 'topalbums', topalbums, 2, 'Top albums'),
        (3, 'topartists', topartists, 2, 'Top artists'),
        (3, 'toptags', toptags, 2, 'Top tags'),
        (3, 'toptracks', toptracks, 2, 'Top tracks'),
        (3, 'tasteometer', tasteometer, 2, 'Music tastes')]

exec_no = [(3, 'lasttracks', no_api, 1, 'Not found LastFM api'),
       (3, 'last', no_api, 1, 'Not found LastFM api'),
       (3, 'lastfriends', no_api, 1, 'Not found LastFM api'),
       (3, 'lastloved', no_api, 1, 'Not found LastFM api'),
       (3, 'lastneighbours', no_api, 1, 'Not found LastFM api'),
       (3, 'lastplaylist', no_api, 1, 'Not found LastFM api'),
       (3, 'topalbums', no_api, 1, 'Not found LastFM api'),
       (3, 'topartists', no_api, 1, 'Not found LastFM api'),
       (3, 'toptags', no_api, 1, 'Not found LastFM api'),
       (3, 'toptracks', no_api, 1, 'Not found LastFM api'),
       (3, 'tasteometer', no_api, 1, 'Not found LastFM api')]

execute = [exec_no,exec_yes][len(GT('lfm_api')) >= 30]