from _add_use_case_scripts.requestPost  import postLayersToSwagger, postTableToSwagger

def add_table(use_case: str, table_name: str):
    '''
    take the columns and add the mappings at the server
    replace all "/"'s in the internal representation with a "_"
    '''

    columns = [
        "travelId"        
    ]    
    columns = { c : c for c in columns }
    columns["UniqueID"] = "travelId+user"    
    columns["latitude"] = "travel//suggestedEndPlaces[0]//endPlace//latitude"
    columns["longitude"] = "travel//suggestedEndPlaces[0]//endPlace//longitude"
    columns["reward"] = "travel//suggestedEndPlaces[0]//reward"
    columns["user"] = "travel//suggestedEndPlaces[0]//suggestedBy"

    table = {
        "name": table_name,
        "mappings": columns
    }

    postTableToSwagger(use_case,table)

def add_layers(use_case:str, table_name: str):
    layers = [
        {
            "use_case": use_case,
            "table": table_name,
            "name": "SuggestedEndPlacesAtLocation_Layer",
            "properties": [
                'UniqueID',
                'travelId',
                'reward',
                'latitude',
                'longitude',
                'user'
            ],
            "cluster_properties": [
                'latitude',
                'latitude'
            ]
        },{
            "use_case": use_case,
            "table": table_name,
            "name": "SuggestedEndPlacesAtReward_Layer",
            "properties": [
                'UniqueID',
                'travelId',
                'reward',
                'latitude',
                'longitude',
                'user'
            ],
            "cluster_properties": [
                'reward'
            ]
        }
    ]

    postLayersToSwagger(use_case,layers)

def main(use_case: str):
    print("TRAVEL_suggestedEndPlaces")
    table_name = "travelSuggestedEndPlaces"
    add_table(use_case,table_name)
    add_layers(use_case, table_name)
