TBD: Manager owner

aiogram_dialog/context/intent_filter.py

@@ -13,7 +13,7 @@
 from .events import DialogUpdateEvent
 from .storage import StorageProxy
 from ..exceptions import InvalidStackIdError, OutdatedIntent
-from ..utils import remove_indent_id, get_chat
+from ..utils import remove_indent_id, get_chat, get_owner_id
 
 STORAGE_KEY = "aiogd_storage_proxy"
 STACK_KEY = "aiogd_stack"
@@ -51,9 +51,10 @@ async def on_pre_process_message(self,
                                      event: Union[Message, ChatMemberUpdated],
                                      data: dict):
         chat = get_chat(event)
+        owner = data['dialog_manager'].owner
         proxy = StorageProxy(
             storage=self.storage,
-            user_id=event.from_user.id,
+            user_id=get_owner_id(event, owner),
             chat_id=chat.id,
             state_groups=self.state_groups,
         )
@@ -74,9 +75,10 @@ async def on_post_process_message(self, _, result, data: dict):
     async def on_pre_process_aiogd_update(self, event: DialogUpdateEvent,
                                           data: dict):
         chat = get_chat(event)
+        owner = data['dialog_manager'].owner
         proxy = StorageProxy(
             storage=self.storage,
-            user_id=event.from_user.id,
+            user_id=get_owner_id(event, owner),
             chat_id=chat.id,
             state_groups=self.state_groups,
         )
@@ -112,9 +114,10 @@ async def on_pre_process_aiogd_update(self, event: DialogUpdateEvent,
     async def on_pre_process_callback_query(self, event: CallbackQuery,
                                             data: dict):
         chat = get_chat(event)
+        owner = data['dialog_manager'].owner
         proxy = StorageProxy(
             storage=self.storage,
-            user_id=event.from_user.id,
+            user_id=get_owner_id(event, owner),
             chat_id=chat.id,
             state_groups=self.state_groups,
         )
@@ -161,10 +164,10 @@ async def on_pre_process_error(self, update: Update, error: Exception,
             return
 
         chat = get_chat(event)
-
+        owner = data['dialog_manager'].owner
         proxy = StorageProxy(
             storage=self.storage,
-            user_id=event.from_user.id,
+            user_id=get_owner_id(event, owner),
             chat_id=chat.id,
             state_groups=self.state_groups,
         )

aiogram_dialog/manager/manager.py

@@ -9,7 +9,7 @@
 from .protocols import (
     DialogManager, BaseDialogManager, ShowMode, LaunchMode,
     ManagedDialogAdapterProto, ManagedDialogProto, DialogRegistryProto,
-    NewMessage,
+    NewMessage, Owner,
 )
 from ..context.context import Context
 from ..context.events import ChatEvent, StartMode, Data, FakeChat, FakeUser
@@ -30,6 +30,7 @@ def __init__(self, event: ChatEvent, registry: DialogRegistryProto,
         self.event = event
         self.data = data
         self.show_mode: ShowMode = ShowMode.AUTO
+        self.owner: Owner = Owner.USER
 
     @property
     def registry(self) -> DialogRegistryProto:

aiogram_dialog/manager/protocols.py

@@ -19,6 +19,17 @@ class ShowMode(Enum):
     SEND = "send"
 
 
+class Owner(Enum):
+    """
+    `GROUP` any users in a group can use this dialog
+
+    `USER` only one user can use this dialog
+    """
+
+    GROUP = "group"
+    USER = "user"
+
+
 class LaunchMode(Enum):
     """
     `ROOT` dialogs will be always a root dialog in stack.
@@ -213,6 +224,7 @@ class DialogManager(BaseDialogManager):
     event: ChatEvent  # current processing event
     data: Dict  # data from middleware
     show_mode: ShowMode  # mode used to show messages
+    owner: Owner  # mode who can use dialogs
 
     def is_preview(self) -> bool:
         raise NotImplementedError

aiogram_dialog/utils.py

@@ -14,7 +14,7 @@
 from .context.events import (
     DialogUpdateEvent, ChatEvent
 )
-from .manager.protocols import MediaAttachment, NewMessage, ShowMode, MediaId
+from .manager.protocols import MediaAttachment, NewMessage, ShowMode, MediaId, Owner
 
 logger = getLogger(__name__)
 
@@ -38,6 +38,15 @@ def get_chat(event: ChatEvent) -> Chat:
         return event.message.chat
 
 
+def get_owner_id(event: ChatEvent, owner: Owner) -> int:
+    if owner == Owner.GROUP:
+        return get_chat(event).id
+    elif owner == Owner.USER:
+        return event.from_user.id
+    else:
+        raise ValueError('Unknown owner type %s', owner)
+
+
 def is_chat_loaded(chat: Chat) -> bool:
     """
     Checks if chat is correctly loaded from telegram.