Skip to content

Commit

Permalink
Add ctrl-z to suspend
Browse files Browse the repository at this point in the history
  • Loading branch information
pickfire committed Jul 29, 2021
1 parent 8a2fa69 commit 361c67c
Show file tree
Hide file tree
Showing 10 changed files with 127 additions and 22 deletions.
14 changes: 14 additions & 0 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion helix-lsp/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -23,5 +23,5 @@ lsp-types = { version = "0.89", features = ["proposed"] }
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
thiserror = "1.0"
tokio = { version = "1.9", features = ["full"] }
tokio = { version = "1.9", features = ["rt", "rt-multi-thread", "io-util", "io-std", "time", "process", "macros", "fs", "parking_lot"] }
tokio-stream = "0.1.7"
6 changes: 5 additions & 1 deletion helix-term/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -29,10 +29,11 @@ helix-lsp = { version = "0.3", path = "../helix-lsp" }
anyhow = "1"
once_cell = "1.8"

tokio = { version = "1", features = ["full"] }
tokio = { version = "1", features = ["rt", "rt-multi-thread", "io-util", "io-std", "time", "process", "macros", "fs", "parking_lot"] }
num_cpus = "1"
tui = { path = "../helix-tui", package = "helix-tui", default-features = false, features = ["crossterm"] }
crossterm = { version = "0.20", features = ["event-stream"] }
signal-hook = "0.3"

futures-util = { version = "0.3", features = ["std", "async-await"], default-features = false }

Expand All @@ -54,3 +55,6 @@ toml = "0.5"

serde_json = "1.0"
serde = { version = "1.0", features = ["derive"] }

[target.'cfg(not(windows))'.dependencies] # https://github.com/vorner/signal-hook/issues/100
signal-hook-tokio = { version = "0.3", features = ["futures-v0_3"] }
80 changes: 70 additions & 10 deletions helix-term/src/application.rs
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,10 @@ use crossterm::{
event::{Event, EventStream},
execute, terminal,
};
#[cfg(not(windows))]
use signal_hook::{consts::signal, low_level};
#[cfg(not(windows))]
use signal_hook_tokio::Signals;

