#!/usr/bin/env python
# -*- coding: utf-8 -*-


import os
import hashlib
import datetime
import subprocess
import pysftp
import telebot
from dotenv import load_dotenv

# Load environment variables from .env file
load_dotenv()

# Set up variables from environment variables
MARIADB_HOST = os.getenv("MARIADB_HOST", default="localhost")
MARIADB_USER = os.getenv("MARIADB_USER", default="root")
MARIADB_PASSWORD = os.getenv("MARIADB_PASSWORD", default="passwd")
MARIADB_DATABASE = os.getenv("MARIADB_DATABASE", default="DBName")
SQLITE_DATABASE = os.getenv("SQLITE_DATABASE", default="/Path/to/database")
SFTP_HOST = os.getenv("SFTP_HOST", default="localhost")
SFTP_USER = os.getenv("SFTP_USER", default="user")
SFTP_PASSWORD = os.getenv("SFTP_PASSWORD", default="passwd")
SFTP_PATH = os.getenv("SFTP_PATH", default="/Paht/to/sftp")
SFTP_PORT = os.getenv("SFTP_PORT", default="22")
TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN", default="TOKEN")
TELEGRAM_CHAT_ID = os.getenv("TELEGRAM_CHAT_ID", default="ID_CHAT")

# Generate backups
start_time = datetime.datetime.now()
mariadb_backup_filename = f"mariadb_backup_{datetime.datetime.now().strftime('%Y%m%d')}.sql.gz"
subprocess.run(f"mariadb-dump -h {MARIADB_HOST} -u {MARIADB_USER} -p{MARIADB_PASSWORD} {MARIADB_DATABASE} | gzip > {mariadb_backup_filename}", shell=True, check=True)
sqlite_backup_filename = f"sqlite_backup_{datetime.datetime.now().strftime('%Y%m%d')}.sql.gz"
subprocess.run(f"sqlite3 {SQLITE_DATABASE} .dump | gzip > {sqlite_backup_filename}", shell=True, check=True)

# Calculate SHA512 hash of backups
mariadb_hash = hashlib.sha512(open(mariadb_backup_filename, "rb").read()).hexdigest()
sqlite_hash = hashlib.sha512(open(sqlite_backup_filename, "rb").read()).hexdigest()

# Transfer backups to SFTP server
with pysftp.Connection(host=SFTP_HOST, username=SFTP_USER, password=SFTP_PASSWORD, port=SFTP_PORT) as sftp:
    sftp.put(mariadb_backup_filename, f"{SFTP_PATH}/{mariadb_backup_filename}")
    sftp.put(sqlite_backup_filename, f"{SFTP_PATH}/{sqlite_backup_filename}")

# Delete local backup files
os.remove(mariadb_backup_filename)
os.remove(sqlite_backup_filename)

# Send message via Telegram
bot = telebot.TeleBot(TELEGRAM_BOT_TOKEN)
message = f"""
 Respaldo completado! 

* **MariaDB:**  *{mariadb_backup_filename}* (hash: {mariadb_hash})
* **SQLite:**  *{sqlite_backup_filename}* (hash: {sqlite_hash})

⏱️ Tiempo de transferencia: {datetime.datetime.now() - start_time}

 Disponible en SFTP: {SFTP_PATH}
"""
bot.send_message(TELEGRAM_CHAT_ID, message)