1 files changed, 52 insertions, 42 deletions
diff --git a/examples/tokio.rs b/examples/tokio.rs
index 2441c61..0dc007f 100644
--- a/examples/tokio.rs
+++ b/examples/tokio.rs
@@ -1,58 +1,63 @@
-use pty_process::Command as _;
-use std::os::unix::process::ExitStatusExt as _;
-use tokio::io::{AsyncReadExt as _, AsyncWriteExt as _};
-
 mod raw_guard;
 
-async fn run(
-    child: &mut pty_process::tokio::Child,
-) -> std::result::Result<(), Box<dyn std::error::Error>> {
-    let _raw = raw_guard::RawGuard::new();
-
-    let mut in_buf = [0_u8; 4096];
-    let mut out_buf = [0_u8; 4096];
-
-    let mut stdin = tokio::io::stdin();
-    let mut stdout = tokio::io::stdout();
-
-    loop {
-        tokio::select! {
-            bytes = stdin.read(&mut in_buf) => match bytes {
-                Ok(bytes) => {
-                    child.pty_mut().write_all(&in_buf[..bytes]).await.unwrap();
-                }
-                Err(e) => {
-                    eprintln!("stdin read failed: {:?}", e);
-                    break;
-                }
-            },
-            bytes = child.pty_mut().read(&mut out_buf) => match bytes {
-                Ok(bytes) => {
-                    stdout.write_all(&out_buf[..bytes]).await.unwrap();
-                    stdout.flush().await.unwrap();
-                }
-                Err(e) => {
-                    // EIO means that the process closed the other
-                    // end of the pty
-                    if e.raw_os_error() != Some(libc::EIO) {
-                        eprintln!("pty read failed: {:?}", e);
+#[cfg(feature = "backend-tokio")]
+mod main {
+    use tokio::io::{AsyncReadExt as _, AsyncWriteExt as _};
+
+    pub async fn run(
+        child: &mut pty_process::tokio::Child,
+    ) -> std::result::Result<(), Box<dyn std::error::Error>> {
+        let _raw = super::raw_guard::RawGuard::new();
+
+        let mut in_buf = [0_u8; 4096];
+        let mut out_buf = [0_u8; 4096];
+
+        let mut stdin = tokio::io::stdin();
+        let mut stdout = tokio::io::stdout();
+
+        loop {
+            tokio::select! {
+                bytes = stdin.read(&mut in_buf) => match bytes {
+                    Ok(bytes) => {
+                        child.pty_mut().write_all(&in_buf[..bytes]).await.unwrap();
+                    }
+                    Err(e) => {
+                        eprintln!("stdin read failed: {:?}", e);
+                        break;
+                    }
+                },
+                bytes = child.pty_mut().read(&mut out_buf) => match bytes {
+                    Ok(bytes) => {
+                        stdout.write_all(&out_buf[..bytes]).await.unwrap();
+                        stdout.flush().await.unwrap();
                     }
-                    break;
-                }
-            },
+                    Err(e) => {
+                        // EIO means that the process closed the other
+                        // end of the pty
+                        if e.raw_os_error() != Some(libc::EIO) {
+                            eprintln!("pty read failed: {:?}", e);
+                        }
+                        break;
+                    }
+                },
+            }
         }
-    }
 
-    Ok(())
+        Ok(())
+    }
 }
 
+#[cfg(feature = "backend-tokio")]
 #[tokio::main]
 async fn main() {
+    use pty_process::Command as _;
+    use std::os::unix::process::ExitStatusExt as _;
+
     let mut child = tokio::process::Command::new("sleep")
         .args(&["500"])
         .spawn_pty(Some(&pty_process::Size::new(24, 80)))
         .unwrap();
-    run(&mut child).await.unwrap();
+    main::run(&mut child).await.unwrap();
     let status = child.wait().await.unwrap();
     std::process::exit(
         status
@@ -60,3 +65,8 @@ async fn main() {
             .unwrap_or_else(|| status.signal().unwrap_or(0) + 128),
     );
 }
+
+#[cfg(not(feature = "backend-tokio"))]
+fn main() {
+    unimplemented!()
+}