fast-link

Author: Sallat-M

.gitignore

@@ -0,0 +1,2 @@
+/.idea
+/__pycache__

FileToLink/__main__.py

@@ -38,14 +38,13 @@ async def main(_, msg: Message):
         # Else if the file not exist on the server, Send the message to Archive Channel and Create empty file
         gen_msg = await bot.send_message(msg.chat.id, Strings.generating_link,
                                          reply_to_message_id=msg.message_id)
-        worker = Worker(msg)
+
         archived_msg = await archive_msg(msg)
+        worker = Worker(archived_msg)
+        AllWorkers.add(worker)
 
-        archive_id = archived_msg.message_id
-        worker.set_link(archive_id)
-        AllWorkers.add(worker, archive_id)
-        if archive_id in NotFound:
-            NotFound.remove(archive_id)
+        if archived_msg.message_id in NotFound:
+            NotFound.remove(archived_msg.message_id)
 
         await worker.create_file()  # Create empty file
 
@@ -59,18 +58,14 @@ async def main(_, msg: Message):
     if worker.stream:
         st_link = f'{dl_link}?st=1'  # Stream Link
         buttons.append([InlineKeyboardButton(Strings.st_link, url=st_link)])
-
+    buttons.append([InlineKeyboardButton(Strings.update_link, callback_data=f'fast|{worker.archive_id}')])
     reply_markup = InlineKeyboardMarkup(buttons)
 
     if gen_msg is not None:
-        await bot.edit_message_text(msg.chat.id, gen_msg.message_id, text,
-                                    reply_markup=reply_markup,
-                                    disable_web_page_preview=True)
+        await gen_msg.edit_text(text, reply_markup=reply_markup, disable_web_page_preview=True)
     else:
-        await bot.send_message(msg.chat.id, text,
-                               reply_to_message_id=msg.message_id,
-                               reply_markup=reply_markup,
-                               disable_web_page_preview=True)
+        await bot.send_message(msg.chat.id, text, reply_to_message_id=msg.message_id,
+                               reply_markup=reply_markup, disable_web_page_preview=True)
 
 
 async def wait(chat_id: int):

FileToLink/archive.py

@@ -2,14 +2,13 @@
 from pyrogram.types import (Message, CallbackQuery, InlineKeyboardButton, InlineKeyboardMarkup, User)
 from pyrogram.errors.exceptions import ButtonDataInvalid
 
-from FileToLink import bot, Config
+from FileToLink import bot, Config, Strings
 
 
 async def archive_msg(msg: Message):
-    buttons = [[InlineKeyboardButton("🚫", callback_data=f'delete-file')]]
+    buttons = [[InlineKeyboardButton("🚫", callback_data='delete-file')]]
 
     forward = msg.forward_from or msg.forward_from_chat
-
     if forward:
         if forward.username:
             buttons[0].append(InlineKeyboardButton("↪", url=f'https://t.me/{forward.username}'))
@@ -50,5 +49,4 @@ async def from_info(_, cb: CallbackQuery):
 
 @bot.on_callback_query(filters.create(lambda _, __, cb: cb.data == 'time-out'))
 async def time_out(_, cb: CallbackQuery):
-    await cb.answer("The bot can't delete messages older than 48 hours,"
-                    " you can delete this message manually", show_alert=True)
+    await cb.answer(Strings.delete_forbidden, show_alert=True)

FileToLink/client.py

@@ -20,11 +20,8 @@ def __init__(
             bot_token: str = None,
             sleep_threshold: int = Session.SLEEP_THRESHOLD
     ):
