telebagger.py

######
#
# Telebagger - Simple Telegram to Discord relay service
#
######

from discord_hooks import Webhook
from telethon import TelegramClient
from telethon.tl.functions.messages import GetDialogsRequest, GetHistoryRequest
from telethon.tl.functions.channels import GetFullChannelRequest
from telethon.tl.functions.updates import GetChannelDifferenceRequest
from telethon.tl.types import UpdateShortMessage, UpdateNewChannelMessage, PeerUser, PeerChat, PeerChannel, InputPeerEmpty, Channel, ChannelMessagesFilter, ChannelMessagesFilterEmpty
from time import sleep
import json
import logging
#logging.basicConfig(level=logging.DEBUG)
logging.basicConfig(level=logging.INFO, format='%(asctime)s %(name)-12s %(levelname)-8s %(message)s', datefmt='%m-%d %H:%M')
logger = logging.getLogger('telebagger')
lastmessage = 0

with open('config.json') as config_file:
    config = json.load(config_file)

try:
    url = config['discord']['url']
    api_id = config['telegram']['api_id']
    api_hash = config['telegram']['api_hash']
    phone = config['telegram']['phone']
    channel_id = config['telegram']['channel_id']
    everyone = config['telegram']['everyone']
    loglevel = config['telegram']['loglevel']
except:
    logger.error('Error processing config file')

logger.setLevel(loglevel)

print('Connecting to Telegram...')

tclient = TelegramClient('session_name', api_id, api_hash, update_workers=4, spawn_read_thread=False)
tclient.connect()
if not tclient.is_user_authorized():
    tclient.send_code_request(phone)
    myself = tclient.sign_in(phone, input('Enter code: '))

def callback(update):
    if type(update) is UpdateNewChannelMessage:
        try:
            logger.debug(update)
            if update.message.to_id.channel_id == channel_id:
                logger.info("Relaying Message from Channel ID: {}".format(update.message.to_id.channel_id))
                if not update.message.message == '':
                    if everyone:
                        msgText = "*{}*: @everyone {}".format(channel_name, update.message.message)
                    else:
                        msgText = "*{}*: {}".format(channel_name, update.message.message)
                    msg = Webhook(url,msg=msgText)
                    msg.post()
                else:
                    logger.debug('ignoring empty message: {}'.format(update.message))
            else:
                logger.info("Ignoring Message from Channel ID: {}".format(update.message.to_id.channel_id))
        except:
            logger.debug('no message')

#msg = Webhook(url,msg="Telebagger ready to bag yo telegrams")
#msg.post()

tclient.add_update_handler(callback)

last_date = None
chunk_size = 20
result = tclient(GetDialogsRequest(
                 offset_date=last_date,
                 offset_id=0,
                 offset_peer=InputPeerEmpty(),
                 limit=chunk_size
             ))
print("\nAvailable Channels:")
for p in result.chats:
    if type(p) is Channel:
        print(str(p.id)+": "+p.title)
        if p.id == channel_id:
            channel_name = p.title

try:
    logger.info("\nListening for messages from channel '{}' with ID '{}'".format(channel_name,channel_id))
except:
    logger.error("Whoops! Couldn't find channel ID '{}'".format(channel_id))

full_channel = tclient(GetHistoryRequest(peer=tclient.get_input_entity(PeerChannel(channel_id)),
                                            offset_date=last_date,
                                            offset_id=0,
                                            add_offset=0,
                                            limit=10,
                                            max_id=0,
                                            min_id=0
                                        ))
full_channel.messages.reverse()
logger.info("\nLast 10 Messages:\n")
for m in full_channel.messages:
    datetime = m.date.strftime('%Y-%m-%d %H:%M:%S')
    logger.info(datetime+" "+str(m.id)+": "+m.message)
    if m.id > lastmessage:
        lastmessage = m.id

tclient.idle()
tclient.disconnect()