import sys

from graphql_gateway import GraphQlGateway
from fixture_builder import FixtureBuilder
from input_formatter import InputFormatter

arguments = len(sys.argv) - 1
application = sys.argv[1]
if arguments == 1:
    source_application = application
else:
    source_application = sys.argv[2]
input_formatter = InputFormatter(source_application)
query_list = input_formatter.get_application_queries()
graph_ql_gateway = GraphQlGateway(application)
responses = []

for query in query_list:
    var_list = input_formatter.get_operation_variable_list(
        query["operationName"])
    if not isinstance(var_list, list):
        var_list = [var_list]
    for var_set in var_list:
        if "generateProfileUUID" in var_set.keys():
            var_set = input_formatter.add_profile_uuid(var_set, application)
        response = graph_ql_gateway.get_response(query, var_set)
        responses.append(response)

file_builder = FixtureBuilder(responses, source_application)
file_builder.build()