You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.6 KiB
84 lines
2.6 KiB
6 years ago
|
from __future__ import unicode_literals
|
||
|
|
||
|
from prompt_toolkit.utils import is_windows
|
||
|
from prompt_toolkit.renderer import Output
|
||
|
from prompt_toolkit.win32_types import STD_OUTPUT_HANDLE
|
||
|
from .vt100 import Vt100_Output
|
||
|
from .win32 import Win32Output
|
||
|
|
||
|
from ctypes import windll, byref
|
||
|
from ctypes.wintypes import DWORD
|
||
|
|
||
|
__all__ = [
|
||
|
'Windows10_Output',
|
||
|
]
|
||
|
|
||
|
# See: https://msdn.microsoft.com/pl-pl/library/windows/desktop/ms686033(v=vs.85).aspx
|
||
|
ENABLE_PROCESSED_INPUT = 0x0001
|
||
|
ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004
|
||
|
|
||
|
|
||
|
class Windows10_Output(object):
|
||
|
"""
|
||
|
Windows 10 output abstraction. This enables and uses vt100 escape sequences.
|
||
|
"""
|
||
|
def __init__(self, stdout):
|
||
|
self.win32_output = Win32Output(stdout)
|
||
|
self.vt100_output = Vt100_Output(stdout, lambda: None)
|
||
|
self._hconsole = windll.kernel32.GetStdHandle(STD_OUTPUT_HANDLE)
|
||
|
|
||
|
def flush(self):
|
||
|
"""
|
||
|
Write to output stream and flush.
|
||
|
"""
|
||
|
original_mode = DWORD(0)
|
||
|
|
||
|
# Remember the previous console mode.
|
||
|
windll.kernel32.GetConsoleMode(self._hconsole, byref(original_mode))
|
||
|
|
||
|
# Enable processing of vt100 sequences.
|
||
|
windll.kernel32.SetConsoleMode(self._hconsole, DWORD(
|
||
|
ENABLE_PROCESSED_INPUT | ENABLE_VIRTUAL_TERMINAL_PROCESSING))
|
||
|
|
||
|
try:
|
||
|
self.vt100_output.flush()
|
||
|
finally:
|
||
|
# Restore console mode.
|
||
|
windll.kernel32.SetConsoleMode(self._hconsole, original_mode)
|
||
|
|
||
|
def __getattr__(self, name):
|
||
|
if name in ('get_size', 'get_rows_below_cursor_position',
|
||
|
'enable_mouse_support', 'disable_mouse_support',
|
||
|
'scroll_buffer_to_prompt', 'get_win32_screen_buffer_info',
|
||
|
'enable_bracketed_paste', 'disable_bracketed_paste'):
|
||
|
return getattr(self.win32_output, name)
|
||
|
else:
|
||
|
return getattr(self.vt100_output, name)
|
||
|
|
||
|
|
||
|
Output.register(Windows10_Output)
|
||
|
|
||
|
|
||
|
def is_win_vt100_enabled():
|
||
|
"""
|
||
|
Returns True when we're running Windows and VT100 escape sequences are
|
||
|
supported.
|
||
|
"""
|
||
|
if not is_windows():
|
||
|
return False
|
||
|
|
||
|
hconsole = windll.kernel32.GetStdHandle(STD_OUTPUT_HANDLE)
|
||
|
|
||
|
# Get original console mode.
|
||
|
original_mode = DWORD(0)
|
||
|
windll.kernel32.GetConsoleMode(hconsole, byref(original_mode))
|
||
|
|
||
|
try:
|
||
|
# Try to enable VT100 sequences.
|
||
|
result = windll.kernel32.SetConsoleMode(hconsole, DWORD(
|
||
|
ENABLE_PROCESSED_INPUT | ENABLE_VIRTUAL_TERMINAL_PROCESSING))
|
||
|
|
||
|
return result == 1
|
||
|
finally:
|
||
|
windll.kernel32.SetConsoleMode(hconsole, original_mode)
|