fixed dm issue

Author: kvsxxx

discord_ui/client.py

@@ -169,6 +169,8 @@ async def _on_slash_response(self, msg):
             return
 
         user = discord.Member(data=data["member"], guild=self._discord._connection._get_guild(int(data["guild_id"])), state=self._discord._connection) if data.get("member") is not None else discord.User(state=self._discord._connection, data=data["user"])
+        if user.dm_channel is None:
+            await user.create_dm()
 
         interaction = Interaction(self._discord._connection, data, user)
         if self.auto_defer[0] is True:
@@ -1320,6 +1322,8 @@ async def _on_component_response(self, msg):
             return
 
         user = discord.Member(data=data["member"], guild=self._discord._connection._get_guild(int(data["guild_id"])), state=self._discord._connection) if data.get("member") is not None else discord.User(state=self._discord._connection, data=data["user"])
+        if user.dm_channel is None:
+            await user.create_dm()
         msg = await getMessage(self._discord._connection, data=data, response=True)
 
         interaction = Interaction(self._discord._connection, data, user, msg)

discord_ui/receive.py

@@ -68,7 +68,7 @@ def channel(self) -> Union[discord.TextChannel, discord.DMChannel]:
         
         :type: :class:`discord.TextChannel` | :class:`discord.DMChannel`
         """
-        return self._state.get_channel(self.channel_id)
+        return self._state.get_channel(self.channel_id) or self._state.get_channel(self.author.id)
 
     async def defer(self, hidden=False):
         """
@@ -376,15 +376,17 @@ async def getMessage(state: ConnectionState, data, response = True):
     :class:`~Message` | :class:`~EphemeralMessage`
         The sent message
     """
-    channel = state.get_channel(int(data["channel_id"])) or state._get_private_channel_by_user(data["author"]["id"])
+    msg_base = data.get("message", data)
+
+    channel = state.get_channel(int(data["channel_id"])) or state.get_channel(int(msg_base["author"]["id"]))
     if response:
-        if data.get("message") is not None and data.get("message", data)["flags"] == 64:
+        if data.get("message") is not None and msg_base["flags"] == 64:
             return EphemeralResponseMessage(state=state, channel=channel, data=data.get("message", data))
-        return Message(state=state, channel=channel, data=data.get("message", data))
+        return Message(state=state, channel=channel, data=msg_base)
 
-    if data.get("message") is not None and data["message"]["flags"] == 64:
-        return EphemeralMessage(state=state, channel=channel, data=data.get("message", data))
-    return Message(state=state, channel=channel, data=data.get("message", data))
+    if data.get("message") is not None and msg_base["flags"] == 64:
+        return EphemeralMessage(state=state, channel=channel, data=msg_base)
+    return Message(state=state, channel=channel, data=msg_base)
 
 class Message(discord.Message):
     """A fixed :class:`discord.Message` optimized for components"""

discord_ui/slash/tools.py

@@ -52,7 +52,7 @@ def resolve(data, _state):
             resolved["messages"] = {}
             for message_id in data["data"]["resolved"]["messages"]:
                 message_data = data["data"]["resolved"]["messages"][message_id]
-                resolved["messages"][message_id] = discord.Message(data=message_data, channel=_state._get_channel(data["channel_id"]), state=_state)
+                resolved["messages"][message_id] = discord.Message(data=message_data, channel=_state.get_channel(data["channel_id"]), state=_state)
         elif x == "channels":
             resolved["channels"] = {}
             for channel_id in data["data"]["resolved"]["channels"]:
@@ -123,7 +123,7 @@ def cache_data(value, typ, data, _state):
     elif typ == OptionType.MEMBER:
         return _state._get_guild(int(data["guild_id"])).get_member(int(value))
     elif typ == OptionType.CHANNEL:
-        return _state._get_channel(int(value))
+        return _state.get_channel(int(value))
     elif typ == OptionType.ROLE:
         return _state._get_guild(int(data["guild_id"])).get_role(int(value))
     elif typ == AdditionalType.MESSAGE: