i"dZddlmZddlmZddlmZddlZddlZddlmZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdS)z*OS specific console_attr helper functions.)absolute_import)division)unicode_literalsN)encodingctd}ttttfD]} |}|rn#YxYw|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_sizes c/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/fire/console/console_attr_os.py GetTermSizersd "-/3,.       b    d xs /3cnddlddlddlfd}|dp|dp |d}|s{d} tjtjtj}||}n #d}YnxYw|tj|n#|tj|wwxYw|S)z.Returns the Posix terminal x and y dimensions.rNc d|jd}|r|d|dfndS#YdSxYw)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. shhsjunkrN)unpackioctl TIOCGWINSZ)fdrcfcntlstructtermioss r_GetXYz!_GetTermSizePosix.._GetXY=s[ == B0BG L L M Mb! +beRU^^t+ TTs AAA r)rrrosopenctermidO_RDONLYclose)rrrrrrs @@@rr r 4s,,,---..."vayy*FF1II*"  B 72:<< - -b 6"::bb bbb        )s;A98B9A?=BB2c ddl}ddlm}ddlm}|jd}|d}|j||sdS|d|j\ }}}}} } } } } }}| | z dz}| | z dz}||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_maxyxys rr r \s---))))))  o""3''!  b ! !$  3 3At < < 4 &}}^TXFF; [+|eV  dlQ! slQ! Q-c~ttjdttjdfS)z=Returns the terminal x and y dimensions from the environment.COLUMNSLINES)intrenvironr<rr r us+ bj# $ $c"*W*=&>&> ??r<cddl}tj|ddg|j}t |}tj|ddg|j}t |}||fS)z5Returns the terminal x and y dimensions from tput(1).rNtputcols)stderrlines) subprocessrDecode check_outputSTDOUTr@)rHoutputrErowss rr r zs ?:22FF3C:D:K3MM N N& V$ ?:22FG3D:D:K3MM N N& V$ r<àcPttfD]} |cS#YxYwdS)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. cdSNrBrBr<rz#GetRawKeyFunction..sr<)_GetRawKeyFunctionPosix_GetRawKeyFunctionWindows)get_raw_key_functions rGetRawKeyFunctionrZsL 78:   ! ! # ####  d s "c&ddlddlfd}|S)z+_GetRawKeyFunction helper using Posix APIs.rNc dddddddddddd }tjtjfd }} |}|tkr|} |tkr|j |S| rn|}|}|d kr|}nT| |d j |Sn #d}YnxYwj |n!#j |wxYw|ttfvrdn|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. ) ABDC56HFMSTcRtjtjdS)Nr)rrIrread)rsr _GetKeyCharzE_GetRawKeyFunctionPosix.._GetRawKeyPosix.._GetKeyChars _RWR^^ , ,,r<T~N)sysstdoutflushstdinfileno tcgetattrsetraw _ANSI_CSI tcsetattr TCSADRAINisalphaget _CONTROL_D _CONTROL_Z) ansi_to_keyrr old_settingscprev_crrttys @r_GetRawKeyPosixz0_GetRawKeyFunctionPosix.._GetRawKeyPosixs            K J     B-----$$R((L= jjnnn +--a i KMM  )^^G-|<<<<YY[[  &kmm! #XXA  q"%%G-|<<< for function keys, otherwise a character. r]r^r_r`rarbrcrd)rkPKrmIQGOcPtjSrU)rrIgetchmsvcrtsrrrzI_GetRawKeyFunctionWindows.._GetRawKeyWindows.._GetKeyChars _V\\^^ , ,,r<rtN)rurvrw_WINDOWS_CSI_1_WINDOWS_CSI_2rrr)windows_to_keyrrrrs r_GetRawKeyWindowsz4_GetRawKeyFunctionWindows.._GetRawKeyWindowss         NJ-----  A^^ ,,,    r 2 22Z00044a7r<r)rrs @rrXrXs/---88888> r<)__doc__ __future__rrrrru fire.consolerrr r r r r|rrrrrZrWrXrBr<rrs  10&&&&&&'''''' !!!!!!0% % % P2@@@       "<<<~$$$$$r<