add rules preview & separate prisma funcs

ryukaizen · Mar 8, 2024 · 442c095 · 442c095
1 parent da01c17
commit 442c095
Show file tree

Hide file tree

Showing 2 changed files with 76 additions and 50 deletions.
diff --git a/src/database/rules_sql.ts b/src/database/rules_sql.ts
@@ -7,4 +7,42 @@ export async function get_rules(chatId: string) {
         }
     });
     return rules?.rules;
+};
+
+export async function set_rules(chatId: string, rulesText: string) {
+    try {
+        let rules = await prisma.rules.upsert({
+            where: {
+                chat_id: chatId,
+            },
+            update: {
+                rules: rulesText,
+            },
+            create: {
+                chat_id: chatId,
+                rules: rulesText,
+            }
+        });
+        return true;
+    }
+    catch (e) {
+        console.error(e);
+        return false;
+    }
+};
+
+export async function reset_rules(chatId: string) {
+    try {
+        let resetRules = await prisma.rules.delete({
+            where: {
+                chat_id: chatId,
+            }
+        });
+        return true;
+    }
+    catch (e) {
+        console.error(e);
+        return false;
+    }
+
 }
diff --git a/src/modules/rules.ts b/src/modules/rules.ts
@@ -1,11 +1,10 @@
 import bot from "../bot";
 import constants from "../config";
-import { prisma } from "../database";
-import { get_rules } from "../database/rules_sql";
+import { get_rules, set_rules, reset_rules } from "../database/rules_sql";
 import { InlineKeyboard } from "grammy";
 import { elevatedUsersOnly } from "../helpers/helper_func";
 
-bot.chatType("supergroup" || "group").command("rules", (async(ctx: any) => {
+bot.chatType("supergroup" || "group").command(["rule", "rules"], (async(ctx: any) => {
     let chatId = ctx.chat.id.toString();
     let rules_button = new InlineKeyboard()
         .url("Rules❕", `https://telegram.me/${constants.BOT_USERNAME}?start=rules_${chatId}`);
@@ -24,68 +23,57 @@ bot.chatType("supergroup" || "group").command("rules", (async(ctx: any) => {
     }
 }));
 
-bot.chatType("supergroup" || "group").command("setrules", elevatedUsersOnly(async(ctx: any) => {
+bot.chatType("supergroup" || "group").command(["setrule", "setrules", "addrule", "addrules"], elevatedUsersOnly(async(ctx: any) => {
     let chatId = ctx.chat.id.toString();
-    if (ctx.message.text.replace("/setrules" || "!setrules" || "?setrules", "").trim() == "") {
-        if (ctx.message.reply_to_message != undefined) {
-            await prisma.rules.upsert({
-                where: {
-                    chat_id: chatId,
-                },
-                update: {
-                    rules: ctx.message.reply_to_message.text,
-                },
-                create: {
-                    chat_id: chatId,
-                    rules: ctx.message.reply_to_message.text,
-                }
-        });
-        await ctx.reply("Rules have been set!");
+
+    const rulesPreview = new InlineKeyboard()
+    .url("Show Preview", `https://telegram.me/${constants.BOT_USERNAME}?start=rules_${chatId}`);
+
+    if (ctx.message.reply_to_message != undefined) {
+        let setRules = await set_rules(chatId, ctx.message.reply_to_message.text)
+        if (setRules == true) {
+            await ctx.reply("Rules have been set!", {reply_markup: rulesPreview, reply_parameters: {message_id: ctx.message.message_id}});
         }
         else {
-            await ctx.reply("Please type the rules next to /setrules command or reply to a message with /setrules command.", {reply_parameters: {message_id: ctx.message.message_id}});
-            return;
+            await ctx.reply("An error occurred while setting the rules. Please re-check the provided data.", {reply_parameters: {message_id: ctx.message.message_id}});
         }
     }
     else {
-        let rules_text = ctx.message.text.replace("/setrules" || "!setrules" || "?setrules", "").trim();
-        if (rules_text.length > 4096) {
-            rules_text = rules_text.substring(0, 4096);
-            await prisma.rules.upsert({
-                where: {
-                    chat_id: chatId,
-                },
-                update: {
-                    rules: rules_text,
-                },
-                create: {
-                    chat_id: chatId,
-                    rules: rules_text,
-                }
-            });
+        let args = ctx.match;
+        if (args) {
+            let rules_text = args
+            if (rules_text.length > 4096) {
+                rules_text = rules_text.substring(0, 4096);
+            }
+            let setRules = await set_rules(chatId, rules_text)
+            if (setRules == true) {
+                await ctx.reply("Rules have been set!", {reply_markup: rulesPreview, reply_parameters: {message_id: ctx.message.message_id}});
+            }
+            else {
+                await ctx.reply("An error occurred while setting the rules. Please re-check the provided data.", {reply_parameters: {message_id: ctx.message.message_id}});
+            }
+        }
+        else {
+            await ctx.reply("Please type the rules next to /setrules command or reply to a message with /setrules command.", {reply_parameters: {message_id: ctx.message.message_id}});
+            return;
         }
-        await ctx.reply("Rules have been set!", {reply_parameters: {message_id: ctx.message.message_id}});
     }
 }));
 
-bot.chatType("supergroup" || "group").command("resetrules", elevatedUsersOnly(async(ctx: any) => {
+bot.chatType("supergroup" || "group").command(["resetrule", "resetrules", "rmrule", "rmrules"], elevatedUsersOnly(async(ctx: any) => {
     let chatId = ctx.chat.id.toString();
-    // check if rules have been set or not, and only then delete
-    let rules = await prisma.rules.findUnique({
-        where: {
-            chat_id: chatId,
-        }
-    });
+    let rules = await get_rules(chatId);
     if (rules == null) {
         await ctx.reply("No rules have been set yet!", {reply_parameters: {message_id: ctx.message.message_id}});
         return;
     }
     else {
-        await prisma.rules.delete({
-            where: {
-                chat_id: chatId,
-            }
-    });
-    await ctx.reply("Rules have been reset!", {reply_parameters: {message_id: ctx.message.message_id}});
+        let resetRules = await reset_rules(chatId);
+        if (resetRules == true) {
+            await ctx.reply("Rules have been reset!", {reply_parameters: {message_id: ctx.message.message_id}});
+        }
+        else {
+            await ctx.reply("An error occurred while resetting the rules.", {reply_parameters: {message_id: ctx.message.message_id}});
+        }       
     }
 }));