# AOC Day Script Day XX
# Date : 20XX.12.XX
# Python Code
# Developer : David Bandeira

import sys,time
from itertools import permutations

from helpingFunctions import *

setSampleMode(False)

aocDay = identifyDay(sys.argv[0])
aocYear = identifyYear(sys.argv[0])

path = getPath2Data(aocDay,aocYear)
filename = getFilename2Data(aocDay)

def calcDistances(data):
    trips={}
    locations=[]
    for dataRow in data:
        trip,distance = dataRow.split(' = ')
        locationA,locationB = trip.split(' to ')
        locations.append(locationA)
        locations.append(locationB)
        locations = list(set(locations))
        trips[(locationA.strip(),locationB.strip())]=int(distance.strip())
        trips[(locationB.strip(),locationA.strip())]=int(distance.strip())
    allTrips = list(permutations(locations))
    distanceTrips={}
    for run in allTrips:
        allDistance=0
        for i in range(len(run)-1):
            allDistance += trips[(run[i],run[i+1])]
        distanceTrips[run]=allDistance
    return distanceTrips 

def taskA (data) -> int:
    global distanceTrips
    gameScoreA = 0
    tripSearchDistance = 0
    distanceTrips = calcDistances(data) 
    for tripList,tripDistance in distanceTrips.items():
        if tripSearchDistance == 0 or tripSearchDistance > tripDistance: tripSearchDistance = tripDistance
    gameScoreA=tripSearchDistance
    return gameScoreA

def taskB (data) -> int:
    gameScoreB = 0
    tripSearchDistance = 0
    for tripList,tripDistance in distanceTrips.items():
        if tripSearchDistance == 0 or tripSearchDistance < tripDistance: tripSearchDistance = tripDistance
    gameScoreB=tripSearchDistance
    return gameScoreB
def task(task: int,data) -> int:
    score=0
    if task == 1:
        score = taskA(data)
    elif task == 2:
        score = taskB(data)
    return score

def main():
    showSampleMode()
    try:
        data = open(path+'/'+filename).read().strip().split('\n')
    except:
        print ('no inputfile found')
        print ('please check file %s on path %s' % (filename, path))
        quit()
    st=time.time()
    print ('Day '+aocDay+': Tasks 1: '+ str(task(1,data))+ ' executation time: ' + str(int((time.time()-st)*1000)) + 'ms')
    st=time.time()
    print ('Day '+aocDay+': Tasks 2: '+ str(task(2,data))+ ' executation time: ' + str(int((time.time()-st)*1000)) + 'ms')

if __name__ == "__main__":
    main()