Added SSH Command

Author: RainbowRain

requirements.txt

@@ -4,4 +4,5 @@ aioping~=0.3.1
 aiohttp~=3.8.1
 asyncoro~=4.5.6
 yarl~=1.7.2
-aiocfscrape~=1.0.0
+aiocfscrape~=1.0.0
+asyncssh~=2.10.0

tools/impl/Pinger.py

@@ -1,7 +1,6 @@
 from asyncio import sleep, open_connection, TimeoutError, wait_for, create_task
 from contextlib import suppress
 from itertools import cycle
-from time import time
 
 from aioconsole import aprint
 from aiohttp import ClientSession

tools/impl/SSH.py

@@ -0,0 +1,32 @@
+from contextlib import suppress
+from getpass import getpass
+from aioconsole import aprint
+from asyncssh import SSHClientConnection, connect
+from tools import Tool
+
+
+class SSH(Tool):
+    using = False
+
+    @staticmethod
+    async def run(*args):
+        assert len(args) == 2, "bad args"
+        ip, username = str(args[0]).split(":"), args[1]  # SSH 185.855.855.690:22 root
+        password = getpass("Enter Password: ")
+        await aprint(f"Connecting To {ip[0]}")
+        try:
+            with suppress(KeyboardInterrupt):
+                SSH.using = True
+                try:
+                    Connection: SSHClientConnection
+
+                    async with connect(ip[0], port=int(ip[1]), username=username, password=password,
+                                       known_hosts=None, connect_timeout=5) as Connection:
+                        while True:
+                            inputcmd = input(f"{username}@{ip[0]}> ")
+                            output = await Connection.run(inputcmd)
+                            await aprint(output.stdout)
+                except Exception as e:
+                    await aprint(str(e))
+        finally:
+            SSH.using = False

tools/impl/__init__.py

@@ -1,5 +1,6 @@
 from tools.impl.Pinger import Pinger
 from tools.impl.cfxFinder import Cfxfinder
+from tools.impl.SSH import SSH
 
 __all__ = ["handle"]
 
@@ -11,4 +12,7 @@ async def handle(cmd, *args):
     elif {cmd} & {"CFX"}:
         await Cfxfinder.run(*args)
         return True
+    elif {cmd} & {"SSH"}:
+        await SSH.run(*args)
+        return True
     return False