admin.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836
  1. """
  2. MIT License
  3. Copyright (c) 2021 TheHamkerCat
  4. Permission is hereby granted, free of charge, to any person obtaining a copy
  5. of this software and associated documentation files (the "Software"), to deal
  6. in the Software without restriction, including without limitation the rights
  7. to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  8. copies of the Software, and to permit persons to whom the Software is
  9. furnished to do so, subject to the following conditions:
  10. The above copyright notice and this permission notice shall be included in all
  11. copies or substantial portions of the Software.
  12. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  13. IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  14. FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  15. AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  16. LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  17. OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  18. SOFTWARE.
  19. """
  20. import asyncio
  21. import re
  22. from time import time
  23. from pyrogram.errors import FloodWait
  24. from pyrogram import filters
  25. from pyrogram.types import (
  26. CallbackQuery,
  27. ChatMemberUpdated,
  28. ChatPermissions,
  29. Message,
  30. )
  31. from wbb import BOT_ID, SUDOERS, app, log
  32. from wbb.core.decorators.errors import capture_err
  33. from wbb.core.keyboard import ikb
  34. from wbb.utils.dbfunctions import (
  35. add_warn,
  36. get_warn,
  37. int_to_alpha,
  38. remove_warns,
  39. save_filter,
  40. )
  41. from wbb.utils.functions import (
  42. extract_user,
  43. extract_user_and_reason,
  44. time_converter,
  45. )
  46. __MODULE__ = "Admin"
  47. __HELP__ = """/ban - Ban A User
  48. /dban - Delete the replied message banning its sender
  49. /tban - Ban A User For Specific Time
  50. /unban - Unban A User
  51. /listban - Ban a user from groups listed in a message
  52. /listunban - Unban a user from groups listed in a message
  53. /warn - Warn A User
  54. /dwarn - Delete the replied message warning its sender
  55. /rmwarns - Remove All Warning of A User
  56. /warns - Show Warning Of A User
  57. /kick - Kick A User
  58. /dkick - Delete the replied message kicking its sender
  59. /purge - Purge Messages
  60. /purge [n] - Purge "n" number of messages from replied message
  61. /del - Delete Replied Message
  62. /promote - Promote A Member
  63. /fullpromote - Promote A Member With All Rights
  64. /demote - Demote A Member
  65. /pin - Pin A Message
  66. /mute - Mute A User
  67. /tmute - Mute A User For Specific Time
  68. /unmute - Unmute A User
  69. /ban_ghosts - Ban Deleted Accounts
  70. /report | @admins | @admin - Report A Message To Admins.
  71. /invite - Send Group/SuperGroup Invite Link."""
  72. async def member_permissions(chat_id: int, user_id: int):
  73. perms = []
  74. try:
  75. member = await app.get_chat_member(chat_id, user_id)
  76. except Exception:
  77. return []
  78. if member.can_post_messages:
  79. perms.append("can_post_messages")
  80. if member.can_edit_messages:
  81. perms.append("can_edit_messages")
  82. if member.can_delete_messages:
  83. perms.append("can_delete_messages")
  84. if member.can_restrict_members:
  85. perms.append("can_restrict_members")
  86. if member.can_promote_members:
  87. perms.append("can_promote_members")
  88. if member.can_change_info:
  89. perms.append("can_change_info")
  90. if member.can_invite_users:
  91. perms.append("can_invite_users")
  92. if member.can_pin_messages:
  93. perms.append("can_pin_messages")
  94. if member.can_manage_voice_chats:
  95. perms.append("can_manage_voice_chats")
  96. return perms
  97. from wbb.core.decorators.permissions import adminsOnly
  98. admins_in_chat = {}
  99. async def list_admins(chat_id: int):
  100. global admins_in_chat
  101. if chat_id in admins_in_chat:
  102. interval = time() - admins_in_chat[chat_id]["last_updated_at"]
  103. if interval < 3600:
  104. return admins_in_chat[chat_id]["data"]
  105. admins_in_chat[chat_id] = {
  106. "last_updated_at": time(),
  107. "data": [
  108. member.user.id
  109. async for member in app.iter_chat_members(
  110. chat_id, filter="administrators"
  111. )
  112. ],
  113. }
  114. return admins_in_chat[chat_id]["data"]
  115. # Admin cache reload
  116. @app.on_chat_member_updated()
  117. async def admin_cache_func(_, cmu: ChatMemberUpdated):
  118. if cmu.old_chat_member and cmu.old_chat_member.promoted_by:
  119. admins_in_chat[cmu.chat.id] = {
  120. "last_updated_at": time(),
  121. "data": [
  122. member.user.id
  123. async for member in app.iter_chat_members(
  124. cmu.chat.id, filter="administrators"
  125. )
  126. ],
  127. }
  128. log.info(f"Updated admin cache for {cmu.chat.id} [{cmu.chat.title}]")
  129. # Purge Messages
  130. @app.on_message(filters.command("purge") & ~filters.edited & ~filters.private)
  131. @adminsOnly("can_delete_messages")
  132. async def purgeFunc(_, message: Message):
  133. repliedmsg = message.reply_to_message
  134. await message.delete()
  135. if not repliedmsg:
  136. return await message.reply_text("Reply to a message to purge from.")
  137. cmd = message.command
  138. if len(cmd) > 1 and cmd[1].isdigit():
  139. purge_to = repliedmsg.message_id + int(cmd[1])
  140. if purge_to > message.message_id:
  141. purge_to = message.message_id
  142. else:
  143. purge_to = message.message_id
  144. chat_id = message.chat.id
  145. message_ids = []
  146. for message_id in range(
  147. repliedmsg.message_id,
  148. purge_to,
  149. ):
  150. message_ids.append(message_id)
  151. # Max message deletion limit is 100
  152. if len(message_ids) == 100:
  153. await app.delete_messages(
  154. chat_id=chat_id,
  155. message_ids=message_ids,
  156. revoke=True, # For both sides
  157. )
  158. # To delete more than 100 messages, start again
  159. message_ids = []
  160. # Delete if any messages left
  161. if len(message_ids) > 0:
  162. await app.delete_messages(
  163. chat_id=chat_id,
  164. message_ids=message_ids,
  165. revoke=True,
  166. )
  167. # Kick members
  168. @app.on_message(
  169. filters.command(["kick", "dkick"]) & ~filters.edited & ~filters.private
  170. )
  171. @adminsOnly("can_restrict_members")
  172. async def kickFunc(_, message: Message):
  173. user_id, reason = await extract_user_and_reason(message)
  174. if not user_id:
  175. return await message.reply_text("I can't find that user.")
  176. if user_id == BOT_ID:
  177. return await message.reply_text(
  178. "I can't kick myself, i can leave if you want."
  179. )
  180. if user_id in SUDOERS:
  181. return await message.reply_text("You Wanna Kick The Elevated One?")
  182. if user_id in (await list_admins(message.chat.id)):
  183. return await message.reply_text(
  184. "I can't kick an admin, You know the rules, so do i."
  185. )
  186. mention = (await app.get_users(user_id)).mention
  187. msg = f"""
  188. **Kicked User:** {mention}
  189. **Kicked By:** {message.from_user.mention if message.from_user else 'Anon'}
  190. **Reason:** {reason or 'No Reason Provided.'}"""
  191. if message.command[0][0] == "d":
  192. await message.reply_to_message.delete()
  193. await message.chat.ban_member(user_id)
  194. await message.reply_text(msg)
  195. await asyncio.sleep(1)
  196. await message.chat.unban_member(user_id)
  197. # Ban members
  198. @app.on_message(
  199. filters.command(["ban", "dban", "tban"])
  200. & ~filters.edited
  201. & ~filters.private
  202. )
  203. @adminsOnly("can_restrict_members")
  204. async def banFunc(_, message: Message):
  205. user_id, reason = await extract_user_and_reason(message, sender_chat=True)
  206. if not user_id:
  207. return await message.reply_text("I can't find that user.")
  208. if user_id == BOT_ID:
  209. return await message.reply_text(
  210. "I can't ban myself, i can leave if you want."
  211. )
  212. if user_id in SUDOERS:
  213. return await message.reply_text(
  214. "You Wanna Ban The Elevated One?, RECONSIDER!"
  215. )
  216. if user_id in (await list_admins(message.chat.id)):
  217. return await message.reply_text(
  218. "I can't ban an admin, You know the rules, so do i."
  219. )
  220. try:
  221. mention = (await app.get_users(user_id)).mention
  222. except IndexError:
  223. mention = (
  224. message.reply_to_message.sender_chat.title
  225. if message.reply_to_message
  226. else "Anon"
  227. )
  228. msg = (
  229. f"**Banned User:** {mention}\n"
  230. f"**Banned By:** {message.from_user.mention if message.from_user else 'Anon'}\n"
  231. )
  232. if message.command[0][0] == "d":
  233. await message.reply_to_message.delete()
  234. if message.command[0] == "tban":
  235. split = reason.split(None, 1)
  236. time_value = split[0]
  237. temp_reason = split[1] if len(split) > 1 else ""
  238. temp_ban = await time_converter(message, time_value)
  239. msg += f"**Banned For:** {time_value}\n"
  240. if temp_reason:
  241. msg += f"**Reason:** {temp_reason}"
  242. try:
  243. if len(time_value[:-1]) < 3:
  244. await message.chat.ban_member(user_id, until_date=temp_ban)
  245. await message.reply_text(msg)
  246. else:
  247. await message.reply_text("You can't use more than 99")
  248. except AttributeError:
  249. pass
  250. return
  251. if reason:
  252. msg += f"**Reason:** {reason}"
  253. await message.chat.ban_member(user_id)
  254. await message.reply_text(msg)
  255. # Unban members
  256. @app.on_message(filters.command("unban") & ~filters.edited & ~filters.private)
  257. @adminsOnly("can_restrict_members")
  258. async def unban_func(_, message: Message):
  259. # we don't need reasons for unban, also, we
  260. # don't need to get "text_mention" entity, because
  261. # normal users won't get text_mention if the user
  262. # they want to unban is not in the group.
  263. reply = message.reply_to_message
  264. if reply and reply.sender_chat and reply.sender_chat != message.chat.id:
  265. return await message.reply_text("You cannot unban a channel")
  266. if len(message.command) == 2:
  267. user = message.text.split(None, 1)[1]
  268. elif len(message.command) == 1 and reply:
  269. user = message.reply_to_message.from_user.id
  270. else:
  271. return await message.reply_text(
  272. "Provide a username or reply to a user's message to unban."
  273. )
  274. await message.chat.unban_member(user)
  275. umention = (await app.get_users(user)).mention
  276. await message.reply_text(f"Unbanned! {umention}")
  277. # Ban users listed in a message
  278. @app.on_message(
  279. SUDOERS & filters.command("listban") & ~filters.edited & ~filters.private
  280. )
  281. async def list_ban_(c, message: Message):
  282. userid, msglink_reason = await extract_user_and_reason(message)
  283. if not userid or not msglink_reason:
  284. return await message.reply_text(
  285. "Provide a userid/username along with message link and reason to list-ban"
  286. )
  287. if (
  288. len(msglink_reason.split(" ")) == 1
  289. ): # message link included with the reason
  290. return await message.reply_text(
  291. "You must provide a reason to list-ban"
  292. )
  293. # seperate messge link from reason
  294. lreason = msglink_reason.split()
  295. messagelink, reason = lreason[0], " ".join(lreason[1:])
  296. if not re.search(
  297. r"(https?://)?t(elegram)?\.me/\w+/\d+", messagelink
  298. ): # validate link
  299. return await message.reply_text("Invalid message link provided")
  300. if userid == BOT_ID:
  301. return await message.reply_text("I can't ban myself.")
  302. if userid in SUDOERS:
  303. return await message.reply_text(
  304. "You Wanna Ban The Elevated One?, RECONSIDER!"
  305. )
  306. splitted = messagelink.split("/")
  307. uname, mid = splitted[-2], int(splitted[-1])
  308. m = await message.reply_text(
  309. "`Banning User from multiple groups. \
  310. This may take some time`"
  311. )
  312. try:
  313. msgtext = (await app.get_messages(uname, mid)).text
  314. gusernames = re.findall("@\w+", msgtext)
  315. except:
  316. return await m.edit_text("Could not get group usernames")
  317. count = 0
  318. for username in gusernames:
  319. try:
  320. await app.ban_chat_member(username.strip("@"), userid)
  321. await asyncio.sleep(1)
  322. except FloodWait as e:
  323. await asyncio.sleep(e.x)
  324. except:
  325. continue
  326. count += 1
  327. mention = (await app.get_users(userid)).mention
  328. msg = f"""
  329. **List-Banned User:** {mention}
  330. **Banned User ID:** `{userid}`
  331. **Admin:** {message.from_user.mention}
  332. **Affected chats:** `{count}`
  333. **Reason:** {reason}
  334. """
  335. await m.edit_text(msg)
  336. # Unban users listed in a message
  337. @app.on_message(
  338. SUDOERS & filters.command("listunban") & ~filters.edited & ~filters.private
  339. )
  340. async def list_unban_(c, message: Message):
  341. userid, msglink = await extract_user_and_reason(message)
  342. if not userid or not msglink:
  343. return await message.reply_text(
  344. "Provide a userid/username along with message link to list-unban"
  345. )
  346. if not re.search(
  347. r"(https?://)?t(elegram)?\.me/\w+/\d+", msglink
  348. ): # validate link
  349. return await message.reply_text("Invalid message link provided")
  350. splitted = msglink.split("/")
  351. uname, mid = splitted[-2], int(splitted[-1])
  352. m = await message.reply_text(
  353. "`Unbanning User from multiple groups. \
  354. This may take some time`"
  355. )
  356. try:
  357. msgtext = (await app.get_messages(uname, mid)).text
  358. gusernames = re.findall("@\w+", msgtext)
  359. except:
  360. return await m.edit_text("Could not get the group usernames")
  361. count = 0
  362. for username in gusernames:
  363. try:
  364. await app.unban_chat_member(username.strip("@"), userid)
  365. await asyncio.sleep(1)
  366. except FloodWait as e:
  367. await asyncio.sleep(e.x)
  368. except:
  369. continue
  370. count += 1
  371. mention = (await app.get_users(userid)).mention
  372. msg = f"""
  373. **List-Unbanned User:** {mention}
  374. **Unbanned User ID:** `{userid}`
  375. **Admin:** {message.from_user.mention}
  376. **Affected chats:** `{count}`
  377. """
  378. await m.edit_text(msg)
  379. # Delete messages
  380. @app.on_message(filters.command("del") & ~filters.edited & ~filters.private)
  381. @adminsOnly("can_delete_messages")
  382. async def deleteFunc(_, message: Message):
  383. if not message.reply_to_message:
  384. return await message.reply_text("Reply To A Message To Delete It")
  385. await message.reply_to_message.delete()
  386. await message.delete()
  387. # Promote Members
  388. @app.on_message(
  389. filters.command(["promote", "fullpromote"])
  390. & ~filters.edited
  391. & ~filters.private
  392. )
  393. @adminsOnly("can_promote_members")
  394. async def promoteFunc(_, message: Message):
  395. user_id = await extract_user(message)
  396. umention = (await app.get_users(user_id)).mention
  397. if not user_id:
  398. return await message.reply_text("I can't find that user.")
  399. bot = await app.get_chat_member(message.chat.id, BOT_ID)
  400. if user_id == BOT_ID:
  401. return await message.reply_text("I can't promote myself.")
  402. if not bot.can_promote_members:
  403. return await message.reply_text("I don't have enough permissions")
  404. if message.command[0][0] == "f":
  405. await message.chat.promote_member(
  406. user_id=user_id,
  407. can_change_info=bot.can_change_info,
  408. can_invite_users=bot.can_invite_users,
  409. can_delete_messages=bot.can_delete_messages,
  410. can_restrict_members=bot.can_restrict_members,
  411. can_pin_messages=bot.can_pin_messages,
  412. can_promote_members=bot.can_promote_members,
  413. can_manage_chat=bot.can_manage_chat,
  414. can_manage_voice_chats=bot.can_manage_voice_chats,
  415. )
  416. return await message.reply_text(f"Fully Promoted! {umention}")
  417. await message.chat.promote_member(
  418. user_id=user_id,
  419. can_change_info=False,
  420. can_invite_users=bot.can_invite_users,
  421. can_delete_messages=bot.can_delete_messages,
  422. can_restrict_members=False,
  423. can_pin_messages=False,
  424. can_promote_members=False,
  425. can_manage_chat=bot.can_manage_chat,
  426. can_manage_voice_chats=bot.can_manage_voice_chats,
  427. )
  428. await message.reply_text(f"Promoted! {umention}")
  429. # Demote Member
  430. @app.on_message(filters.command("demote") & ~filters.edited & ~filters.private)
  431. @adminsOnly("can_promote_members")
  432. async def demote(_, message: Message):
  433. user_id = await extract_user(message)
  434. if not user_id:
  435. return await message.reply_text("I can't find that user.")
  436. if user_id == BOT_ID:
  437. return await message.reply_text("I can't demote myself.")
  438. if user_id in SUDOERS:
  439. return await message.reply_text(
  440. "You wanna demote the elevated one?, RECONSIDER!"
  441. )
  442. await message.chat.promote_member(
  443. user_id=user_id,
  444. can_change_info=False,
  445. can_invite_users=False,
  446. can_delete_messages=False,
  447. can_restrict_members=False,
  448. can_pin_messages=False,
  449. can_promote_members=False,
  450. can_manage_chat=False,
  451. can_manage_voice_chats=False,
  452. )
  453. umention = (await app.get_users(user_id)).mention
  454. await message.reply_text(f"Demoted! {umention}")
  455. # Pin Messages
  456. @app.on_message(
  457. filters.command(["pin", "unpin"]) & ~filters.edited & ~filters.private
  458. )
  459. @adminsOnly("can_pin_messages")
  460. async def pin(_, message: Message):
  461. if not message.reply_to_message:
  462. return await message.reply_text("Reply to a message to pin/unpin it.")
  463. r = message.reply_to_message
  464. if message.command[0][0] == "u":
  465. await r.unpin()
  466. return await message.reply_text(
  467. f"**Unpinned [this]({r.link}) message.**",
  468. disable_web_page_preview=True,
  469. )
  470. await r.pin(disable_notification=True)
  471. await message.reply(
  472. f"**Pinned [this]({r.link}) message.**",
  473. disable_web_page_preview=True,
  474. )
  475. msg = "Please check the pinned message: ~ " + f"[Check, {r.link}]"
  476. filter_ = dict(type="text", data=msg)
  477. await save_filter(message.chat.id, "~pinned", filter_)
  478. # Mute members
  479. @app.on_message(
  480. filters.command(["mute", "tmute"]) & ~filters.edited & ~filters.private
  481. )
  482. @adminsOnly("can_restrict_members")
  483. async def mute(_, message: Message):
  484. user_id, reason = await extract_user_and_reason(message)
  485. if not user_id:
  486. return await message.reply_text("I can't find that user.")
  487. if user_id == BOT_ID:
  488. return await message.reply_text("I can't mute myself.")
  489. if user_id in SUDOERS:
  490. return await message.reply_text(
  491. "You wanna mute the elevated one?, RECONSIDER!"
  492. )
  493. if user_id in (await list_admins(message.chat.id)):
  494. return await message.reply_text(
  495. "I can't mute an admin, You know the rules, so do i."
  496. )
  497. mention = (await app.get_users(user_id)).mention
  498. keyboard = ikb({"🚨 Unmute 🚨": f"unmute_{user_id}"})
  499. msg = (
  500. f"**Muted User:** {mention}\n"
  501. f"**Muted By:** {message.from_user.mention if message.from_user else 'Anon'}\n"
  502. )
  503. if message.command[0] == "tmute":
  504. split = reason.split(None, 1)
  505. time_value = split[0]
  506. temp_reason = split[1] if len(split) > 1 else ""
  507. temp_mute = await time_converter(message, time_value)
  508. msg += f"**Muted For:** {time_value}\n"
  509. if temp_reason:
  510. msg += f"**Reason:** {temp_reason}"
  511. try:
  512. if len(time_value[:-1]) < 3:
  513. await message.chat.restrict_member(
  514. user_id,
  515. permissions=ChatPermissions(),
  516. until_date=temp_mute,
  517. )
  518. await message.reply_text(msg, reply_markup=keyboard)
  519. else:
  520. await message.reply_text("You can't use more than 99")
  521. except AttributeError:
  522. pass
  523. return
  524. if reason:
  525. msg += f"**Reason:** {reason}"
  526. await message.chat.restrict_member(user_id, permissions=ChatPermissions())
  527. await message.reply_text(msg, reply_markup=keyboard)
  528. # Unmute members
  529. @app.on_message(filters.command("unmute") & ~filters.edited & ~filters.private)
  530. @adminsOnly("can_restrict_members")
  531. async def unmute(_, message: Message):
  532. user_id = await extract_user(message)
  533. if not user_id:
  534. return await message.reply_text("I can't find that user.")
  535. await message.chat.unban_member(user_id)
  536. umention = (await app.get_users(user_id)).mention
  537. await message.reply_text(f"Unmuted! {umention}")
  538. # Ban deleted accounts
  539. @app.on_message(
  540. filters.command("ban_ghosts")
  541. & ~filters.private
  542. & ~filters.edited
  543. )
  544. @adminsOnly("can_restrict_members")
  545. async def ban_deleted_accounts(_, message: Message):
  546. chat_id = message.chat.id
  547. deleted_users = []
  548. banned_users = 0
  549. m = await message.reply("Finding ghosts...")
  550. async for i in app.iter_chat_members(chat_id):
  551. if i.user.is_deleted:
  552. deleted_users.append(i.user.id)
  553. if len(deleted_users) > 0:
  554. for deleted_user in deleted_users:
  555. try:
  556. await message.chat.ban_member(deleted_user)
  557. except Exception:
  558. pass
  559. banned_users += 1
  560. await m.edit(f"Banned {banned_users} Deleted Accounts")
  561. else:
  562. await m.edit("There are no deleted accounts in this chat")
  563. @app.on_message(
  564. filters.command(["warn", "dwarn"]) & ~filters.edited & ~filters.private
  565. )
  566. @adminsOnly("can_restrict_members")
  567. async def warn_user(_, message: Message):
  568. user_id, reason = await extract_user_and_reason(message)
  569. chat_id = message.chat.id
  570. if not user_id:
  571. return await message.reply_text("I can't find that user.")
  572. if user_id == BOT_ID:
  573. return await message.reply_text(
  574. "I can't warn myself, i can leave if you want."
  575. )
  576. if user_id in SUDOERS:
  577. return await message.reply_text(
  578. "You Wanna Warn The Elevated One?, RECONSIDER!"
  579. )
  580. if user_id in (await list_admins(chat_id)):
  581. return await message.reply_text(
  582. "I can't warn an admin, You know the rules, so do i."
  583. )
  584. user, warns = await asyncio.gather(
  585. app.get_users(user_id),
  586. get_warn(chat_id, await int_to_alpha(user_id)),
  587. )
  588. mention = user.mention
  589. keyboard = ikb({"🚨 Remove Warn 🚨": f"unwarn_{user_id}"})
  590. if warns:
  591. warns = warns["warns"]
  592. else:
  593. warns = 0
  594. if message.command[0][0] == "d":
  595. await message.reply_to_message.delete()
  596. if warns >= 2:
  597. await message.chat.ban_member(user_id)
  598. await message.reply_text(
  599. f"Number of warns of {mention} exceeded, BANNED!"
  600. )
  601. await remove_warns(chat_id, await int_to_alpha(user_id))
  602. else:
  603. warn = {"warns": warns + 1}
  604. msg = f"""
  605. **Warned User:** {mention}
  606. **Warned By:** {message.from_user.mention if message.from_user else 'Anon'}
  607. **Reason:** {reason or 'No Reason Provided.'}
  608. **Warns:** {warns + 1}/3"""
  609. await message.reply_text(msg, reply_markup=keyboard)
  610. await add_warn(chat_id, await int_to_alpha(user_id), warn)
  611. @app.on_callback_query(filters.regex("unwarn_"))
  612. async def remove_warning(_, cq: CallbackQuery):
  613. from_user = cq.from_user
  614. chat_id = cq.message.chat.id
  615. permissions = await member_permissions(chat_id, from_user.id)
  616. permission = "can_restrict_members"
  617. if permission not in permissions:
  618. return await cq.answer(
  619. "You don't have enough permissions to perform this action.\n"
  620. + f"Permission needed: {permission}",
  621. show_alert=True,
  622. )
  623. user_id = cq.data.split("_")[1]
  624. warns = await get_warn(chat_id, await int_to_alpha(user_id))
  625. if warns:
  626. warns = warns["warns"]
  627. if not warns or warns == 0:
  628. return await cq.answer("User has no warnings.")
  629. warn = {"warns": warns - 1}
  630. await add_warn(chat_id, await int_to_alpha(user_id), warn)
  631. text = cq.message.text.markdown
  632. text = f"~~{text}~~\n\n"
  633. text += f"__Warn removed by {from_user.mention}__"
  634. await cq.message.edit(text)
  635. # Rmwarns
  636. @app.on_message(
  637. filters.command("rmwarns") & ~filters.edited & ~filters.private
  638. )
  639. @adminsOnly("can_restrict_members")
  640. async def remove_warnings(_, message: Message):
  641. if not message.reply_to_message:
  642. return await message.reply_text(
  643. "Reply to a message to remove a user's warnings."
  644. )
  645. user_id = message.reply_to_message.from_user.id
  646. mention = message.reply_to_message.from_user.mention
  647. chat_id = message.chat.id
  648. warns = await get_warn(chat_id, await int_to_alpha(user_id))
  649. if warns:
  650. warns = warns["warns"]
  651. if warns == 0 or not warns:
  652. await message.reply_text(f"{mention} have no warnings.")
  653. else:
  654. await remove_warns(chat_id, await int_to_alpha(user_id))
  655. await message.reply_text(f"Removed warnings of {mention}.")
  656. # Warns
  657. @app.on_message(filters.command("warns") & ~filters.edited & ~filters.private)
  658. @capture_err
  659. async def check_warns(_, message: Message):
  660. user_id = await extract_user(message)
  661. if not user_id:
  662. return await message.reply_text("I can't find that user.")
  663. warns = await get_warn(message.chat.id, await int_to_alpha(user_id))
  664. mention = (await app.get_users(user_id)).mention
  665. if warns:
  666. warns = warns["warns"]
  667. else:
  668. return await message.reply_text(f"{mention} has no warnings.")
  669. return await message.reply_text(f"{mention} has {warns}/3 warnings.")
  670. # Report
  671. @app.on_message(
  672. (
  673. filters.command("report")
  674. | filters.command(["admins", "admin"], prefixes="@")
  675. )
  676. & ~filters.edited
  677. & ~filters.private
  678. )
  679. @capture_err
  680. async def report_user(_, message):
  681. if not message.reply_to_message:
  682. return await message.reply_text(
  683. "Reply to a message to report that user."
  684. )
  685. reply = message.reply_to_message
  686. reply_id = reply.from_user.id if reply.from_user else reply.sender_chat.id
  687. user_id = message.from_user.id if message.from_user else message.sender_chat.id
  688. if reply_id == user_id:
  689. return await message.reply_text("Why are you reporting yourself ?")
  690. list_of_admins = await list_admins(message.chat.id)
  691. linked_chat = (await app.get_chat(message.chat.id)).linked_chat
  692. if linked_chat is not None:
  693. if reply_id in list_of_admins or reply_id == message.chat.id or reply_id == linked_chat.id:
  694. return await message.reply_text(
  695. "Do you know that the user you are replying is an admin ?"
  696. )
  697. else:
  698. if reply_id in list_of_admins or reply_id == message.chat.id:
  699. return await message.reply_text(
  700. "Do you know that the user you are replying is an admin ?"
  701. )
  702. user_mention = reply.from_user.mention if reply.from_user else reply.sender_chat.title
  703. text = f"Reported {user_mention} to admins!"
  704. admin_data = await app.get_chat_members(
  705. chat_id=message.chat.id, filter="administrators"
  706. ) # will it giv floods ?
  707. for admin in admin_data:
  708. if admin.user.is_bot or admin.user.is_deleted:
  709. # return bots or deleted admins
  710. continue
  711. text += f"[\u2063](tg://user?id={admin.user.id})"
  712. await message.reply_to_message.reply_text(text)
  713. @app.on_message(filters.command("invite") & ~filters.edited)
  714. @adminsOnly("can_invite_users")
  715. async def invite(_, message):
  716. if message.chat.type in ["group", "supergroup"]:
  717. link = (await app.get_chat(message.chat.id)).invite_link
  718. if not link:
  719. link = await app.export_chat_invite_link(message.chat.id)
  720. text = f"Here's This Group Invite Link.\n\n{link}"
  721. if message.reply_to_message:
  722. await message.reply_to_message.reply_text(text, disable_web_page_preview=True)
  723. else:
  724. await message.reply_text(text, disable_web_page_preview=True)