# -*- coding: utf-8 -*-
import telegram
import steam
import random
import osu
import hearthstone
import sys


# Check per la modalità votazione del bot, corrisponde al numero della chat in cui è attiva la votazione
# 0 per disattivare la votazione
class Votazione:
    chat = int()
    domanda = str()
    # 0: non votato
    # 1: sì
    # 2: no
    # 3: astenuto
    voto = {
        'steffo': int(0),
        'alby1': int(0),
        'boni3099': int(0),
        'maxsensei': int(0),
        'cosimo03': int(0),
        'frankrekt': int(0),
        'heisendoc': int(0),
        'acterryg': int(0),
        'adry99': int(0),
        'alleanderl': int(0),
        'thevagginadestroyer': int(0),
        'tiztiztiz': int(0),
        'fultz': int(0),
        'gotob': int(0),
        'enribenassati': int(0),
        'iemax': int(0),
        'peraemela99': int(0),
        'ilgattopardo': int(0),
        'mrdima98': int(0),
        'ruozir': int(0),
        'supersmurf': int(0),
        'tauei': int(0),
        'voltaggio': int(0),
        'gibait': int(0),
    }

    def __init__(self, question, askin):
        self.domanda = question
        self.chat = askin

    def ask(self):
        telegram.sendmessage(self.domanda, self.chat)

    def register(self, utente, voto):
        self.voto[utente] = voto

    def showresults(self):
        lista = str()
        si = 0
        no = 0
        astenuti = 0
        for membro in self.voto:
            if self.voto[membro] == 0:
                lista += chr(9898)
            elif self.voto[membro] == 1:
                si += 1
                lista += chr(128309)
            elif self.voto[membro] == 2:
                no += 1
                lista += chr(128308)
            elif self.voto[membro] == 3:
                astenuti += 1
                lista += chr(9899)
            lista += " @" + membro + "\n"
        telegram.sendmessage(self.domanda + "\n"
                             "*Risultati:*\n"
                             "Sì: " + str(si) + " (" + str(round(si / (si + no + astenuti) * 100, 2)) + "%)\n"
                             "No: " + str(no) + " (" + str(round(no / (si + no + astenuti) * 100, 2)) + "%)\n"
                             "Astenuti: " + str(astenuti) + "\n\n" + lista, self.chat)

# Votazione in corso
incorso = None

# Playlist di /rage, si riempie quando è vuota
rage = []

# TODO: Rimettere gli audio di Wololo
# wololo = []

# Dizionario con i nomi utenti di osu!
# Se qualcuno cambia nome utente di Telegram, lo cambi anche QUI.
osunames = {
    'steffo': 'SteffoRYG',
    'evilbalu': 'NemesisRYG',
    'fultz': 'ftz99',
    'ilgattopardo': 'gattopardo',
    'frankrekt': 'FrankezRYG',
    'tiztiztiz': 'fedececco',
    'acterryg': 'Acter1',
    'maxsensei': 'MaxSensei',
    'heisendoc': 'ImHeisenberg',
    'thevagginadestroyer': 'barboll',
    'cosimo03': 'Cosimo03',
    'albertino04': 'Alby1',
    'voltaggio': 'voltaggio',
    'tauei': 'tauei',
    'boni3099': 'boni3099',
    'mrdima98': 'MRdima98',
}

# Dizionario con gli steamID
# Vedi sopra
steamids = {
    'steffo': 76561198034314260,
    'alby1': 76561198071383448,
    'boni3099': 76561198131868211,
    'maxsensei': 76561198121094516,
    'cosimo03': 76561198062778224,
    'frankrekt': 76561198071099951,
    'heisendoc': 76561198080377213,
    'acterryg': 76561198146704979,
    'adry99': 76561198230034568,
    'alleanderl': 76561198154175301,
    'thevagginadestroyer': 76561198128738388,
    'tiztiztiz': 76561198109189938,
    'fultz': 76561198035547490,
    'gattino02': 76561198071069550,
    'gotob': 76561198096658890,
    'enribenassati': 76561198123018368,
    'iemax': 76561198149695151,
    'peraemela99': 76561198161867082,
    'ilgattopardo': 76561198111021344,
    'mrdima98': 76561198140863530,
    'ruozir': 76561198117708290,
    'supersmurf': 76561198115852550,
    'tauei': 76561198104305298,
    'voltaggio': 76561198147601821,
    'gibait': 76561198157721704,
}

