import sys
import os
import urllib.request
import shutil
import subprocess

if not os.path.exists("firefox"):
    os.mkdir("firefox")

firefox_profiles = [
    {
        "name": "125",
        "url": "https://ftp.mozilla.org/pub/firefox/nightly/2024/03/2024-03-17-21-27-40-mozilla-central/firefox-125.0a1.en-US.win64.zip",
    },
    {
        "name": "126",
        "url": "https://ftp.mozilla.org/pub/firefox/nightly/2024/04/2024-04-14-21-22-02-mozilla-central/firefox-126.0a1.en-US.win64.zip",
    },
    {
        "name": "127",
        "url": "https://ftp.mozilla.org/pub/firefox/nightly/2024/04/2024-04-15-08-26-38-mozilla-central/firefox-127.0a1.en-US.win64.zip",
    },
    {
        "name": "135",
        "url": "https://ftp.mozilla.org/pub/firefox/nightly/2024/12/2024-12-29-09-34-11-mozilla-central/firefox-135.0a1.en-US.win64.zip",
    },
]


for i, e in enumerate(firefox_profiles):
    firefox_profiles[i]["fname"] = os.path.join("firefox", "firefox-" + e["name"])
    firefox_profiles[i]["zip"] = e["fname"] + ".zip"
    if not os.path.exists(e["zip"]):
        urllib.request.urlretrieve(e["url"], e["zip"])
        print("Firefox Download: {}".format(e["name"]))

for i, e in enumerate(firefox_profiles):
    if sys.argv[1] == e["name"]:
        if os.path.exists(e["fname"]):
            shutil.rmtree(e["fname"])
        shutil.unpack_archive(e["zip"], e["fname"])
        subprocess.run([os.path.join(e["fname"], "firefox", "firefox"), "poc.pdf"])