import os
from dotenv import load_dotenv
import inquirer
import zipfile
import cloudscraper

load_dotenv()

scraper = cloudscraper.create_scraper()

headers = {
    'x-api-key': os.getenv("CURSEFORGE_API_KEY")
}

input_name = input("Enter modpack name: ")
response = scraper.get(f'https://api.curseforge.com/v1/mods/search?gameId=432&searchFilter={input_name}', headers=headers)
data = response.json()

found = False
selected_modpack_id = -1

modpack_names = []
for mod in data['data']:
    modpack_url = mod['links']['websiteUrl']
    if modpack_url.startswith("https://www.curseforge.com/minecraft/modpacks/"):
        modpack_names.append(mod['name'])
        found = True
    else:
        pass

if not found:
    print("No modpack found with that name.")
else:
    pass

if found == True:
    questions = [
    inquirer.List('name',
                  message="Select a modpack",
                  choices=modpack_names,
            ),
]

    answers = inquirer.prompt(questions)

    selected_modpack_name = answers['name']
    for mod in data['data']:
        if mod['name'] == selected_modpack_name:
            selected_modpack_id = mod['id']
            selected_url = mod['links']['websiteUrl']

    r = scraper.get(f"https://api.curseforge.com/v1/mods/{selected_modpack_id}/files", headers = headers)
    file_data = r.json()
    for file in file_data['data']:
        server_file_id = file['serverPackFileId']
        file = scraper.get(f"https://www.api.curseforge.com/v1/mods/{selected_modpack_id}/files/{server_file_id}", headers=headers)
        download_url = file.json()['data']['downloadUrl']
        print(f"Download Url: {download_url}")
        
        dl_response = scraper.get(download_url, stream=True)
        with open(f"{selected_modpack_name}-server.zip", 'wb') as f:
            for chunk in dl_response.iter_content(chunk_size=8192):
                f.write(chunk)

        with zipfile.ZipFile(f"{selected_modpack_name}-server.zip", 'r') as zip_ref:
            zip_ref.extractall(f"{selected_modpack_name}-server")
        os.remove(f"{selected_modpack_name}-server.zip")
        break
    else:
        print("No server pack found for this modpack.")