-        super(TelegramClient, self).__init__(session_name,
-                                             api_id=api_id,
-                                             api_hash=api_hash,
-                                             bot_token=bot_token,
-                                             sleep_threshold=sleep_threshold)
+        super(TelegramClient, self).__init__(session_name, api_id=api_id, api_hash=api_hash,
+                                             bot_token=bot_token, sleep_threshold=sleep_threshold)
 
     async def get_part_file(
             self,
@@ -253,8 +250,5 @@ async def download_part(
         return rng
 
 
-bot = TelegramClient(Config.Session,
-                     api_id=Config.API_ID,
-                     api_hash=Config.API_HASH,
-                     bot_token=Config.Token,
-                     sleep_threshold=Config.Sleep_Threshold)
+bot = TelegramClient(Config.Session, api_id=Config.API_ID, api_hash=Config.API_HASH,
+                     bot_token=Config.Token, sleep_threshold=Config.Sleep_Threshold)

FileToLink/config.py

@@ -12,15 +12,17 @@ class Config:
     Port = int(os.environ.get("PORT"))
     Archive_Channel_ID = int(os.environ.get("ARCHIVE_CHANNEL_ID"))
     Start_Message = os.environ.get("Start_Message")
+
+    Link_Root = f"https://{App_Name}.herokuapp.com/"
+    Download_Folder = "Files"
     Bot_Channel = "shadow_bots"
     Bot_UserName = None  # The bot will set it after starting
     Part_size = 10 * 1024 * 1024  # (10MB) For Pyrogram
     Buffer_Size = 256 * 1024  # For Quart
     Pre_Dl = 3  # How many parts to download from telegram before client request them
     Separate_Time = 4  # (seconds)  wait time between messages if user send more than one
     Sleep_Threshold = 60  # (Seconds) sleep threshold for flood wait exceptions
-    Link_Root = f"https://{App_Name}.herokuapp.com/"
-    Download_Folder = "Files"
+    Max_Fast_Processes = 1  # How many links user can update them to fast links at the same time
 
 
 class Strings:
@@ -29,3 +31,15 @@ class Strings:
     st_link = "🎞 Stream LINK"
     generating_link = "**⏳ Generating Link...**"
     join_channel = "📢 Bot Channel"
+    fast = "⚡️**The link has been updated to a fast link**"
+    update_link = "⚡ Update To Fast Link"
+    update_limited = (f"⛔ You can update just {Config.Max_Fast_Processes} link in one time, "
+                      "please wait until previous update to complete")
+    re_update_link = "🔄 Re-Updating the link"
+    already_updated = "The link is already updated"
+    wait_update = "⏳ Updating the link..."
+    wait = "⏳ Please wait..."
+    progress = "⏳ Progress"
+    file_not_found = "⚠️File Not Found, Please resend it again"
+    delete_manually_button = "⚠️You can delete it"
+    delete_forbidden = "The bot can't delete messages older than 48 hours, you can delete this message manually"

FileToLink/server.py

@@ -68,6 +68,7 @@ async def download(archive_id: int):
         try:
             worker: Worker = await create_worker(archive_id)
         except (ValueError, MessageIdInvalid):
+            # This Message not found in Archive Channel
             NotFound.append(archive_id)
             return abort(404)  # Not Found
 

FileToLink/worker.py

@@ -4,22 +4,26 @@
 import os
 
 from pyrogram import filters
-from pyrogram.errors import MessageDeleteForbidden
+from pyrogram.errors import MessageDeleteForbidden, MessageIdInvalid
 from pyrogram.types import Message, CallbackQuery, InlineKeyboardMarkup, InlineKeyboardButton
 
-from FileToLink import bot, Config
+from FileToLink import bot, Config, Strings
 
 
 class Worker:
     def __init__(self, msg: Message):
+        """
+        :param msg: Message from Archive Channel
+        """
         if msg.empty:
             raise ValueError
         self.msg = msg
+        self.archive_id = msg.message_id
         self.media = (msg.video or msg.document or msg.photo or msg.audio or
                       msg.voice or msg.video_note or msg.sticker or msg.animation)
         self.size = self.media.file_size
         self.id = self.media.file_unique_id
-        self.link = None
+        self.link = f'{Config.Link_Root}dl/{self.archive_id}'
         self.current_dl: int = 0  # Number of currently downloading parts
 
         if hasattr(self.media, 'mime_type') and self.media.mime_type not in (None, ''):
@@ -53,11 +57,9 @@ def __init__(self, msg: Message):
         self.parts = [False for _ in
                       range(int(self.size / Config.Part_size) + (1 if self.size % Config.Part_size else 0))]
         self.done = False  # If All parts are downloaded
+        self.fast = False  # If User update to Fast Link
         AllWorkers.add(self)
 
-    def set_link(self, archive_id: int):
-        self.link = f'{Config.Link_Root}dl/{archive_id}'
-
     async def create_file(self):
         """
         Create empty file with same size of real file
@@ -151,36 +153,38 @@ def part_number(self, byte_number: int):
 
 class Workers:
     def __init__(self):
-        """
-        Store Workers by file_unique_id and by archive_id
-        """
+        """ Store Workers by file_unique_id and by archive_id """
         self.by_file_id = {}
         self.by_archive_id = {}
 
     def get(self, archive_id: int = None, file_id: str = None):
+        """Get the worker by archive_id or by file_id"""
         if archive_id is not None and archive_id in self.by_archive_id:
             return self.by_archive_id[archive_id]
         elif file_id is not None and file_id in self.by_file_id:
             return self.by_file_id[file_id]
         else:
             return None
 
-    def add(self, worker, archive_id: int = None):
+    def add(self, worker):
+        """Add the worker to <self.by_file_id> and <self.by_archive_id>"""
         if worker.id not in self.by_file_id:
             self.by_file_id[worker.id] = worker
-        if archive_id is not None and archive_id not in self.by_archive_id:
-            self.by_archive_id[archive_id] = worker
-
-    def remove(self, archive_id: int = None, file_id: str = None):
-        if archive_id is not None and archive_id in self.by_archive_id:
-            del self.by_file_id[self.by_archive_id[archive_id].id]
+        if worker.msg.message_id not in self.by_archive_id:
+            self.by_archive_id[worker.archive_id] = worker
+
+    def remove(self, archive_id: int):
+        """Remove the worker from <self.by_file_id> and <self.by_archive_id>"""
+        if archive_id in self.by_archive_id:
+            file_id = self.by_archive_id[archive_id].id
+            if file_id in self.by_file_id:
+                del self.by_file_id[file_id]
             del self.by_archive_id[archive_id]
-        elif file_id is not None and file_id in self.by_file_id:
-            del self.by_file_id[file_id]
 
 
 AllWorkers = Workers()
 NotFound = []  # Store IDs of messages that not exist in Archive Channel
+FastProcesses = {}  # {User_ID: Number_Of_Link_Updating}
 
 
 async def create_worker(archive_msg_id):
@@ -190,17 +194,84 @@ async def create_worker(archive_msg_id):
     if msg.empty or not msg.media:
         raise ValueError
     worker = Worker(msg)
-    AllWorkers.add(worker, archive_id=archive_msg_id)
+    AllWorkers.add(worker)
     await worker.create_file()
     return worker
 
 
+@bot.on_callback_query(filters.create(lambda _, __, cb: cb.data.split('|')[0] == 'fast'))
+async def update_to_fast_link(_, cb: CallbackQuery):
+    msg = cb.message
+    user_id = msg.chat.id
+    if user_id in FastProcesses and FastProcesses[user_id] >= Config.Max_Fast_Processes:
+        await cb.answer(Strings.update_limited, show_alert=True)
+        return
+
+    archive_id = int(cb.data.split('|')[1])
+    worker: Worker = AllWorkers.get(archive_id=archive_id)
+    if worker is None:
+        try:
+            worker = await create_worker(archive_id)
+        except (ValueError, MessageIdInvalid):
+            await cb.answer(Strings.file_not_found, show_alert=True)
+            return
+    if worker.fast:
+        await cb.answer(Strings.already_updated, show_alert=True)
+        return
+
+    buttons = cb.message.reply_markup.inline_keyboard
+    update_button_row = -1  # Last Row
+    old_data = buttons[update_button_row][0].callback_data
+    buttons[update_button_row][0].text = Strings.wait_update
+    new_data = f"fast-prog|{archive_id}"
+    buttons[update_button_row][0].callback_data = new_data
+
+    await cb.answer(Strings.wait)
+    await cb.message.edit_reply_markup(InlineKeyboardMarkup(buttons))
+    progress = await msg.reply_text(
+        Strings.wait_update, reply_to_message_id=msg.message_id,
+        reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton(Strings.progress, callback_data=new_data)]]))
+
+    if user_id in FastProcesses:
+        FastProcesses[user_id] += 1
+    else:
+        FastProcesses[user_id] = 1
+
+    await worker.dl_all()
+    worker.fast = True
+
+    FastProcesses[user_id] -= 1
+
+    buttons[update_button_row][0].text = Strings.re_update_link
+    buttons[update_button_row][0].callback_data = old_data
+    await msg.edit_reply_markup(InlineKeyboardMarkup(buttons))
+    await progress.edit_text(Strings.fast)
+
+
+@bot.on_callback_query(filters.create(lambda _, __, cb: cb.data.split('|')[0] == 'fast-prog'))
+async def fast_progress(_, cb: CallbackQuery):
+    archive_id = int(cb.data.split('|')[1])
+    worker = AllWorkers.get(archive_id=archive_id)
+    if worker is None:
+        await cb.answer(Strings.file_not_found, show_alert=True)
+        return
+    downloaded = len([i for i in worker.parts if i])
+    total = len(worker.parts)
+    await cb.answer(progress_bar(downloaded, total), show_alert=True)
+
+
 @bot.on_callback_query(filters.create(lambda _, __, cb: cb.data == 'delete-file'))
 async def delete_file_handler(_, cb: CallbackQuery):
     msg = cb.message
     AllWorkers.remove(msg.message_id)
     try:
         await msg.delete()
     except MessageDeleteForbidden:
-        button = InlineKeyboardButton("⚠️You can delete it", callback_data='time-out')
+        button = InlineKeyboardButton(Strings.delete_manually_button, callback_data='time-out')
         await msg.edit_reply_markup(InlineKeyboardMarkup([[button]]))
+
+
+def progress_bar(current, total, length=16, finished='█', unfinished='░'):
+    rate = current / total
+    finished_len = int(length * rate) if rate <= 1 else length
+    return f'{finished * finished_len}{unfinished * (length - finished_len)} {int(rate * 100)}%'

README.md

@@ -24,7 +24,7 @@ Press the below button to deploy to Heroku
 
 `ARCHIVE_CHANNEL_ID` : Create a new channel (private/public), post anything in the channel, forward the post to @message_ids_bot, now copy and paste chat_forward in this field. 
 
-`Session_String` : [Not Required] Use session_generator.py to generate Pyrogram Session String, If you don't add it the bot will create a new session every Restarting.
+`Session_String` : [Optional] Use session_generator.py to generate Pyrogram Session String, If you don't add it the bot will create a new session every Restarting.
 
 
 ## How to use the bot

app.json

@@ -32,7 +32,7 @@
             "value": "Hello, Send me any file to get download and Stream links"
         },
         "Session_String": {
-            "description": "[Not Required] Pyrogram Session String, use session_generator.py to generate it, If you don't add it the bot will create a new session every Restarting.",
+            "description": "[Optional] Pyrogram Session String, use session_generator.py to generate it, If you don't add it the bot will create a new session every Restarting.",
             "required": false
         }
     },