handle typing when already responded

cogs/mod.py

@@ -4179,6 +4179,7 @@ async def start_lockdown(
         records = []
         success, failures = [], []
         reason = f'Lockdown request by {ctx.author} (ID: {ctx.author.id})'
+        await ctx.defer()
         async with ctx.typing():
             for channel in channels:
                 overwrite = channel.overwrites_for(default_role)

utils/context.py

@@ -20,6 +20,8 @@
     from types import TracebackType
 
     from aiohttp import ClientSession
+    from discord.context_managers import Typing
+    from discord.ext.commands.context import DeferTyping
 
     from bot import Ayaka
 
@@ -28,6 +30,16 @@
 
 T = TypeVar('T')
 
+# async with typing() when already responded should early return
+
+
+class NoOpTyping:
+    async def __aenter__(self) -> None:
+        ...
+
+    async def __aexit__(self, *_: Any) -> None:
+        ...
+
 
 # For typing purposes, `Context.db` returns a Protocol type
 # that allows us to properly type the return values via narrowing
@@ -36,25 +48,33 @@
 
 
 class ConnectionContextManager(Protocol):
-    async def __aenter__(self) -> asyncpg.Connection: ...
+    async def __aenter__(self) -> asyncpg.Connection:
+        ...
 
     async def __aexit__(
         self, exc_type: type[BaseException] | None, exc_value: BaseException | None, traceback: TracebackType | None
-    ) -> None: ...
+    ) -> None:
+        ...
 
 
 class DatabaseProtocol(Protocol):
-    async def execute(self, query: str, *args: Any, timeout: float | None = None) -> str: ...
+    async def execute(self, query: str, *args: Any, timeout: float | None = None) -> str:
+        ...
 
-    async def fetch(self, query: str, *args: Any, timeout: float | None = None) -> list[asyncpg.Record]: ...
+    async def fetch(self, query: str, *args: Any, timeout: float | None = None) -> list[asyncpg.Record]:
+        ...
 
-    async def fetchrow(self, query: str, *args: Any, timeout: float | None = None) -> asyncpg.Record | None: ...
+    async def fetchrow(self, query: str, *args: Any, timeout: float | None = None) -> asyncpg.Record | None:
+        ...
 
-    async def fetchval(self, query: str, *args: Any, timeout: float | None = None) -> Any | None: ...
+    async def fetchval(self, query: str, *args: Any, timeout: float | None = None) -> Any | None:
+        ...
 
-    def acquire(self, *, timeout: float | None = None) -> ConnectionContextManager: ...
+    def acquire(self, *, timeout: float | None = None) -> ConnectionContextManager:
+        ...
 
-    def release(self, connection: asyncpg.Connection) -> None: ...
+    def release(self, connection: asyncpg.Connection) -> None:
+        ...
 
 
 class Context(commands.Context['Ayaka']):
@@ -162,6 +182,11 @@ async def safe_send(self, content: str, *, escape_mentions: bool = True, **kwarg
         else:
             return await self.send(content)
 
+    def typing(self, *, ephemeral: bool = False) -> Typing | DeferTyping | NoOpTyping:
+        if self.interaction and not self.interaction.response.is_done():
+            return super().typing(ephemeral=ephemeral)
+        return NoOpTyping()
+
 
 class GuildContext(Context):
     author: discord.Member