pub struct Application {
compositor: Compositor,
Expand All @@ -36,6 +40,8 @@ pub struct Application {
#[allow(dead_code)]
syn_loader: Arc<syntax::Loader>,

#[cfg(not(windows))]
signals: Signals,
jobs: Jobs,
lsp_progress: LspProgressMap,
}
Expand Down Expand Up @@ -102,6 +108,9 @@ impl Application {

editor.set_theme(theme);

#[cfg(not(windows))]
let signals = Signals::new(&[signal::SIGTSTP, signal::SIGCONT])?;

let app = Self {
compositor,
editor,
Expand All @@ -111,6 +120,8 @@ impl Application {
theme_loader,
syn_loader,

#[cfg(not(windows))]
signals,
jobs: Jobs::new(),
lsp_progress: LspProgressMap::new(),
};
Expand Down Expand Up @@ -147,6 +158,51 @@ impl Application {

use futures_util::StreamExt;

#[cfg(not(windows))]
tokio::select! {
biased;

event = reader.next() => {
self.handle_terminal_events(event)
}
Some(signal) = self.signals.next() => {
use helix_view::graphics::Rect;
match signal {
signal::SIGTSTP => {
self.compositor.save_cursor();
self.restore_term().unwrap();
low_level::emulate_default_handler(signal::SIGTSTP).unwrap();
}
signal::SIGCONT => {
self.claim_term().await.unwrap();
// redraw the terminal
let Rect { width, height, .. } = self.compositor.size();
self.compositor.resize(width, height);
self.compositor.load_cursor();
self.render();
}
_ => unreachable!(),
}
}
Some((id, call)) = self.editor.language_servers.incoming.next() => {
self.handle_language_server_message(call, id).await;
// limit render calls for fast language server messages
let last = self.editor.language_servers.incoming.is_empty();
if last || last_render.elapsed() > deadline {
self.render();
last_render = Instant::now();
}
}
Some(callback) = self.jobs.futures.next() => {
self.jobs.handle_callback(&mut self.editor, &mut self.compositor, callback);
self.render();
}
Some(callback) = self.jobs.wait_futures.next() => {
self.jobs.handle_callback(&mut self.editor, &mut self.compositor, callback);
self.render();
}
}
#[cfg(windows)]
tokio::select! {
biased;

Expand Down Expand Up @@ -443,12 +499,23 @@ impl Application {
}
}

pub async fn run(&mut self) -> Result<(), Error> {
async fn claim_term(&mut self) -> Result<(), Error> {
terminal::enable_raw_mode()?;
self.editor.close_language_servers(None).await?;
Ok(())
}

fn restore_term(&mut self) -> Result<(), Error> {
let mut stdout = stdout();
// reset cursor shape
write!(stdout, "\x1B[2 q")?;
execute!(stdout, terminal::LeaveAlternateScreen)?;
terminal::disable_raw_mode()?;
Ok(())
}

execute!(stdout, terminal::EnterAlternateScreen)?;
pub async fn run(&mut self) -> Result<(), Error> {
self.claim_term().await?;

// Exit the alternate screen and disable raw mode before panicking
let hook = std::panic::take_hook();
Expand All @@ -463,14 +530,7 @@ impl Application {

self.event_loop().await;

self.editor.close_language_servers(None).await?;

// reset cursor shape
write!(stdout, "\x1B[2 q")?;

execute!(stdout, terminal::LeaveAlternateScreen)?;

terminal::disable_raw_mode()?;
self.restore_term()?;

Ok(())
}
Expand Down
8 changes: 7 additions & 1 deletion helix-term/src/commands.rs
Original file line number Diff line number Diff line change
Expand Up @@ -290,7 +290,8 @@ impl Command {
surround_replace, "Surround replace",
surround_delete, "Surround delete",
select_textobject_around, "Select around object",
select_textobject_inner, "Select inside object"
select_textobject_inner, "Select inside object",
suspend, "Suspend"
);
}

Expand Down Expand Up @@ -3780,3 +3781,8 @@ fn surround_delete(cx: &mut Context) {
}
})
}

fn suspend(_cx: &mut Context) {
#[cfg(not(windows))]
signal_hook::low_level::raise(signal_hook::consts::signal::SIGTSTP).unwrap();
}
17 changes: 16 additions & 1 deletion helix-term/src/compositor.rs
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ pub trait Component: Any + AnyComponent {

use anyhow::Error;
use std::io::stdout;
use tui::backend::CrosstermBackend;
use tui::backend::{Backend, CrosstermBackend};
type Terminal = tui::terminal::Terminal<CrosstermBackend<std::io::Stdout>>;

pub struct Compositor {
Expand Down Expand Up @@ -99,6 +99,21 @@ impl Compositor {
.expect("Unable to resize terminal")
}

pub fn save_cursor(&mut self) {
if self.terminal.cursor_kind() == CursorKind::Hidden {
self.terminal
.backend_mut()
.show_cursor(CursorKind::Block)
.ok();
}
}

pub fn load_cursor(&mut self) {
if self.terminal.cursor_kind() == CursorKind::Hidden {
self.terminal.backend_mut().hide_cursor().ok();
}
}

pub fn push(&mut self, mut layer: Box<dyn Component>) {
let size = self.size();
// trigger required_size on init
Expand Down
1 change: 1 addition & 0 deletions helix-term/src/keymap.rs
Original file line number Diff line number Diff line change
Expand Up @@ -502,6 +502,7 @@ impl Default for Keymaps {
},

"\"" => select_register,
"C-z" => suspend,
});
// TODO: decide whether we want normal mode to also be select mode (kakoune-like), or whether
// we keep this separate select mode. More keys can fit into normal mode then, but it's weird
Expand Down
17 changes: 11 additions & 6 deletions helix-tui/src/terminal.rs
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,8 @@ where
buffers: [Buffer; 2],
/// Index of the current buffer in the previous array
current: usize,
/// Whether the cursor is currently hidden
hidden_cursor: bool,
/// Kind of cursor (hidden or others)
cursor_kind: CursorKind,
/// Viewport
viewport: Viewport,
}
Expand All @@ -57,7 +57,7 @@ where
{
fn drop(&mut self) {
// Attempt to restore the cursor state
if self.hidden_cursor {
if self.cursor_kind == CursorKind::Hidden {
if let Err(err) = self.show_cursor(CursorKind::Block) {
eprintln!("Failed to show the cursor: {}", err);
}
Expand Down Expand Up @@ -93,7 +93,7 @@ where
Buffer::empty(options.viewport.area),
],
current: 0,
hidden_cursor: false,
cursor_kind: CursorKind::Block,
viewport: options.viewport,
})
}
Expand Down Expand Up @@ -185,15 +185,20 @@ where
Ok(())
}

#[inline]
pub fn cursor_kind(&self) -> CursorKind {
self.cursor_kind
}

pub fn hide_cursor(&mut self) -> io::Result<()> {
self.backend.hide_cursor()?;
self.hidden_cursor = true;
self.cursor_kind = CursorKind::Hidden;
Ok(())
}

pub fn show_cursor(&mut self, kind: CursorKind) -> io::Result<()> {
self.backend.show_cursor(kind)?;
self.hidden_cursor = false;
self.cursor_kind = kind;
Ok(())
}

Expand Down
2 changes: 1 addition & 1 deletion helix-view/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ crossterm = { version = "0.20", optional = true }
once_cell = "1.8"
url = "2"

tokio = { version = "1", features = ["full"] }
tokio = { version = "1", features = ["rt", "rt-multi-thread", "io-util", "io-std", "time", "process", "macros", "fs", "parking_lot"] }
futures-util = { version = "0.3", features = ["std", "async-await"], default-features = false }

slotmap = "1"
Expand Down
2 changes: 1 addition & 1 deletion helix-view/src/graphics.rs
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
use bitflags::bitflags;
use std::cmp::{max, min};

#[derive(Debug)]
#[derive(Debug, Clone, Copy, PartialEq)]
/// UNSTABLE
pub enum CursorKind {
/// █
Expand Down

0 comments on commit 361c67c

Please sign in to comment.