o  @sdZddlmZddlmZddlmZddlZddlZddlmZddlm Z dd Z d d Z d d Z ddZ ddZdZdZdZdZdZddZddZddZddZdS)z*OS specific console_attr helper functions.)absolute_import)division)unicode_literalsN)encoding) platformscCs<d}ttttfD]}z |}|rWnWqYq|pdS)zGets the terminal x and y dimensions in characters. _GetTermSize*() helper functions taken from: http://stackoverflow.com/questions/263890/ Returns: (columns, lines): A tuple containing the terminal x and y dimensions. N)P)_GetTermSizePosix_GetTermSizeWindows_GetTermSizeEnvironment_GetTermSizeTput)xyget_terminal_sizerH/tmp/google-cloud-sdk/lib/googlecloudsdk/core/console/console_attr_os.py GetTermSizes rc sddlddlddlfdd}|dp|dp|d}|sRd}z"ztttj}||}Wnd}YW|durEt||S|durQt|ww|S)z.Returns the Posix terminal x and y dimensions.rNcsBzd|jd}|r|d|dfWSdWSYdS)zReturns the terminal (x,y) size for fd. Args: fd: The terminal file descriptor. Returns: The terminal (x,y) size for fd or None on error. shhjunkrN)unpackioctl TIOCGWINSZ)fdrcfcntlstructtermiosrr_GetXY>s  z!_GetTermSizePosix.._GetXYr)rrrosopenctermidO_RDONLYclose)rr rrrrr 5s(   r c Csddl}ddlm}ddlm}|jd}|d}|j||s#dS|d|j\ }}}}} } } } } }}| | d}| | d}||fS) z0Returns the Windows terminal x and y dimensions.rN)create_string_buffer)windllis hhhhHhhhhhhr) rctypesr$r%kernel32 GetStdHandleGetConsoleScreenBufferInforraw)rr$r%hcsbi unused_bufx unused_bufy unused_curx unused_cury unused_wattrlefttoprightbottom unused_maxx unused_maxyxyrrrr ]s       r cCsttjdttjdfS)z=Returns the terminal x and y dimensions from the environment.COLUMNSLINES)intrenvironrrrrr vsr cCsTddl}t|jddg|jd}t|}t|jddg|jd}t|}||fS)z5Returns the terminal x and y dimemsions from tput(1).rNtputcols)stderrlines) subprocessrDecode check_outputSTDOUTr=)rCoutputr@rowsrrrr {sr àcCs,ttfD] }z|WSYqddS)zReturns a function that reads one keypress from stdin with no echo. Returns: A function that reads one keypress from stdin with no echo or a function that always returns None if stdin does not support it. cSsdSNrrrrrsz#GetRawKeyFunction..)_GetRawKeyFunctionPosix_GetRawKeyFunctionWindows)get_raw_key_functionrrrGetRawKeyFunctions rScs"ddlddlfdd}|S)z+_GetRawKeyFunction helper using Posix APIs.rNc sdddddddddddd }tjtjfd d }}zUzD|}|tkrg|} |tkrE|WWj |S| rJn |}|}|d krV|}nq5| |dWWj |SWnd}YWj |n j |w|t t fvrdS|S)zReads and returns one keypress from stdin, no echo, using Posix APIs. Returns: The key name, None for EOF, <*> for function keys, otherwise a character. ) ABDC56HFMSTcsttdS)Nr)rrDrreadrrrr _GetKeyCharszE_GetRawKeyFunctionPosix.._GetRawKeyPosix.._GetKeyCharT~N)sysstdoutflushstdinfileno tcgetattrsetraw _ANSI_CSI tcsetattr TCSADRAINisalphaget _CONTROL_D _CONTROL_Z) ansi_to_keyri old_settingscprev_crttyrhr_GetRawKeyPosixsR        $z0_GetRawKeyFunctionPosix.._GetRawKeyPosix)rr)rrr~rrPs5rPcsddlfdd}|S)z-_GetRawKeyFunction helper using Windows APIs.rNc s`ddddddddd }tjfd d }|}|ttfvr&||d S|ttfvr.d S|S)zReads and returns one keypress from stdin, no echo, using Windows APIs. Returns: The key name, None for EOF, <*> for function keys, otherwise a character. rTrUrVrWrXrYrZr[)rbPKrdIQGOcstSrN)rrDgetchrmsvcrtrrriszI_GetRawKeyFunctionWindows.._GetRawKeyWindows.._GetKeyCharrkN)rlrmrn_WINDOWS_CSI_1_WINDOWS_CSI_2rwrxry)windows_to_keyrir|rrr_GetRawKeyWindowss  z4_GetRawKeyFunctionWindows.._GetRawKeyWindowsr)rrrrrQs rQc Cstjtjjkr dSz1ddl}d}|jjd}|j }|jj || |r7|jj ||j |Br:WdSWdSWdSttfyHYdSw)zAttempts to enable virtual terminal processing on Windows. Returns: bool: True if ANSI support is now active; False otherwise. FrNiT)rOperatingSystemCurrentWINDOWSr'r%r(r)wintypesDWORDGetConsoleModebyrefSetConsoleModevalueOSErrorAttributeError)r'"enable_virtual_terminal_processingr,old_moderrrForceEnableAnsis(  r)__doc__ __future__rrrrrlgooglecloudsdk.core.utilrrrr r r r rsrxryrrrSrPrQrrrrrs,     ( ? '