1 files changed, 156 insertions, 0 deletions
diff --git a/input_msg.py b/input_msg.py
new file mode 100644
index 0000000..9bae20d
--- /dev/null
+++ b/input_msg.py
@@ -0,0 +1,156 @@
+import string
+
+from random import choices
+
+from vars import *
+from config import *
+from helpers import get_user_config, set_user_setting
+
+from aiogram.types import (
+    InlineKeyboardButton,
+    InlineKeyboardMarkup,
+    InlineQueryResultArticle,
+    InputTextMessageContent,
+    Message,
+)
+
+async def da(bot, message, user_id):
+    config = get_user_config(user_id)
+    print(config)
+    if user_id in alo_command:
+        if message.chat.id == alo_command[user_id]["chat_id"]:
+            if alo_command[user_id]["action"] == "p_name":
+                alo_command[user_id]["action"] = "p_prompt"
+                alo_command[user_id]["name"] = message.text[:16]
+                return await bot.edit_message_text(
+                    text=f"установи промпт для `{alo_command[user_id]['name']}`:",
+                    chat_id=alo_command[user_id]["chat_id"],
+                    message_id=alo_command[user_id]["message_id"],
+                    reply_markup=InlineKeyboardMarkup(
+                        inline_keyboard=[
+                            [
+                                InlineKeyboardButton(
+                                    text=f"❌ отменить", callback_data=f"p_cancel%{user_id}"
+                                ),
+                            ],
+                        ]
+                    )
+                )
+            elif alo_command[user_id]["action"] == "p_prompt":
+                alo_command[user_id]["action"] = None
+                alo_command[user_id]["prompt"] = message.text
+
+                if len(config["prompts"]) >= MAX_PROMPTS:
+                    return await bot.edit_message_text(
+                        text=f"невозможно создать новый промпт так как ты превысил лимит промптов da.",
+                        chat_id=alo_command[user_id]["chat_id"],
+                        message_id=alo_command[user_id]["message_id"],
+                        reply_markup=InlineKeyboardMarkup(
+                            inline_keyboard=[
+                                [
+                                    InlineKeyboardButton(
+                                        text=f"ок", callback_data=f"user_prompts%{user_id}"
+                                    ),
+                                ],
+                            ]
+                        )
+                    )
+                
+                while 1:
+                    alo_command[user_id]["id"] = ''.join(choices(string.ascii_letters, k=10))
+                    if alo_command[user_id]["id"] not in config["prompts"]: 
+                        break
+
+                config["prompts"] = config["prompts"] | {
+                    alo_command[user_id]["id"]: {
+                        "name": alo_command[user_id]["name"],
+                        "text": alo_command[user_id]["prompt"],
+                        "filer": False,
+                        "public": False
+                    }
+                }
+
+                set_user_setting(
+                    user_id, 
+                    prompts=config["prompts"]
+                )
+                
+                return await bot.edit_message_text(
+                    text=f"промпт `{alo_command[user_id]['name']}` создан!",
+                    chat_id=alo_command[user_id]["chat_id"],
+                    message_id=alo_command[user_id]["message_id"],
+                    reply_markup=InlineKeyboardMarkup(
+                        inline_keyboard=[
+                            [
+                                InlineKeyboardButton(
+                                    text=f"✅ посмотреть", callback_data=f"vp%{user_id}%{alo_command[user_id]['id']}"
+                                ),
+                            ],
+                        ]
+                    )
+                )
+            elif alo_command[user_id]["action"] == "p_e_name":
+                alo_command[user_id]["action"] = None
+                alo_command[user_id]["name"] = message.text[:16]
+                
+                config["prompts"] = config["prompts"] | {
+                    alo_command[user_id]["prompt_id"]: {
+                        "name": alo_command[user_id]["name"],
+                        "text": config["prompts"][alo_command[user_id]["prompt_id"]]["text"],
+                        "filer": config["prompts"][alo_command[user_id]["prompt_id"]]["filer"],
+                        "public": config["prompts"][alo_command[user_id]["prompt_id"]]["public"],
+                    }
+                }
+
+                set_user_setting(
+                    user_id, 
+                    prompts = config["prompts"]
+                )
+                
+                return await bot.edit_message_text(
+                    text=f"имя промпта было изменено на `{alo_command[user_id]['name']}`.",
+                    chat_id=alo_command[user_id]["chat_id"],
+                    message_id=alo_command[user_id]["message_id"],
+                    reply_markup=InlineKeyboardMarkup(
+                        inline_keyboard=[
+                            [
+                                InlineKeyboardButton(
+                                    text=f"✅ посмотреть", callback_data=f"vp%{user_id}%{alo_command[user_id]['prompt_id']}"
+                                ),
+                            ],
+                        ]
+                    )
+                )
+            elif alo_command[user_id]["action"] == "p_e_prompt":
+                alo_command[user_id]["action"] = None
+                alo_command[user_id]["prompt"] = message.text
+                
+                config["prompts"] = config["prompts"] | {
+                    alo_command[user_id]["prompt_id"]: {
+                        "name": config["prompts"][alo_command[user_id]["prompt_id"]]["name"],
+                        "text": alo_command[user_id]["prompt"],
+                        "filer": config["prompts"][alo_command[user_id]["prompt_id"]]["filer"],
+                        "public": config["prompts"][alo_command[user_id]["prompt_id"]]["public"],
+                    }
+                }
+
+                set_user_setting(
+                    user_id, 
+                    prompts=config["prompts"]
+                )
+
+                return await bot.edit_message_text(
+                    text=f"текст промпта был изменён.",
+                    chat_id=alo_command[user_id]["chat_id"],
+                    message_id=alo_command[user_id]["message_id"],
+                    reply_markup=InlineKeyboardMarkup(
+                        inline_keyboard=[
+                            [
+                                InlineKeyboardButton(
+                                    text=f"✅ посмотреть", callback_data=f"vp%{user_id}%{alo_command[user_id]['prompt_id']}"
+                                ),
+                            ],
+                        ]
+                    )
+                )
+    return None