1 files changed, 52 insertions, 50 deletions

diff --git a/handlers/gen.py b/handlers/gen.py
index b618fec..8f19e96 100644
--- a/handlers/gen.py
+++ b/handlers/gen.py
@@ -1,67 +1,69 @@
-import os
+import random
 
 import mc
 from aiogram import types as t
 
-from shared import config
-from shared.instances import bot, dp
+from shared.database import Message
+from shared.instances import bot, config, dp, session
 from utils import filters as f
 
 
-@dp.message_handler(commands=["gen"])
-async def сгенерировать_хуету(msg: t.Message):
-    await msg.answer(получить_говно(msg.chat.id))
+def get_text(chat_id: int) -> str:
+    with session() as s:
+        samples = [
+            m.tuple()[0]
+            for m in s.query(Message.message).filter(Message.chat_id == chat_id).all()
+        ]
 
+    assert (
+        len(samples) != 0
+    ), "Нету сообщений на основе которых можно сгенерировать сообщение"
 
-@dp.message_handler(commands=["del"])
-async def удалить_хуету(msg: t.Message):
-    await msg.delete()
+    generator = mc.PhraseGenerator(samples)
+    gen_config = config.get_config(chat_id).gen
+    validators = []
 
-    if msg.reply_to_message:
-        if msg.reply_to_message.from_user.id in [bot.id, msg.from_user.id]:
-            await msg.reply_to_message.delete()
-        else:
-            await msg.answer("Ты умник, можно только свои или мои удалять")
-    else:
-        await msg.answer("Ты умник, ответь на сообщение")
-
-
-@dp.message_handler(commands=["void"])
-async def лоботомия(msg: t.Message):
-    if msg.get_args() == "Я знаю что делаю":
-        os.remove(f"data/{msg.chat.id}")
-        await msg.answer("Лоботомия проведена успешно")
-    else:
-        await msg.answer(
-            "Напишите <code>/void Я знаю что делаю</code>", parse_mode=t.ParseMode.HTML
+    if gen_config.max_word_count is not None or gen_config.min_word_count is not None:
+        validators.append(
+            mc.builtin.validators.words_count(
+                minimal=gen_config.min_word_count,
+                maximal=gen_config.max_word_count,
+            )
         )
 
+    while True:
+        message = generator.generate_phrase_or_none(1, validators=validators)
+        if message is not None:
+            return message
 
-@dp.message_handler(commands=["chance"])
-async def изменить_шанс_срания(msg: t.Message):
-    if msg.get_args():
-        try:
-            chance = int(msg.get_args().split()[0])
-            if 0 <= chance <= 100:
-                config.chances[str(msg.chat.id)] = chance
-                config.save()
-            else:
-                raise RuntimeError()
-
-            await msg.answer(f"Теперь я сру с шансом в: {chance}%")
-        except Exception:
-            await msg.answer(
-                "Я хз что не так, но я знаю что ты дебил \n    /chance <ЧИСЛО ОТ 0 ДО 100>"
-            )
-    else:
-        await msg.answer(f"Я сру с шансом в: {config.chances.get(str(msg.chat.id), 10)}%")
 
+@dp.message_handler(commands=["gen"])
+async def gen_command(msg: t.Message) -> None:
+    await msg.delete()
+    message = get_text(msg.chat.id)
+    if message is not None:
+        await msg.answer(message)
 
-@dp.message_handler(f.message.chance, content_types=[t.ContentType.ANY])
-async def срать_сообщение_с_шансом(msg: t.Message):
-    await msg.answer(получить_говно(msg.chat.id))
 
+@dp.message_handler(commands=["del"])
+async def del_command(msg: t.Message) -> None:
+    await msg.delete()
 
-def получить_говно(id: int) -> str:
-    samples = mc.util.load_txt_samples(f"data/{id}", separator="§")
-    return mc.StringGenerator(samples=samples).generate_string()
+    if msg.reply_to_message:
+        if msg.reply_to_message.from_user.id == bot.id:
+            await msg.reply_to_message.delete()
+        else:
+            await msg.reply("Можно удалять только сообщения бота")
+    else:
+        await msg.reply("Вы не ответили на сообщение")
+
+
+@dp.message_handler(
+    f.message.is_chat,
+    f.message.chance,
+    content_types=[t.ContentType.ANY],
+)
+async def chance_message(msg: t.Message) -> None:
+    message = get_text(msg.chat.id)
+    if message is not None:
+        await msg.reply(message)