fix(userbot): presence, story and profile history toggles of the capture policy take effect for each user peer

backend/src/userbot/modules/capture/context.py

@@ -6,7 +6,7 @@ from userbot.modules.capture.pending import PendingDeletes
 from userbot.modules.contacts import ContactCache
 from userbot.modules.folders import FolderCache
 from userbot.modules.watches import WatchCache
-from utils.policy.models import CaptureToggles, ChatMeta, PolicySet
+from utils.policy.models import CaptureToggles, ChatKind, ChatMeta, PolicySet
 from utils.policy.repository import load_policy_set
 from utils.policy.resolver import resolve
 from utils.storage import ContentAddressedStorage
@@ -38,6 +38,16 @@ class CaptureContext:
     def resolve(self, chat: ChatMeta) -> CaptureToggles:
         return resolve(chat, self.folders.folders, self.policies)
 
+    def resolve_peer(self, peer_id: int, *, is_bot: bool = False) -> CaptureToggles:
+        return self.resolve(
+            ChatMeta(
+                chat_id=peer_id,
+                kind=ChatKind.DM,
+                is_bot=is_bot,
+                is_contact=peer_id in self.contacts.ids,
+            )
+        )
+
 
 async def build_capture_context(
     client: Client,

backend/src/userbot/modules/capture/identity.py

@@ -58,7 +58,14 @@ async def _capture_peer(message: Message, ctx: CaptureContext) -> None:
     fields, photo_file_id, photo_unique_id = snapshot_from_high_level(user)
     if not await ctx.peer_identity.changed(ctx.pool, ctx.account_id, user.id, fields):
         return
-    await write_profile(ctx.pool, ctx.account_id, user.id, fields, raw_json(user))
+    await write_profile(
+        ctx.pool,
+        ctx.account_id,
+        user.id,
+        fields,
+        raw_json(user),
+        history=ctx.resolve_peer(user.id, is_bot=bool(user.is_bot)).profile_history,
+    )
     if photo_file_id and photo_unique_id:
         from userbot.modules.avatars import note_avatar  # noqa: PLC0415
 

backend/src/userbot/modules/jobs/handlers/enrich_chat.py

@@ -27,7 +27,14 @@ LIMIT 100
 
 async def _save_user(ctx: CaptureContext, user: User) -> None:
     fields, photo_file_id, photo_unique_id = snapshot_from_high_level(user)
-    await write_profile(ctx.pool, ctx.account_id, user.id, fields, raw_json(user))
+    await write_profile(
+        ctx.pool,
+        ctx.account_id,
+        user.id,
+        fields,
+        raw_json(user),
+        history=ctx.resolve_peer(user.id, is_bot=bool(user.is_bot)).profile_history,
+    )
     if photo_file_id and photo_unique_id:
         await note_avatar(
             ctx.pool, ctx.account_id, user.id, "peer", photo_unique_id, photo_file_id

backend/src/userbot/modules/jobs/handlers/sync_contacts.py

@@ -22,7 +22,16 @@ async def sync_contacts(ctx: JobContext) -> None:
         if not isinstance(user, User):
             continue
         fields, photo_file_id, photo_unique_id = snapshot_from_high_level(user)
-        await write_profile(ctx.pool, ctx.account_id, user.id, fields, raw_json(user))
+        await write_profile(
+            ctx.pool,
+            ctx.account_id,
+            user.id,
+            fields,
+            raw_json(user),
+            history=capture.resolve_peer(
+                user.id, is_bot=bool(user.is_bot)
+            ).profile_history,
+        )
         if photo_file_id and photo_unique_id:
             await note_avatar(
                 ctx.pool,

backend/src/userbot/modules/jobs/handlers/sync_dialogs.py

@@ -34,7 +34,14 @@ async def _enrich_users(client: Client, ctx: CaptureContext, ids: list[int]) ->
                 continue
             fields, photo_file_id, photo_unique_id = snapshot_from_high_level(user)
             await write_profile(
-                ctx.pool, ctx.account_id, user.id, fields, raw_json(user)
+                ctx.pool,
+                ctx.account_id,
+                user.id,
+                fields,
+                raw_json(user),
+                history=ctx.resolve_peer(
+                    user.id, is_bot=bool(user.is_bot)
+                ).profile_history,
             )
             if photo_file_id and photo_unique_id:
                 await note_avatar(

backend/src/userbot/modules/profiles/repository.py

@@ -49,8 +49,14 @@ async def get_peer(
     )
 
 
-async def write_profile(
-    pool: asyncpg.Pool, account_id: int, peer_id: int, fields: ProfileFields, raw: str
+async def write_profile(  # noqa: PLR0913
+    pool: asyncpg.Pool,
+    account_id: int,
+    peer_id: int,
+    fields: ProfileFields,
+    raw: str,
+    *,
+    history: bool,
 ) -> None:
     args = (
         fields.first_name,
@@ -63,4 +69,5 @@ async def write_profile(
     )
     async with pool.acquire() as conn, conn.transaction():
         await conn.execute(_UPSERT_PEER, account_id, peer_id, *args)
-        await conn.execute(_INSERT_HISTORY, account_id, peer_id, *args)
+        if history:
+            await conn.execute(_INSERT_HISTORY, account_id, peer_id, *args)

backend/src/userbot/modules/profiles/snapshots.py

@@ -1,5 +1,6 @@
 from datetime import UTC, datetime
 
+from pyrogram.enums import ChatType
 from pyrogram.types import Chat
 
 from userbot.modules.avatars import note_avatar
@@ -13,7 +14,15 @@ from userbot.modules.raw_json import raw_json
 async def save_private(ctx: CaptureContext, chat: Chat) -> bool:
     chat_id = chat.id or 0
     fields, photo_file_id, photo_unique_id = snapshot_from_chat(chat)
-    await write_profile(ctx.pool, ctx.account_id, chat_id, fields, raw_json(chat))
+    is_bot = chat.type is ChatType.BOT
+    await write_profile(
+        ctx.pool,
+        ctx.account_id,
+        chat_id,
+        fields,
+        raw_json(chat),
+        history=ctx.resolve_peer(chat_id, is_bot=is_bot).profile_history,
+    )
     if photo_file_id and photo_unique_id:
         await note_avatar(
             ctx.pool, ctx.account_id, chat_id, "peer", photo_unique_id, photo_file_id