From 84461290033bfb72111c05c31cf96f6470d1ab95 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Sat, 14 Apr 2018 19:48:15 +0200
Subject: [PATCH] Enhance send_document by accepting file_ids and URLs

---
 pyrogram/client/client.py | 59 +++++++++++++++++++++++++++++++--------
 1 file changed, 48 insertions(+), 11 deletions(-)

diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
index 51f63896..90cdac10 100644
--- a/pyrogram/client/client.py
+++ b/pyrogram/client/client.py
@@ -1394,7 +1394,9 @@ class Client:
 
             document (``str``):
                 File to send.
-                Pass a file path as string to send a file that exists on your local machine.
+                Pass a file_id as string to send a file that exists on the Telegram servers,
+                pass an HTTP URL as a string for Telegram to get a file from the Internet, or
+                pass a file path as string to upload a new file that exists on your local machine.
 
             caption (``str``, optional):
                 Document caption, 0-200 characters.
@@ -1423,26 +1425,56 @@ class Client:
                 The size of the file.
 
         Returns:
-            On success, the sent Message is returned.
+            On success, the sent :obj:`Message <pyrogram.Message>` is returned.
 
         Raises:
             :class:`Error <pyrogram.Error>`
         """
+        file = None
         style = self.html if parse_mode.lower() == "html" else self.markdown
-        file = self.save_file(document, progress=progress)
+
+        if os.path.exists(document):
+            file = self.save_file(document, progress=progress)
+            media = types.InputMediaUploadedDocument(
+                mime_type=mimetypes.types_map.get("." + document.split(".")[-1], "text/plain"),
+                file=file,
+                attributes=[
+                    types.DocumentAttributeFilename(os.path.basename(document))
+                ]
+            )
+        elif document.startswith("http"):
+            media = types.InputMediaDocumentExternal(
+                url=document
+            )
+        else:
+            try:
+                decoded = utils.decode(document)
+                fmt = "<iiqqqqi" if len(decoded) > 24 else "<iiqq"
+                unpacked = struct.unpack(fmt, decoded)
+            except (AssertionError, binascii.Error, struct.error):
+                raise FileIdInvalid from None
+            else:
+                if unpacked[0] not in (5, 10):
+                    media_type = Client.MEDIA_TYPE_ID.get(unpacked[0], None)
+
+                    if media_type:
+                        raise FileIdInvalid("The file_id belongs to a {}".format(media_type))
+                    else:
+                        raise FileIdInvalid("Unknown media type: {}".format(unpacked[0]))
+
+                media = types.InputMediaDocument(
+                    id=types.InputDocument(
+                        id=unpacked[2],
+                        access_hash=unpacked[3]
+                    )
+                )
 
         while True:
             try:
                 r = self.send(
                     functions.messages.SendMedia(
                         peer=self.resolve_peer(chat_id),
-                        media=types.InputMediaUploadedDocument(
-                            mime_type=mimetypes.types_map.get("." + document.split(".")[-1], "text/plain"),
-                            file=file,
-                            attributes=[
-                                types.DocumentAttributeFilename(os.path.basename(document))
-                            ]
-                        ),
+                        media=media,
                         silent=disable_notification or None,
                         reply_to_msg_id=reply_to_message_id,
                         random_id=self.rnd_id(),
@@ -1452,7 +1484,12 @@ class Client:
             except FilePartMissing as e:
                 self.save_file(document, file_id=file.id, file_part=e.x)
             else:
-                return r
+                for i in r.updates:
+                    if isinstance(i, (types.UpdateNewMessage, types.UpdateNewChannelMessage)):
+                        users = {i.id: i for i in r.users}
+                        chats = {i.id: i for i in r.chats}
+
+                        return message_parser.parse_message(self, i.message, users, chats)
 
     def send_sticker(self,
                      chat_id: int or str,