import os
import subprocess
import random
import datetime
import requests

OVPN_DIR = "ovpn_configs"
OVPN_ZIP_URL = "https://downloads.nordcdn.com/configs/archives/servers/ovpn.zip"
OVPN_LEGACY_URL = "https://downloads.nordcdn.com/configs/files/ovpn_legacy/servers/"

def download_and_extract_ovpn_configs(cursor, conn):
    """
    Downloads the OVPN ZIP archive, extracts its contents, 
    adds new entries to the database, and cleans up.
    Returns a list of OVPN filenames that need to be checked.
    """

    if not os.path.exists(OVPN_DIR):
        os.makedirs(OVPN_DIR)

    subprocess.run(["curl", "-s", "-L", "-o", f"{OVPN_DIR}/ovpn.zip", OVPN_ZIP_URL])
    subprocess.run(["unzip", "-q", "-o", f"{OVPN_DIR}/ovpn.zip", "-d", OVPN_DIR])
    os.remove(f"{OVPN_DIR}/ovpn.zip")

    # Find all OVPN files
    ovpn_files = [f for f in os.listdir(OVPN_DIR) if f.endswith(".ovpn")]

    # Add new entries to the database
    for filename in ovpn_files:
        cursor.execute("SELECT 1 FROM ovpn_files WHERE file_name = %s", (filename,))
        if cursor.fetchone() is None:  # If no record found, insert a new one
            cursor.execute("""
                INSERT INTO ovpn_files (file_name, last_observed)
                VALUES (%s, CURRENT_TIMESTAMP)
            """, (filename,))
    conn.commit()

    # Fetch the list of VPN servers that need to be checked (last check older than 3 days)
    three_days_ago = datetime.datetime.now() - datetime.timedelta(days=3)
    cursor.execute("SELECT file_name FROM ovpn_files WHERE last_exit_ip_check < %s OR last_exit_ip_check IS NULL", (three_days_ago,))
    ovpn_files_to_check = [row[0] for row in cursor.fetchall()]

    return ovpn_files_to_check

def download_ovpn_if_needed(filename, cursor, conn):
    """
    Downloads the specified OVPN file from the legacy URL if it doesn't exist in the OVPN_DIR.
    Removes the entry from the database if the file returns a 404 error.
    Returns True if the file was downloaded or already exists, False if it was not found and removed from the database
    """
    filepath = os.path.join(OVPN_DIR+"/ovpn_tcp/", filename)

    if os.path.exists(filepath):
        print(f"{filename} already exists. Skipping download.")
        return True

    url = OVPN_LEGACY_URL + filename
    print(f"Attempting to download {filename} from legacy URL...")
    response = requests.get(url)

    if response.status_code == 200:
        with open(filepath, 'wb') as f:
            f.write(response.content)
        print(f"Downloaded {filename} from legacy URL.")
        return True
    elif response.status_code == 404:
        print(f"{filename} not found on legacy URL: {filepath}. Removing from database.") # this is the updated line
        cursor.execute("DELETE FROM ovpn_files WHERE file_name = %s", (filename,))
        conn.commit()
        return False
    else:
        print(f"Error downloading {filename}: {response.status_code}")
        return False