def start(self):

super().start()

try:

self.send_message(chat_id=int(Config.OWNER), text="<b>Bot Started!</b>")

except Exception as err:

LOGGER.error("Boot alert failed! Please start bot in PM")

return LOGGER.info("Bot Started!")

def stop(self):

super().stop()

name="merge-bot",

api_hash=Config.API_HASH,

api_id=int(Config.TELEGRAM_API),

bot_token=Config.BOT_TOKEN,

workers=300,

plugins=dict(root="plugins"),

app_version="5.0+yash-mergebot",

user = UserSettings(m.from_user.id, m.from_user.first_name)

if user.banned:

await m.reply_text(text=f"**Banned User Detected!**\n 🛡️ Unfortunately you can't use me\n\nContact: 🈲 @{Config.OWNER_USERNAME}", quote=True)

return

if user.user_id == int(Config.OWNER):

user.allowed = True

if user.allowed:

await m.reply_text(text=f"**Dont Spam**\n ⚡ You can use me!!", quote=True)

else:

try:

passwd = m.text.split(" ", 1)[1]

except:

await m.reply_text("**Command:**\n `/login <password>`\n\n**Usage:**\n `password`: Get the password from owner",quote=True,parse_mode=enums.parse_mode.ParseMode.MARKDOWN)

passwd = passwd.strip()

if passwd == Config.PASSWORD:

user.allowed = True

await m.reply_text(

text=f"**Login passed ✅,**\n ⚡ Now you can use me!!", quote=True

)

else:

await m.reply_text(

text=f"**Login failed ❌,**\n 🛡️ Unfortunately you can't use me\n\nContact: 🈲 @{Config.OWNER_USERNAME}",

quote=True,

)

user.set()

del user

currentTime = get_readable_time(time.time() - botStartTime)

total, used, free = shutil.disk_usage(".")

total = get_readable_file_size(total)

used = get_readable_file_size(used)

free = get_readable_file_size(free)

sent = get_readable_file_size(psutil.net_io_counters().bytes_sent)

recv = get_readable_file_size(psutil.net_io_counters().bytes_recv)

cpuUsage = psutil.cpu_percent(interval=0.5)

memory = psutil.virtual_memory().percent

disk = psutil.disk_usage("/").percent

stats = (

f"<b>╭「 💠 BOT STATISTICS 」</b>\n"

f"<b>│</b>\n"

f"<b>├⏳ Bot Uptime : {currentTime}</b>\n"

f"<b>├💾 Total Disk Space : {total}</b>\n"

f"<b>├📀 Total Used Space : {used}</b>\n"

f"<b>├💿 Total Free Space : {free}</b>\n"

f"<b>├🔺 Total Upload : {sent}</b>\n"

f"<b>├🔻 Total Download : {recv}</b>\n"

f"<b>├🖥 CPU : {cpuUsage}%</b>\n"

f"<b>├⚙️ RAM : {memory}%</b>\n"

f"<b>╰💿 DISK : {disk}%</b>"

)

msg = m.reply_to_message

userList = await database.broadcast()

len = userList.collection.count_documents({})

status = await m.reply_text(text=BROADCAST_MSG.format(str(len), "0"), quote=True)

success = 0

for i in range(len):

try:

uid = userList[i]["_id"]

if uid != int(Config.OWNER):

await msg.copy(chat_id=uid)

success = i + 1

await status.edit_text(text=BROADCAST_MSG.format(len, success))

LOGGER.info(f"Message sent to {userList[i]['name']} ")

except FloodWait as e:

await asyncio.sleep(e.x)

await msg.copy(chat_id=userList[i]["_id"])

LOGGER.info(f"Message sent to {userList[i]['name']} ")

except InputUserDeactivated:

await database.deleteUser(userList[i]["_id"])

LOGGER.info(f"{userList[i]['_id']} - {userList[i]['name']} : deactivated\n")

except UserIsBlocked:

await database.deleteUser(userList[i]["_id"])

LOGGER.info(

f"{userList[i]['_id']} - {userList[i]['name']} : blocked the bot\n"

)

except PeerIdInvalid:

await database.deleteUser(userList[i]["_id"])

