Update the ssh server to provide the client terminal type to the prompt-toolkit output object

Author: vxgmichel

prompt_toolkit/contrib/ssh/server.py

@@ -9,7 +9,7 @@
 
 from prompt_toolkit.application.current import AppSession, create_app_session
 from prompt_toolkit.data_structures import Size
-from prompt_toolkit.input.posix_pipe import PosixPipeInput
+from prompt_toolkit.input import create_pipe_input
 from prompt_toolkit.output.vt100 import Vt100_Output
 
 __all__ = [
@@ -27,7 +27,8 @@ def __init__(self, interact: Callable[[], Awaitable[None]]) -> None:
  # PipInput object, for sending input in the CLI.
  # (This is something that we can use in the prompt_toolkit event loop,
  # but still write date in manually.)
- self._input = PosixPipeInput()
+ self._input = create_pipe_input()
+ self._output = None
 
  # Output object. Don't render to the real stdout, but write everything
  # in the SSH channel.
@@ -39,9 +40,7 @@ def write(s, data):
  def flush(s):
  pass
 
- self._output = Vt100_Output(
- cast(TextIO, Stdout()), self._get_size, write_binary=False
- )
+ self.stdout = cast(TextIO, Stdout())
 
  def _get_size(self) -> Size:
  """
@@ -70,7 +69,11 @@ async def _interact(self) -> None:
  # Disable the line editing provided by asyncssh. Prompt_toolkit
  # provides the line editing.
  self._chan.set_line_mode(False)
+ term = self._chan.get_terminal_type()
 
+ self._output = Vt100_Output(
+ self.stdout, self._get_size, term=term, write_binary=False
+ )
  with create_app_session(input=self._input, output=self._output) as session:
  self.app_session = session
  try: