diff options
Diffstat (limited to 'src/cmd/stream.rs')
| -rw-r--r-- | src/cmd/stream.rs | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/cmd/stream.rs b/src/cmd/stream.rs index 1e60a25..3afc9eb 100644 --- a/src/cmd/stream.rs +++ b/src/cmd/stream.rs @@ -195,11 +195,14 @@ impl<S: tokio::io::AsyncRead + tokio::io::AsyncWrite + Send + 'static> crate::client::Event::Disconnect => { Ok(crate::component_future::Poll::DidWork) } - crate::client::Event::Connect(size) => { - self.sent_remote = 0; + crate::client::Event::Start(size) => { self.process.resize(size); Ok(crate::component_future::Poll::DidWork) } + crate::client::Event::Connect() => { + self.sent_remote = 0; + Ok(crate::component_future::Poll::DidWork) + } crate::client::Event::ServerMessage(..) => { // we don't expect to ever see a server message once we // start streaming, so if one comes through, assume @@ -239,6 +242,7 @@ impl<S: tokio::io::AsyncRead + tokio::io::AsyncWrite + Send + 'static> .context(crate::error::ToRawMode)?, ); } + self.process.resize(crate::term::Size::get()?); } crate::process::Event::CommandExit(..) => { self.done = true; |