import os
import zipfile
import paramiko

HOST = '23.140.140.70'
PASSWORD = '33nKDS80L6ipmedJ3y0j'
PORT = 22
USER = 'root'
REMOTE_DIR = '/opt/codex-console'

BASE_DIR = os.path.dirname(os.path.abspath(__file__))
ZIP_FILE = os.path.join(BASE_DIR, 'deploy.zip')
EXCLUDE_DIRS = {'.git', '.venv', 'data', 'logs', '__pycache__', '.pytest_cache', '.vscode', '.idea'}

def create_zip():
    print(f"Creating zip file {ZIP_FILE}...")
    with zipfile.ZipFile(ZIP_FILE, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(BASE_DIR):
            dirs[:] = [d for d in dirs if d not in EXCLUDE_DIRS]
            for file in files:
                if file.endswith('.zip') or file.endswith('.pyc') or file == 'logs*.txt':
                    continue
                file_path = os.path.join(root, file)
                arcname = os.path.relpath(file_path, BASE_DIR)
                zipf.write(file_path, arcname)
    print("Zip created successfully.")

def deploy():
    print(f"\n--- Deploying to {HOST} ---")
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(HOST, port=PORT, username=USER, password=PASSWORD, timeout=30, banner_timeout=30)
    print("Connected.")

    ssh.exec_command(f'mkdir -p {REMOTE_DIR}')
    ssh.exec_command(f'rm -rf {REMOTE_DIR}/logs/*.log')
    ssh.exec_command(f'rm -rf {REMOTE_DIR}/logs/*.txt')
    ssh.exec_command(f'mkdir -p {REMOTE_DIR}/data')
    ssh.exec_command(f'mkdir -p {REMOTE_DIR}/logs')

    print("Uploading zip file...")
    sftp = ssh.open_sftp()
    sftp.put(ZIP_FILE, f'{REMOTE_DIR}/deploy.zip')
    sftp.close()
    print("Upload complete.")

    print("Extracting and running docker-compose...")
    commands = [
        f"cd {REMOTE_DIR}",
        "unzip -o deploy.zip > /dev/null",
        "rm deploy.zip",
        "(docker-compose down || docker compose down || true)",
        "(docker-compose up -d --build || docker compose up -d --build)"
    ]
    command = ' && '.join(commands)
    stdin, stdout, stderr = ssh.exec_command(command, timeout=120)
    
    for line in stdout:
        print(line.strip())
    for line in stderr:
        print("ERR:", line.strip())
        
    exit_status = stdout.channel.recv_exit_status()
    if exit_status == 0:
        print(f"Deployment to {HOST} successful!")
    else:
        print(f"Deployment to {HOST} failed with exit status {exit_status}")
    
    ssh.close()

if __name__ == '__main__':
    create_zip()
    deploy()
    if os.path.exists(ZIP_FILE):
        os.remove(ZIP_FILE)
        print("Cleaned up local zip file.")