def mention_answer(client: Client, message: Message, aid: int, uid: int,
action_type: str) -> str:
# Mention abuse
try:
# Basic data
gid = message.chat.id
mid = message.message_id
# Abuse
if action_type == "abuse":
message.reply_to_message = message
message.reply_to_message.from_user.id = uid
message.reply_to_message.from_user.is_self = lang("abuse_mention")
text, markup = warn_user(client, message, uid, aid)
text += f"{lang('reason')}{lang('colon')}{code(lang('reason_abuse'))}\n"
# Edit the report message
thread(edit_message_text, (client, gid, mid, text, markup))
delay(180, delete_message, [client, gid, mid])
# Delete
elif action_type == "delete":
glovar.message_ids[gid] = (0, 0)
save("message_ids")
delete_message(client, gid, mid)
except Exception as e:
logger.warning(f"Mention answer error: {e}", exc_info=True)
return ""
def save_admins(gid: int, admin_members: List[ChatMember]) -> bool:
# Save the group's admin list
result = False
try:
# Admin list
glovar.admin_ids[gid] = {admin.user.id for admin in admin_members
if (((not admin.user.is_bot and not admin.user.is_deleted)
and admin.can_delete_messages
and admin.can_restrict_members)
or admin.status == "creator"
or admin.user.id in glovar.bot_ids)}
save("admin_ids")
# Trust list
glovar.trust_ids[gid] = {admin.user.id for admin in admin_members
if ((not admin.user.is_bot and not admin.user.is_deleted)
or admin.user.id in glovar.bot_ids)}
save("trust_ids")
result = True
except Exception as e:
logger.warning(f"Save admins error: {e}", exc_info=True)
return result
def receive_rollback(client: Client, message: Message, data: dict) -> bool:
# Receive rollback data
try:
# Basic data
aid = data["admin_id"]
the_type = data["type"]
the_data = receive_file_data(client, message)
if not the_data:
return True
exec(f"glovar.{the_type} = the_data")
save(the_type)
# Send debug message
text = (
f"{lang('project')}{lang('colon')}{general_link(glovar.project_name, glovar.project_link)}\n"
f"{lang('admin_project')}{lang('colon')}{mention_id(aid)}\n"
f"{lang('action')}{lang('colon')}{code(lang('rollback'))}\n"
f"{lang('more')}{lang('colon')}{code(the_type)}\n")
thread(send_message, (client, glovar.debug_channel_id, text))
except Exception as e:
logger.warning(f"Receive rollback error: {e}", exc_info=True)
return False
def undo_user(client: Client, message: Message, aid: int, uid: int,
action_type: str) -> str:
try:
# Basic
gid = message.chat.id
mid = message.message_id
# Init user data
if not init_user_id(uid):
return ""
# Check the user's lock
if gid in glovar.user_ids[uid]["lock"]:
return lang("answer_proceeded")
# Proceed
glovar.user_ids[uid]["lock"].add(gid)
try:
if action_type == "ban":
text = unban_user(client, message, uid, aid)
else:
text = unwarn_user(client, message, uid, aid)
thread(edit_message_text, (client, gid, mid, text))
finally:
glovar.user_ids[uid]["lock"].discard(gid)
# Save data
save("user_ids")
except Exception as e:
logger.warning(f"Undo user error: {e}", exc_info=True)
return ""
def receive_watch_user(data: dict) -> bool:
# Receive watch users that other bots shared
try:
# Basic data
the_type = data["type"]
uid = data["id"]
until = data["until"]
# Decrypt the data
until = crypt_str("decrypt", until, glovar.key)
until = get_int(until)
# Add to list
if the_type == "ban":
glovar.watch_ids["ban"][uid] = until
elif the_type == "delete":
glovar.watch_ids["delete"][uid] = until
else:
return False
save("watch_ids")
return True
except Exception as e:
logger.warning(f"Receive watch user error: {e}", exc_info=True)
return False
def reset_data(client: Client) -> bool:
# Reset data every month
try:
glovar.bad_ids = {"users": set()}
save("bad_ids")
glovar.left_group_ids = set()
save("left_group_ids")
glovar.user_ids = {}
save("user_ids")
glovar.watch_ids = {"ban": {}, "delete": {}}
save("watch_ids")
glovar.reports = {}
save("reports")
# Send debug message
text = (
f"{lang('project')}{lang('colon')}{general_link(glovar.project_name, glovar.project_link)}\n"
f"{lang('action')}{lang('colon')}{code(lang('reset'))}\n")
thread(send_message, (client, glovar.debug_channel_id, text))
return True
except Exception as e:
logger.warning(f"Reset data error: {e}", exc_info=True)
return False
def init_user_id(uid: int) -> bool:
# Init user data
try:
if glovar.user_ids.get(uid) is None:
glovar.user_ids[uid] = deepcopy(glovar.default_user_status)
save("user_ids")
return True
except Exception as e:
logger.warning(f"Init user id {uid} error: {e}", exc_info=True)
return False
def receive_config_commit(data: dict) -> bool:
# Receive config commit
try:
# Basic data
gid = data["group_id"]
config = data["config"]
glovar.configs[gid] = config
save("configs")
return True
except Exception as e:
logger.warning(f"Receive config commit error: {e}", exc_info=True)
return False
def receive_help_report(client: Client, data: dict) -> bool:
# Receive help report requests
try:
# Basic data
gid = data["group_id"]
uid = data["user_id"]
mid = data["message_id"]
# Check declared status
if is_declared_message_id(gid, mid):
return True
# Check group
if gid not in glovar.admin_ids:
return True
if not init_group_id(gid):
return True
if not glovar.configs[gid]["report"]["auto"]:
return True
if not (init_user_id(0) and init_user_id(uid)
and gid not in glovar.user_ids[uid]["lock"]
and gid not in glovar.user_ids[uid]["waiting"]
and gid not in glovar.user_ids[uid]["ban"]):
return True
the_message = get_message(client, gid, mid)
if not the_message:
return True
text, markup, key = report_user(gid, the_message.from_user, 0, mid)
result = send_message(client, gid, text, mid, markup)
if result:
glovar.reports[key]["report_id"] = result.message_id
else:
glovar.reports[key].pop(key, {})
save("reports")
return True
except Exception as e:
logger.warning(f"Receive help report error: {e}", exc_info=True)
return False
def receive_remove_watch(data: int) -> bool:
# Receive removed watching users
try:
# Basic data
uid = data
# Reset watch status
glovar.watch_ids["ban"].pop(uid, 0)
glovar.watch_ids["delete"].pop(uid, 0)
save("watch_ids")
return True
except Exception as e:
logger.warning(f"Receive remove watch error: {e}", exc_info=True)
return False
def receive_add_bad(data: dict) -> bool:
# Receive bad users or channels that other bots shared
try:
# Basic data
the_id = data["id"]
the_type = data["type"]
# Receive bad user
if the_type == "user":
glovar.bad_ids["users"].add(the_id)
save("bad_ids")
return True
except Exception as e:
logger.warning(f"Receive add bad error: {e}", exc_info=True)
return False
def forgive(client: Client, message: Message) -> bool:
# Forgive users
if not message or not message.chat:
return True
# Basic data
gid = message.chat.id
mid = message.message_id
try:
# Check permission
if not is_class_c(None, None, message):
return True
# Get user id
uid, _ = get_class_d_id(message)
# Check user status
if not uid or uid in glovar.admin_ids[gid]:
return True
# Forgive the user
reason = get_command_type(message)
text, success = forgive_user(client, message, uid, reason)
glovar.user_ids[uid]["lock"].discard(gid)
save("user_ids")
if success:
secs = 180
else:
secs = 15
# Send the report message
thread(send_report_message, (secs, client, gid, text, None))
return True
except Exception as e:
logger.warning(f"Forgive error: {e}", exc_info=True)
finally:
delete_message(client, gid, mid)
return False
def receive_remove_score(data: int) -> bool:
# Receive remove user's score
glovar.locks["message"].acquire()
try:
# Basic data
uid = data
if not glovar.user_ids.get(uid):
return True
glovar.user_ids[uid] = deepcopy(glovar.default_user_status)
save("user_ids")
return True
except Exception as e:
logger.warning(f"Receive remove score error: {e}", exc_info=True)
finally:
glovar.locks["message"].release()
return False
def init_group_id(gid: int) -> bool:
# Init group data
try:
if gid == glovar.test_group_id:
return False
if gid in glovar.left_group_ids:
return False
if glovar.admin_ids.get(gid) is None:
glovar.admin_ids[gid] = set()
save("admin_ids")
if glovar.message_ids.get(gid) is None:
glovar.message_ids[gid] = (0, 0)
save("message_ids")
if glovar.configs.get(gid) is None:
glovar.configs[gid] = deepcopy(glovar.default_config)
save("configs")
if glovar.counts.get(gid) is None:
glovar.counts[gid] = {}
if glovar.declared_message_ids.get(gid) is None:
glovar.declared_message_ids[gid] = set()
return True
except Exception as e:
logger.warning(f"Init group id {gid} error: {e}", exc_info=True)
return False
def receive_user_score(project: str, data: dict) -> bool:
# Receive and update user's score
glovar.locks["message"].acquire()
try:
# Basic data
project = project.lower()
uid = data["id"]
if not init_user_id(uid):
return True
score = data["score"]
glovar.user_ids[uid]["score"][project] = score
save("user_ids")
return True
except Exception as e:
logger.warning(f"Receive user score error: {e}", exc_info=True)
finally:
glovar.locks["message"].release()
return False
def update_score(client: Client, uid: int) -> bool:
# Update a user's score, share it
try:
ban_count = len(glovar.user_ids[uid]["ban"])
kick_count = len(glovar.user_ids[uid]["kick"])
warn_count = len(glovar.user_ids[uid]["warn"])
score = ban_count * 1 + kick_count * 0.3 + warn_count * 0.4
glovar.user_ids[uid]["score"][glovar.sender.lower()] = score
save("user_ids")
share_data(client=client,
receivers=glovar.receivers["score"],
action="update",
action_type="score",
data={
"id": uid,
"score": round(score, 1)
})
return True
except Exception as e:
logger.warning(f"Update score error: {e}", exc_info=True)
return False
def leave_group(client: Client, gid: int) -> bool:
# Leave a group, clear it's data
try:
glovar.left_group_ids.add(gid)
save("left_group_ids")
thread(leave_chat, (client, gid))
glovar.admin_ids.pop(gid, None)
save("admin_ids")
glovar.message_ids.pop(gid, (0, 0))
save("message_ids")
glovar.configs.pop(gid, None)
save("configs")
return True
except Exception as e:
logger.warning(f"Leave group error: {e}", exc_info=True)
return False
def receive_clear_data(client: Client, data_type: str, data: dict) -> bool:
# Receive clear data command
glovar.locks["message"].acquire()
try:
# Basic data
aid = data["admin_id"]
the_type = data["type"]
# Clear bad data
if data_type == "bad":
if the_type == "channels":
glovar.bad_ids["channels"] = set()
elif the_type == "users":
glovar.bad_ids["users"] = set()
save("bad_ids")
# Clear user data
if data_type == "user":
if the_type == "all":
glovar.user_ids = {}
save("user_ids")
# Clear watch data
if data_type == "watch":
if the_type == "all":
glovar.watch_ids = {"ban": {}, "delete": {}}
elif the_type == "ban":
glovar.watch_ids["ban"] = {}
elif the_type == "delete":
glovar.watch_ids["delete"] = {}
save("watch_ids")
# Send debug message
text = (
f"{lang('project')}{lang('colon')}{general_link(glovar.project_name, glovar.project_link)}\n"
f"{lang('admin_project')}{lang('colon')}{mention_id(aid)}\n"
f"{lang('action')}{lang('colon')}{code(lang('clear'))}\n"
f"{lang('more')}{lang('colon')}{code(f'{data_type} {the_type}')}\n"
)
thread(send_message, (client, glovar.debug_channel_id, text))
except Exception as e:
logger.warning(f"Receive clear data: {e}", exc_info=True)
finally:
glovar.locks["message"].release()
return False
def receive_remove_bad(data: dict) -> bool:
# Receive removed bad objects
try:
# Basic data
the_id = data["id"]
the_type = data["type"]
# Remove bad user
if the_type == "user":
glovar.bad_ids["users"].discard(the_id)
glovar.watch_ids["ban"].pop(the_id, {})
glovar.watch_ids["delete"].pop(the_id, {})
save("watch_ids")
glovar.user_ids[the_id] = deepcopy(glovar.default_user_status)
save("user_ids")
save("bad_ids")
return True
except Exception as e:
logger.warning(f"Receive remove bad error: {e}", exc_info=True)
return False
def config_directly(client: Client, message: Message) -> bool:
# Config the bot directly
if not message or not message.chat:
return True
# Basic data
gid = message.chat.id
mid = message.message_id
try:
# Check permission
if not is_class_c(None, None, message):
return True
aid = message.from_user.id
success = True
reason = lang("config_updated")
new_config = deepcopy(glovar.configs[gid])
text = f"{lang('admin_group')}{lang('colon')}{code(aid)}\n"
# Check command format
command_type, command_context = get_command_context(message)
if command_type:
if command_type == "show":
text += f"{lang('action')}{lang('colon')}{code(lang('config_show'))}\n"
text += get_config_text(new_config)
thread(send_report_message, (30, client, gid, text))
return True
now = get_now()
if now - new_config["lock"] > 310:
if command_type == "default":
new_config = deepcopy(glovar.default_config)
else:
if command_context:
if command_type in {"delete", "mention"}:
if command_context == "off":
new_config[command_type] = False
elif command_context == "on":
new_config[command_type] = True
else:
success = False
reason = lang("command_para")
elif command_type == "limit":
limit = get_int(command_context)
if 2 <= limit <= 5:
new_config["limit"] = limit
else:
success = False
reason = lang("command_para")
elif command_type == "report":
if not new_config.get("report"):
new_config["report"] = {}
if command_context == "off":
new_config["report"]["auto"] = False
new_config["report"]["manual"] = False
elif command_context == "auto":
new_config["report"]["auto"] = True
new_config["report"]["manual"] = False
elif command_context == "manual":
new_config["report"]["auto"] = False
new_config["report"]["manual"] = True
elif command_context == "both":
new_config["report"]["auto"] = True
new_config["report"]["manual"] = True
else:
success = False
reason = lang("command_para")
else:
success = False
reason = lang("command_type")
else:
success = False
reason = lang("command_lack")
if success:
new_config["default"] = False
else:
success = False
reason = lang("config_locked")
else:
success = False
reason = lang("command_usage")
if success and new_config != glovar.configs[gid]:
# Save new config
glovar.configs[gid] = new_config
save("configs")
# Send debug message
debug_text = get_debug_text(client, message.chat)
debug_text += (
f"{lang('admin_group')}{lang('colon')}{code(message.from_user.id)}\n"
f"{lang('action')}{lang('colon')}{code(lang('config_change'))}\n"
f"{lang('more')}{lang('colon')}{code(f'{command_type} {command_context}')}\n"
)
thread(send_message, (client, glovar.debug_channel_id, debug_text))
text += (
f"{lang('action')}{lang('colon')}{code(lang('config_change'))}\n"
f"{lang('status')}{lang('colon')}{code(reason)}\n")
thread(send_report_message,
((lambda x: 10 if x else 5)(success), client, gid, text))
return True
except Exception as e:
logger.warning(f"Config directly error: {e}", exc_info=True)
finally:
delete_message(client, gid, mid)
return False
def interval_hour_01(client: Client) -> bool:
# Execute every hour
result = False
glovar.locks["message"].acquire()
try:
# Clear old calling messages
now = get_now()
for gid in list(glovar.message_ids):
mid, time = glovar.message_ids[gid]
if not time:
continue
if now - time < 86400:
continue
glovar.message_ids[gid] = (0, 0)
delete_message(client, gid, mid)
save("message_ids")
# Clear old reports
now = get_now()
for key in list(glovar.reports):
report_record = glovar.reports[key]
time = report_record["time"]
if not time:
glovar.reports.pop(key, {})
continue
if now - time < 86400:
continue
gid = report_record["group_id"]
mid = report_record["report_id"]
thread(delete_message, (client, gid, mid))
glovar.reports.pop(key, {})
save("reports")
# Clear user's waiting status
reported_users = {
glovar.reports[key]["user_id"]
for key in glovar.reports
}
for uid in set(glovar.user_ids) - reported_users:
glovar.user_ids[uid]["waiting"] = set()
save("user_ids")
result = True
except Exception as e:
logger.warning(f"Interval hour 01 error: {e}", exc_info=True)
finally:
glovar.locks["message"].release()
return result
def forgive_user(client: Client,
message: Message,
uid: int,
reason: str = None) -> (str, bool):
# Forgive user
text = ""
success = False
try:
# Basic data
gid = message.chat.id
aid = message.from_user.id
# Init user data
if not init_user_id(uid):
return "", False
# Check users' locks
if gid in glovar.user_ids[uid]["lock"]:
return "", False
# Proceed
glovar.user_ids[uid]["lock"].add(gid)
try:
# Text prefix
text += f"{lang('user_id')}{lang('colon')}{mention_id(uid)}\n"
if gid in glovar.user_ids[uid]["ban"]:
glovar.user_ids[uid]["ban"].discard(gid)
thread(unban_chat_member, (client, gid, uid))
text += (
f"{lang('action')}{lang('colon')}{code(lang('action_unban'))}\n"
f"{lang('status')}{lang('colon')}{code(lang('status_succeeded'))}\n"
)
success = True
elif glovar.user_ids[uid]["warn"].get(gid, 0):
glovar.user_ids[uid]["warn"].pop(gid, 0)
text += (
f"{lang('action')}{lang('colon')}{code(lang('action_unwarns'))}\n"
f"{lang('status')}{lang('colon')}{code(lang('status_succeeded'))}\n"
)
success = True
elif gid in glovar.user_ids[uid]["waiting"]:
glovar.user_ids[uid]["waiting"].discard(gid)
text += (
f"{lang('action')}{lang('colon')}{code(lang('action_unwait'))}\n"
f"{lang('status')}{lang('colon')}{code(lang('status_succeeded'))}\n"
)
success = True
else:
text += (
f"{lang('action')}{lang('colon')}{code(lang('action_forgive'))}\n"
f"{lang('status')}{lang('colon')}{code(lang('status_failed'))}\n"
f"{lang('reason')}{lang('colon')}{code(lang('reason_none'))}\n"
)
success = False
text += f"{lang('description')}{lang('colon')}{code(lang('description_by_admin'))}\n"
if not success:
return text, success
save("user_ids")
if reason:
text += f"{lang('reason')}{lang('colon')}{code(reason)}\n"
update_score(client, uid)
send_debug(client=client,
message=message,
action=lang("action_forgive"),
uid=uid,
aid=aid,
reason=reason)
finally:
glovar.user_ids[uid]["lock"].discard(gid)
except Exception as e:
logger.warning(f"Forgive user error: {e}")
return text, success
def report_answer(client: Client,
message: Message,
gid: int,
aid: int,
mid: int,
action_type: str,
key: str,
reason: str = None) -> str:
# Answer the user's report
try:
report_record = glovar.reports.get(key)
if not report_record:
message_text = get_text(message)
uid = get_int(message_text.split("\n")[0].split(lang("colon"))[1])
text = (
f"{lang('description')}{lang('colon')}{code(lang('description_by_admin'))}\n"
f"{lang('status')}{lang('colon')}{code(lang('status_failed'))}\n"
f"{lang('reason')}{lang('colon')}{code(lang('expired'))}\n")
thread(edit_message_text, (client, gid, mid, text))
delay(15, delete_message, [client, gid, mid])
glovar.user_ids[uid]["waiting"].discard(gid)
save("user_ids")
return ""
if not report_record["time"]:
return ""
rid = report_record["reporter_id"]
uid = report_record["user_id"]
r_mid = report_record["message_id"]
record_reason = report_record["reason"]
if not reason:
reason = record_reason
if not (init_user_id(rid) and init_user_id(uid)):
return ""
# Check users' locks
if gid in glovar.user_ids[uid]["lock"] or gid in glovar.user_ids[rid][
"lock"]:
return lang("answer_proceeded")
# Lock the report status
glovar.reports[key]["time"] = 0
try:
if action_type == "ban":
text, markup = ban_user(client, message, uid, aid, 0, reason)
thread(delete_message, (client, gid, r_mid))
elif action_type == "warn":
text, markup = warn_user(client, message, uid, aid, reason)
thread(delete_message, (client, gid, r_mid))
elif action_type == "abuse":
if not rid:
return ""
message.reply_to_message.from_user.id = rid
message.reply_to_message.from_user.is_self = lang(
"abuse_report")
text, markup = warn_user(client, message, rid, aid)
text += f"{lang('reason')}{lang('colon')}{code(lang('reason_abuse'))}\n"
else:
reported_link = general_link(
r_mid, f'{get_channel_link(message)}/{r_mid}')
if rid:
reporter_text = code(rid)
else:
reporter_text = code(lang("auto_triggered"))
text = (
f"{lang('reported_user')}{lang('colon')}{mention_id(uid)}\n"
f"{lang('reported_message')}{lang('colon')}{reported_link}\n"
f"{lang('reporter')}{lang('colon')}{reporter_text}\n"
f"{lang('action')}{lang('colon')}{code(lang('action_cancel'))}\n"
f"{lang('status')}{lang('colon')}{code(lang('status_succeeded'))}\n"
f"{lang('description')}{lang('colon')}{code(lang('description_by_admin'))}\n"
)
markup = None
if markup:
secs = 180
else:
secs = 15
thread(edit_message_text, (client, gid, mid, text, markup))
delay(secs, delete_message, [client, gid, mid])
finally:
glovar.user_ids[uid]["lock"].discard(gid)
glovar.user_ids[rid]["lock"].discard(gid)
glovar.user_ids[uid]["waiting"].discard(gid)
glovar.user_ids[rid]["waiting"].discard(gid)
save("user_ids")
except Exception as e:
logger.warning(f"Report answer error: {e}", exc_info=True)
return ""
def report_user(gid: int,
user: User,
rid: int,
mid: int,
name: str = None,
reason: str = None) -> (str, InlineKeyboardMarkup, str):
# Report a user
text = ""
markup = None
key = ""
try:
if user:
uid = user.id
else:
return "", None
glovar.user_ids[uid]["waiting"].add(gid)
glovar.user_ids[rid]["waiting"].add(gid)
save("user_ids")
key = random_str(8)
while glovar.reports.get(key):
key = random_str(8)
glovar.reports[key] = {
"time": get_now(),
"group_id": gid,
"reporter_id": rid,
"user_id": uid,
"message_id": mid,
"report_id": 0,
"reason": reason
}
if rid:
reporter_text = code("██████")
else:
reporter_text = code(lang("auto_triggered"))
text = f"{lang('reported_user')}{lang('colon')}{mention_id(uid)}\n"
if name:
text += f"{lang('reported_name')}{lang('colon')}{code(name)}\n"
text += (
f"{lang('reported_message')}{lang('colon')}{general_link(mid, f'{get_channel_link(gid)}/{mid}')}\n"
f"{lang('reporter')}{lang('colon')}{reporter_text}\n"
f"{lang('mention_admins')}{lang('colon')}{get_admin_text(gid)}\n"
f"{lang('description')}{lang('colon')}{code(lang('description_wait_admin'))}\n"
)
if reason:
text += f"{lang('reason')}{lang('colon')}{code(reason)}\n"
warn_data = button_data("report", "warn", key)
ban_data = button_data("report", "ban", key)
cancel_data = button_data("report", "cancel", key)
markup_list = [[
InlineKeyboardButton(text=lang("warn"), callback_data=warn_data),
InlineKeyboardButton(text=lang("ban"), callback_data=ban_data)
],
[
InlineKeyboardButton(text=lang("cancel"),
callback_data=cancel_data)
]]
if rid:
abuse_data = button_data("report", "abuse", key)
markup_list[1].append(
InlineKeyboardButton(text=lang("abuse"),
callback_data=abuse_data))
markup = InlineKeyboardMarkup(markup_list)
except Exception as e:
logger.warning(f"Report user error: {e}", exc_info=True)
return text, markup, key
def config(client: Client, message: Message) -> bool:
# Request CONFIG session
if not message or not message.chat:
return True
# Basic data
gid = message.chat.id
mid = message.message_id
try:
# Check permission
if not is_class_c(None, None, message):
return True
# Check command format
command_type = get_command_type(message)
if not command_type or not re.search(f"^{glovar.sender}$",
command_type, re.I):
return True
now = get_now()
# Check the config lock
if now - glovar.configs[gid]["lock"] < 310:
return True
# Set lock
glovar.configs[gid]["lock"] = now
save("configs")
# Ask CONFIG generate a config session
group_name, group_link = get_group_info(client, message.chat)
share_data(client=client,
receivers=["CONFIG"],
action="config",
action_type="ask",
data={
"project_name": glovar.project_name,
"project_link": glovar.project_link,
"group_id": gid,
"group_name": group_name,
"group_link": group_link,
"user_id": message.from_user.id,
"config": glovar.configs[gid],
"default": glovar.default_config
})
# Send debug message
text = get_debug_text(client, message.chat)
text += (
f"{lang('admin_group')}{lang('colon')}{code(message.from_user.id)}\n"
f"{lang('action')}{lang('colon')}{code(lang('config_create'))}\n")
thread(send_message, (client, glovar.debug_channel_id, text))
return True
except Exception as e:
logger.warning(f"Config error: {e}", exc_info=True)
finally:
if is_class_c(None, None, message):
delay(3, delete_message, [client, gid, mid])
else:
delete_message(client, gid, mid)
return False
def update_admins(client: Client) -> bool:
# Update admin list every day
result = False
glovar.locks["admin"].acquire()
try:
# Basic data
group_list = list(glovar.admin_ids)
# Check groups
for gid in group_list:
group_name, group_link = get_group_info(client, gid)
admin_members = get_admins(client, gid)
# Bot is not in the chat, leave automatically without approve
if admin_members is False or any(
admin.user.is_self for admin in admin_members) is False:
leave_group(client, gid)
share_data(client=client,
receivers=["MANAGE"],
action="leave",
action_type="info",
data={
"group_id": gid,
"group_name": group_name,
"group_link": group_link
})
project_text = general_link(glovar.project_name,
glovar.project_link)
debug_text = (
f"{lang('project')}{lang('colon')}{project_text}\n"
f"{lang('group_name')}{lang('colon')}{general_link(group_name, group_link)}\n"
f"{lang('group_id')}{lang('colon')}{code(gid)}\n"
f"{lang('status')}{lang('colon')}{code(lang('leave_auto'))}\n"
f"{lang('reason')}{lang('colon')}{code(lang('reason_leave'))}\n"
)
thread(send_message,
(client, glovar.debug_channel_id, debug_text))
continue
# Check the admin list
if not (admin_members
and any([admin.user.is_self for admin in admin_members])):
continue
# Save the admin list
save_admins(gid, admin_members)
# Ignore the group
if gid in glovar.lack_group_ids:
continue
# Check the permissions
if glovar.user_id not in glovar.admin_ids[gid]:
reason = "user"
elif any(admin.user.is_self and admin.can_delete_messages
and admin.can_restrict_members
for admin in admin_members):
glovar.lack_group_ids.discard(gid)
save("lack_group_ids")
continue
else:
reason = "permissions"
glovar.lack_group_ids.add(gid)
save("lack_group_ids")
# Send the leave request
share_data(client=client,
receivers=["MANAGE"],
action="leave",
action_type="request",
data={
"group_id": gid,
"group_name": group_name,
"group_link": group_link,
"reason": reason
})
reason = lang(f"reason_{reason}")
project_link = general_link(glovar.project_name,
glovar.project_link)
debug_text = (
f"{lang('project')}{lang('colon')}{project_link}\n"
f"{lang('group_name')}{lang('colon')}{general_link(group_name, group_link)}\n"
f"{lang('group_id')}{lang('colon')}{code(gid)}\n"
f"{lang('status')}{lang('colon')}{code(reason)}\n")
thread(send_message, (client, glovar.debug_channel_id, debug_text))
result = True
except Exception as e:
logger.warning(f"Update admin error: {e}", exc_info=True)
finally:
glovar.locks["admin"].release()
return result
def report(client: Client, message: Message) -> bool:
# Report spam messages
if not message or not message.chat:
return True
# Basic data
gid = message.chat.id
mid = message.message_id
try:
# Normal user
if not is_class_c(None, None, message):
# Check config
if not glovar.configs[gid]["report"]["manual"]:
return True
rid = message.from_user.id
now = message.date or get_now()
# Init user data
if not init_user_id(rid):
return True
# Get user id
uid, r_mid = get_class_d_id(message)
# Init user data
if not uid or not init_user_id(uid):
return True
# Check user status
bad_user = (gid in glovar.user_ids[rid]["lock"]
or gid in glovar.user_ids[uid]["lock"]
or gid in glovar.user_ids[rid]["waiting"]
or gid in glovar.user_ids[uid]["waiting"]
or gid in glovar.user_ids[uid]["ban"]
or is_watch_user(message.from_user, "ban", now)
or is_watch_user(message.from_user, "delete", now)
or is_high_score_user(message.from_user))
good_user = (is_class_e_user(message.from_user)
and uid not in glovar.admin_ids[gid])
# Users can not self-report
if uid == rid or (bad_user and not good_user):
return True
# Reporter cannot report someone by replying WARN's report
r_message = message.reply_to_message
if r_message.from_user.is_self:
return True
# Proceed
if r_message.service:
name = get_full_name(r_message.from_user)
else:
name = None
reason = get_command_type(message)
text, markup, key = report_user(gid, r_message.from_user, rid,
r_mid, name, reason)
result = send_message(client, gid, text, r_mid, markup)
if result:
glovar.reports[key]["report_id"] = result.message_id
else:
glovar.reports.pop(key, {})
save("reports")
# Admin
else:
aid = message.from_user.id
action_type, reason = get_command_context(message)
# Text prefix
text = (
f"{lang('admin')}{lang('colon')}{code(aid)}\n"
f"{lang('action')}{lang('colon')}{code(lang('action_answer'))}\n"
)
# Check command format
if action_type not in {"warn", "ban", "cancel", "abuse"
} or not message.reply_to_message:
text += (
f"{lang('status')}{lang('colon')}{code(lang('status_failed'))}\n"
f"{lang('reason')}{lang('colon')}{code(lang('command_usage'))}\n"
)
thread(send_report_message, (15, client, gid, text))
return True
# Check the evidence message
r_message = get_message(client, gid,
message.reply_to_message.message_id)
if not r_message or not r_message.reply_to_message:
text += (
f"{lang('status')}{lang('colon')}{code(lang('status_failed'))}\n"
f"{lang('reason')}{lang('colon')}{code(lang('reason_deleted'))}\n"
)
thread(send_report_message, (15, client, gid, text))
return True
# Check the report message
callback_data_list = get_callback_data(r_message)
if not callback_data_list or callback_data_list[0]["a"] != "report":
text += (
f"{lang('status')}{lang('colon')}{code(lang('status_failed'))}\n"
f"{lang('reason')}{lang('colon')}{code(lang('command_reply'))}\n"
)
thread(send_report_message, (15, client, gid, text))
return True
# Proceed
key = callback_data_list[0]["d"]
report_answer(client=client,
message=r_message,
gid=gid,
aid=aid,
mid=r_message.message_id,
action_type=action_type,
key=key,
reason=reason)
return True
except Exception as e:
logger.warning(f"Report error: {e}", exc_info=True)
finally:
delete_message(client, gid, mid)
return False
def admin(client: Client, message: Message) -> bool:
# Mention admins
if not message or not message.chat:
return True
# Basic data
gid = message.chat.id
mid = message.message_id
try:
# Check permission
if is_class_c(None, None, message):
return True
# Check config
if not glovar.configs[gid].get("mention"):
return True
uid = message.from_user.id
# Init user data
if not init_user_id(uid):
return True
# Warned user and the user having report status can't mention admins
if (gid in glovar.user_ids[uid]["waiting"]
or gid in glovar.user_ids[uid]["ban"]
or glovar.user_ids[uid]["warn"].get(gid)):
return True
# Generate report text
text = (
f"{lang('from_user')}{lang('colon')}{mention_id(uid)}\n"
f"{lang('mention_admins')}{lang('colon')}{get_admin_text(gid)}\n")
reason = get_command_type(message)
if reason:
text += f"{lang('reason')}{lang('colon')}{code(reason)}\n"
# Generate report markup
button_abuse = button_data("mention", "abuse", uid)
button_delete = button_data("mention", "delete", uid)
markup = InlineKeyboardMarkup([[
InlineKeyboardButton(text=lang("abuse"),
callback_data=button_abuse),
InlineKeyboardButton(text=lang("del"), callback_data=button_delete)
]])
# Send the report message
if message.reply_to_message:
rid = message.reply_to_message.message_id
else:
rid = None
result = send_message(client, gid, text, rid, markup)
if not result:
return True
old_mid, _ = glovar.message_ids.get(gid, (0, 0))
old_mid and thread(delete_message, (client, gid, old_mid))
sent_mid = result.message_id
glovar.message_ids[gid] = (sent_mid, get_now())
save("message_ids")
return True
except Exception as e:
logger.warning(f"Admin error: {e}", exc_info=True)
finally:
delete_message(client, gid, mid)
return False
