o >@s"dZddlmZddlmZddlmZddlmZddlZddlZddlZddl Z ddl m Z ddl m Z dd lmZdd lmZdd lmZddlZd Zed ddgZdd iZddZddZdZdZd(ddZGdddeZefddZddZddZd d!Z d"Z!d#Z"d$d%Z#d&d'Z$dS))z9Utilities for configuring platform specific installation.)absolute_import)division)print_function)unicode_literalsN) properties) console_io)encoding)files) platformsbashzshkshfishcCst|tjddS)z Prints action to standard error.)fileN)printsysstderr)actionrB/tmp/google-cloud-sdk/lib/googlecloudsdk/core/platforms_install.py _TraceAction+srcszddlddlddlmWnty"tdj|dYdSwfddfdd d d fd d }|d|gtdj|ddS)zUpdate the Windows system path to include bin_path. Args: bin_path: str, The absolute path to the directory that will contain Cloud SDK binaries. rNwinregzThe installer is unable to automatically update your system PATH. Please add {path} to your system PATH to enable easy use of the Cloud SDK Command Line Tools. )pathcsHj}d}||dj}z ||\}}W|Sty#YdSw)N Environmentr)HKEY_CURRENT_USEROpenKeyKEY_READ QueryValueEx WindowsError)namerootsubkeykeyvalue_rrrGetEnvGs z%_UpdatePathForWindows..GetEnvcsJjddj}||dj||jjdd|S)Nrr) rrKEY_ALL_ACCESS SetValueEx REG_EXPAND_SZCloseKey SendMessageHWND_BROADCASTWM_SETTINGCHANGE)r!r%r$)win32conwin32guirrrSetEnvRs   z%_UpdatePathForWindows..SetEnvcSs"||vr||||vsdSdSN)remove)pathsr%rrrRemove[s z%_UpdatePathForWindows..RemovecsJ|d}|D]}||vr|||d|q |d|dS)N;r)splitinsertjoin)r!valuesr4r%)r'r5r1rr PrependEnv_s  z)_UpdatePathForWindows..PrependEnvPathz~The following directory has been added to your PATH. {bin_path} Create a new command shell for the changes to take effect. )bin_path)r/r0 six.movesr ImportErrorrformat)r=r;r)r'r5r1r/r0rr_UpdatePathForWindows1s$       rAz9 {comment} if [ -f '{rc_path}' ]; then . '{rc_path}'; fi z5 {comment} if [ -f '{rc_path}' ]; . '{rc_path}'; end c Cs`|st|}td|dtj}|dkrtnt}|j||d}|d|}dj||d}|S)aGenerates the RC file contents with new comment and `source rc_path` lines. Args: comment: The shell comment string that precedes the source line. rc_path: The path of the rc file to source. rc_contents: The current contents. shell: The shell base name, specific to this module. pattern: A regex pattern that matches comment, None for exact match on comment. Returns: The comment and `source rc_path` lines to be inserted into a shell rc file.  z (source '.*'|if .*; then source .* fi|if .*; then (\.|source) .*; fi|if .*; (\.|source) .*; end|if .*; if type source .*; end) r)commentrc_pathrz{filtered_contents}{line})filtered_contentsline)reescapecompile MULTILINE _INJECT_FISH _INJECT_SHr@sub) rCrD rc_contentsshellpatternsubreinjectrFrErrr_GetRcContents}s     rSc@s(eZdZdZddZddZddZdS) _RcUpdaterz7Updates the RC file completion and PATH code injection.cCsX||_||_||_t||}tj|dj|d|_ tj|dj|d|_||_ dS)Nzcompletion.{shell}.incrOzpath.{shell}.inc) completion_update path_updaterD_COMPATIBLE_INC_SHELLgetosrr9r@ completionrO)selfrVrWrOrDsdk_rootcompatible_shellrrr__init__s    z_RcUpdater.__init__cCstj|jSr2)rZrexistsr[)r\rrr_CompletionExistssz_RcUpdater._CompletionExistsc Cs|jrtj|jrt|j}|}ntj|jr%tdj|jddSd}d}|j r5t d|j||j }|j rG| rGt d|j||j dd}||krVtd j|jd dStj|jrs|jd }td j|j|d t|j|tj|j}zt|Wntjttfytdj|jdYdSwz t|j|Wntjttfytdj|jdYdSwtdj|jdttdtjjj}|rdnd}|j s| rt|dj|jd |j st|dj|jd dSdS)zCreates or updates the RC file.z>[{rc_path}] exists and is not a file, so it cannot be updated.rDNrz6# The next line updates PATH for the Google Cloud SDK.z<# The next line enables shell command completion for gcloud.zB# The next line enables [a-z][a-z]* command completion for gcloud.)rPz No changes necessary for [{rc}].)rcz.backupz Backing up [{rc}] to [{backup}].)rcbackupzFCould not create directories for [{rc_path}], so it cannot be updated.zLCould not update [{rc_path}]. Ensure you have write access to this location.z[{rc_path}] has been updated.z1Start a new shell for the changes to take effect.z==> zLSource [{rc}] in your profile to enable shell command completion for gcloud.z[Source [{rc}] in your profile to add the Google Cloud SDK command line tools to your $PATH.)rDrZrisfiler ReadFileContentsr`rr@rWrSrOrVrar[shutilcopyfiledirnameMakeDirErrorIOErrorOSErrorWriteFileContentsrFormatRequiredUserActionrVALUES accessibility screen_readerGetBool)r\rNoriginal_rc_contents rc_backuprc_dirrrprefixrrrUpdates      z_RcUpdater.UpdateN)__name__ __module__ __qualname____doc__r_rarxrrrrrTs  rTcCs8|s|Stj|}tD] }|t|vr|Sq |S)a*Returns the preferred shell name based on the base file name in path. Args: path: str, The file path to check. default: str, The default value to return if a preferred name cannot be determined. Returns: The preferred user shell name or default if none can be determined. )rZrbasename_SUPPORTED_SHELLSsix text_type)rdefaultr!rOrrr_GetPreferredShell s  rcCs||dkrttjddp dS|dkrtjdddS|dkr$d j|d S|tjj kr,d S|tjj kr4d S|tjj krt|S|S)a4Returns an rc path based on the default rc path or user input. Gets default rc path based on environment. If prompts are enabled, allows user to update to preferred file path. Otherwise, prints a warning that the default rc path will be updated. Args: completion_update: bool, Whether or not to do command completion. path_update: bool, Whether or not to update PATH. rc_path: str, the rc path given by the user, from --rc-path arg. host_os: str, The host os identification string. Returns: str, A path to the rc file to update. NSHELL/bin/shzcYou specified that you wanted to update your rc file. The default file will be updated: [{rc_path}]rbzThe Google Cloud SDK installer will now prompt you to update an rc file to bring the Google Cloud CLIs into your environment. Enter a path to an rc file to update, or leave blank to use [{rc_path}]: )rrrrZrrr9r GetHomeDirrr CanPromptrr@PromptResponse ExpandHomeDir)rVrWrDrpreferred_shelldefault_rc_pathrc_path_updaterrr_GetAndUpdateRcPath6s,rcCs:t||||}t|tttjddd}t|||||S)aReturns an _RcUpdater object for the preferred user shell. Args: completion_update: bool, Whether or not to do command completion. path_update: bool, Whether or not to update PATH. rc_path: str, The path to the rc file to update. If None, ask. sdk_root: str, The path to the Cloud SDK root. host_os: str, The host os identification string. Returns: An _RcUpdater() object for the preferred user shell. rr)r)rrrrrZrrT)rVrWrDr]rrrrr _GetRcUpdater`s  rzupdate your $PATHzenable shell command completioncCs|dur |dur ||fSg}|dur|t|dur |tdd|}t|}|dur3|n|}|dur;|n|}||fS)agPrompt the user to update path or command completion if unspecified. Args: path_update: bool, Value of the --update-path arg. completion_update: bool, Value of the --command-completion arg. Returns: (path_update, completion_update) (bool, bool) Whether to update path and enable completion, respectively, after prompting the user. Nz Modify profile to {}?z and )append _PATH_PROMPT_COMPLETION_PROMPTr@r9rPromptContinue)rWrVactionspromptresponserrr_PromptToUpdate|s     rcCstj}|tjjkr|durtjdd}|rt|dStr)t||\}}n|r?|dur?|dur?d}d}t d t t t |||||dS)a-Update the system path to include bin_path. Args: completion_update: bool, Whether or not to do command completion. From --command-completion arg during install. If None, ask. path_update: bool, Whether or not to update PATH. From --path-update arg during install. If None, ask. rc_path: str, The path to the rc file to update. From --rc-path during install. If None, ask. bin_path: str, The absolute path to the directory that will contain Cloud SDK binaries. sdk_root: str, The path to the Cloud SDK root. Nz,Update %PATH% to include Cloud SDK binaries?) prompt_stringTz&Profile will be modified to {} and {}.)r rCurrentWINDOWSrrrArrrr@rrrrx)rVrWrDr=r]rrrrUpdateRCs.     rr2)%r| __future__rrrrrZrGrgrgooglecloudsdk.corergooglecloudsdk.core.consolergooglecloudsdk.core.utilrr r r_DEFAULT_SHELLr~rXrrArLrKrSobjectrTrrrrrrrrrrrrs@          @ . ^* !