From 0bf5ae70ebc99354013883abe3962e59154e3218 Mon Sep 17 00:00:00 2001
From: Profitroll <47523801+profitrollgame@users.noreply.github.com>
Date: Sat, 7 Jan 2023 10:20:58 +0100
Subject: [PATCH] Improved inline query usage filters

---
 modules/inline.py | 28 ++++++++++++++++++----------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/modules/inline.py b/modules/inline.py
index d50ec8d..3f5f50a 100644
--- a/modules/inline.py
+++ b/modules/inline.py
@@ -11,7 +11,7 @@ from pyrogram.enums.chat_members_filter import ChatMembersFilter
 from dateutil.relativedelta import relativedelta
 from classes.errors.holo_user import UserNotFoundError, UserInvalidError
 from classes.holo_user import HoloUser
-from modules.utils import configGet, locale
+from modules.utils import configGet, jsonLoad, locale
 from modules.database import col_applications, col_spoilers
 from bson.objectid import ObjectId
 from bson.errors import InvalidId
@@ -64,19 +64,27 @@ async def inline_answer(client: Client, inline_query: InlineQuery):
         )
         return
 
+    results_forbidden = [
+        InlineQueryResultArticle(
+            title=locale("title", "inline", "forbidden", locale=inline_query.from_user),
+            input_message_content=InputTextMessageContent(
+                locale("message_content", "inline", "forbidden", locale=inline_query.from_user)
+            ),
+            description=locale("description", "inline", "forbidden", locale=inline_query.from_user)
+        )
+    ]
+
     try:
         holo_user = HoloUser(inline_query.from_user)
     except (UserNotFoundError, UserInvalidError):
         await inline_query.answer(
-        results=[
-            InlineQueryResultArticle(
-                title=locale("title", "inline", "forbidden", locale=inline_query.from_user),
-                input_message_content=InputTextMessageContent(
-                    locale("message_content", "inline", "forbidden", locale=inline_query.from_user)
-                ),
-                description=locale("description", "inline", "forbidden", locale=inline_query.from_user)
-                )
-            ]
+            results=results_forbidden
+        )
+        return
+
+    if path.exists(path.join(configGet("cache", "locations"), "group_members")) and (inline_query.from_user.id not in jsonLoad(path.join(configGet("cache", "locations"), "group_members"))):
+        await inline_query.answer(
+            results=results_forbidden
         )
         return