Better handling for exit in pty shells

Handled a condition where pty shells would hang on exit. This is
currently resolved by explicitly passing "exit" to the channel when
exiting the context manager. I'm sure there are better ways to handle
this, but this works for now.

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "hussh"
-version = "0.1.3"
+version = "0.1.4"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

pyproject.toml

@@ -40,6 +40,7 @@ dev = [
     "pexpect",
     "pre-commit",
     "pytest",
+    "pytest-randomly",
     "ruff",
 ]
 

src/connection.rs

@@ -503,6 +503,7 @@ impl Connection {
         channel.shell().unwrap();
         Ok(InteractiveShell {
             channel: ChannelWrapper { channel },
+            pty: pty.unwrap_or(false),
             exit_result: None,
         })
     }
@@ -518,16 +519,18 @@ pub struct ChannelWrapper {
 #[derive(Clone)]
 struct InteractiveShell {
     channel: ChannelWrapper,
+    pty: bool,
     #[pyo3(get)]
     exit_result: Option<SSHResult>,
 }
 
 #[pymethods]
 impl InteractiveShell {
     #[new]
-    fn new(channel: ChannelWrapper) -> Self {
+    fn new(channel: ChannelWrapper, pty: bool) -> Self {
         InteractiveShell {
             channel,
+            pty,
             exit_result: None,
         }
     }
@@ -569,6 +572,9 @@ impl InteractiveShell {
         _exc_value: Option<&Bound<'_, PyAny>>,
         _traceback: Option<&Bound<'_, PyAny>>,
     ) -> PyResult<()> {
+        if self.pty {
+            self.send("exit\n".to_string(), Some(false)).unwrap();
+        }
         self.exit_result = Some(self.read());
         Ok(())
     }

tests/test_connection.py

@@ -149,6 +149,16 @@ def test_shell_context(conn):
     assert sh.exit_result.status != 0
 
 
+def test_pty_shell_context(conn):
+    """Test that we can run multiple commands in a pty shell context."""
+    with conn.shell(pty=True) as sh:
+        sh.send("echo test shell")
+        sh.send("bad command")
+    assert "test shell" in sh.exit_result.stdout
+    assert "command not found" in sh.exit_result.stdout
+    assert sh.exit_result.status != 0
+
+
 def test_connection_timeout():
     """Test that we can trigger a timeout on connect."""
     with pytest.raises(TimeoutError):