modules/open.py

#  Moon-Userbot - telegram userbot
#  Copyright (C) 2020-present Moon Userbot Organization
#
#  This program is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.

#  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 <https://www.gnu.org/licenses/>.

import datetime
import os
import time

import aiofiles
from pyrogram import Client, filters
from pyrogram.errors import MessageTooLong
from pyrogram.types import Message

from utils.misc import modules_help, prefix
from utils.scripts import edit_or_reply, format_exc, progress
from utils.rentry import paste as rentry_paste


async def read_file(file_path):
    async with aiofiles.open(file_path, mode="r") as file:
        content = await file.read()
    return content


def check_extension(file_path):
    extensions = {
        ".txt": "<pre lang='plaintext'>",
        ".py": "<pre lang='python'>",
        ".js": "<pre lang='javascript'>",
        ".json": "<pre lang='json'>",
        ".smali": "<pre lang='smali'>",
        ".sh": "<pre lang='shell'>",
        ".c": "<pre lang='c'>",
        ".java": "<pre lang='java'>",
        ".php": "<pre lang='php'>",
        ".doc": "<pre lang='doc'>",
        ".docx": "<pre lang='docx'>",
        ".rtf": "<pre lang='rtf'>",
        ".s": "<pre lang='asm'>",
        ".dart": "<pre lang='dart'>",
        ".cfg": "<pre lang='cfg'>",
        ".swift": "<pre lang='swift'>",
        ".cs": "<pre lang='csharp'>",
        ".vb": "<pre lang='vb'>",
        ".css": "<pre lang='css'>",
        ".htm": "<pre lang='html'>",
        ".html": "<pre lang='html'>",
        ".rss": "<pre lang='xml'>",
        ".xhtml": "<pre lang='xtml'>",
        ".cpp": "<pre lang='cpp'>",
    }

    ext = os.path.splitext(file_path)[1].lower()

    return extensions.get(ext, "<pre>")


@Client.on_message(filters.command("open", prefix) & filters.me)
async def openfile(client: Client, message: Message):
    if not message.reply_to_message:
        return await message.edit_text("Kindly Reply to a File")

    try:
        ms = await edit_or_reply(message, "<b>Downloading...</b>")
        ct = time.time()
        file_path = await message.reply_to_message.download(
            progress=progress, progress_args=(ms, ct, "Downloading...")
        )
        await ms.edit_text("<code>Trying to open file...</code>")
        file_info = os.stat(file_path)
        file_name = file_path.split("/")[-1:]
        file_size = file_info.st_size
        last_modified = datetime.datetime.fromtimestamp(file_info.st_mtime).strftime(
            "%Y-%m-%d %H:%M:%S"
        )
        code_start = check_extension(file_path=file_path)
        content = await read_file(file_path=file_path)
        await ms.edit_text(
            f"<b>File Name:</b> <code>{file_name[0]}</code>\n<b>Size:</b> <code>{file_size} bytes</code>\n<b>Last Modified:</b> <code>{last_modified}</code>\n<b>Content:</b> {code_start}{content}</pre>",
        )

    except MessageTooLong:
        await ms.edit_text(
            "<code>File Content is too long... Pasting to rentry...</code>"
        )
        content_new = f"```{code_start[11:-2]}\n{content}```"
        try:
            rentry_url, edit_code = await rentry_paste(
                text=content_new, return_edit=True
            )
        except RuntimeError:
            await ms.edit_text("<b>Error:</b> <code>Failed to paste to rentry</code>")
            return
        await client.send_message(
            "me",
            f"Here's your edit code for Url: {rentry_url}\nEdit code:  <code>{edit_code}</code>",
            disable_web_page_preview=True,
        )
        await ms.edit_text(
            f"<b>File Name:</b> <code>{file_name[0]}</code>\n<b>Size:</b> <code>{file_size} bytes</code>\n<b>Last Modified:</b> <code>{last_modified}</code>\n<b>Content:</b> {rentry_url}\n<b>Note:</b> <code>Edit Code has been sent to your saved messages</code>",
            disable_web_page_preview=True,
        )

    except Exception as e:
        await ms.edit_text(format_exc(e))

    finally:
        if os.path.exists(file_path):
            os.remove(file_path)


modules_help["open"] = {
    "open": "Open content of any text supported filetype and show it's raw data"
}