#  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 os
import shutil
import subprocess
import sys

from pyrogram import Client, filters
from pyrogram.types import Message
from utils.db import db
from utils.misc import modules_help, prefix, requirements_list
from utils.scripts import format_exc, restart


def check_command(command):
    return shutil.which(command) is not None


@Client.on_message(filters.command('restart', prefix) & filters.me)
async def restart_cmd(_, message: Message):
    db.set(
        'core.updater',
        'restart_info',
        {
            'type': 'restart',
            'chat_id': message.chat.id,
            'message_id': message.id,
        },
    )

    if 'LAVHOST' in os.environ:
        await message.edit('<b>Your lavHost is restarting...</b>')
        os.system('lavhost restart')  # noqa: S605, S607
        return

    await message.edit('<b>Restarting...</b>')
    if os.path.exists('moonlogs.txt'):
        os.remove('moonlogs.txt')
    restart()


@Client.on_message(filters.command('update', prefix) & filters.me)
async def update(_, message: Message):
    db.set(
        'core.updater',
        'restart_info',
        {
            'type': 'update',
            'chat_id': message.chat.id,
            'message_id': message.id,
        },
    )

    if 'LAVHOST' in os.environ:
        await message.edit('<b>Your lavHost is updating...</b>')
        os.system('lavhost update')  # noqa: S605, S607
        return

    await message.edit('<b>Updating...</b>')
    try:
        if not check_command('termux-setup-storage'):
            subprocess.run([sys.executable, '-m', 'pip', 'install', '-U', 'pip'], check=True)
        subprocess.run(['git', 'pull'], check=True)  # noqa: S607

        if os.path.exists('requirements.txt') and os.path.getsize('requirements.txt') > 0:
            subprocess.run(
                [
                    sys.executable,
                    '-m',
                    'pip',
                    'install',
                    '-U',
                    '-r',
                    'requirements.txt',
                ],
                check=True,
            )

        if requirements_list:
            subprocess.run(  # noqa: S603
                [sys.executable, '-m', 'pip', 'install', '-U', *requirements_list],
                check=True,
            )
    except Exception as e:
        await message.edit(format_exc(e))
        db.remove('core.updater', 'restart_info')
    else:
        await message.edit('<b>Updating: done! Restarting...</b>')
        if os.path.exists('moonlogs.txt'):
            os.remove('moonlogs.txt')
        restart()


modules_help['updater'] = {
    'update': 'Update the userbot. If new core modules are avaliable, they will be installed',
    'restart': 'Restart userbot',
}