import os
import requests

# URLs da API
version_url = 'https://anablock.lapsus.com.br/api/domain/version'
domain_list_url = 'https://anablock.lapsus.com.br/api/domain'

# Caminhos locais
base_path = '/home/halley/anablock'
version_file = os.path.join(base_path, 'version')
domain_list_file = os.path.join(base_path, 'domain_list.txt')

# Garante que a pasta exista
os.makedirs(base_path, exist_ok=True)

def download_file(url, path):
    response = requests.get(url)
    response.raise_for_status()
    with open(path, 'w') as f:
        f.write(response.text)

def get_version(path):
    with open(path, 'r') as f:
        return int(f.read().strip())

def main():
    if os.path.exists(version_file):
        current_version = get_version(version_file)
        temp_version_path = '/tmp/version_temp'
        download_file(version_url, temp_version_path)
        new_version = get_version(temp_version_path)

        if new_version > current_version:
            print(f"🔄 Nova versão detectada: {new_version}")
            os.rename(temp_version_path, version_file)
            download_file(domain_list_url, domain_list_file)
            print("✅ Lista de domínios atualizada.")
        else:
            print("✔️ Versão atual já está atualizada.")
            os.remove(temp_version_path)
    else:
        print("📥 Primeira execução. Baixando arquivos...")
        download_file(version_url, version_file)
        download_file(domain_list_url, domain_list_file)
        print("✅ Lista de domínios salva com sucesso.")

if __name__ == "__main__":
    main()