Ssh key connection added

README.md

@@ -41,4 +41,19 @@ script:
         user: pi
         pass: raspberry
         command: ls -la
+```
+
+if you want connect with ssh key, use this
+
+```yaml
+script:
+  run_on_host:
+    alias: Run shell command on host
+    sequence:
+    - service: ssh_command.exec_command
+      data:
+        host: 192.168.1.123
+        user: pi
+        ssh_private_key_path: /config/ssh/id_rsa
+        command: ls -la
 ```

custom_components/ssh_command/__init__.py

@@ -1,7 +1,7 @@
 import logging
 
 from homeassistant.core import ServiceCall
-from paramiko import SSHClient, AutoAddPolicy
+from paramiko import AutoAddPolicy, RSAKey, SSHClient
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -15,10 +15,22 @@ async def exec_command(call: ServiceCall):
         username = call.data.get('user', 'pi')
         password = call.data.get('pass', 'raspberry')
         command = call.data.get('command')
+        ssh_private_key_path = call.data.get('ssh_private_key_path')
 
         client = SSHClient()
         client.set_missing_host_key_policy(AutoAddPolicy())
-        client.connect(host, port, username, password)
+
+        if ssh_private_key_path:
+            try:
+                key = RSAKey.from_private_key_file(ssh_private_key_path)
+                client.connect(host, port, username, pkey=key)
+            except Exception as e:
+                _LOGGER.error(f"Failed to connect using SSH key: {e}")
+                return
+        else:
+            # Use password for authentication if SSH key is not provided
+            client.connect(host, port, username, password)
+
         stdin, stdout, stderr = client.exec_command(command)
         data = stdout.read()
         stderr.read()
@@ -32,4 +44,4 @@ async def exec_command(call: ServiceCall):
 
 
 async def async_setup_entry(hass, entry):
-    return True
+    return True

custom_components/ssh_command/services.yaml

@@ -33,3 +33,8 @@ exec_command:
       required: true
       selector:
         text:
+
+    ssh_private_key_path:
+      example: /config/ssh/id_rsa
+      selector:
+        text: