import json
import requests

# lit un json listant les id de photo de chaque séquence et va
# chercher la séquence par API.

import argparse


def parse_args(argv=None):
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--username",
        type=str,
        help="Username to get the sequences id of",
        required=True,
    )
    parser.add_argument("--dev_token", type=str, help="Your mapillary developer token")
    parser.add_argument(
        "--max_sequence", type=str, help="Limit the amount of retrieved sequence ids"
    )

    global args
    args = parser.parse_args(argv)
    print(args)


# Initialisation de la liste pour stocker les réponses
responses = []
sequences = []


def get_image_data_from_sequences():
    username = args.username
    input_file = "out_" + username + ".json"

    print('get_image_data_from_sequences input_file', input_file)
    # Chargement du fichier JSON d'entrée
    with open(input_file, "r") as file:
        input_data = json.load(file)

    # Itération sur les noeuds pour collectionner les image_ids
    # Vérification de l'existence des clés dans le chemin d'accès
    if "data" not in input_data:
        print("Erreur: 'data' n'existe pas dans le fichier JSON")
        return
    if "fetch__User" not in input_data["data"]:
        print("Erreur: 'fetch__User' n'existe pas dans les données")
        return
    if "feed" not in input_data["data"]["fetch__User"]:
        print("Erreur: 'feed' n'existe pas pour cet utilisateur")
        return
    if "nodes" not in input_data["data"]["fetch__User"]["feed"]:
        print("Erreur: 'nodes' n'existe pas dans le feed")
        return
    
    nodelist = input_data["data"]["fetch__User"]["feed"]["nodes"]
    print("séquences : ", len(nodelist))
    image_ids = [node["image_id"] for node in nodelist if "image_id" in node]
    print(image_ids)

    dev_token = args.dev_token

    # Préparation de la tête d'autorisation pour toutes les futures requêtes
    header = {"Access-Token": dev_token}

    ii = 0
    limit_requests = 1000000000

    FRANCE_MIN_LAT = 42.25
    FRANCE_MIN_LON = -4.77
    FRANCE_MAX_LAT = 51.10
    FRANCE_MAX_LON = 9.57

    bbox_filter = f"&bbox={FRANCE_MIN_LAT},{FRANCE_MIN_LON},{FRANCE_MAX_LAT},{FRANCE_MAX_LON}"

    # Boucle sur chaque image_id pour interroger l'API Mapillary
    for image_id in image_ids:
        ii += 1
        if limit_requests >= ii and image_id:
            params = {"id": image_id, "fields": "id,sequence"}
            request_url = (
                "https://graph.mapillary.com/"
                + str(image_id)
                + "?access_token="
                + dev_token
                + "&fields=id,sequence"
                + bbox_filter
            )
            print("requete graphql : "+request_url)

            response = requests.get(request_url)

            # Analyse de la réponse
            parsed_response = {}
            if response.ok and response.status_code == 200:
                raw_response = response.json()

                parsed_response["id"] = raw_response["id"]
                parsed_response["sequence"] = raw_response["sequence"]
                sequences.append(parsed_response["sequence"])

                print(
                    "séquence trouvée: "
                    + str(ii)
                   # TODO fix calculation of max sequence
                   # + "/"
                   # + args.max_sequence
                    + " : "
                    + raw_response["sequence"]
                )
            else:
                print(response)

            responses.append(parsed_response)


def persist_files():
    # Sauvegarde des nouveaux résultats dans le fichier output.json
    output_file = "sequences_" + args.username + ".json"

    with open(output_file, "w") as file:
        json.dump(responses, file)

    sequence_filename = "sequences_" + args.username + ".txt"
    with open(sequence_filename, "w") as file:
        json.dump(sequences, file)
        print("fichier sauvegardé: " + sequence_filename)


parse_args()
get_image_data_from_sequences()
persist_files()

# si la requete donne moins du max de noeuds on prévoit d'en relancer une nouvelle pour avoir la suite