import requests

faces = []

# get array of face asset ids
with open("faces.csv", mode="r") as faces_file:
    for row in faces_file:
        faces.append(row.strip())

bundle_face_dict = {}

for face in faces:
    response = requests.get(url=f"https://roblox.com/catalog/{face}")
    # some faces don't have bundles
    redirected_to_bundle = any(
        r.status_code == 302 and "/bundles/" in r.headers.get("Location", "")
        for r in response.history
    )
    if redirected_to_bundle:
        bundle_id = response.url.split("/")[4]
    else:
        bundle_id = None
    bundle_face_dict[face] = bundle_id

lines = ["local faceByBundle: {[number]: number} = {"]
for face, bundle_id in bundle_face_dict.items():
    if bundle_id is not None:
        lines.append(f"\t[{bundle_id}] = {face},")
lines.append("}")
lines.append("")
lines.append("return faceByBundle")

with open("../src/faceByBundle.luau", mode="w") as f:
    f.write("\n".join(lines) + "\n")