LOGGER.info(

f"{userList[i]['_id']} - {userList[i]['name']} : user id invalid\n"

)

except Exception as err:

LOGGER.warning(f"{err}\n")

await asyncio.sleep(3)

await status.edit_text(

text=BROADCAST_MSG.format(len, success)

+ f"**Failed: {str(len-success)}**\n\n__🤓 Broadcast completed sucessfully__",

user = UserSettings(m.from_user.id, m.from_user.first_name)

if m.from_user.id != int(Config.OWNER):

if user.allowed is False:

res = await m.reply_text(

text=f"Hi **{m.from_user.first_name}**\n\n 🛡️ Unfortunately you can't use me\n\n**Contact: 🈲 @{Config.OWNER_USERNAME}** ",

quote=True,

)

return

else:

user.allowed = True

user.set()

res = await m.reply_text(

text=f"Hi **{m.from_user.first_name}**\n\n ⚡ I am a file/video merger bot\n\n😎 I can merge Telegram files!, And upload it to telegram\n\n**Owner: 🈲 @{Config.OWNER_USERNAME}** ",

quote=True,

)

user_id = m.from_user.id

user = UserSettings(user_id, m.from_user.first_name)

if user_id != int(Config.OWNER):

if user.allowed is False:

res = await m.reply_text(

text=f"Hi **{m.from_user.first_name}**\n\n 🛡️ Unfortunately you can't use me\n\n**Contact: 🈲 @{Config.OWNER_USERNAME}** ",

quote=True,

)

return

if user.merge_mode == 4: # extract_mode

return

input_ = f"downloads/{str(user_id)}/input.txt"

if os.path.exists(input_):

await m.reply_text("Sorry Bro,\nAlready One process in Progress!\nDon't Spam.")

return

media = m.video or m.document or m.audio

if media.file_name is None:

await m.reply_text("File Not Found")

return

currentFileNameExt = media.file_name.rsplit(sep=".")[-1].lower()

if currentFileNameExt in "conf":

await m.reply_text(

text="**💾 Config file found, Do you want to save it?**",

reply_markup=InlineKeyboardMarkup(

[

[

InlineKeyboardButton("✅ Yes", callback_data=f"rclone_save"),

InlineKeyboardButton("❌ No", callback_data="rclone_discard"),

]

]

),

quote=True,

)

return

# if MERGE_MODE.get(user_id) is None:

# userMergeMode = database.getUserMergeSettings(user_id)

# if userMergeMode is not None:

# MERGE_MODE[user_id] = userMergeMode

# else:

# database.setUserMergeMode(uid=user_id, mode=1)

# MERGE_MODE[user_id] = 1

if user.merge_mode == 1:

if queueDB.get(user_id, None) is None:

formatDB.update({user_id: currentFileNameExt})

if formatDB.get(

user_id, None

) is not None and currentFileNameExt != formatDB.get(user_id):

await m.reply_text(

f"First you sent a {formatDB.get(user_id).upper()} file so now send only that type of file.",

quote=True,

)

return

if currentFileNameExt not in VIDEO_EXTENSIONS:

await m.reply_text(

"This Video Format not Allowed!\nOnly send MP4 or MKV or WEBM.",

quote=True,

)

return

editable = await m.reply_text("Please Wait ...", quote=True)

MessageText = "Okay,\nNow Send Me Next Video or Press **Merge Now** Button!"

if queueDB.get(user_id, None) is None:

queueDB.update({user_id: {"videos": [], "subtitles": [], "audios": []}})

if (

len(queueDB.get(user_id)["videos"]) >= 0

and len(queueDB.get(user_id)["videos"]) < 10

):

queueDB.get(user_id)["videos"].append(m.id)

queueDB.get(m.from_user.id)["subtitles"].append(None)

# LOGGER.info(

# queueDB.get(user_id)["videos"], queueDB.get(m.from_user.id)["subtitles"]

# )

if len(queueDB.get(user_id)["videos"]) == 1:

reply_ = await editable.edit(

"**Send me some more videos to merge them into single file**",

reply_markup=InlineKeyboardMarkup(

bMaker.makebuttons(["Cancel"], ["cancel"])

),

)

replyDB.update({user_id: reply_.id})

return

if queueDB.get(user_id, None)["videos"] is None:

formatDB.update({user_id: currentFileNameExt})

if replyDB.get(user_id, None) is not None:

await c.delete_messages(

chat_id=m.chat.id, message_ids=replyDB.get(user_id)

)

if len(queueDB.get(user_id)["videos"]) == 10:

MessageText = "Okay, Now Just Press **Merge Now** Button Plox!"

markup = await makeButtons(c, m, queueDB)

reply_ = await editable.edit(

text=MessageText, reply_markup=InlineKeyboardMarkup(markup)

)

replyDB.update({user_id: reply_.id})

elif len(queueDB.get(user_id)["videos"]) > 10:

markup = await makeButtons(c, m, queueDB)

await editable.text(

"Max 10 videos allowed", reply_markup=InlineKeyboardMarkup(markup)

)

elif user.merge_mode == 2:

editable = await m.reply_text("Please Wait ...", quote=True)

MessageText = (

"Okay,\nNow Send Me Some More <u>Audios</u> or Press **Merge Now** Button!"

)

if queueDB.get(user_id, None) is None:

queueDB.update({user_id: {"videos": [], "subtitles": [], "audios": []}})

if len(queueDB.get(user_id)["videos"]) == 0:

queueDB.get(user_id)["videos"].append(m.id)

# if len(queueDB.get(user_id)["videos"])==1:

reply_ = await editable.edit(

text="Now, Send all the audios you want to merge",

reply_markup=InlineKeyboardMarkup(

bMaker.makebuttons(["Cancel"], ["cancel"])

),

)

replyDB.update({user_id: reply_.id})

return

elif (

len(queueDB.get(user_id)["videos"]) >= 1

and currentFileNameExt in AUDIO_EXTENSIONS

):

queueDB.get(user_id)["audios"].append(m.id)

if replyDB.get(user_id, None) is not None:

await c.delete_messages(

chat_id=m.chat.id, message_ids=replyDB.get(user_id)

)

markup = await makeButtons(c, m, queueDB)

reply_ = await editable.edit(

text=MessageText, reply_markup=InlineKeyboardMarkup(markup)

)

replyDB.update({user_id: reply_.id})

else:

await m.reply("This Filetype is not valid")

return

elif user.merge_mode == 3:

editable = await m.reply_text("Please Wait ...", quote=True)

MessageText = "Okay,\nNow Send Me Some More <u>Subtitles</u> or Press **Merge Now** Button!"

if queueDB.get(user_id, None) is None:

queueDB.update({user_id: {"videos": [], "subtitles": [], "audios": []}})

if len(queueDB.get(user_id)["videos"]) == 0:

queueDB.get(user_id)["videos"].append(m.id)

# if len(queueDB.get(user_id)["videos"])==1:

reply_ = await editable.edit(

text="Now, Send all the subtitles you want to merge",

reply_markup=InlineKeyboardMarkup(

bMaker.makebuttons(["Cancel"], ["cancel"])

),

)

replyDB.update({user_id: reply_.id})

return

elif (

len(queueDB.get(user_id)["videos"]) >= 1

and currentFileNameExt in SUBTITLE_EXTENSIONS

):

queueDB.get(user_id)["subtitles"].append(m.id)

if replyDB.get(user_id, None) is not None:

await c.delete_messages(

chat_id=m.chat.id, message_ids=replyDB.get(user_id)

)

markup = await makeButtons(c, m, queueDB)

reply_ = await editable.edit(

text=MessageText, reply_markup=InlineKeyboardMarkup(markup)

)

replyDB.update({user_id: reply_.id})

else:

await m.reply("This Filetype is not valid")

user = UserSettings(m.chat.id, m.from_user.first_name)

# if m.from_user.id != int(Config.OWNER):

if not user.allowed:

res = await m.reply_text(

text=f"Hi **{m.from_user.first_name}**\n\n 🛡️ Unfortunately you can't use me\n\n**Contact: 🈲 @{Config.OWNER_USERNAME}** ",

quote=True,

)

del user

return

thumbnail = m.photo.file_id

msg = await m.reply_text("Saving Thumbnail. . . .", quote=True)

user.thumbnail = thumbnail

user.set()

# await database.saveThumb(m.from_user.id, thumbnail)

LOCATION = f"downloads/{m.from_user.id}_thumb.jpg"

await c.download_media(message=m, file_name=LOCATION)

await msg.edit_text(text="✅ Custom Thumbnail Saved!")

user = UserSettings(uid=m.from_user.id, name=m.from_user.first_name)

if not user.allowed:

return

if user.merge_mode == 4:

if m.reply_to_message is None:

await m.reply(text="Reply /extract to a video or document file")

return

rmess = m.reply_to_message

if rmess.video or rmess.document:

media = rmess.video or rmess.document

mid=rmess.id

file_name = media.file_name

if file_name is None:

await m.reply("File name not found; goto @yashoswalyo")

return

markup = bMaker.makebuttons(

set1=["Audio", "Subtitle", "Cancel"],

set2=[f"extract_audio_{mid}", f"extract_subtitle_{mid}", 'cancel'],

isCallback=True,

rows=2,

)

await m.reply(

text="Choose from below what you want to extract?",

quote=True,

reply_markup=InlineKeyboardMarkup(markup),

)

else:

await m.reply(

text="Change settings and set mode to extract\nthen use /extract command"

await m.reply_text(

text="""**Follow These Steps:

5) Select rename if you want to give custom file name else press default**""",

quote=True,

reply_markup=InlineKeyboardMarkup(

[[InlineKeyboardButton("Close 🔐", callback_data="close")]]

),

await m.reply_text(

text="""

""",

quote=True,

reply_markup=InlineKeyboardMarkup(

[

[InlineKeyboardButton("👨‍💻Developer👨‍💻", url="https://t.me/yashoswalyo")],

[

InlineKeyboardButton(

"🏘Source Code🏘", url="https://github.com/yashoswalyo/MERGE-BOT"

),

InlineKeyboardButton(

"🤔Deployed By🤔", url=f"https://t.me/{Config.OWNER_USERNAME}"

),

],

[InlineKeyboardButton("Close 🔐", callback_data="close")],

]

),

if m.reply_to_message:

if m.reply_to_message.photo:

await photo_handler(c, m.reply_to_message)

else:

await m.reply(text="Please reply to a valid photo")

else:

await m.reply(text="Please reply to a message")

try:

user = UserSettings(m.from_user.id, m.from_user.first_name)

thumb_id = user.thumbnail

LOCATION = f"downloads/{str(m.from_user.id)}_thumb.jpg"

if os.path.exists(LOCATION):

await m.reply_photo(

photo=LOCATION, caption="🖼️ Your custom thumbnail", quote=True

)

elif thumb_id is not None :

await c.download_media(message=str(thumb_id), file_name=LOCATION)

await m.reply_photo(

photo=LOCATION, caption="🖼️ Your custom thumbnail", quote=True

)

else:

await m.reply_text(text="❌ Custom thumbnail not found", quote=True)

del user

except Exception as err:

LOGGER.info(err)

try:

user = UserSettings(m.from_user.id, m.from_user.first_name)

user.thumbnail = None

user.set()

if os.path.exists(f"downloads/{str(m.from_user.id)}"):

os.remove(f"downloads/{str(m.from_user.id)}")

await m.reply_text("✅ Deleted Sucessfully", quote=True)

del user

else: raise Exception("Thumbnail file not found")

except Exception as err:

incoming=m.text.split(' ')[0]

if incoming == '/ban':

if m.from_user.id == int(Config.OWNER):

try:

abuser_id = int(m.text.split(" ")[1])

if abuser_id == int(Config.OWNER):

await m.reply_text("I can't ban you master,\nPlease don't abandon me. ",quote=True)

else:

try:

user_obj: User = await c.get_users(abuser_id)

udata = UserSettings(uid=abuser_id,name=user_obj.first_name)

udata.banned=True

udata.allowed=False

udata.set()

await m.reply_text(f"Pooof, {user_obj.first_name} has been **BANNED**",quote=True)

acknowledgement = f"""

try:

await c.send_message(

chat_id=abuser_id,

text=acknowledgement

)

except Exception as e:

await m.reply_text(f"An error occured while sending acknowledgement\n\n`{e}`",quote=True)

LOGGER.error(e)

except Exception as e:

LOGGER.error(e)

except:

await m.reply_text("**Command:**\n `/ban <user_id>`\n\n**Usage:**\n `user_id`: User ID of the user",quote=True,parse_mode=enums.parse_mode.ParseMode.MARKDOWN)

else:

await m.reply_text("**(Only for __OWNER__)\nCommand:**\n `/ban <user_id>`\n\n**Usage:**\n `user_id`: User ID of the user",quote=True,parse_mode=enums.parse_mode.ParseMode.MARKDOWN)

return

elif incoming == '/unban':

if m.from_user.id == int(Config.OWNER):

try:

abuser_id = int(m.text.split(" ")[1])

if abuser_id == int(Config.OWNER):

await m.reply_text("I can't ban you master,\nPlease don't abandon me. ",quote=True)

else:

try:

user_obj: User = await c.get_users(abuser_id)

udata = UserSettings(uid=abuser_id,name=user_obj.first_name)

udata.banned=False

udata.allowed=True

udata.set()

await m.reply_text(f"Pooof, {user_obj.first_name} has been **UN_BANNED**",quote=True)

release_notice = f"""

try:

await c.send_message(

chat_id=abuser_id,

text=release_notice

)

except Exception as e:

await m.reply_text(f"An error occured while sending release notice\n\n`{e}`",quote=True)

LOGGER.error(e)

except Exception as e:

LOGGER.error(e)

except:

await m.reply_text("**Command:**\n `/unban <user_id>`\n\n**Usage:**\n `user_id`: User ID of the user",quote=True,parse_mode=enums.parse_mode.ParseMode.MARKDOWN)

else:

await m.reply_text("**(Only for __OWNER__)\nCommand:**\n `/unban <user_id>`\n\n**Usage:**\n `user_id`: User ID of the user",quote=True,parse_mode=enums.parse_mode.ParseMode.MARKDOWN)

try:

markup = await makeButtons(c, cb.message, queueDB)

await cb.message.edit(

text="Okay,\nNow Send Me Next Video or Press **Merge Now** Button!",

reply_markup=InlineKeyboardMarkup(markup),

)

except ValueError:

await cb.message.edit("Send Some more videos")

try:

shutil.rmtree(root)

except Exception as e:

markup = []

user = UserSettings(m.chat.id, m.chat.first_name)

if user.merge_mode == 1:

for i in await bot.get_messages(

chat_id=m.chat.id, message_ids=db.get(m.chat.id)["videos"]

):

media = i.video or i.document or None

if media is None:

continue

else:

markup.append(

[

InlineKeyboardButton(

f"{media.file_name}",

callback_data=f"showFileName_{i.id}",

)

]

)

elif user.merge_mode == 2:

msgs: list[Message] = await bot.get_messages(

chat_id=m.chat.id, message_ids=db.get(m.chat.id)["audios"]

)

msgs.insert(

0,

await bot.get_messages(

chat_id=m.chat.id, message_ids=db.get(m.chat.id)["videos"][0]

),

)

for i in msgs:

media = i.audio or i.document or i.video or None

if media is None:

continue

else:

markup.append(

[

InlineKeyboardButton(

f"{media.file_name}",

callback_data=f"tryotherbutton",

)

]

)

elif user.merge_mode == 3:

msgs: list[Message] = await bot.get_messages(

chat_id=m.chat.id, message_ids=db.get(m.chat.id)["subtitles"]

)

msgs.insert(

0,

await bot.get_messages(

chat_id=m.chat.id, message_ids=db.get(m.chat.id)["videos"][0]

),

)

for i in msgs:

media = i.video or i.document or None

if media is None:

continue

else:

markup.append(

[

InlineKeyboardButton(

f"{media.file_name}",

callback_data=f"tryotherbutton",

)

]

)

markup.append([InlineKeyboardButton("🔗 Merge Now", callback_data="merge")])

markup.append([InlineKeyboardButton("💥 Clear Files", callback_data="cancel")])