a 6i@s"dZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl m Z dd lmZdd lmZeed d d ZeedddZeed ddZee edddZd$eeeedddZd%eeedddZd&eeeedddZeddd d!Zdd d"d#ZdS)'z+Gets the game id and applies a fix if foundN) lru_cache) import_module)Optional) ProtonVersion)config) run_checks)log)returncCspdtjvrtjdSdtjvr(tjdSdtjvr|d|kr>|d |kr>|d Wd WSq>Wd n1s0YWnTtyt d |Yn4t y}zt d |WYd }~n d }~00t d|S)z-Get the game name from the local umu databaser STOREnoneUNKNOWNutf-8)newlineencodingrrNzCSV file not found: zError reading CSV file: zGame title not found in CSV) rrgetopencsvreaderlenFileNotFoundErrorr warn Exceptiondebug)rZumu_idstoretitleZcsvfileZ csvreaderrowexrrrget_game_title"s  $@ & r.c Cstjdr.tjtjtd}t|Szt dt dtjdt j d}tj |dtd}t|d d V}|D]<}d |vrt d |t jd}|Wd WSqWd n1s0YWntttfyYn0dS)z8Tries to return the game name from environment variablesr z/umu-database.csvzUMU_ID is not in environmentz .*/steamappsZPWDrZ appmanifest_z.acfrrnamez"[^"]+"Nr)rrr!pathdirnameabspath__file__r.r r)rr IGNORECASEjoinrr" readlinesstripUNICODEOSError IndexErrorUnicodeDecodeError)rZ game_libraryZ game_manifestZ appmanifestZxliner0rrr get_game_name9s    <r=)r*r c Cs"ddddddddd d d |d S) z*Mapping for store identifier to store nameZAmazonz Battle.netZEAZEGSZGOGZHumblezItch.ioZSteamZUbisoftz ZOOM Platform) Zamazon battleneteaZegsZgogZhumbleZitchiosteamubisoftZ zoomplatformN)r!)r*rrrget_store_namePs rBF)game_iddefaultlocalr cCsd}|rd}ntjdr0tjdd}|dkrtdtd|dt|}|rvt|d|d ntd d}|sd |d nd |s|ndS)z;Creates the name of a gamefix module, which can be importedZumur@rrzNon-steam game  ()z store specified, using z databasez&No store specified, using UMU databasezprotonfixes.gamefixes-.z localfixes.rD) isnumericrrr!lowerr infor=rB)rCrDrEr*Z store_namerrrget_module_name`s   rL)rCrDr cCstjd}|s|nd}tjtj||ds6dSttj|dddd$tjtjd Wd n1sx0Yt||d S) z6Check if a local gamefix is available first and run itz ~/.config/protonfixes/localfixesrDz.pyFz __init__.pyarr/z~/.config/protonfixesNT) rr1 expanduserisfiler6r"sysappend_run_fix)rCrDZ localpath module_namerrr_run_fix_localws  2rTcCs|sdnd}|sdnd}z\t|||}t|}td|d|dtd|d t|d rj||n|Wn:tytd |d|d td|d Yd S0dS)z3Private function, which actually executes gamefixesZ protonfixdefaultsglobalrEzUsing  z for rFrG main_with_idzNo z found for FT) rLrr rKr=hasattrrXmain ImportError)rCrDrEZfix_typeZscoperSZ game_modulerrrrRs   &    &rR)rCr cCsV|dur dStjjrtt|ds6tjjr6t|dt|dsRtjjrRt|ddS)zdLoads a gamefix module by it's gameid local fixes prevent global fixes from being executed NTF)rrZZ enable_checksrrTZenable_global_fixesrR)rCrrrrun_fixs r\cCsdtjdvdtjdvdtjdvg}t|rNtttjtddSt}td|j d |j d t t dS) zRuns the gamefixziscriptevaluator.exe getcompatpathr getnativepathz&Not running protonfixes for setup runsNzRunning protonfixes on "z ", build at rH) rPargvanyr r)strrZfrom_version_filerKZ version_nameZ build_dater\r)Z check_argsversionrrrrZs    rZ)FF)F)FF)__doc__rrrPr# functoolsr importlibrtypingrZutilrrZchecksrloggerr rbrr.r=rBboolrLrTrRr\rZrrrrs,