random.seed()

# Ciclo principale del bot
print("Bot avviato!")
while True:
    # Guarda il comando ricevuto.
    msg = telegram.getupdates()
    # Se il messaggio non è una notifica di servizio...
    if 'text' in msg:
        # Salvatelo in una stringa
        text = msg['text']
        # Guarda l'ID della chat in cui è stato inviato
        sentin = msg['chat']['id']
        # Nome da visualizzare nella console per capire chi accidenti è che invia messaggi strani
        if 'username' in msg['from']:
            # Salva l'username se esiste
            username = msg['from']['username']
        else:
            # Altrimenti, salva l'userID
            username = str(msg['from']['id'])
        # Riconosci il comando.
        # Viene usato startswith perchè il comando potrebbe anche essere inviato in forma /ciao@RoyalBot.
        if text.startswith('/ahnonlosoio'):
            print("@" + username + ": /ahnonlosoio")
            telegram.sendmessage("Ah, non lo so nemmeno io!", sentin)
        elif text.startswith('/ehoh'):
            print("@" + username + ": /ehoh")
            telegram.sendmessage("Eh, oh. Sono cose che capitano.", sentin)
        elif text.startswith('/playing'):
            print("@" + username + ": /playing")
            cmd = text.split(" ")
            # Se è stato specificato un AppID...
            if len(cmd) >= 2:
                n = steam.getnumberofcurrentplayers(cmd[1])
                # Se viene ricevuta una risposta...
                if n is None:
                    telegram.sendmessage(chr(9888) + " L'app specificata non esiste!", sentin)
                else:
                    telegram.sendmessage('In questo momento, ' + str(n) + ' persone stanno giocando a [' + cmd[1] +
                                         '](https://steamdb.info/app/' + cmd[1] + '/graphs/)', sentin)
            else:
                telegram.sendmessage(chr(9888) + ' Non hai specificato un AppID!\n'
                                     'La sintassi corretta è /playing <AppID>.', sentin)
        elif text.startswith('/saldi'):
            print("@" + username + ": /saldi")
            cmd = text.split(" ", 1)
            if len(cmd) == 2:
                telegram.sendmessage(
                    'Visualizza le offerte di '
                    '[' + cmd[1] + '](https://isthereanydeal.com/#/search:' + cmd[1].replace(' ', '%20') +
                    ";/scroll:%23gamelist).", sentin)
            else:
                telegram.sendmessage(chr(9888) +
                                     " Non hai specificato un gioco!"
                                     "[Visualizza tutte le offerte]"
                                     "(https://isthereanydeal.com/#/search:.;/scroll:%23gamelist).",
                                     sentin)
        # elif text.startswith('/audio'):
        #     # Se qualcuno ne ha voglia, qui si potrebbe aggiungere la selezione degli audio come argomento,
        #     # invece che fare una playlist casuale...
        #     # Se non ci sono rage nella playlist, riempila e mescolala!
        #     if len(rage) <= 0:
        #         # TODO: Rimettere gli audio di /audio
        #         rage = []
        #         random.shuffle(rage)
        #     # Estrai un audio a caso tra quelli nella playlist e rimuovilo.
        #     ragesend = rage.pop()
        #     print("@" + username + ": /audio")
        #     telegram.senddocument(ragesend, sentin)
        elif text.startswith('/sbam'):
            print("@" + username + ": /sbam")
            telegram.senddocument('BQADAgADBwMAAh8GgAGSsR4rwmk_LwI', sentin)
        # elif text.startswith('/wololo'):
        #     print("@" + username + ": /wololo")
        #     if len(cmd) >= 2:
        #         telegram.senddocument(wololo[int(cmd[1]) - 1], sentin)
        #     else:
        #         telegram.senddocument('BQADAgADZwIAAh8GgAF3etjqkzFDxAI', sentin)
        elif text.startswith('/osu'):
            print("@" + username + ": /osu")
            # Trova il nome utente
            cmd = text.split(' ', 1)
            if len(cmd) >= 2:
                # Trova la modalità
                cmd = text.split(' ', 2)
                if len(cmd) >= 3:
                    # Modalità specificata
                    mode = int(cmd[2])
                else:
                    # Osu! normale
                    mode = 0
                try:
                    r = osu.getuserrecent(cmd[1], mode)
                except NameError:
                    telegram.sendmessage(chr(9888) + " Errore nella richiesta ai server di Osu!", sentin)
                else:
                    if mode == 0:
                        telegram.sendmessage("*Osu!*\n"
                                             "[Beatmap " + r['beatmap_id'] + "](" + 'https://osu.ppy.sh/b/' + r[
                                                 'beatmap_id'] +
                                             ")\n*" + r['rank'] + "*\n*Punti*: " + r['score'] + "\n"
                                             "*Combo* x" + r['maxcombo'] + "\n"
                                             "*300*: " + r['count300'] + "\n"
                                             "*100*: " + r['count100'] + "\n"
                                             "*50*: " + r['count50'] + "\n"
                                             "*Awesome*: " + r['countkatu'] + "\n"
                                             "*Good*: " + r['countgeki'] + "\n"
                                             "*Miss*: " + r['countmiss'], sentin)
                    elif mode == 1:
                        telegram.sendmessage("*Taiko*\n"
                                             "[Beatmap " + r['beatmap_id'] + "](" + 'https://osu.ppy.sh/b/' + r[
                                                 'beatmap_id'] +
                                             ")\n*" + r['rank'] + "*\n*Punti*: " + r['score'] + "\n"
                                             "*Combo* x" + r['maxcombo'] + "\n"
                                             "*Great*: " + r['count300'] + "\n"
                                             "*Good*: " + r['count100'] + "\n"
                                             "_Large_ *Great*: " + r['countkatu'] + "\n"
                                             "_Large_ *Good*: " + r['countgeki'] + "\n"
                                             "*Bad*: " + r['countmiss'], sentin)
                    elif mode == 2:
                        telegram.sendmessage("*Catch the Beat*\n"
                                             "[Beatmap " + r['beatmap_id'] + "](" + 'https://osu.ppy.sh/b/' + r[
                                                 'beatmap_id'] +
                                             ")\n*" + r['rank'] + "*\n*Punti*: " + r['score'] + "\n"
                                             "*Combo* x" + r['maxcombo'] + "\n"
                                             "*Fruit*: " + r['count300'] + "\n"
                                             "*Droplet* _tick_: " + r['count100'] + "\n"
                                             "*Droplet* _trail_: " + r['count50'] + "\n"
                                             "*Miss*: " + r['countmiss'], sentin)
                    elif mode == 3:
                        telegram.sendmessage("*Osu!mania*\n" +
                                             "[Beatmap " + r['beatmap_id'] + "](" + 'https://osu.ppy.sh/b/' + r[
                                                 'beatmap_id'] + ")\n*" + r['rank'] +
                                             "*\n*Punti*: " + r['score'] + "\n"
                                             "*Combo* x" + r['maxcombo'] + "\n"
                                             "_Rainbow_ *300*: " + r['countgeki'] + "\n"
                                             "*300*: " + r['count300'] + "\n"
                                             "*100*: " + r['count100'] + "\n"
                                             "*200*: " + r['countkatu'] + "\n"
                                             "*50*: " + r['count50'] + "\n"
                                             "*Miss*: " + r['countmiss'], sentin)
                    else:
                        # E' un po' una scorciatoia eh, peeerò...
                        if username.lower() in osunames:
                            r = osu.getuserrecent(osunames[username.lower()], 0)
                            telegram.sendmessage("*Osu!*\n"
                                                 "[Beatmap " + r['beatmap_id'] + "](" + 'https://osu.ppy.sh/b/' + r[
                                                     'beatmap_id'] +
                                                 ")\n*" + r['rank'] + "*\n*Punti*: " + r['score'] + "\n"
                                                 "*Combo* x" + r['maxcombo'] + "\n"
                                                 "*300*: " + r['count300'] + "\n"
                                                 "*100*: " + r['count100'] + "\n"
                                                 "*50*: " + r['count50'] + "\n"
                                                 "*Awesome*: " + r['countgeki'] + "\n"
                                                 "*Good*: " + r['countkatu'] + "\n"
                                                 "*Miss*: " + r['countmiss'], sentin)
        elif text.startswith('/roll'):
            print("@" + username + ": /roll")
            cmd = text.split(' ', 1)
            if len(cmd) >= 2:
                m = int(cmd[1])
            else:
                m = 100
            n = random.randrange(m) + 1
            telegram.sendmessage("Numero casuale da 1 a " + str(m) + ":\n*" + str(n) + "*", sentin)
        elif text.startswith('/automah'):
            print("@" + username + ": /automah")
            # TODO: Mettere l'audio di Tobia
            telegram.sendmessage("Automaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa! Devi funzionare, cavolo!", sentin)
        elif text.startswith('/hs'):
            print("@" + username + ": /hs")
            cmd = text.split(" ", 1)
            r = None
            try:
                r = hearthstone.card(cmd[1])
                # Se ci sono più carte, prendine una a caso!
                r = r[random.randrange(len(r))]
            except ValueError:
                telegram.sendmessage(chr(9888) + "La carta specificata non esiste!", sentin)
            else:
                # Si trova nelle bustine
                if 'howToGet' not in r:
                    if 'cardSet' in r:
                        r['howToGet'] = "Trovala in " + r['cardSet'] + "."
                    else:
                        r['howToGet'] = "Inottenibile."
                # Nessuna classe
                if 'playerClass' not in r:
                    r['playerClass'] = "Neutral"
                # Nessun effetto
                if 'text' not in r:
                    r['text'] = ""
                # HTML nella descrizione
                r['text'] = r['text'].replace("<b>", "*")
                r['text'] = r['text'].replace("</b>", "*")
                r['text'] = r['text'].replace("<i>", "_")
                r['text'] = r['text'].replace("</i>", "_")
                r['text'] = r['text'].replace("$", "")
                # Nessuna rarità
                if 'rarity' not in r:
                    r['rarity'] = "None"
                # Nessuna descrizione
                if 'flavor' not in r:
                    r['flavor'] = ""
                # Testo principale
                text = None
                # Magie
                if r['type'] == "Spell":
                    text = str("[" + r['name'] + "](" + r['img'] + ") "
                               "(" + r['rarity'] + ")\n" +
                               r['playerClass'] + "\n" +
                               str(r['cost']) + " mana\n" +
                               r['text'] + "\n" +
                               r['howToGet'] + "\n\n_" +
                               r['flavor'] + "_\n")
                # Servitori
                elif r['type'] == "Minion":
                    text = str("[" + r['name'] + "](" + r['img'] + ") "
                               "(" + r['rarity'] + ")\n" +
                               r['playerClass'] + "\n" +
                               str(r['cost']) + " mana\n" +
                               str(r['attack']) + " attacco\n" +
                               str(r['health']) + " salute\n" +
                               r['text'] + "\n" +
                               r['howToGet'] + "\n\n_" +
                               r['flavor'] + "_\n")
                # Armi
                elif r['type'] == "Weapon":
                    text = str("[" + r['name'] + "](" + r['img'] + ") "
                               "(" + r['rarity'] + ")\n" +
                               r['playerClass'] + "\n" +
                               str(r['cost']) + " mana\n" +
                               str(r['attack']) + " attacco\n" +
                               str(r['durability']) + " integrita'\n" +
                               r['text'] + "\n" +
                               r['howToGet'] + "\n\n_" +
                               r['flavor'] + "_\n")
                # Potere Eroe
                elif r['type'] == "Hero Power":
                    text = str("[" + r['name'] + "](" + r['img'] + ")\n" +
                               r['playerClass'] + "\n" +
                               str(r['cost']) + " mana\n" +
                               r['text'] + "\n")
                # Eroe
                elif r['type'] == "Hero":
                    text = str("[" + r['name'] + "](" + r['img'] + ")\n" +
                               str(r['health']) + " salute\n")
                telegram.sendmessage(text, sentin)
        elif text.startswith('/online'):
            # Elenco di tutte le persone online su Steam
            print("@" + username + ": /online ")
            # Stringa utilizzata per ottenere informazioni su tutti gli utenti in una sola richiesta a steam
            userids = str()
            for nome in steamids:
                userids += str(steamids[nome]) + ','
            tosend = "*Online ora:*\n"
            telegram.sendchataction(sentin)
            r = steam.getplayersummaries(userids)
            for player in r:
                # In gioco
                if 'gameextrainfo' in player:
                    tosend += chr(128308) + " _" + player['gameextrainfo'] + "_ |"
                elif 'gameid' in player:
                    tosend += chr(128308) + " _" + player['gameid'] + "_ |"
                # Online
                elif player['personastate'] == 1:
                    tosend += chr(128309)
                # Occupato
                elif player['personastate'] == 2:
                    tosend += chr(128308)
                # Assente o Inattivo
                elif player['personastate'] == 3 or player['personastate'] == 4:
                    tosend += chr(9899)
                # Disponibile per scambiare
                elif player['personastate'] == 5:
                    tosend += chr(128310)
                # Disponibile per giocare
                elif player['personastate'] == 6:
                    tosend += chr(128311)
                if player['personastate'] != 0:
                    tosend += " " + player['personaname'] + "\n"
            else:
                telegram.sendmessage(tosend, sentin)
        elif text.startswith('/shrekt'):
            print("@" + username + ": /shrekt ")
            telegram.senddocument("BQADBAADsQADiBjiAqYN-EBXASyhAg", sentin)
        elif text.startswith('/restart') and username == "Steffo":
            print("@" + username + ": /restart ")
            telegram.sendmessage("Riavvio accettato.", sentin)
            sys.exit(0)
        elif text.startswith('/nuovavotazione') and username == "Steffo":
            print("@" + username + ": /nuovavotazione ")
            cmd = text.split(" ", 1)
            incorso = Votazione(cmd[1], sentin)
        elif text.startswith('/si') and incorso is not None:
            if incorso.chat == sentin:
                print("@" + username + ": /si ")
                incorso.register(username.lower(), 1)
                telegram.sendmessage("Votazione registrata!", sentin)
        elif text.startswith('/no') and incorso is not None:
            if incorso.chat == sentin:
                print("@" + username + ": /no ")
                incorso.register(username.lower(), 2)
                telegram.sendmessage("Votazione registrata!", sentin)
        elif text.startswith('/astieniti') and incorso is not None:
            if incorso.chat == sentin:
                print("@" + username + ": /astieniti ")
                incorso.register(username.lower(), 3)
                telegram.sendmessage("Votazione registrata!", sentin)
        elif text.startswith('/domanda') and incorso is not None:
            if incorso.chat == sentin:
                print("@" + username + ": /domanda ")
                incorso.ask()
        elif text.startswith('/risultati') and incorso is not None:
            if incorso.chat == sentin:
                print("@" + username + ": /risultati ")
                incorso.showresults()