use crate::error::{Error, Result};
use crate::highlight::Highlighting;
use crate::input::{InputSeq, KeySeq};
use crate::row::Row;
use crate::signal::SigwinchWatcher;
use crate::status_bar::StatusBar;
use crate::term_color::{Color, TermColor};
use crate::text_buffer::TextBuffer;
use std::cmp;
use std::io::Write;
use std::time::SystemTime;
use unicode_width::UnicodeWidthChar;
pub const VERSION: &str = env!("CARGO_PKG_VERSION");
pub const HELP: &str = "\
Ctrl-Q : Quit
Ctrl-S : Save to file
Ctrl-O : Open text buffer
Ctrl-X : Next text buffer
Alt-X : Previous text buffer
Ctrl-P or UP : Move cursor up
Ctrl-N or DOWN : Move cursor down
Ctrl-F or RIGHT : Move cursor right
Ctrl-B or LEFT : Move cursor left
Ctrl-A or Alt-LEFT or HOME : Move cursor to head of line
Ctrl-E or Alt-RIGHT or END : Move cursor to end of line
Ctrl-[ or Ctrl-V or PAGE DOWN : Next page
Ctrl-] or Alt-V or PAGE UP : Previous page
Alt-F or Ctrl-RIGHT : Move cursor to next word
Alt-B or Ctrl-LEFT : Move cursor to previous word
Alt-N or Ctrl-DOWN : Move cursor to next paragraph
Alt-P or Ctrl-UP : Move cursor to previous paragraph
Alt-< : Move cursor to top of file
Alt-> : Move cursor to bottom of file
Ctrl-H or BACKSPACE : Delete character
Ctrl-D or DELETE : Delete next character
Ctrl-W : Delete a word
Ctrl-J : Delete until head of line
Ctrl-K : Delete until end of line
Ctrl-U : Undo last change
Ctrl-R : Redo last undo change
Ctrl-G : Search text
Ctrl-M : New line
Ctrl-L : Refresh screen
Ctrl-? : Show this help";
#[derive(PartialEq)]
enum StatusMessageKind {
Info,
Error,
}
struct StatusMessage {
text: String,
timestamp: SystemTime,
kind: StatusMessageKind,
}
impl StatusMessage {
fn new<S: Into<String>>(message: S, kind: StatusMessageKind) -> StatusMessage {
StatusMessage {
text: message.into(),
timestamp: SystemTime::now(),
kind,
}
}
}
fn get_window_size<I, W>(input: I, mut output: W) -> Result<(usize, usize)>
where
I: Iterator<Item = Result<InputSeq>>,
W: Write,
{
if let Some(s) = term_size::dimensions_stdout() {
return Ok(s);
}
output.write(b"\x1b[9999C\x1b[9999B\x1b[6n")?;
output.flush()?;
for seq in input {
if let KeySeq::Cursor(r, c) = seq?.key {
return Ok((c, r));
}
}
Err(Error::UnknownWindowSize)
}
fn too_small_window(width: usize, height: usize) -> bool {
width < 1 || height < 3
}
#[derive(PartialEq, Clone, Copy, Debug)]
enum DrawMessage {
Open,
Close,
Update,
DoNothing,
}
impl DrawMessage {
fn fold(self, rhs: Self) -> Self {
use DrawMessage::*;
match (self, rhs) {
(Open, Open) => unreachable!(),
(Open, Close) => DoNothing,
(Open, Update) => Open,
(Close, Open) => Update,
(Close, Close) => unreachable!(),
(Close, Update) => unreachable!(),
(Update, Open) => unreachable!(),
(Update, Close) => Close,
(Update, Update) => Update,
(lhs, DoNothing) => lhs,
(DoNothing, rhs) => rhs,
}
}
}
pub struct Screen<W: Write> {
output: W,
rx: usize,
num_cols: usize,
num_rows: usize,
message: Option<StatusMessage>,
draw_message: DrawMessage,
dirty_start: Option<usize>,
sigwinch: SigwinchWatcher,
term_color: TermColor,
pub cursor_moved: bool,
pub rowoff: usize,
pub coloff: usize,
}
impl<W: Write> Screen<W> {
pub fn new<I>(size: Option<(usize, usize)>, input: I, mut output: W) -> Result<Self>
where
I: Iterator<Item = Result<InputSeq>>,
{
let (w, h) = if let Some(s) = size {
s
} else {
get_window_size(input, &mut output)?
};
if too_small_window(w, h) {
return Err(Error::TooSmallWindow(w, h));
}
output.write(b"\x1b[?47h")?;
Ok(Self {
output,
rx: 0,
num_cols: w,
num_rows: h.saturating_sub(2),
message: Some(StatusMessage::new(
"Ctrl-? for help",
StatusMessageKind::Info,
)),
draw_message: DrawMessage::Open,
dirty_start: Some(0),
sigwinch: SigwinchWatcher::new()?,
term_color: TermColor::from_env(),
cursor_moved: true,
rowoff: 0,
coloff: 0,
})
}
fn write_flush(&mut self, bytes: &[u8]) -> Result<()> {
self.output.write(bytes)?;
self.output.flush()?;
Ok(())
}
fn trim_line<'a, S: AsRef<str>>(&self, line: &'a S) -> String {
let line = line.as_ref();
if line.len() <= self.coloff {
return "".to_string();
}
line.chars().skip(self.coloff).take(self.num_cols).collect()
}
fn draw_status_bar<B: Write>(&self, mut buf: B, status_bar: &StatusBar) -> Result<()> {
write!(buf, "\x1b[{}H", self.rows() + 1)?;
buf.write(self.term_color.sequence(Color::Invert))?;
let left = status_bar.left();
let left = &left[..cmp::min(left.len(), self.num_cols)];
buf.write(left.as_bytes())?;
let rest_len = self.num_cols - left.len();
if rest_len == 0 {
buf.write(self.term_color.sequence(Color::Reset))?;
return Ok(());
}
let right = status_bar.right();
if right.len() > rest_len {
for _ in 0..rest_len {
buf.write(b" ")?;
}
buf.write(self.term_color.sequence(Color::Reset))?;
return Ok(());
}
for _ in 0..rest_len - right.len() {
buf.write(b" ")?;
}
buf.write(right.as_bytes())?;
buf.write(self.term_color.sequence(Color::Reset))?;
Ok(())
}
fn draw_message_bar<B: Write>(&self, mut buf: B, message: &StatusMessage) -> Result<()> {
let text = &message.text[..cmp::min(message.text.len(), self.num_cols)];
write!(buf, "\x1b[{}H", self.num_rows + 2)?;
if message.kind == StatusMessageKind::Error {
buf.write(self.term_color.sequence(Color::RedBG))?;
}
buf.write(text.as_bytes())?;
if message.kind != StatusMessageKind::Info {
buf.write(self.term_color.sequence(Color::Reset))?;
}
buf.write(b"\x1b[K")?;
Ok(())
}
pub fn render_welcome(&mut self, status_bar: &StatusBar) -> Result<()> {
self.write_flush(b"\x1b[?25l")?;
let mut buf = Vec::with_capacity((self.rows() + 2 + self.num_cols) * 3);
buf.write(self.term_color.sequence(Color::Reset))?;
for y in 0..self.rows() {
write!(buf, "\x1b[{}H", y + 1)?;
if y == self.rows() / 3 {
let msg_buf = format!("Kiro editor -- version {}", VERSION);
let welcome = self.trim_line(&msg_buf);
let padding = (self.num_cols - welcome.len()) / 2;
if padding > 0 {
buf.write(self.term_color.sequence(Color::NonText))?;
buf.write(b"~")?;
buf.write(self.term_color.sequence(Color::Reset))?;
for _ in 0..padding - 1 {
buf.write(b" ")?;
}
}
buf.write(welcome.as_bytes())?;
} else {
buf.write(self.term_color.sequence(Color::NonText))?;
buf.write(b"~")?;
}
buf.write(b"\x1b[K")?;
}
buf.write(self.term_color.sequence(Color::Reset))?;
self.draw_status_bar(&mut buf, status_bar)?;
if let Some(message) = &self.message {
self.draw_message_bar(&mut buf, message)?;
}
write!(buf, "\x1b[H")?;
buf.write(b"\x1b[?25h")?;
self.write_flush(&buf)?;
self.after_render();
Ok(())
}
fn draw_rows<B: Write>(
&self,
mut buf: B,
dirty_start: usize,
rows: &[Row],
hl: &Highlighting,
) -> Result<()> {
let row_len = rows.len();
buf.write(self.term_color.sequence(Color::Reset))?;
for y in 0..self.rows() {
let file_row = y + self.rowoff;
if file_row < dirty_start {
continue;
}
write!(buf, "\x1b[{}H", y + 1)?;
if file_row >= row_len {
buf.write(self.term_color.sequence(Color::NonText))?;
buf.write(b"~")?;
} else {
let row = &rows[file_row];
let mut col = 0;
let mut prev_color = Color::Reset;
for (c, hl) in row.render_text().chars().zip(hl.lines[file_row].iter()) {
col += c.width_cjk().unwrap_or(1);
if col <= self.coloff {
continue;
} else if col > self.num_cols + self.coloff {
break;
}
let color = hl.color();
if color != prev_color {
if prev_color.has_bg_color() {
buf.write(self.term_color.sequence(Color::Reset))?;
}
buf.write(self.term_color.sequence(color))?;
prev_color = color;
}
write!(buf, "{}", c)?;
}
}
buf.write(self.term_color.sequence(Color::Reset))?;
buf.write(b"\x1b[K")?;
}
Ok(())
}
fn redraw(
&mut self,
text_buf: &TextBuffer,
hl: &Highlighting,
status_bar: &StatusBar,
) -> Result<()> {
let cursor_row = text_buf.cy() - self.rowoff + 1;
let cursor_col = self.rx - self.coloff + 1;
let draw_message = self.draw_message;
if self.dirty_start.is_none()
&& !status_bar.redraw
&& draw_message == DrawMessage::DoNothing
{
if self.cursor_moved {
write!(self.output, "\x1b[{};{}H", cursor_row, cursor_col)?;
self.output.flush()?;
}
return Ok(());
}
self.write_flush(b"\x1b[?25l")?;
let mut buf = Vec::with_capacity((self.rows() + 2) * self.num_cols);
if let Some(s) = self.dirty_start {
self.draw_rows(&mut buf, s, text_buf.rows(), hl)?;
}
if status_bar.redraw
|| draw_message == DrawMessage::Open
|| draw_message == DrawMessage::Close
{
self.draw_status_bar(&mut buf, status_bar)?;
}
if draw_message == DrawMessage::Update || draw_message == DrawMessage::Open {
if let Some(message) = &self.message {
self.draw_message_bar(&mut buf, message)?;
}
}
write!(buf, "\x1b[{};{}H", cursor_row, cursor_col)?;
buf.write(b"\x1b[?25h")?;
self.write_flush(&buf)?;
Ok(())
}
fn next_coloff(&self, want_stop: usize, row: &Row) -> usize {
let mut coloff = 0;
for c in row.render_text().chars() {
coloff += c.width_cjk().unwrap_or(1);
if coloff >= want_stop {
break;
}
}
coloff
}
fn do_scroll(&mut self, rows: &[Row], (cx, cy): (usize, usize)) {
let prev_rowoff = self.rowoff;
let prev_coloff = self.coloff;
if cy < rows.len() {
self.rx = rows[cy].rx_from_cx(cx);
} else {
self.rx = 0;
}
if cy < self.rowoff {
self.rowoff = cy;
}
if cy >= self.rowoff + self.rows() {
self.rowoff = cy - self.rows() + 1;
}
if self.rx < self.coloff {
self.coloff = self.rx;
}
if self.rx >= self.coloff + self.num_cols {
self.coloff = self.next_coloff(self.rx - self.num_cols + 1, &rows[cy]);
}
if prev_rowoff != self.rowoff || prev_coloff != self.coloff {
self.set_dirty_start(self.rowoff);
}
}
fn update_message_bar(&mut self) -> Result<()> {
if let Some(m) = &self.message {
if SystemTime::now().duration_since(m.timestamp)?.as_secs() > 5 {
self.unset_message();
}
}
if self.draw_message == DrawMessage::Close {
self.set_dirty_start(self.num_rows);
}
Ok(())
}
fn after_render(&mut self) {
self.dirty_start = None;
self.cursor_moved = false;
self.draw_message = DrawMessage::DoNothing;
}
pub fn render(
&mut self,
buf: &TextBuffer,
hl: &mut Highlighting,
status_bar: &StatusBar,
) -> Result<()> {
self.do_scroll(buf.rows(), buf.cursor());
self.update_message_bar()?;
hl.update(buf.rows(), self.rowoff + self.rows());
self.redraw(buf, hl, status_bar)?;
self.after_render();
Ok(())
}
pub fn render_help(&mut self) -> Result<()> {
let help: Vec<_> = HELP
.split('\n')
.skip_while(|s| !s.contains(':'))
.map(str::trim_start)
.collect();
let rows = self.rows();
let vertical_margin = if help.len() < rows {
(rows - help.len()) / 2
} else {
0
};
let help_max_width = help.iter().map(|l| l.len()).max().unwrap();
let left_margin = if help_max_width < self.num_cols {
(self.num_cols - help_max_width) / 2
} else {
0
};
let mut buf = Vec::with_capacity(rows * self.num_cols);
for y in 0..vertical_margin {
write!(buf, "\x1b[{}H", y + 1)?;
buf.write(b"\x1b[K")?;
}
let left_pad = " ".repeat(left_margin);
let help_height = cmp::min(vertical_margin + help.len(), rows);
for y in vertical_margin..help_height {
let idx = y - vertical_margin;
write!(buf, "\x1b[{}H", y + 1)?;
buf.write(left_pad.as_bytes())?;
let help = &help[idx][..cmp::min(help[idx].len(), self.num_cols)];
buf.write(self.term_color.sequence(Color::Cyan))?;
let mut cols = help.split(':');
if let Some(col) = cols.next() {
buf.write(col.as_bytes())?;
}
buf.write(self.term_color.sequence(Color::Reset))?;
if let Some(col) = cols.next() {
write!(buf, ":{}", col)?;
}
buf.write(b"\x1b[K")?;
}
for y in help_height..rows {
write!(buf, "\x1b[{}H", y + 1)?;
buf.write(b"\x1b[K")?;
}
self.write_flush(&buf)
}
pub fn set_dirty_start(&mut self, start: usize) {
if let Some(s) = self.dirty_start {
if s < start {
return;
}
}
self.dirty_start = Some(start);
}
pub fn maybe_resize<I>(&mut self, input: I) -> Result<bool>
where
I: Iterator<Item = Result<InputSeq>>,
{
if !self.sigwinch.notified() {
return Ok(false);
}
let (w, h) = get_window_size(input, &mut self.output)?;
if too_small_window(w, h) {
return Err(Error::TooSmallWindow(w, h));
}
self.num_rows = h.saturating_sub(2);
self.num_cols = w;
self.dirty_start = Some(0);
Ok(true)
}
fn set_message(&mut self, m: Option<StatusMessage>) {
let op = match (&self.message, &m) {
(Some(p), Some(n)) if p.text == n.text => DrawMessage::DoNothing,
(Some(_), Some(_)) => DrawMessage::Update,
(None, Some(_)) => DrawMessage::Open,
(Some(_), None) => DrawMessage::Close,
(None, None) => DrawMessage::DoNothing,
};
self.draw_message = self.draw_message.fold(op);
self.message = m;
}
pub fn set_info_message<S: Into<String>>(&mut self, message: S) {
self.set_message(Some(StatusMessage::new(message, StatusMessageKind::Info)));
}
pub fn set_error_message<S: Into<String>>(&mut self, message: S) {
self.set_message(Some(StatusMessage::new(message, StatusMessageKind::Error)));
}
pub fn unset_message(&mut self) {
self.set_message(None);
}
pub fn rows(&self) -> usize {
if self.message.is_some() {
self.num_rows
} else {
self.num_rows + 1
}
}
pub fn cols(&self) -> usize {
self.num_cols
}
pub fn message_text(&self) -> &'_ str {
self.message.as_ref().map(|m| m.text.as_str()).unwrap_or("")
}
pub fn force_set_cursor(&mut self, row: usize, col: usize) -> Result<()> {
write!(self.output, "\x1b[{};{}H", row, col)?;
self.output.flush()?;
Ok(())
}
}
impl<W: Write> Drop for Screen<W> {
fn drop(&mut self) {
self.write_flush(b"\x1b[?47l\x1b[H")
.expect("Back to normal screen buffer");
}
}