Initial commit

bot/discovery.py

@@ -0,0 +1,52 @@
+import logging
+from dataclasses import dataclass
+from typing import List, Optional
+
+import discord
+
+log = logging.getLogger(__name__)
+
+
+@dataclass
+class Target:
+    target_id: int
+    target_type: str  # "guild" | "group_dm"
+    name: str
+
+
+async def discover_targets(client: discord.Client, friend_id: int, manual_guild_ids: Optional[List[int]] = None) -> List[Target]:
+    targets: List[Target] = []
+    guild_ids_seen = set()
+
+    try:
+        friend = client.get_user(friend_id) or await client.fetch_user(friend_id)
+        profile = await friend.profile(with_mutual_guilds=True, with_mutual_friends=False, with_mutual_friends_count=False)
+        for mutual in (profile.mutual_guilds or []):
+            guild = client.get_guild(mutual.id)
+            name = guild.name if guild else f"Unknown Guild {mutual.id}"
+            targets.append(Target(mutual.id, "guild", name))
+            guild_ids_seen.add(mutual.id)
+    except discord.NotFound:
+        log.warning(
+            "No mutual guild/friend relationship visible for user %s "
+            "(profile lookup requires already being friends or sharing a server). "
+            "Relying on manual_guild_ids and group DMs only.",
+            friend_id,
+        )
+    except discord.HTTPException:
+        log.warning("Mutual-guilds profile lookup failed; falling back to manual_guild_ids only", exc_info=True)
+
+    for guild_id in (manual_guild_ids or []):
+        if guild_id in guild_ids_seen:
+            continue
+        guild = client.get_guild(guild_id)
+        name = guild.name if guild else f"Manual Guild {guild_id}"
+        targets.append(Target(guild_id, "guild", name))
+        guild_ids_seen.add(guild_id)
+
+    for channel in client.private_channels:
+        if isinstance(channel, discord.GroupChannel) and any(u.id == friend_id for u in channel.recipients):
+            name = channel.name or ", ".join(u.name for u in channel.recipients)
+            targets.append(Target(channel.id, "group_dm", name))
+
+    return targets