reborn.py

# encoding=utf-8

import api
import importlib
import logging
import re
import config
import threading
import traceback
import sys
from time import gmtime
import datetime as dt
from calendar import timegm
from random import random,randint


logging.basicConfig(format='%(asctime)s.%(msecs)d %(levelname)-8s [%(filename)s:%(lineno)d] %(message)s',
                    datefmt='%d-%m-%Y:%H:%M:%S',
                    level=logging.INFO, handlers=[logging.FileHandler("log.log", "a", "utf-8")])


def msg_type(msg):
    """ Retorna se é texto, foto, áudio, vídeo """
    if "text" in msg:
        return "text"
    elif "photo" in msg:
        return "photo"
    elif "voice" in msg:
        return "voice"
    elif "video" in msg:
        return "video"
    elif "document" in msg:
        return "document"
    elif "audio" in msg:
        return "audio"
    elif "sticker" in msg:
        return "sticker"
    elif "video_note" in msg:
        return "video note"
    else:
        return "outra coisa"


def msg_origin(msg):
    """ Mensagem privada ou mensagem de grupo"""
    return msg["chat"]["type"]


def log(msg):
    """ Loga pra arquivo tudo o que acontecer. """

    if type(msg) is str:
        logging.info(msg)
        print(msg)
    else:
        origin = msg_origin(msg)
        message_type = msg_type(msg)

        log_str = "[" + dt.datetime.fromtimestamp(int(msg["date"])).strftime('%Y-%m-%d %H:%M:%S') + "] "
        log_str += msg["from"]["first_name"] + "(" + str(msg["from"]["id"]) + ")" + " enviou " + message_type + " "

        # Suportando supergroups e outros no log
        if origin == "group" or origin == "supergroup" or origin == "channel":
            log_str += "em \"" + msg["chat"]["title"] + "\" (" + str(msg["chat"]["id"]) + ")"
        elif origin == "private":
            log_str += "em PRIVADO"

        if message_type == "text":
            log_str += ": " + msg["text"]

        #if message_type == "audio":
        #    print(msg)

        logging.info(log_str)
        print(log_str)


def is_sudoer(id):
    return id in config.config["sudoers"]


def is_authorized(msg):

    if msg["from"]["id"] in config.config["authorized_users"]:
        return True
    elif msg["from"]["id"] in config.config["automatically_authorized_users"]:
        return True # Para usuários autorizados automaticamente com base no pertencimento
                    # a grupos. Por ora, só podem ser removidos manualmente
    else:
        # Como eu não sei a ID de todos os indivíduos romanos, resolvi fazer uma query para ver se o caboclo
        # pertence ao chat...

        if isAuthorizedGroupMember(msg["from"]["id"]) is True:
            # Adiciona o indivíduo de vez, economizando essa consulta burra e redundante
            config.config["automatically_authorized_users"].append(msg["from"]["id"])
            config.save_config()
            log(msg["from"]["first_name"] + "(" + str(msg["from"]["id"]) + ")" + \
                " foi adicionado à lista de usuários autorizados. ")
            return True
        else:
            return False


# Verifica se o caboclo faz parte de algum dos chats autorizados
def isAuthorizedGroupMember(user_id):

    for gid in config.config["authorized_chats"]:
        if api.isGroupMember(gid, user_id):
            return True

    return False # Solução estranha, porém funcional e razoavelmente elegante



def msg_matches(msg_text):
    for query, plugin in config.plugins.items():
        pattern = re.compile(query)
        ###print(query)
        match = pattern.search(msg_text)

        if match:
            if query != "^(.*)$":
                log("MATCH! Plugin: " + plugin)

            return plugin, match

    # Hardcodando o plugin multipurpose, para evitar
    # que ele continue causando problemas ao comando !plugin
    return "multipurpose", ""

def on_msg_received(msg):
    """ Callback pra quando uma mensagem é recebida. """

    if is_authorized(msg):
        log(msg)

        tipo = msg_type(msg)
        if tipo == "text":
            plugin_match, matches = msg_matches(msg["text"])

            if plugin_match is not None and matches is not None:
                loaded = importlib.import_module("plugins." + plugin_match)
                try:
                    loaded.on_msg_received(msg, matches)
                except:
                    # Hardcodando aviso de erro. Não serve pra prevenir.
                    exc_type, exc_value, exc_traceback = sys.exc_info()

                    printable = repr(traceback.format_exception(exc_type, exc_value, exc_traceback))

                    for sudoer in config.config["sudoers"]:
                        api.send_message(sudoer, "ME CAPOTARO AQUI PORRA \n\n" + printable)

                    raise
        elif tipo != "text" and tipo != "outra coisa":
            # Garante as estatísticas para o que não for texto
            stt = importlib.import_module("plugins.stats")
            stt.do_statistics(msg)
            # Eu provavelmente deveria fazer um tratamento de erros aqui, mas o
            # plugin em questão é meio que parte do core, então tanto faz...


    else:
        log("Mensagem não autorizada de " + msg["from"]["first_name"] + " (" + str(msg["from"]["id"]) + ")")

def on_old_msg_received(msg):
    """ Callback pra quando uma mensagem ANTIGA é recebida. """

    if is_authorized(msg):
        log(msg)

        tipo = msg_type(msg)

        # Tudo vira estatística...
        if tipo != "outra coisa":
            # Mensagens antigas são postas nas estatísticas
            stt = importlib.import_module("plugins.stats")
            stt.do_statistics(msg)

    else:
        log("Mensagem não autorizada de " + msg["from"]["first_name"] + " (" + str(msg["from"]["id"]) + ")")

def on_msg_edited(msg):
    """ Callback que define o que acontecerá quando uma mensagem for editada. """
    pass


def on_callback_query(msg):
    """ Callback que define o que acontecerá quando um dado de um InlineKeyboardButton for recebido. """
    for plugin in config.config["callback_query_plugins"]:
        loaded = importlib.import_module("plugins." + plugin)
        loaded.on_callback_query(msg)


def start_longpoll():
    """ Inicia longpolling do get_updates. """
    most_recent = 0

    while True:
        updates = api.get_updates(offset=most_recent)

        if updates is not None:
            for update in updates:
                if "message" in update and timegm(gmtime()) - update["message"]["date"] < 10:
                    on_msg_received(update["message"])
                elif "edited_message" in update and timegm(gmtime()) - update["edited_message"]["date"] < 10:
                    on_msg_edited(update["edited_message"])
                elif "callback_query" in update:
                    on_callback_query(update["callback_query"])
                elif "message" in update and timegm(gmtime()) - update["message"]["date"] > 10:
                    # Mensagens antigas poderão ser processadas...
                    # A priori, isto servirá para manter estatísticas mesmo
                    # quanto o bot não conseguir se manter online
                    on_old_msg_received(update["message"])
                else:
                    log("Mensagem desconhecida; ignorando.")

                most_recent = update["update_id"] + 1

        # De improviso, implementa um pedido de WS
        fenestrao = "-1001049208929"
        agora = dt.datetime.now()
        probab = randint(0,9999)

        #if probab == 9999: 
            # send_message(fenestrao, "Já assistiu Diamond is Ubreakable, @Daniele_Harai?")



def start_plugins():
    for match, plugin in config.plugins.items():
        loaded = importlib.import_module("plugins." + plugin)

        if hasattr(loaded, "run"):
            thread = threading.Thread(name=plugin, target=loaded.run)
            thread.start()


def main():
    """ Entry point né porra. """
    log("Iniciando sessão")
    start_plugins()
    start_longpoll()


if __name__ == "__main__":
    main()