From 6694a264fde414c2f35911c95f49138a6bdb3509 Mon Sep 17 00:00:00 2001
From: none <none>
Date: Thu, 8 Feb 2024 18:05:45 +0300
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?=
 =?UTF-8?q?=D0=BD=D0=BE=20/key=5Fres=20=D0=B8=20=D0=BF=D0=BE=D0=B4=D1=82?=
 =?UTF-8?q?=D0=B2=D0=B5=D1=80=D0=B6=D0=B4=D0=B5=D0=BD=D0=B8=D0=B5=20=D0=BA?=
 =?UTF-8?q?=D0=BB=D1=8E=D1=87=D0=B5=D0=B9=20=D0=BF=D1=80=D0=B8=20=D0=BE?=
 =?UTF-8?q?=D1=82=D0=BF=D1=80=D0=B0=D0=B2=D0=BA=D0=B5=20=D1=81=D0=BE=D0=BE?=
 =?UTF-8?q?=D0=B1=D1=89=D0=B5=D0=BD=D0=B8=D0=B9.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 anon.py | 125 ++++++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 94 insertions(+), 31 deletions(-)

diff --git a/anon.py b/anon.py
index c44f919..b04fe6a 100644
--- a/anon.py
+++ b/anon.py
@@ -170,7 +170,8 @@ def nick(message):
 				del db[old_nick]
 				write_db(db)
 				bot.reply_to(message,f"Вы успешно сменили ник с {telebot.formatting.hcode(old_nick)} на {telebot.formatting.hcode(new_nick)}",parse_mode="HTML")
-				bot.reply_to(message, "Ваша аватарка сброшена до стандартной: ♿️")
+				bot.reply_to(message, """Ваша аватарка сброшена до стандартной: ♿️
+Также вы можете сбросить публичный ключ: /key_res""")
 			else:
 				bot.reply_to(message,"Данный ник уже занят")
 	except:
@@ -200,32 +201,92 @@ def av(message):
 
 @bot.message_handler(commands=['key'])
 def key(message):
-	if len(message.text.split()) == 2:
-		nick = message.text.split()[1]
-		if nick[0] == ':':
-			nick = nick[1:]
-		key = db[nick]["pkey"]
-		bot.reply_to(message,f"Ключ пользователя: {telebot.formatting.hcode(key)}", parse_mode="HTML")
-	else:
-		bot.reply_to(message,"/key ник")
+	try:
+		if len(message.text.split()) == 2:
+			db = read_db()
+			nick = message.text.split()[1]
+			if nick[0] == ':':
+				nick = nick[1:]
+			key = db[nick]["pkey"]
+			bot.reply_to(message,f"Ключ пользователя: {telebot.formatting.hcode(key)}", parse_mode="HTML")
+		else:
+			bot.reply_to(message,"/key ник")
+	except:
+		catch_error(message)
 
 @bot.message_handler(commands=['ver'])
 def ver(message):
-	if len(message.text.split()) == 3:
-		nick = message.text.split()[1]
-		if nick[0] == ':':
-			nick = nick[1:]
-		key = message.text.split()[2]
-		if not nick in db:
-			bot.reply_to(message,"Не существует такого пользователя")
-			return 0
-
-		if key == db[nick]["pkey"]:
-			bot.reply_to(message,"✅ Ключи совпадают")
+	try:
+		if len(message.text.split()) == 3:
+			db = read_db()
+			nick = message.text.split()[1]
+			if nick[0] == ':':
+				nick = nick[1:]
+			key = message.text.split()[2]
+			if not nick in db:
+				bot.reply_to(message,"Не существует такого пользователя")
+				return 0
+
+			if key == db[nick]["pkey"]:
+				bot.reply_to(message,"✅ Ключи совпадают")
+			else:
+				bot.reply_to(message,"❌ Ключи не совпадают")
+		else:
+			bot.reply_to(message,"/ver ник ключ")
+	except:
+		catch_error(message)
+
+@bot.message_handler(commands=['key_res'])
+def key_res(message):
+	try:
+		if is_auth(message):
+			db = read_db()
+
+			key = hash(randint(74287, 5747962))
+			nick = db[str(message.chat.id)]
+			old_key = db[nick]["pkey"]
+
+			db[nick]["pkey"] = key
+			write_db(db)
+
+			bot.reply_to(message,f"""🔑 Ключ успешно сброшен.
+
+Старый ключ: {telebot.formatting.hcode(old_key)}
+Новый ключ: {telebot.formatting.hcode(key)}""",parse_mode="HTML")
+	except:
+		catch_error(message)
+
+# Проверяем совпадение ключей при отправке сообщений
+def key_valid(message, channel):
+	try:
+		db = read_db()
+		our_nick = db[str(message.chat.id)]
+		# Добавляем ключ если его нету в нашей БД
+		if channel not in db[our_nick]["keys"]:
+			db[our_nick]["keys"][channel] = db[channel]["pkey"]
+			write_db(db)
+
+		db = read_db()
+		our_key = db[our_nick]["keys"][channel]
+		dest_key = db[channel]["pkey"]
+
+		if our_key == dest_key:
+			print("Valid: ", channel)
+			return True
 		else:
-			bot.reply_to(message,"❌ Ключи не совпадают")
-	else:
-		bot.reply_to(message,"/ver ник ключ")
+			print("Not valid: ", channel)
+			db[our_nick]["keys"][channel] = dest_key
+			write_db(db)
+
+			bot.reply_to(message, f"""⚠️ Публичные ключи не совпадают ⚠️
+Ожидаемый ключ: {telebot.formatting.hcode(our_key)}
+
+Отправка сообщения отклонена.
+Если вы уверены - повторите отправку.
+""", parse_mode="HTML")
+			return False
+	except:
+		catch_error(message)
 
 ####################################
 
@@ -261,20 +322,22 @@ def catch_all_messages(message):
 		if message.content_type == "text" and message.text[:1].lower() == ":":
 			channel = message.text[1:]
 			if channel in db:
-				db[nick]["channel"] = channel
-
-				# Обрабатываем публичные ключи
-				if channel in db[nick]["keys"]:
-					pass
-				else:
-					db[nick]["keys"][channel] = db[channel]["pkey"]
+				# Проверяем ключи
+				if not key_valid(message, channel):
+					return 0
+				db = read_db()
 
+				db[nick]["channel"] = channel
 				bot.reply_to(message, "Установлен адресат: " + telebot.formatting.hcode(channel), parse_mode="HTML")
 				write_db(db)
 			else:
 				bot.reply_to(message, "Не существует данного пользователя.")
 		elif db[nick]["channel"] != None:
 			channel = db[nick]["channel"]
+			# Проверяем ключи
+			if not key_valid(message, channel):
+				return 0
+			db = read_db()
 
 			if message.chat.id not in db[channel]["blocks"]:
 				try:
@@ -302,7 +365,7 @@ def catch_all_messages(message):
 		catch_error(message)
 
 #### POLLING ####
-mode = 0
+mode = 1
 # Normal - 0, debug - 1
 
 if mode == 0: