Changing so that it's keyfile only auth. Will make it bidirectional in a future PR.

Author: jamfish

src/hackingBuddyGPT/capabilities/ssh_test_credential.py

@@ -4,7 +4,7 @@
 import paramiko
 
 from hackingBuddyGPT.utils import SSHConnection
-from .capability import Capability
+from hackingBuddyGPT.capabilities import Capability
 
 
 @dataclass
@@ -17,8 +17,8 @@ def describe(self) -> str:
     def get_name(self):
         return "test_credential"
 
-    def __call__(self, username: str, password: str) -> Tuple[str, bool]:
-        test_conn = self.conn.new_with(username=username, password=password)
+    def __call__(self, username: str, keyfilename: str, ) -> Tuple[str, bool]:
+        test_conn = self.conn.new_with(username=username, keyfilename=keyfilename)
         try:
             test_conn.init()
             user = test_conn.run("whoami")[0].strip('\n\r ')

src/hackingBuddyGPT/cli/stats.py

@@ -2,7 +2,7 @@
 
 import argparse
 
-from utils.db_storage import DbStorage
+from hackingBuddyGPT.utils.db_storage import DbStorage
 from rich.console import Console
 from rich.table import Table
 

src/hackingBuddyGPT/utils/ssh_connection/ssh_connection.py

@@ -12,29 +12,29 @@ class SSHConnection:
     host: str
     hostname: str
     username: str
-    password: str
-    port: int = 22
     keyfilename: str
+    port: int = 22
+    
 
     _conn: Connection = None
 
     def init(self):
         # create the SSH Connection
         conn = Connection(
             f"{self.username}@{self.host}:{self.port}",
-            connect_kwargs={"password": self.password, "key_filename": self.keyfilename, "allow_agent": False},
+            connect_kwargs={ "key_filename": self.keyfilename, "allow_agent": False},
         )
         self._conn = conn
         self._conn.open()
 
-    def new_with(self, *, host=None, hostname=None, username=None, password=None, port=None, keyfilename=None) -> "SSHConnection":
+    def new_with(self, *, host=None, hostname=None, username=None, keyfilename=None, port=None) -> "SSHConnection":
         return SSHConnection(
             host=host or self.host,
             hostname=hostname or self.hostname,
             username=username or self.username,
-            password=password or self.password,
+            keyfilename=keyfilename or self.keyfilename,
             port=port or self.port,
-            keyfilename=keyfilename or self.keyfilename
+            
         )
 
     def run(self, cmd, *args, **kwargs) -> Tuple[str, str, int]: