Upload 6 files

unzipper/modules/__init__.py

@@ -0,0 +1,11 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #

unzipper/modules/admin.py

@@ -0,0 +1,137 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from asyncio import sleep
+from shutil import disk_usage as sdisk_usage
+
+from config import Config
+from pyrogram import filters
+from unzipper import unzip_client
+from pyrogram.types import Message
+
+from pyrogram.errors import FloodWait
+from unzipper.helpers_nexa.utils import humanbytes
+
+from unzipper.database.users import (add_banned_user, count_banned_users,
+                                     count_users, del_banned_user, del_user,
+                                     get_users_list)
+from psutil import cpu_percent, disk_usage, net_io_counters, virtual_memory
+
+
+@unzip_client.on_message(filters.private & filters.command("stats"))
+@unzip_client.handle_erros
+async def send_stats(_, message: Message, texts):
+    stats_msg = await message.reply(texts["processing"])
+    # Is message from owner?
+    frmow = False
+    if message.from_user and message.from_user.id == Config.BOT_OWNER:
+        frmow = True
+    # Disk usage
+    total, used, free = sdisk_usage(".")
+    total = humanbytes(total)
+    used = humanbytes(used)
+    free = humanbytes(free)
+    # Hardware usage
+    cpu_usage = cpu_percent()
+    ram_usage = virtual_memory().percent
+    cdisk_usage = disk_usage('/').percent
+    # Bandwith usage
+    net_usage = net_io_counters()
+    # Users count
+    total_users = await count_users()
+    total_banned_users = await count_banned_users()
+    usrtxt = f"""
+**👥 Users:** 
+ ↳**Users in Database:** `{total_users}`
+ ↳**Total Banned Users:** `{total_banned_users}`
+
+"""
+    # Show status
+    await stats_msg.edit(f"""
+**💫 Current Bot Stats 💫**
+{usrtxt if frmow else ""}
+**🌐 Bandwith Usage,**
+ ↳ **Sent:** `{humanbytes(net_usage.bytes_sent)}`
+ ↳ **Received:** `{humanbytes(net_usage.bytes_recv)}`
+
+
+**💾 Disk Usage,**
+ ↳**Total Disk Space:** `{total}`
+ ↳**Used:** `{used}({cdisk_usage}%)`
+ ↳**Free:** `{free}`
+
+
+**🎛 Hardware Usage,**
+ ↳**CPU Usage:** `{cpu_usage}%`
+ ↳**RAM Usage:** `{ram_usage}%`""")
+
+
+async def _do_broadcast(message, user):
+    try:
+        await message.copy(chat_id=int(user))
+        return 200
+    except FloodWait as e:
+        await sleep(e.x)
+        return _do_broadcast(message, user)
+    except Exception:
+        await del_user(int(user))
+
+
+@unzip_client.on_message(filters.private & filters.command("broadcast") & filters.user(Config.BOT_OWNER))
+@unzip_client.handle_erros
+async def broadcast_dis(_, message: Message, texts):
+    bc_msg = await message.reply(texts["processing"])
+    r_msg = message.reply_to_message
+    if not r_msg:
+        return await bc_msg.edit(texts["no_replied_msg"])
+    # Starting the broadcast
+    await bc_msg.edit(texts["broadcast_started"])
+    success_no = 0
+    failed_no = 0
+    total_users = await count_users()
+    async for user in await get_users_list():
+        b_cast = await _do_broadcast(r_msg, user)
+        if b_cast == 200:
+            success_no += 1
+        else:
+            failed_no += 1
+    await bc_msg.edit((texts["boradcast_results"]).format(total_users, success_no, failed_no))
+
+
+@unzip_client.on_message(filters.private & filters.command("ban") & filters.user(Config.BOT_OWNER))
+@unzip_client.handle_erros
+async def ban_user(_, message: Message, texts):
+    ban_msg = await message.reply(texts["processing"])
+    try:
+        user_id = message.text.split(None, 1)[1]
+    except:
+        return await ban_msg.edit(texts["no_userid"])
+    # Return if user_id string is not numeric
+    if not user_id.isnumeric():
+        return await ban_msg.edit(texts["no_userid"])
+    await add_banned_user(int(user_id))
+    await ban_msg.edit(texts["ok_ban"].format(user_id))
+
+
+@unzip_client.on_message(filters.private & filters.command("unban") & filters.user(Config.BOT_OWNER))
+@unzip_client.handle_erros
+async def unban_user(_, message: Message, texts):
+    unban_msg = await message.reply(texts["processing"])
+    try:
+        user_id = message.text.split(None, 1)[1]
+    except:
+        return await unban_msg.edit(texts["no_userid"])
+    # Return if user_id string is not numeric
+    if not user_id.isnumeric():
+        return await unban_msg.edit(texts["no_userid"])
+    await del_banned_user(int(user_id))
+    await unban_msg.edit(texts["ok_unban"].format(user_id))

unzipper/modules/callbacks.py

@@ -0,0 +1,249 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+import logging
+
+from time import time
+from shutil import rmtree
+from os import makedirs, path
+
+from config import Config
+from aiohttp import ClientSession
+from unzipper import unzip_client, Buttons
+from unzipper.client.caching import USER_LANG
+
+from pyrogram.types import CallbackQuery
+from unzipper.database.cloud import GofileDB
+from pyrogram.errors import ReplyMarkupTooLong
+from unzipper.database.language import set_language
+from unzipper.database.upload_mode import set_upload_mode
+
+from unzipper.helpers_nexa.utils import (TimeFormatter, get_files, humanbytes,
+                                         progress_for_pyrogram)
+from unzipper.database.split_arc import add_split_arc_user, del_split_arc_user
+from unzipper.lib.downloader import Downloader
+from unzipper.lib.backup_tool import CloudBackup
+from unzipper.lib.extractor import Extractor, ExtractionFailed
+
+
+# Callbacks
+@unzip_client.on_callback_query()
+@unzip_client.handle_query
+async def unzipper_cb(_, query: CallbackQuery, texts):
+    qdat = query.data
+
+    if qdat == "megoinhome":
+        await query.edit_message_text(texts["start"].format(query.from_user.mention), reply_markup=Buttons.START)
+
+    elif qdat == "helpcallback":
+        await query.edit_message_text(texts["help_head"], reply_markup=Buttons.HELP)
+
+    elif qdat == "extracthelp":
+        await query.edit_message_text(texts["help_extract"], reply_markup=Buttons.HELP_BACK)
+
+    elif qdat == "upmodhelp":
+        await query.edit_message_text(texts["help_upmode"], reply_markup=Buttons.HELP_BACK)
+
+    elif qdat == "backuphelp":
+        await query.edit_message_text(texts["help_backup"], reply_markup=Buttons.HELP_BACK)
+
+    elif qdat == "thumbhelp":
+        await query.edit_message_text(texts["help_thumb"], reply_markup=Buttons.HELP_BACK)
+
+    elif qdat == "langhelp":
+        await query.edit_message_text(texts["help_lang"], reply_markup=Buttons.HELP_BACK)
+
+    elif qdat == "aboutcallback":
+        await query.edit_message_text(texts["about"].format(unzip_client.version), reply_markup=Buttons.BACK, disable_web_page_preview=True)
+
+    elif qdat.startswith("extract_file"):
+        splitted_data = qdat.split("|")
+        user_id = query.from_user.id
+        r_message = query.message.reply_to_message
+        arc_name = ""
+        download_path = f"{Config.DOWNLOAD_LOCATION}/{user_id}"
+        ext_files_dir = f"{download_path}/extracted"
+
+        try:
+            if splitted_data[1] == "url":
+                url = r_message.text
+                async with ClientSession() as ses:
+                    # Get the file size
+                    cleng = (await ses.head(url)).headers.get("Content-Length")
+                    fsize = humanbytes(int(cleng)) if cleng else "undefined"
+                    # Makes download dir
+                    makedirs(download_path)
+                    # Send logs
+                    await unzip_client.send_message(Config.LOGS_CHANNEL, texts["log"].format(user_id, "N/A", "N/A", url, fsize))
+                s_time = time()
+                arc_name = f"{download_path}/archive_from_{user_id}_{path.basename(url)}"
+                await Downloader().download(url, arc_name, query.message)
+                e_time = time()
+
+            elif splitted_data[1] == "tg_file":
+                # Makes download dir
+                makedirs(download_path)
+                # Send Logs
+                rdoc = r_message.document
+                rchat = r_message.forward_from_chat
+                await r_message.copy(Config.LOGS_CHANNEL, texts["log"].format(
+                    user_id,
+                    rchat.title if rchat else "N/A",
+                    rchat.id if rchat else "N/A",
+                    rdoc.file_name,
+                    humanbytes(rdoc.file_size))
+                )
+                s_time = time()
+                arc_name = f"{download_path}/archive_from_{user_id}_{rdoc.file_name}"
+                await r_message.download(
+                    file_name=arc_name,
+                    progress=progress_for_pyrogram, progress_args=(
+                        "**Trying to Download!** \n", query.message, s_time)
+                )
+                e_time = time()
+
+            else:
+                return await unzip_client.answer_query(query, "Can't Find Details! Please contact support group!", answer_only=True)
+
+            await unzip_client.answer_query(query, texts["ok_download"].format(arc_name, TimeFormatter(round(e_time-s_time) * 1000)))
+
+            # Checks if the archive is a splitted one
+            arc_ext = path.splitext(arc_name)[1]
+            if arc_ext.replace(".", "").isnumeric():
+                password = ""
+                if splitted_data[2] == "with_pass":
+                    password = (await unzip_client.ask(query.message.chat.id, texts["ask_password"])).text
+                await unzip_client.answer_query(query, texts["alert_splitted_arc"])
+                await add_split_arc_user(user_id, arc_name, password)
+                return
+
+            # Extract
+            exter = Extractor()
+            if splitted_data[2] == "with_pass":
+                password = (await unzip_client.ask(query.message.chat.id, texts["ask_password"])).text
+                ext_s_time = time()
+                await exter.extract(arc_name, ext_files_dir, password)
+                ext_e_time = time()
+            else:
+                ext_s_time = time()
+                await exter.extract(arc_name, ext_files_dir)
+                ext_e_time = time()
+
+            await unzip_client.answer_query(query, texts["ok_extract"].format(TimeFormatter(round(ext_e_time-ext_s_time) * 1000)))
+
+            # Upload extracted files
+            files = await get_files(ext_files_dir)
+            i_e_buttons = await Buttons.make_files_keyboard(files, user_id, query.message.chat.id)
+            try:
+                await unzip_client.answer_query(query, texts["select_files"], reply_markup=i_e_buttons)
+            except ReplyMarkupTooLong:
+                i_e_buttons = await Buttons.make_files_keyboard(files, user_id, query.message.chat.id, False)
+                await unzip_client.answer_query(query, texts["select_files"], reply_markup=i_e_buttons)
+
+        except ExtractionFailed:
+            await unzip_client.answer_query(query, texts["failed_extract"])
+        except BaseException as e:
+            try:
+                await unzip_client.answer_query(query, texts["failed_main"].format(e))
+                rmtree(download_path)
+            except Exception as er:
+                logging.warn(er)
+
+    elif qdat.startswith("ext_f"):
+        spl_data = qdat.split("|")
+        file_path = f"{Config.DOWNLOAD_LOCATION}/{spl_data[1]}/extracted"
+        files = await get_files(file_path)
+        # Next level logic lmao
+        if not files:
+            if path.isdir(f"{Config.DOWNLOAD_LOCATION}/{spl_data[1]}"):
+                rmtree(f"{Config.DOWNLOAD_LOCATION}/{spl_data[1]}")
+            return await unzip_client.answer_query(query, texts["alert_empty_files"])
+
+        await unzip_client.answer_query(query, texts["alert_sending_file"], True)
+        await unzip_client.send_file(spl_data[2], files[int(spl_data[3])], query, texts["this_lang"])
+
+        # Refreshing Inline keyboard
+        await unzip_client.answer_query(query, texts["refreshing"])
+        files = await get_files(file_path)
+        # There are no files let's die
+        if not files:
+            try:
+                rmtree(f"{Config.DOWNLOAD_LOCATION}/{spl_data[1]}")
+            except:
+                pass
+            return await unzip_client.answer_query(query, texts["ok_upload_basic"])
+        i_e_buttons = await Buttons.make_files_keyboard(files, query.from_user.id, query.message.chat.id)
+        try:
+            await unzip_client.answer_query(query, texts["select_files"], reply_markup=i_e_buttons)
+        except ReplyMarkupTooLong:
+            i_e_buttons = await Buttons.make_files_keyboard(files, user_id, query.message.chat.id, False)
+            await unzip_client.answer_query(query, texts["select_files"], reply_markup=i_e_buttons)
+
+    elif qdat.startswith("ext_a"):
+        spl_data = qdat.split("|")
+        file_path = f"{Config.DOWNLOAD_LOCATION}/{spl_data[1]}/extracted"
+        paths = await get_files(file_path)
+        if not paths:
+            if path.isdir(f"{Config.DOWNLOAD_LOCATION}/{spl_data[1]}"):
+                rmtree(f"{Config.DOWNLOAD_LOCATION}/{spl_data[1]}")
+            return await unzip_client.answer_query(query, texts["alert_empty_files"])
+
+        await unzip_client.answer_query(query, texts["alert_sending_files"], True)
+        for file in paths:
+            await unzip_client.send_file(spl_data[2], file, query, texts["this_lang"], True)
+
+        await unzip_client.answer_query(query, texts["ok_upload_basic"])
+        rmtree(f"{Config.DOWNLOAD_LOCATION}/{spl_data[1]}")
+
+    elif qdat.startswith("set_mode"):
+        mode = qdat.split("|")[1]
+        await set_upload_mode(query.from_user.id, mode)
+        await unzip_client.answer_query(query, texts["changed_upmode"].format(mode))
+
+    elif qdat.startswith("set_lang"):
+        qlang = qdat.split("|")[1]
+        chid = query.message.chat.id
+        USER_LANG[chid] = qlang
+        await set_language(chid, qlang)
+        await unzip_client.answer_query(query, texts["changed_lang"].format(qlang))
+
+    elif qdat.startswith("gf_setting"):
+        gf = GofileDB(query.from_user.id)
+        mode = qdat.split("-")[1]
+        if mode == "set":
+            tkn = await unzip_client.ask(query.message.chat.id, texts["ask_gofile_token"])
+            await gf.save_token(tkn.text)
+            await tkn.delete()
+        elif mode == "del":
+            await gf.del_token()
+        elif mode == "get":
+            return await unzip_client.answer_query(query, texts["gofile_token"].format(await gf.get_token()))
+        await unzip_client.answer_query(query, "**Done ✅!**")
+
+    elif qdat.startswith("cloudbackup"):
+        clb = CloudBackup(query.from_user.id)
+        to = qdat.split("|")[1]
+        if to == "gofile":
+            await unzip_client.answer_query(query, texts["alert_uploading_to_gofile"])
+            glnk = await clb.gofile_backup()
+            await unzip_client.answer_query(query, texts["ok_backup"].format(glnk), reply_markup=Buttons.make_button("Gofile link", url=glnk))
+
+    elif qdat == "cancel_dis":
+        await del_split_arc_user(query.from_user.id)
+        try:
+            rmtree(f"{Config.DOWNLOAD_LOCATION}/{query.from_user.id}")
+        except:
+            pass
+        await unzip_client.answer_query(query, texts["canceled"].format("Process cancelled"))
+
+    elif qdat == "nobully":
+        await unzip_client.answer_query(query, texts["ok_wont_delete"])

unzipper/modules/extract.py

@@ -0,0 +1,131 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from re import match
+from time import time
+from os import path, remove
+
+from config import Config
+from pyrogram import filters
+from aiohttp import ClientSession
+from pyrogram.types import Message
+from unzipper import unzip_client, Buttons
+from pyrogram.errors import ReplyMarkupTooLong
+
+from unzipper.lib.extractor import Extractor
+from unzipper.lib.downloader import Downloader, dl_regex
+from unzipper.helpers_nexa.utils import (TimeFormatter, get_files,
+                                         progress_for_pyrogram, humanbytes)
+from unzipper.database.split_arc import del_split_arc_user, get_split_arc_user
+
+
+@unzip_client.on_message(filters.incoming & filters.private & filters.regex(dl_regex) | filters.document)
+@unzip_client.handle_erros
+async def extract_dis_archive(_, message: Message, texts):
+    unzip_msg = await message.reply(texts["processing"], reply_to_message_id=message.id)
+    user_id = message.from_user.id
+    download_path = f"{Config.DOWNLOAD_LOCATION}/{user_id}"
+    is_url = message.text and (match(dl_regex, message.text))
+    is_doc = message.document
+
+    # Splitted files
+    is_spl, lfn, ps = await get_split_arc_user(user_id)
+    if is_spl:
+        file_name = is_doc.file_name if is_doc else path.basename(message.text)
+        await unzip_msg.edit(texts["alert_downloading_part"])
+        # Check file extension
+        if not path.splitext(file_name)[1].replace(".", "").isnumeric():
+            return await unzip_msg.edit(texts["no_splitted_arc"])
+        arc_name = f"{download_path}/archive_from_{user_id}_{file_name}"
+        if path.isfile(arc_name):
+            return await unzip_msg.edit(texts["alert_part_exists"])
+        # Download the file
+        s_time = time()
+        if is_url:
+            async with ClientSession() as ses:
+                cleng = (await ses.head(message.text)).headers.get("Content-Length")
+                fsize = humanbytes(cleng) if cleng else "undefined"
+                # Send logs
+                await unzip_client.send_message(Config.LOGS_CHANNEL, texts["log"].format(
+                    user_id,
+                    "N/A",
+                    "N/A",
+                    file_name,
+                    fsize)
+                )
+            await Downloader().download(message.text, arc_name, unzip_msg)
+        else:
+            # Send logs
+            rchat = message.forward_from_chat
+            await unzip_client.send_message(Config.LOGS_CHANNEL, texts["log"].format(
+                user_id,
+                rchat.title if rchat else "N/A",
+                rchat.id if rchat else "N/A",
+                file_name,
+                humanbytes(is_doc.file_size))
+            )
+            await message.download(
+                file_name=arc_name,
+                progress=progress_for_pyrogram, progress_args=(
+                    "**Trying to Download!** \n", unzip_msg, s_time)
+            )
+        e_time = time()
+        await unzip_msg.edit(texts["ok_download"].format(file_name, TimeFormatter(round(e_time-s_time) * 1000)))
+        return
+
+    if path.isdir(download_path):
+        return await unzip_msg.edit(texts["alert_process_running_already"])
+    if is_url:
+        await unzip_msg.edit(texts["ask_what_you_want"], reply_markup=Buttons.EXTRACT_URL)
+    elif is_doc:
+        await unzip_msg.edit(texts["ask_what_you_want"], reply_markup=Buttons.EXTRACT_FILE)
+    else:
+        await unzip_msg.edit(texts["no_archive"])
+
+
+@unzip_client.on_message(filters.private & filters.command("done"))
+@unzip_client.handle_erros
+async def extracted_dis_spl_archive(_, message: Message, texts):
+    spl_umsg = await message.reply(texts["processing"], reply_to_message_id=message.id)
+    user_id = message.from_user.id
+    # Retrive data from database
+    is_spl, lfn, ps = await get_split_arc_user(user_id)
+    # Path checks
+    if not is_spl:
+        return await spl_umsg.edit("`Bruh, why are you sending this command 🤔?`")
+    ext_path = f"{Config.DOWNLOAD_LOCATION}/{user_id}/extracted"
+    arc_path = path.dirname(lfn)
+    if not path.isdir(arc_path):
+        await spl_umsg.edit(texts["alert_empty_files"])
+        return await del_split_arc_user(user_id)
+    # Remove user record from the database
+    await del_split_arc_user(user_id)
+    # Extract the archive
+    ext = Extractor()
+    s_time = time()
+    await ext.extract(lfn, ext_path, ps, True)
+    extdarc = f"{ext_path}/{path.splitext(path.basename(lfn))[0]}"
+    await ext.extract(extdarc, ext_path, ps)
+    e_time = time()
+    await spl_umsg.edit(texts["ok_extract"].format(TimeFormatter(round(e_time-s_time) * 1000)))
+    # Try to remove merged archive
+    try:
+        remove(extdarc)
+    except:
+        pass
+    paths = await get_files(ext_path)
+    i_e_btns = await Buttons.make_files_keyboard(paths, user_id, message.chat.id)
+    try:
+        await spl_umsg.edit(texts["select_files"], reply_markup=i_e_btns)
+    except ReplyMarkupTooLong:
+        i_e_btns = await Buttons.make_files_keyboard(paths, user_id, message.chat.id, False)
+        await spl_umsg.edit(texts["select_files"], reply_markup=i_e_btns)

unzipper/modules/settings.py

@@ -0,0 +1,40 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from pyrogram import filters
+from pyrogram.types import Message
+from unzipper import unzip_client, Buttons
+from unzipper.database.language import get_language
+from unzipper.database.upload_mode import get_upload_mode
+
+
+@unzip_client.on_message(filters.private & filters.command(["gofile", "gfsets"]))
+@unzip_client.handle_erros
+async def gofile_settings(_, message: Message, texts):
+    prs_msg = await message.reply(texts["processing"], reply_to_message_id=message.id)
+    await prs_msg.edit("**Gofile.io settings ⚙️**", reply_markup=Buttons.SETTINGS_GOFILE)
+
+
+@unzip_client.on_message(filters.private & filters.command(["mode", "setmode"]))
+@unzip_client.handle_erros
+async def set_up_mode_for_user(_, message: Message, texts):
+    prs_msg = await message.reply(texts["processing"], reply_to_message_id=message.id)
+    upload_mode = await get_upload_mode(message.from_user.id)
+    await prs_msg.edit(texts["select_upmode"].format(upload_mode), reply_markup=Buttons.UPLOAD_MODE)
+
+
+@unzip_client.on_message(filters.private & filters.command(["lang", "set_lang"]))
+@unzip_client.handle_erros
+async def language_settings(_, message: Message, texts):
+    prs_msg = await message.reply(texts["processing"])
+    clng = await get_language(message.from_user.id)
+    await prs_msg.edit(texts["select_lang"].format(clng), reply_markup=Buttons.LANGUAGES)

unzipper/modules/user_utils.py

@@ -0,0 +1,72 @@
+# ===================================================================== #
+#                      Copyright (c) 2022 Itz-fork                      #
+#                                                                       #
+# This program is distributed in the hope that it will be useful,       #
+# but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.                  #
+# See the GNU General Public License for more details.                  #
+#                                                                       #
+# You should have received a copy of the GNU General Public License     #
+# along with this program. If not, see <http://www.gnu.org/licenses/>   #
+# ===================================================================== #
+
+from os import remove
+from pyrogram import filters
+from pyrogram.types import Message
+from unzipper import unzip_client, Buttons
+from unzipper.database.thumbnail import save_thumbnail, get_thumbnail, del_thumbnail
+
+
+@unzip_client.on_message(filters.private & filters.command("start"))
+@unzip_client.handle_erros
+async def start_bot(_, message: Message, texts):
+    await message.reply_text(texts["start"].format(message.from_user.mention), reply_markup=Buttons.START, disable_web_page_preview=True)
+
+
+@unzip_client.on_message(filters.private & filters.command(["save", "set_thumb"]))
+@unzip_client.handle_erros
+async def save_dis_thumb(_, message: Message, texts):
+    prs_msg = await message.reply(texts["processing"], reply_to_message_id=message.id)
+    rply = message.reply_to_message
+    if not rply or not rply.photo:
+        return await prs_msg.edit(texts["no_replied_msg"])
+    await save_thumbnail(message.from_user.id, rply)
+    await prs_msg.edit(texts["ok_saving_thumb"])
+
+
+@unzip_client.on_message(filters.private & filters.command(["thget", "get_thumb"]))
+@unzip_client.handle_erros
+async def give_my_thumb(_, message: Message, texts):
+    prs_msg = await message.reply(texts["processing"], reply_to_message_id=message.id)
+    gthumb = await get_thumbnail(message.from_user.id, True)
+    if not gthumb:
+        return await prs_msg.edit(texts["no_thumb"])
+    await prs_msg.delete()
+    await message.reply_photo(gthumb)
+    remove(gthumb)
+
+
+@unzip_client.on_message(filters.private & filters.command(["thdel", "del_thumb"]))
+@unzip_client.handle_erros
+async def delete_my_thumb(_, message: Message, texts):
+    prs_msg = await message.reply(texts["processing"], reply_to_message_id=message.id)
+    texist = await get_thumbnail(message.from_user.id)
+    if not texist:
+        return await prs_msg.edit(texts["no_thumb"])
+    await del_thumbnail(message.from_user.id)
+    remove(texist)
+    await prs_msg.edit(texts["ok_deleting_thumb"])
+
+
+@unzip_client.on_message(filters.private & filters.command("backup"))
+@unzip_client.handle_erros
+async def do_backup_files(_, message: Message, texts):
+    prs_msg = await message.reply(texts["processing"], reply_to_message_id=message.id)
+    await prs_msg.edit(texts["select_provider"], reply_markup=Buttons.BACKUP)
+
+
+@unzip_client.on_message(filters.private & filters.command("clean"))
+@unzip_client.handle_erros
+async def clean_ma_files(_, message: Message, texts):
+    prs_msg = await message.reply(texts["processing"], reply_to_message_id=message.id)
+    await prs_msg.edit(texts["ask_clean"], reply_markup=Buttons.CLEAN)