82 lines
3.1 KiB
Python
82 lines
3.1 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
from dotenv import load_dotenv
|
|
import os
|
|
import logging
|
|
import asyncio
|
|
|
|
from tgbridge.forwarder import TelegramForwarder
|
|
from tgbridge.discord_to_telegram import DiscordToTelegramForwarder
|
|
|
|
from twitterbridge.rss import TwitterRSSForwarder
|
|
|
|
load_dotenv('credentials.env')
|
|
|
|
intents = discord.Intents.default()
|
|
intents.message_content = True
|
|
intents.members = True
|
|
|
|
bot = commands.Bot(command_prefix='!', intents=intents, help_command=None)
|
|
|
|
# Telegram → Discord
|
|
forwarder = TelegramForwarder(token=os.getenv('TELEGRAM_BOT_TOKEN'))
|
|
if os.getenv('TELEGRAM_GROUP_ID') and os.getenv('DISCORD_WEBHOOK_URL'):
|
|
forwarder.add_route(int(os.getenv('TELEGRAM_GROUP_ID')), os.getenv('DISCORD_WEBHOOK_URL'))
|
|
if os.getenv('TELEGRAM_GROUP_ID_SHITPOST') and os.getenv('DISCORD_WEBHOOK_URL_SHITPOST'):
|
|
forwarder.add_route(int(os.getenv('TELEGRAM_GROUP_ID_SHITPOST')), os.getenv('DISCORD_WEBHOOK_URL_SHITPOST'))
|
|
|
|
# Discord → Telegram (reverse bridge)
|
|
d_to_tg = DiscordToTelegramForwarder(bot=bot, tg_token=os.getenv('TELEGRAM_BOT_TOKEN'))
|
|
if os.getenv('DISCORD_CHANNEL_ID') and os.getenv('TELEGRAM_GROUP_ID'):
|
|
d_to_tg.add_route(int(os.getenv('DISCORD_CHANNEL_ID')), int(os.getenv('TELEGRAM_GROUP_ID')))
|
|
if os.getenv('DISCORD_CHANNEL_ID_SHITPOST') and os.getenv('TELEGRAM_GROUP_ID_SHITPOST'):
|
|
d_to_tg.add_route(int(os.getenv('DISCORD_CHANNEL_ID_SHITPOST')), int(os.getenv('TELEGRAM_GROUP_ID_SHITPOST')))
|
|
|
|
twitter = TwitterRSSForwarder(
|
|
rss_url=os.getenv('TWITTER_RSS_URL'),
|
|
webhook_url=os.getenv('TWITTER_DISCORD_WEBHOOK'),
|
|
)
|
|
|
|
twitter_mao = TwitterRSSForwarder(
|
|
rss_url=os.getenv('TWITTER_RSS_URL'),
|
|
webhook_url=os.getenv('TWITTER_DISCORD_WEBHOOK_MAO'),
|
|
guid_file='data/last_tweet_mao.txt',
|
|
)
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
print(f'[+] {bot.user} is online!')
|
|
asyncio.create_task(forwarder.start())
|
|
asyncio.create_task(twitter.start())
|
|
asyncio.create_task(twitter_mao.start())
|
|
|
|
async def main():
|
|
async with bot:
|
|
await bot.add_cog(d_to_tg)
|
|
await bot.load_extension('cogs.general')
|
|
await bot.load_extension('cogs.chance')
|
|
await bot.load_extension('cogs.gifs')
|
|
await bot.load_extension('cogs.copypasta')
|
|
await bot.load_extension('cogs.sysinfo')
|
|
await bot.load_extension('music.music_wrapper')
|
|
await bot.load_extension('cogs.statchannel')
|
|
await bot.load_extension('cogs.ffmpeg_things')
|
|
await bot.load_extension('cogs.archive')
|
|
await bot.load_extension('cogs.roles')
|
|
await bot.load_extension('cogs.gitpull')
|
|
await bot.load_extension('cogs.currency_converter')
|
|
await bot.load_extension('ai.ai_roleplay')
|
|
await bot.load_extension('twitterbridge.leaks_cmd')
|
|
await bot.load_extension("rio.ssh_connection_rio")
|
|
await bot.load_extension('blue_archive.gacha')
|
|
|
|
try:
|
|
await bot.start(os.getenv('DISCORD_TOKEN'))
|
|
finally:
|
|
await forwarder.stop()
|
|
twitter.stop()
|
|
twitter_mao.stop()
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
logging.getLogger('httpx').setLevel(logging.WARNING)
|
|
asyncio.run(main()) |