Ultroid-fork/update_script.py

#!/usr/bin/env python3
"""
Ultroid Update Script
This script handles updating the bot while it's not running.
"""

import os
import sys
import subprocess
import time
from pathlib import Path

def run_command(cmd, shell=True):
    """Run a command and return success status."""
    try:
        result = subprocess.run(cmd, shell=shell, capture_output=True, text=True)
        print(f"Command: {cmd}")
        print(f"Output: {result.stdout}")
        if result.stderr:
            print(f"Error: {result.stderr}")
        return result.returncode == 0
    except Exception as e:
        print(f"Error running command '{cmd}': {e}")
        return False

def main():
    """Main update function."""
    print("🔄 Starting Ultroid update process...")
    
    # Get script directory
    script_dir = Path(__file__).parent.absolute()
    os.chdir(script_dir)
    
    print(f"📁 Working directory: {script_dir}")
    
    # Check if we're in a git repository
    if not (script_dir / ".git").exists():
        print("❌ Not a git repository. Cannot update.")
        return False    
    # Get the repository URL from command line args or default to user's fork
    repo_url = sys.argv[1] if len(sys.argv) > 1 else "https://github.com/overspend1/Ultroid-fork.git"
    
    # Fetch and pull updates
    print("📥 Fetching updates from repository...")
    
    if repo_url:
        print(f"🔗 Using repository: {repo_url}")
        # Set up remote if needed
        if not run_command("git remote get-url origin"):
            run_command(f"git remote add origin {repo_url}")
        else:
            run_command(f"git remote set-url origin {repo_url}")
    
    # Fetch latest changes
    if not run_command("git fetch origin"):
        print("❌ Failed to fetch updates")
        return False
      # Get current branch
    result = subprocess.run("git branch --show-current", shell=True, capture_output=True, text=True)
    current_branch = result.stdout.strip() or "main"
    
    print(f"🌿 Current branch: {current_branch}")
    
    # Check for untracked files that might conflict
    print("🔍 Checking for conflicting files...")
    untracked_result = subprocess.run("git ls-files --others --exclude-standard", shell=True, capture_output=True, text=True)
    untracked_files = untracked_result.stdout.strip().split('\n') if untracked_result.stdout.strip() else []
    
    # If update_script.py is untracked and would conflict, temporarily move it
    script_moved = False
    if "update_script.py" in untracked_files:
        print("📦 Temporarily moving update script to avoid conflicts...")
        if run_command("move update_script.py update_script_temp.py"):
            script_moved = True
    
    # Stash any local changes first
    print("💾 Stashing local changes...")
    run_command("git stash push -m 'Auto-stash before update'")
      # Pull updates
    print("⬇️ Pulling updates...")
    if not run_command(f"git pull origin {current_branch}"):
        print("❌ Failed to pull updates")
        # Try to restore stashed changes
        print("🔄 Attempting to restore stashed changes...")
        run_command("git stash pop")
        # Restore moved script if it was moved
        if script_moved and os.path.exists("update_script_temp.py"):
            print("🔄 Restoring update script...")
            run_command("move update_script_temp.py update_script.py")
        return False
      # Restore stashed changes if any
    print("🔄 Restoring local changes...")
    stash_result = subprocess.run("git stash list", shell=True, capture_output=True, text=True)
    if "Auto-stash before update" in stash_result.stdout:
        if not run_command("git stash pop"):
            print("⚠️ Warning: Could not restore some local changes. Check git stash list.")
        else:
            print("✅ Local changes restored successfully")
    
    # Restore moved script if it was moved
    if script_moved and os.path.exists("update_script_temp.py"):
        print("🔄 Restoring update script...")
        run_command("move update_script_temp.py update_script.py")
    
    # Update dependencies
    print("📦 Installing/updating dependencies...")
    if not run_command("pip3 install -r requirements.txt --upgrade"):
        print("⚠️ Warning: Failed to update some dependencies")
    
    # Try alternative pip command
    run_command("pip3 install -r requirements.txt --break-system-packages --upgrade")
    
    print("✅ Update completed successfully!")
    return True

def restart_bot():
    """Restart the bot after update."""
    print("🔄 Restarting Ultroid...")
    
    # Check if we have a virtual environment
    venv_python = None
    if os.path.exists("venv/bin/python"):
        venv_python = "venv/bin/python"
    elif os.path.exists("venv/Scripts/python.exe"):
        venv_python = "venv/Scripts/python.exe"
    
    # Determine how to start the bot
    if len(sys.argv) > 1 and sys.argv[-1] == "main.py":
        # Started with main.py
        if venv_python:
            os.execv(venv_python, [venv_python, "main.py"])
        else:
            os.execv(sys.executable, [sys.executable, "main.py"])
    else:
        # Started as module
        if venv_python:
            os.execv(venv_python, [venv_python, "-m", "pyUltroid"])
        else:
            os.execv(sys.executable, [sys.executable, "-m", "pyUltroid"])

if __name__ == "__main__":
    print("🚀 Ultroid Update Script")
    print("=" * 40)
    
    # Wait a moment for the bot to fully shutdown
    time.sleep(2)
    
    # Perform update
    if main():
        print("=" * 40)
        restart_bot()
    else:
        print("❌ Update failed. Please check the errors above.")
        sys.exit(1)