1 files changed, 120 insertions, 0 deletions
diff --git a/examples/ssh/asyncssh-server.py b/examples/ssh/asyncssh-server.py
new file mode 100755
index 0000000..27d0dd2
--- /dev/null
+++ b/examples/ssh/asyncssh-server.py
@@ -0,0 +1,120 @@
+#!/usr/bin/env python
+"""
+Example of running a prompt_toolkit application in an asyncssh server.
+"""
+import asyncio
+import logging
+
+import asyncssh
+from pygments.lexers.html import HtmlLexer
+
+from prompt_toolkit.completion import WordCompleter
+from prompt_toolkit.contrib.ssh import PromptToolkitSSHServer, PromptToolkitSSHSession
+from prompt_toolkit.lexers import PygmentsLexer
+from prompt_toolkit.shortcuts import ProgressBar, print_formatted_text
+from prompt_toolkit.shortcuts.dialogs import input_dialog, yes_no_dialog
+from prompt_toolkit.shortcuts.prompt import PromptSession
+
+animal_completer = WordCompleter(
+    [
+        "alligator",
+        "ant",
+        "ape",
+        "bat",
+        "bear",
+        "beaver",
+        "bee",
+        "bison",
+        "butterfly",
+        "cat",
+        "chicken",
+        "crocodile",
+        "dinosaur",
+        "dog",
+        "dolphin",
+        "dove",
+        "duck",
+        "eagle",
+        "elephant",
+        "fish",
+        "goat",
+        "gorilla",
+        "kangaroo",
+        "leopard",
+        "lion",
+        "mouse",
+        "rabbit",
+        "rat",
+        "snake",
+        "spider",
+        "turkey",
+        "turtle",
+    ],
+    ignore_case=True,
+)
+
+
+async def interact(ssh_session: PromptToolkitSSHSession) -> None:
+    """
+    The application interaction.
+
+    This will run automatically in a prompt_toolkit AppSession, which means
+    that any prompt_toolkit application (dialogs, prompts, etc...) will use the
+    SSH channel for input and output.
+    """
+    prompt_session = PromptSession()
+
+    # Alias 'print_formatted_text', so that 'print' calls go to the SSH client.
+    print = print_formatted_text
+
+    print("We will be running a few prompt_toolkit applications through this ")
+    print("SSH connection.\n")
+
+    # Simple progress bar.
+    with ProgressBar() as pb:
+        for i in pb(range(50)):
+            await asyncio.sleep(0.1)
+
+    # Normal prompt.
+    text = await prompt_session.prompt_async("(normal prompt) Type something: ")
+    print("You typed", text)
+
+    # Prompt with auto completion.
+    text = await prompt_session.prompt_async(
+        "(autocompletion) Type an animal: ", completer=animal_completer
+    )
+    print("You typed", text)
+
+    # prompt with syntax highlighting.
+    text = await prompt_session.prompt_async(
+        "(HTML syntax highlighting) Type something: ", lexer=PygmentsLexer(HtmlLexer)
+    )
+    print("You typed", text)
+
+    # Show yes/no dialog.
+    await prompt_session.prompt_async("Showing yes/no dialog... [ENTER]")
+    await yes_no_dialog("Yes/no dialog", "Running over asyncssh").run_async()
+
+    # Show input dialog
+    await prompt_session.prompt_async("Showing input dialog... [ENTER]")
+    await input_dialog("Input dialog", "Running over asyncssh").run_async()
+
+
+async def main(port=8222):
+    # Set up logging.
+    logging.basicConfig()
+    logging.getLogger().setLevel(logging.DEBUG)
+
+    await asyncssh.create_server(
+        lambda: PromptToolkitSSHServer(interact),
+        "",
+        port,
+        server_host_keys=["/etc/ssh/ssh_host_ecdsa_key"],
+    )
+
+    # Run forever.
+    await asyncio.Future()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())