import os
import requests
import telebot
from datetime import datetime

# Bot information
BOT_TOKEN = 'YOUR_TELEGRAM_BOT_TOKEN'
bot = telebot.TeleBot(BOT_TOKEN)
BOT_NAME = "GitRepo Downloader"
BOT_VERSION = "1.0.1"
LAST_UPDATE = "December 26, 2024"
DEVELOPER = "@Real_OptimusPrime"

# Command: /start
@bot.message_handler(commands=['start'])
def start(message):
    uptime = datetime.now() - bot_start_time
    bot.reply_to(
        message, 
        f"✨ Welcome to **{BOT_NAME}** bot! ✨\n"
        f"👤 User: @{message.chat.username}\n"
        f"🕒 Date and time: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n"
        f"⏳ Uptime: {str(uptime).split('.')[0]}\n\n"
        f"👨‍💻 Developed by: **{DEVELOPER}**\n"
        f"📌 Version: {BOT_VERSION}\n"
        f"🗓 Last update: {LAST_UPDATE}\n\n"
        f"Send a GitHub link to download repositories. Use `/help` for more information."
    )

# Command: /help
@bot.message_handler(commands=['help'])
def help(message):
    bot.reply_to(
        message,
        f"**Bot Help** 🛠\n"
        f"📥 **How to use**:\n"
        f"1. Send a GitHub link of a public repository.\n"
        f"2. The bot will download the repository and send you a zip file.\n"
        f"3. You will receive information about the repository (if it has a README.md).\n\n"
    )

# Process GitHub links
@bot.message_handler(func=lambda msg: "github.com" in msg.text)
def download_repo(message):
    repo_url = message.text.strip()
    bot.reply_to(message, "⏳ Processing the link...")

    try:
        repo_name = repo_url.split('/')[-1]
        if not repo_name:
            repo_name = repo_url.split('/')[-2]

        zip_url = f"{repo_url}/archive/refs/heads/main.zip"
        response = requests.get(zip_url, stream=True)
        if response.status_code != 200:
            raise Exception("Could not download the repository. Make sure the link is public.")
        
        zip_file = f"{repo_name}.zip"
        with open(zip_file, 'wb') as file:
            for chunk in response.iter_content(chunk_size=1024):
                file.write(chunk)

        file_size = os.path.getsize(zip_file) / (1024 * 1024)
        bot.reply_to(message, "📤 Sending the repository...")
        with open(zip_file, 'rb') as file:
            bot.send_document(
                message.chat.id, 
                file, 
                caption=f"✅ File downloaded.\n📂 Name: {repo_name}\n📦 Size: {file_size:.2f} MB"
            )

        os.remove(zip_file)
        repo_info = f"🔗 **Repository:** {repo_url}\n"
        try:
            readme_url = f"{repo_url}/raw/main/README.md"
            readme_response = requests.get(readme_url)
            if readme_response.status_code == 200:
                repo_info += f"📝 **Description:**\n{readme_response.text[:500]}...\n"
            else:
                repo_info += "📝 **Description:** README.md file not found.\n"
        except Exception:
            repo_info += "📝 **Description:** README.md file not found.\n"

        bot.send_message(message.chat.id, repo_info, parse_mode='Markdown')
    except Exception as e:
        bot.reply_to(message, f"❌ Error: {str(e)}")

# Main function
if __name__ == "__main__":
    bot_start_time = datetime.now()
    print(f"{BOT_NAME} started...")
    bot.polling()