from os import system, path, mkdir, getcwd
from bs4 import BeautifulSoup
import requests
from datetime import datetime, timedelta
from fake_useragent import UserAgent

class cs:
    INFO = '\033[93m'
    GREEN = '\033[92m'
    RED = '\033[91m'
    END = '\033[0m'

User_Agent = f'{UserAgent().random}'

_ = system("cls")
name = input(f"{cs.INFO}Name: ")
offset = input("Offset: ")
now = datetime.now()

# Список ключевых слов для проверки
KEYWORDS = [
    "логин", "login", "пароль", "password",
    "данные", "данные от аккаунта", "аккаунт",
    "аккаунта", "Steam", "Epic Games"
]

def parse(name, day, month, offset):
    HEADERS = {
        'User-Agent': User_Agent
    }
    url = f"https://telegra.ph/{name}-{month}-{day}{offset}"
    response = requests.get(url, headers=HEADERS)

    # Проверка на 404 ошибку
    if response.status_code == 404:
        print(f"{cs.RED}ERROR    | 404 Page Not Found | {day}.{month}{offset}{cs.END}")
        return

    soup = BeautifulSoup(response.content, 'html.parser')

    # Извлечение текстового контента
    text_content = soup.get_text().strip()

    # Проверка на наличие текста и ключевых слов
    if text_content and any(keyword in text_content.lower() for keyword in KEYWORDS):
        # Создание директории для сохранения текста
        base_dir = f"{getcwd()}/media/{name}"
        if not path.isdir(base_dir):
            mkdir(base_dir)

        # Сохранение текста в файл
        try:
            with open(f"{base_dir}/{month}_{day}_{offset[1:] if offset else ''}.txt", "w", encoding="utf-8") as file:
                file.write(text_content)
            print(f"{cs.GREEN}SAVE     | Text saved | {day}.{month}{offset}{cs.END}")
        except Exception as e:
            print(f"Error saving text: {e}")
    else:
        print(f"{cs.RED}SKIP     | No keywords found in text | {day}.{month}{offset}{cs.END}")

def main():
    current_date = now

    while True:
        _month = current_date.month
        _day = current_date.day

        for _offset in range(1, int(offset) + 1):
            if _offset == 1:
                parse(name, f"{_day:02}", f"{_month:02}", "")
            else:
                parse(name, f"{_day:02}", f"{_month:02}", f"-{_offset}")

        # Переход к предыдущему дню
        current_date -= timedelta(days=1)

        # Проверка для завершения по желаемому пределу
        if current_date.year < now.year - 1:
            break

if __name__ == "__main__":
    main()