import os
from urllib.request import urlopen


def download_file(url, filepath):
    response = urlopen(url)

    target_dir = os.path.dirname(filepath)
    if target_dir and not os.path.exists(target_dir):
        os.makedirs(target_dir)

    # Check if the request was successful
    if response.status == 200:
        with open(filepath, 'wb') as f:
            f.write(response.read())
    else:
        print(f"Failed to download file. Status code: {response.status_code}")

    return filepath

def get_demo_data():
    demo_data_url = "https://www.dropbox.com/scl/fi/covbvo6f547kak4em3cjd/sybil_example.zip?rlkey=7a13nhlc9uwga9x7pmtk1cf1c&st=dqi0cf9k&dl=1"

    zip_file_name = "sybil_example.zip"
    cache_dir = os.path.expanduser("~/.sybil")
    zip_file_path = os.path.join(cache_dir, zip_file_name)
    os.makedirs(cache_dir, exist_ok=True)
    if not os.path.exists(zip_file_path):
        print(f"Downloading demo data to {zip_file_path}")
        download_file(demo_data_url, zip_file_path)

    demo_data_dir = os.path.join(cache_dir, "sybil_example")
    image_data_dir = os.path.join(demo_data_dir, "sybil_demo_data")
    if not os.path.exists(demo_data_dir):
        print(f"Extracting demo data to {demo_data_dir}")
        import zipfile
        with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
            zip_ref.extractall(demo_data_dir)

    dicom_files = os.listdir(image_data_dir)
    dicom_files = [os.path.join(image_data_dir, x) for x in dicom_files]
    return dicom_files