import logging
import os
import subprocess
from datetime import datetime

from aiogram.types import BufferedInputFile

from bot import bot
from config import ADMIN_ID, DB_NAME, DB_PASSWORD, DB_USER, BACK_DIR


async def backup_database():

    USER = DB_USER
    HOST = "localhost"
    BACKUP_DIR = BACK_DIR
    DATE = datetime.now().strftime("%Y-%m-%d-%H%M%S")
    BACKUP_FILE = f"{BACKUP_DIR}/{DB_NAME}-backup-{DATE}.sql"

    os.environ['PGPASSWORD'] = DB_PASSWORD

    try:
        subprocess.run(
            ['pg_dump', '-U', USER, '-h', HOST, '-F', 'c', '-f', BACKUP_FILE, DB_NAME],
            check=True
        )
        logging.info(f"Бэкап базы данных создан: {BACKUP_FILE}")
    except subprocess.CalledProcessError as e:
        logging.error(f"Ошибка при создании бэкапа базы данных: {e}")
        return

    try:
        with open(BACKUP_FILE, 'rb') as backup_file:
            backup_input_file = BufferedInputFile(backup_file.read(), filename=os.path.basename(BACKUP_FILE))
            await bot.send_document(ADMIN_ID, backup_input_file)
        logging.info(f"Бэкап базы данных отправлен админу: {ADMIN_ID}")
    except Exception as e:
        logging.error(f"Ошибка при отправке бэкапа в Telegram: {e}")

    try:
        subprocess.run(
            ['find', BACKUP_DIR, '-type', 'f', '-name', '*.sql', '-mtime', '+7', '-exec', 'rm', '{}', ';'],
            check=True
        )
        logging.info("Старые бэкапы удалены.")
    except subprocess.CalledProcessError as e:
        logging.error(f"Ошибка при удалении старых бэкапов: {e}")

    del os.environ['PGPASSWORD']