modalkit_ratatui/

textbox.rs

//! # Text box
//!
//! ## Overview
//!
//! This text box provides a view of a shared editing buffer, and, on top of passing operations
//! through to an [EditBuffer], is capable of doing the following:
//!
//! - Toggling wrapped and non-wrapped views of the buffer's
//! - Scrolling through the buffer's contents
//! - Rendering line annotations in left and right gutters
//!
//! [EditBuffer]: modalkit::editing::buffer::EditBuffer
//!
//! ## Example
//!
//! ```
//! use modalkit::editing::application::EmptyInfo;
//! use modalkit::editing::store::Store;
//! use modalkit_ratatui::textbox::TextBoxState;
//!
//! use ratatui::layout::Rect;
//!
//! let mut store = Store::<EmptyInfo>::default();
//! let buffer = store.load_buffer(String::from("*scratch*"));
//! let mut tbox = TextBoxState::new(buffer);
//!
//! tbox.set_term_info(Rect::new(0, 0, 6, 4));
//!
//! tbox.set_text("a\nb\nc\nd\ne\nf\n");
//! assert_eq!(tbox.get_text(), "a\nb\nc\nd\ne\nf\n");
//! assert_eq!(tbox.get_lines(), 6);
//! assert_eq!(tbox.has_lines(4), 4);
//! assert_eq!(tbox.has_lines(6), 6);
//! assert_eq!(tbox.has_lines(8), 6);
//! ```
use std::convert::TryInto;
use std::iter::Iterator;
use std::marker::PhantomData;

use ratatui::{
    buffer::Buffer,
    layout::Rect,
    style::{Modifier, Style},
    text::Span,
    widgets::{Block, StatefulWidget, Widget},
};

use modalkit::actions::*;
use modalkit::editing::{
    application::{ApplicationInfo, EmptyInfo},
    buffer::{CursorGroupId, FollowersInfo, HighlightInfo},
    completion::CompletionList,
    context::{EditContext, Resolve},
    cursor::Cursor,
    rope::{CharOff, EditRope},
    store::{SharedBuffer, Store},
};
use modalkit::errors::{EditError, EditResult, UIResult};
use modalkit::prelude::*;

use unicode_width::{UnicodeWidthChar, UnicodeWidthStr};

use super::{ScrollActions, TerminalCursor, WindowOps};

/// Line annotation shown in the left gutter.
pub struct LeftGutterInfo {
    text: String,
    style: Style,
}

impl LeftGutterInfo {
    /// Create a new instance.
    pub fn new(text: String, style: Style) -> Self {
        LeftGutterInfo { text, style }
    }

    fn render(&self, area: Rect, buf: &mut Buffer) {
        let _ = buf.set_stringn(area.x, area.y, &self.text, area.width as usize, self.style);
    }
}

/// Line annotation shown in the right gutter.
pub struct RightGutterInfo {
    text: String,
    style: Style,
}

impl RightGutterInfo {
    /// Create a new instance.
    pub fn new(text: String, style: Style) -> Self {
        RightGutterInfo { text, style }
    }

    fn render(&self, area: Rect, buf: &mut Buffer) {
        let _ = buf.set_stringn(area.x, area.y, &self.text, area.width as usize, self.style);
    }
}

/// Persistent state for [TextBox].
pub struct TextBoxState<I: ApplicationInfo = EmptyInfo> {
    buffer: SharedBuffer<I>,
    group_id: CursorGroupId,
    readonly: bool,

    viewctx: ViewportContext<Cursor>,
    term_cursor: (u16, u16),
}

/// Widget for rendering a multi-line text box.
pub struct TextBox<'a, I: ApplicationInfo = EmptyInfo> {
    block: Option<Block<'a>>,
    prompt: Span<'a>,
    oneline: bool,
    style: Style,

    lgutter_width: u16,
    rgutter_width: u16,

    _pc: PhantomData<I>,
}

/*
 * If the cursor has moved outside of the viewport, update the corner of the viewport so that the
 * cursor is visible onscreen again.
 */
fn shift_corner_nowrap(cursor: &Cursor, corner: &mut Cursor, width: usize, height: usize) {
    if cursor.y < corner.y {
        corner.set_y(cursor.y);
    } else if cursor.y >= corner.y + height {
        corner.set_y(cursor.y - height + 1);
    }

    if cursor.x < corner.x {
        corner.set_x(cursor.x);
    } else if cursor.x >= corner.x + width {
        corner.set_x(cursor.x - width + 1);
    }
}

fn shift_corner_wrap(cursor: &Cursor, corner: &mut Cursor, height: usize) {
    if cursor.y < corner.y {
        corner.set_y(cursor.y);
        corner.set_x(0);
    } else if cursor.y >= corner.y + height {
        corner.set_y(cursor.y - height + 1);
        corner.set_x(0);
    } else if cursor.y == corner.y && cursor.x < corner.x {
        corner.set_x(0);
    }
}

fn shift_corner_oneline(cursor: &Cursor, corner: &mut Cursor) {
    if cursor < corner {
        corner.set_y(cursor.y);
        corner.set_x(cursor.x);
    }
}

fn shift_corner(
    viewctx: &mut ViewportContext<Cursor>,
    cursor: &Cursor,
    width: usize,
    height: usize,
) {
    if viewctx.wrap {
        shift_corner_wrap(cursor, &mut viewctx.corner, height);
    } else {
        shift_corner_nowrap(cursor, &mut viewctx.corner, width, height);
    }
}

/*
 * If the cursor has moved outside of the viewport, move the cursor back within the boundaries of
 * the viewport, so it is visible onscreen again.
 */
fn shift_cursor(cursor: &mut Cursor, corner: &Cursor, width: usize, height: usize) {
    if cursor.y < corner.y {
        cursor.set_y(corner.y);
    } else if cursor.y >= corner.y + height {
        cursor.set_y(corner.y + height - 1);
    }

    if cursor.x < corner.x {
        cursor.set_x(corner.x);
    } else if cursor.x >= corner.x + width {
        cursor.set_x(corner.x + width - 1);
    }
}

impl<I> TextBoxState<I>
where
    I: ApplicationInfo,
{
    /// Create state for a new text box.
    pub fn new(buffer: SharedBuffer<I>) -> Self {
        let mut viewctx = ViewportContext::default();
        let group_id = buffer.write().unwrap().create_group();

        viewctx.set_wrap(true);

        TextBoxState {
            buffer,
            group_id,
            readonly: false,

            viewctx,
            term_cursor: (0, 0),
        }
    }

    /// Get a reference to the shared buffer used by this text box.
    pub fn buffer(&self) -> SharedBuffer<I> {
        self.buffer.clone()
    }

    /// Indicates whether the buffer contents are readonly.
    pub fn is_readonly(&self) -> bool {
        self.readonly
    }

    /// Set whether the buffer contents are modifiable through the [Editable] trait.
    pub fn set_readonly(&mut self, readonly: bool) {
        self.readonly = readonly;
    }

    /// Get the contents of the underlying buffer as an [EditRope].
    pub fn get(&self) -> EditRope {
        self.buffer.read().unwrap().get().clone()
    }

    /// Get the contents of the underlying buffer as a [String].
    pub fn get_text(&self) -> String {
        self.buffer.read().unwrap().get_text()
    }

    /// Replace the contents of the text box's underlying buffer.
    pub fn set_text<T: Into<EditRope>>(&mut self, t: T) {
        self.buffer.write().unwrap().set_text(t)
    }

    /// Clear the text box's underlying buffer of its content, and return it.
    pub fn reset(&mut self) -> EditRope {
        self.buffer.write().unwrap().reset()
    }

    /// Clear the text box's underlying buffer of its content, and return it as a [String].
    pub fn reset_text(&mut self) -> String {
        self.buffer.write().unwrap().reset_text()
    }

    /// Create or update a line annotation for the left gutter.
    pub fn set_left_gutter(&mut self, line: usize, s: String, style: Option<Style>) {
        let style = style.unwrap_or_default();
        let info = LeftGutterInfo::new(s, style);

        self.buffer.write().unwrap().set_line_info(line, info);
    }

    /// Create or update a line annotation for the right gutter.
    pub fn set_right_gutter(&mut self, line: usize, s: String, style: Option<Style>) {
        let style = style.unwrap_or_default();
        let info = RightGutterInfo::new(s, style);

        self.buffer.write().unwrap().set_line_info(line, info);
    }

    /// Control whether the text box should wrap long lines when displaying them.
    pub fn set_wrap(&mut self, wrap: bool) {
        self.viewctx.set_wrap(wrap);
    }

    /// Inform the text box what its dimensions and placement on the terminal window is.
    pub fn set_term_info(&mut self, area: Rect) {
        self.viewctx.dimensions = (area.width as usize, area.height as usize);
    }

    /// Get the leader cursor for this text box's cursor group.
    pub fn get_cursor(&mut self) -> Cursor {
        self.buffer.write().unwrap().get_leader(self.group_id)
    }

    /// Calculate how many lines are in this text box.
    pub fn get_lines(&self) -> usize {
        self.buffer.read().unwrap().get_lines()
    }

    /// Check whether this text box is capable of displaying `max` lines.
    ///
    /// If there are fewer lines available than `max`, this returns the same value as
    /// [get_lines()](TextBoxState::get_lines).
    /// Otherwise, this returns `max`.
    ///
    /// This method is useful for building additional widgets that want to create a [TextBox] with a
    /// flexible height up to `max` lines.
    pub fn has_lines(&self, max: usize) -> usize {
        if self.viewctx.wrap {
            let width = self.viewctx.get_width();
            let mut count = 0;

            if width == 0 {
                return count;
            }

            let mut fline = false;

            for line in self.buffer.read().unwrap().lines(0) {
                count += 1;
                let mut line_width = 0;
                for c in line.chars(CharOff::from(0)) {
                    let c_width = c.width_cjk().unwrap_or(1);
                    if line_width + c_width > width {
                        count += 1;
                        line_width = c_width;
                    } else {
                        line_width += c_width;
                    }
                }
                fline |= line_width == width;

                if count >= max {
                    return max;
                }
            }

            if fline {
                // At least one of our lines is the full area width, so
                // we bump the count by 1 to move closer to the max, so
                // that moving the cursor to the line end doesn't move
                // the viewport corner in annoying ways.
                count += 1;
            }

            return count;
        } else {
            self.buffer.read().unwrap().get_lines().min(max)
        }
    }
}

macro_rules! c2cgi {
    ($s: expr, $ctx: expr) => {
        &($s.group_id, &$s.viewctx, $ctx)
    };
}

impl<I> Editable<EditContext, Store<I>, I> for TextBoxState<I>
where
    I: ApplicationInfo,
{
    fn editor_command(
        &mut self,
        act: &EditorAction,
        ctx: &EditContext,
        store: &mut Store<I>,
    ) -> EditResult<EditInfo, I> {
        if self.readonly && !act.is_readonly(ctx) {
            Err(EditError::ReadOnly)
        } else {
            self.buffer.editor_command(act, c2cgi!(self, ctx), store)
        }
    }
}

impl<I> Jumpable<EditContext, I> for TextBoxState<I>
where
    I: ApplicationInfo,
{
    fn jump(
        &mut self,
        list: PositionList,
        dir: MoveDir1D,
        count: usize,
        ctx: &EditContext,
    ) -> UIResult<usize, I> {
        self.buffer.jump(list, dir, count, c2cgi!(self, ctx))
    }
}

impl<I> Promptable<EditContext, Store<I>, I> for TextBoxState<I>
where
    I: ApplicationInfo,
{
    fn prompt(
        &mut self,
        _: &PromptAction,
        _: &EditContext,
        _: &mut Store<I>,
    ) -> EditResult<Vec<(Action<I>, EditContext)>, I> {
        Err(EditError::Failure("Not at a prompt".to_string()))
    }
}

impl<I> Searchable<EditContext, Store<I>, I> for TextBoxState<I>
where
    I: ApplicationInfo,
{
    fn search(
        &mut self,
        dir: MoveDirMod,
        count: Count,
        ctx: &EditContext,
        store: &mut Store<I>,
    ) -> UIResult<EditInfo, I> {
        self.buffer.search(dir, count, c2cgi!(self, ctx), store)
    }
}

impl<I> ScrollActions<EditContext, Store<I>, I> for TextBoxState<I>
where
    I: ApplicationInfo,
{
    fn dirscroll(
        &mut self,
        dir: MoveDir2D,
        size: ScrollSize,
        count: &Count,
        ctx: &EditContext,
        _: &mut Store<I>,
    ) -> EditResult<EditInfo, I> {
        let count = ctx.resolve(count);

        let height = self.viewctx.dimensions.1;
        let rows = match size {
            ScrollSize::Cell => count,
            ScrollSize::HalfPage => count.saturating_mul(height) / 2,
            ScrollSize::Page => count.saturating_mul(height),
        };

        let width = self.viewctx.dimensions.0;
        let cols = match size {
            ScrollSize::Cell => count,
            ScrollSize::HalfPage => count.saturating_mul(width) / 2,
            ScrollSize::Page => count.saturating_mul(width),
        };

        match (dir, self.viewctx.wrap) {
            (MoveDir2D::Up, _) => self.viewctx.corner.up(rows),
            (MoveDir2D::Down, _) => self.viewctx.corner.down(rows),
            (MoveDir2D::Left, false) => self.viewctx.corner.left(cols),
            (MoveDir2D::Right, false) => self.viewctx.corner.right(cols),
            (MoveDir2D::Left | MoveDir2D::Right, true) => (),
        };

        /*
         * We do a quick dance here: moving the viewport should move the cursor so that it stays
         * visible on the screen. The cursor should not be shifted past the last line or last
         * column, though, so we clamp it after shifting it. Since the cursor should never
         * be off-screen, this also sets a boundary of how far we can move the viewport.
         */
        let mut cursor = self.get_cursor();
        let mut buffer = self.buffer.write().unwrap();
        shift_cursor(&mut cursor, &self.viewctx.corner, width, height);
        buffer.clamp(&mut cursor, c2cgi!(self, ctx));
        shift_corner(&mut self.viewctx, &cursor, width, height);
        buffer.set_leader(self.group_id, cursor);

        Ok(None)
    }

    fn cursorpos(
        &mut self,
        pos: MovePosition,
        axis: Axis,
        _: &EditContext,
        _: &mut Store<I>,
    ) -> EditResult<EditInfo, I> {
        if axis == Axis::Horizontal && self.viewctx.wrap {
            return Ok(None);
        }

        let (width, height) = self.viewctx.dimensions;
        let cursor = self.get_cursor();
        shift_corner(&mut self.viewctx, &cursor, width, height);

        match (axis, pos) {
            (Axis::Horizontal, MovePosition::Beginning) => {
                self.viewctx.corner.set_x(cursor.x);
            },
            (Axis::Horizontal, MovePosition::Middle) => {
                let off = cursor.x.saturating_add(1).saturating_sub(width / 2);

                self.viewctx.corner.set_x(off);
            },
            (Axis::Horizontal, MovePosition::End) => {
                let off = cursor.x.saturating_add(1).saturating_sub(width);

                self.viewctx.corner.set_x(off);
            },
            (Axis::Vertical, MovePosition::Beginning) => {
                self.viewctx.corner.set_y(cursor.y);
            },
            (Axis::Vertical, MovePosition::Middle) => {
                let off = cursor.y.saturating_add(1).saturating_sub(height / 2);

                self.viewctx.corner.set_y(off);
            },
            (Axis::Vertical, MovePosition::End) => {
                let off = cursor.y.saturating_add(1).saturating_sub(height);

                self.viewctx.corner.set_y(off);
            },
        }

        Ok(None)
    }

    fn linepos(
        &mut self,
        pos: MovePosition,
        count: &Count,
        ctx: &EditContext,
        _: &mut Store<I>,
    ) -> EditResult<EditInfo, I> {
        let mut buffer = self.buffer.write().unwrap();
        let max = buffer.get_lines();
        let line = ctx.resolve(count).min(max).saturating_sub(1);

        let height = self.viewctx.get_height();

        buffer.set_leader(self.group_id, Cursor::new(line, 0));

        match pos {
            MovePosition::Beginning => {
                self.viewctx.corner.set_y(line);
            },
            MovePosition::Middle => {
                let off = line.saturating_add(1).saturating_sub(height / 2);

                self.viewctx.corner.set_y(off);
            },
            MovePosition::End => {
                let off = line.saturating_add(1).saturating_sub(height);

                self.viewctx.corner.set_y(off);
            },
        }

        Ok(None)
    }
}

impl<I> Scrollable<EditContext, Store<I>, I> for TextBoxState<I>
where
    I: ApplicationInfo,
{
    fn scroll(
        &mut self,
        style: &ScrollStyle,
        ctx: &EditContext,
        store: &mut Store<I>,
    ) -> EditResult<EditInfo, I> {
        match style {
            ScrollStyle::Direction2D(dir, size, count) => {
                return self.dirscroll(*dir, *size, count, ctx, store);
            },
            ScrollStyle::CursorPos(pos, axis) => {
                return self.cursorpos(*pos, *axis, ctx, store);
            },
            ScrollStyle::LinePos(pos, count) => {
                return self.linepos(*pos, count, ctx, store);
            },
        }
    }
}

impl<I> TerminalCursor for TextBoxState<I>
where
    I: ApplicationInfo,
{
    fn get_term_cursor(&self) -> Option<(u16, u16)> {
        if self.viewctx.get_height() == 0 {
            return None;
        }

        self.term_cursor.into()
    }
}

impl<I> WindowOps<I> for TextBoxState<I>
where
    I: ApplicationInfo,
{
    fn dup(&self, _: &mut Store<I>) -> Self {
        let buffer = self.buffer.clone();
        let group_id = buffer.write().unwrap().create_group();

        TextBoxState {
            buffer,
            group_id,
            readonly: self.readonly,

            viewctx: self.viewctx.clone(),
            term_cursor: (0, 0),
        }
    }

    fn close(&mut self, _: CloseFlags, _: &mut Store<I>) -> bool {
        true
    }

    fn write(&mut self, _: Option<&str>, _: WriteFlags, _: &mut Store<I>) -> UIResult<EditInfo, I> {
        if self.readonly {
            return Err(EditError::ReadOnly.into());
        } else {
            return Ok(None);
        }
    }

    fn draw(&mut self, area: Rect, buf: &mut Buffer, _: bool, _: &mut Store<I>) {
        TextBox::new().render(area, buf, self);
    }

    fn get_completions(&self) -> Option<CompletionList> {
        self.buffer.read().unwrap().get_completions(self.group_id)
    }

    fn get_cursor_word(&self, style: &WordStyle) -> Option<String> {
        self.buffer.read().unwrap().get_cursor_word(self.group_id, style)
    }

    fn get_selected_word(&self) -> Option<String> {
        self.buffer.read().unwrap().get_selected_word(self.group_id)
    }
}

impl<'a, I> TextBox<'a, I>
where
    I: ApplicationInfo,
{
    /// Create a new widget.
    pub fn new() -> Self {
        TextBox {
            block: None,
            prompt: Span::default(),
            oneline: false,
            style: Style::default(),

            lgutter_width: 0,
            rgutter_width: 0,

            _pc: PhantomData,
        }
    }

    /// Set the style to use for rendering text within the [TextBoxState].
    pub fn style(mut self, style: Style) -> Self {
        self.style = style;
        self
    }

    /// Wrap this text box in a [Block].
    pub fn block(mut self, block: Block<'a>) -> Self {
        self.block = Some(block);
        self
    }

    /// Force the text to render on a single line. LF and CR will be rendered as ^J and ^M
    /// respectively.
    ///
    /// Any gutters will not be shown.
    pub fn oneline(mut self) -> Self {
        self.oneline = true;
        self
    }

    /// Display a prompt in the top left of the text box when focused.
    pub fn prompt(mut self, prompt: impl Into<Span<'a>>) -> Self {
        self.prompt = prompt.into();
        self
    }

    /// Set the width of the left gutter.
    pub fn left_gutter(mut self, lw: u16) -> Self {
        self.lgutter_width = lw;
        self
    }

    /// Set the width of the right gutter.
    pub fn right_gutter(mut self, rw: u16) -> Self {
        self.rgutter_width = rw;
        self
    }

    #[inline]
    fn _highlight_followers(
        &self,
        line: usize,
        start: usize,
        end: usize,
        (x, y): (u16, u16),
        followers: &FollowersInfo,
        buf: &mut Buffer,
    ) {
        let hlstyled = self.style.add_modifier(Modifier::REVERSED);
        let cs = (line, start);
        let ce = (line, end);

        for follower in followers.query(cs..ce) {
            let fx = x + (follower.value.x - start) as u16;
            let fa = Rect::new(fx, y, 1, 1);
            buf.set_style(fa, hlstyled);
        }
    }

    #[inline]
    fn _set_style(&self, start: usize, h1: usize, h2: usize, (x, y): (u16, u16), buf: &mut Buffer) {
        let tx: u16 = x + (h1 - start) as u16;
        let selwidth: u16 = (h2 - h1 + 1).try_into().unwrap();

        let hlstyled = self.style.add_modifier(Modifier::REVERSED);
        let selarea = Rect::new(tx, y, selwidth, 1);

        buf.set_style(selarea, hlstyled);
    }

    #[inline]
    fn _highlight_line(
        &self,
        line: usize,
        start: usize,
        end: usize,
        (x, y): (u16, u16),
        hls: &HighlightInfo,
        buf: &mut Buffer,
    ) {
        for selection in hls.query_point(line) {
            let (sb, se, shape) = &selection.value;

            let maxcol = end.saturating_sub(1);
            let range = start..end;

            match shape {
                TargetShape::CharWise => {
                    let x1 = if line == sb.y { sb.x.max(start) } else { start };
                    let x2 = if line == se.y {
                        se.x.min(maxcol)
                    } else {
                        maxcol
                    };

                    if range.contains(&x1) && range.contains(&x2) {
                        self._set_style(start, x1, x2, (x, y), buf);
                    }
                },
                TargetShape::LineWise => {
                    let hlstyled = self.style.add_modifier(Modifier::REVERSED);
                    let selwidth: u16 = (end - start).try_into().unwrap();
                    let selarea = Rect::new(x, y, selwidth, 1);

                    buf.set_style(selarea, hlstyled);
                },
                TargetShape::BlockWise => {
                    let lx = sb.x.min(se.x);
                    let rx = sb.x.max(se.x);

                    let x1 = lx.max(start);
                    let x2 = rx.min(maxcol);

                    if range.contains(&x1) && range.contains(&x2) {
                        self._set_style(start, x1, x2, (x, y), buf);
                    }
                },
            }
        }
    }

    fn _render_lines_wrap(
        &mut self,
        area: Rect,
        gutters: (Rect, Rect),
        buf: &mut Buffer,
        hinfo: HighlightInfo,
        finfo: FollowersInfo,
        state: &mut TextBoxState<I>,
    ) {
        let bot = area.bottom();
        let x = area.left();
        let mut y = area.top();

        let height = area.height as usize;
        let width = area.width as usize;

        /*
         * If the cursor has moved off-screen, update the viewport corner.
         *
         * There might be several long wrapped lines between the new corner and the cursor
         * afterwards, though, so we update the corner again after handling wrapping if needed.
         */
        let cursor = state.get_cursor();
        shift_corner_wrap(&cursor, &mut state.viewctx.corner, height);

        let cby = state.viewctx.corner.y;
        let cbx = state.viewctx.corner.x;

        let text = state.buffer.read().unwrap();

        let mut wrapped = Vec::new();
        let mut sawcursor = false;

        for (loff, s) in text.lines_at(cby, cbx).enumerate() {
            if wrapped.len() >= height && sawcursor {
                break;
            }

            let base = if loff == 0 { cbx } else { 0 };
            let line = cby + loff;
            let mut first = true;
            let mut char_offset = 0;
            let s_charlen = s.len();

            while char_offset < s_charlen && (wrapped.len() < height || !sawcursor) {
                let start_char = char_offset;

                let mut full = false;
                let mut num_chars = 0;
                let mut cur_width = 0;
                for (i, c) in s.chars(CharOff::from(start_char)).enumerate() {
                    let c_width = c.width_cjk().unwrap_or(1);
                    if cur_width + c_width > width {
                        full = true;
                        break;
                    }
                    cur_width += c_width;
                    num_chars = i + 1;
                }

                let end_char = start_char + num_chars;
                let swrapped =
                    s.slice(CharOff::from(start_char)..CharOff::from(end_char)).to_string();
                char_offset = end_char;

                let start = base + start_char;
                let end = base + end_char;
                let slen = base + s_charlen;

                full |= cur_width == width;
                let last = end == slen && cursor.x == slen;
                let cursor_line = line == cursor.y && ((start..end).contains(&cursor.x) || last);

                if cursor_line && full && last {
                    wrapped.push((line, start, end, swrapped, false, first));
                    wrapped.push((line, end, end, " ".to_string(), true, first));
                } else {
                    wrapped.push((line, start, end, swrapped, cursor_line, first));
                }

                sawcursor |= cursor_line;

                first = false;
            }

            if s_charlen == 0 {
                let cursor_line = line == cursor.y;
                wrapped.push((line, base, base, s.to_string(), cursor_line, true));
                sawcursor |= cursor_line;
            }
        }

        if wrapped.len() > height {
            let n = wrapped.len() - height;
            let _ = wrapped.drain(..n);
            let (line, start, _, _, _, _) = wrapped.first().unwrap();
            state.viewctx.corner.set_y(*line);
            state.viewctx.corner.set_x(*start);
        }

        for (line, start, end, s, cursor_line, first) in wrapped.into_iter() {
            if y >= bot {
                break;
            }

            if first {
                let lgutter = text.get_line_info::<LeftGutterInfo>(line);
                let rgutter = text.get_line_info::<RightGutterInfo>(line);

                if let Some(lgi) = lgutter {
                    let lga = Rect::new(gutters.0.x, y, gutters.0.width, 0);
                    lgi.render(lga, buf);
                }

                if let Some(rgi) = rgutter {
                    let rga = Rect::new(gutters.1.x, y, gutters.1.width, 0);
                    rgi.render(rga, buf);
                }
            }

            if cursor_line {
                let coff = s[..s
                    .char_indices()
                    .map(|(i, _)| i)
                    .nth(cursor.x.saturating_sub(start))
                    .unwrap_or(s.len())]
                    .width_cjk() as u16;

                state.term_cursor = (x + coff, y);
            }

            let _ = buf.set_stringn(x, y, s, width, self.style);

            self._highlight_followers(line, start, end, (x, y), &finfo, buf);
            self._highlight_line(line, start, end, (x, y), &hinfo, buf);

            y += 1;
        }
    }

    fn _render_lines_oneline(
        &mut self,
        area: Rect,
        buf: &mut Buffer,
        hinfo: HighlightInfo,
        finfo: FollowersInfo,
        state: &mut TextBoxState<I>,
    ) {
        let right = area.right();
        let mut x = area.left();
        let y = area.top();

        let width = area.width as usize;

        // If the cursor has moved off-screen, update the viewport corner.
        let cursor = state.get_cursor();
        shift_corner_oneline(&cursor, &mut state.viewctx.corner);

        let cby = state.viewctx.corner.y;
        let cbx = state.viewctx.corner.x;

        let text = state.buffer.read().unwrap();

        let mut joined = Vec::new();
        let mut sawcursor = false;
        let mut len = 0;
        let mut off = cbx;

        for (loff, s) in text.lines_at(cby, cbx).enumerate() {
            if len >= width && sawcursor {
                break;
            }

            let base = if loff == 0 { cbx } else { 0 };
            let line = cby + loff;
            let slen = s.len();

            while off < slen && (len <= width || !sawcursor) {
                let start = off;
                let end = (start + width).min(slen);
                let swrapped = s.slice(CharOff::from(start)..CharOff::from(end));

                off = end;

                let start = base + start;
                let end = base + end;
                let slen = base + slen;

                let full = end - start == width;
                let last = end == slen && cursor.x == slen;
                let cursor_line = line == cursor.y && ((start..end).contains(&cursor.x) || last);

                let wlen = swrapped.len();

                if cursor_line && full && last {
                    joined.push((line, start, end, swrapped, wlen, false));
                    joined.push((line, end, end, EditRope::from(" "), 1, true));
                    len += wlen + 1;
                } else {
                    joined.push((line, start, end, swrapped, wlen, cursor_line));
                    len += wlen;
                }

                sawcursor |= cursor_line;
            }

            if slen == 0 {
                let cursor_line = line == cursor.y;
                joined.push((line, 0, 0, s, 0, cursor_line));
                sawcursor |= cursor_line;
            }

            joined.push((line, slen, slen, EditRope::from("^J"), 2, false));
            len += 2;

            // Reset for next iteration.
            off = 0;
        }

        if !joined.is_empty() {
            // Remove the last ^J.
            joined.pop();
            len -= 2;
        }

        if len > width {
            let mut n = 0;

            for (idx, (_, ref mut start, _, ref mut s, slen, ref cursor_line)) in
                joined.iter_mut().enumerate()
            {
                if len <= width {
                    break;
                }

                let diff = len - width;
                n = idx;

                if *cursor_line {
                    let into = cursor.x - *start;
                    let rm = diff.min(into);
                    *s = s.slice(CharOff::from(rm)..);
                    *start += rm;
                    break;
                } else if *slen > diff {
                    *s = s.slice(CharOff::from(diff)..);
                    *start += diff;
                    break;
                } else {
                    len -= *slen;
                    continue;
                }
            }

            let _ = joined.drain(..n);
            let (line, start, _, _, _, _) = joined.first().unwrap();
            state.viewctx.corner.set_y(*line);
            state.viewctx.corner.set_x(*start);
        }

        state.term_cursor = (x, y);

        for (line, start, end, s, _, cursor_line) in joined.into_iter() {
            if x >= right {
                break;
            }

            let s = s.to_string();
            let w = (right - x) as usize;

            if cursor_line {
                let coff = s[..s
                    .char_indices()
                    .map(|(i, _)| i)
                    .nth(cursor.x.saturating_sub(start))
                    .unwrap_or(s.len())]
                    .width_cjk() as u16;

                state.term_cursor = (x + coff, y);
            }

            let (xres, _) = buf.set_stringn(x, y, s, w, self.style);

            self._highlight_followers(line, start, end, (x, y), &finfo, buf);
            self._highlight_line(line, start, end, (x, y), &hinfo, buf);

            x = xres;
        }
    }

    fn _render_lines_nowrap(
        &mut self,
        area: Rect,
        gutters: (Rect, Rect),
        buf: &mut Buffer,
        hinfo: HighlightInfo,
        finfo: FollowersInfo,
        state: &mut TextBoxState<I>,
    ) {
        let bot = area.bottom();
        let x = area.left();
        let mut y = area.top();

        let height = area.height as usize;
        let width = area.width as usize;

        // If the cursor has moved off-screen, update the viewport corner.
        let cursor = state.get_cursor();
        shift_corner_nowrap(&cursor, &mut state.viewctx.corner, width, height);

        let cby = state.viewctx.corner.y;
        let cbx = state.viewctx.corner.x;

        let text = state.buffer.read().unwrap();
        let mut line = cby;
        let mut lines = text.lines(line);

        while y < bot {
            if let Some(s) = lines.next() {
                let lgutter = text.get_line_info::<LeftGutterInfo>(line);
                let rgutter = text.get_line_info::<RightGutterInfo>(line);

                let slen = s.len();
                let start = cbx;
                let end = slen;

                if let Some(lgi) = lgutter {
                    let lga = Rect::new(gutters.0.x, y, gutters.0.width, 0);
                    lgi.render(lga, buf);
                }

                let s = s.slice(CharOff::from(start)..CharOff::from(end)).to_string();

                if line == cursor.y && (start..=end).contains(&cursor.x) {
                    let coff = s[..s
                        .char_indices()
                        .map(|(i, _)| i)
                        .nth(cursor.x.saturating_sub(start))
                        .unwrap_or(s.len())]
                        .width_cjk() as u16;

                    state.term_cursor = (x + coff, y);
                }

                if cbx < slen {
                    let _ = buf.set_stringn(x, y, s, width, self.style);
                }

                if let Some(rgi) = rgutter {
                    let rga = Rect::new(gutters.1.x, y, gutters.1.width, 0);
                    rgi.render(rga, buf);
                }

                self._highlight_followers(line, start, end, (x, y), &finfo, buf);
                self._highlight_line(line, start, end, (x, y), &hinfo, buf);

                y += 1;
                line += 1;
            } else {
                break;
            }
        }
    }

    #[inline]
    fn _selection_intervals(&self, state: &mut TextBoxState<I>) -> HighlightInfo {
        state.buffer.write().unwrap().selection_intervals(state.group_id)
    }

    #[inline]
    fn _follower_intervals(&self, state: &mut TextBoxState<I>) -> FollowersInfo {
        state.buffer.write().unwrap().follower_intervals(state.group_id)
    }

    fn _render_lines(&mut self, area: Rect, buf: &mut Buffer, state: &mut TextBoxState<I>) {
        let hinfo = self._selection_intervals(state);
        let finfo = self._follower_intervals(state);

        if self.oneline {
            state.set_term_info(area);
            self._render_lines_oneline(area, buf, hinfo, finfo, state);
            return;
        }

        let (lgw, rgw) = if area.width <= self.lgutter_width + self.rgutter_width {
            (0, 0)
        } else {
            (self.lgutter_width, self.rgutter_width)
        };
        let textw = area.width - lgw - rgw;
        let lga = Rect::new(area.x, area.y, lgw, area.height);
        let texta = Rect::new(area.x + lgw, area.y, textw, area.height);
        let rga = Rect::new(area.x + lgw + textw, area.y, rgw, area.height);
        let gutters = (lga, rga);

        state.set_term_info(texta);

        if state.viewctx.wrap {
            self._render_lines_wrap(texta, gutters, buf, hinfo, finfo, state);
        } else {
            self._render_lines_nowrap(texta, gutters, buf, hinfo, finfo, state);
        }
    }
}

impl<I> Default for TextBox<'_, I>
where
    I: ApplicationInfo,
{
    fn default() -> Self {
        TextBox::new()
    }
}

impl<I> StatefulWidget for TextBox<'_, I>
where
    I: ApplicationInfo,
{
    type State = TextBoxState<I>;

    fn render(mut self, area: Rect, buf: &mut Buffer, state: &mut Self::State) {
        let area = match self.block.take() {
            Some(block) => {
                let inner_area = block.inner(area);
                block.render(area, buf);
                inner_area
            },
            None => area,
        };

        let plen = self.prompt.width() as u16;
        let gutter = Rect::new(area.x, area.y, plen, area.height);

        let text_area =
            Rect::new(area.x + plen, area.y, area.width.saturating_sub(plen), area.height);

        if text_area.width == 0 || text_area.height == 0 {
            return;
        }

        // First, draw the prompt in the gutter.
        let _ = buf.set_span(gutter.left(), gutter.top(), &self.prompt, gutter.width);

        // Now draw the text.
        self._render_lines(text_area, buf, state);
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use modalkit::editing::store::Store;
    use modalkit::env::vim::VimState;

    macro_rules! mv {
        ($mt: expr) => {
            EditTarget::Motion($mt, Count::Contextual)
        };
        ($mt: expr, $c: expr) => {
            EditTarget::Motion($mt, Count::Exact($c))
        };
    }

    macro_rules! dirscroll {
        ($tbox: expr, $d: expr, $s: expr, $c: expr, $ctx: expr, $store: expr) => {
            $tbox
                .scroll(&ScrollStyle::Direction2D($d, $s, $c), $ctx, &mut $store)
                .unwrap()
        };
    }

    macro_rules! cursorpos {
        ($tbox: expr, $pos: expr, $axis: expr, $ctx: expr, $store: expr) => {
            $tbox
                .scroll(&ScrollStyle::CursorPos($pos, $axis), $ctx, &mut $store)
                .unwrap()
        };
    }

    macro_rules! linepos {
        ($tbox: expr, $pos: expr, $c: expr, $ctx: expr, $store: expr) => {
            $tbox.scroll(&ScrollStyle::LinePos($pos, $c), $ctx, &mut $store).unwrap()
        };
    }

    fn mkbox() -> (TextBoxState, Store<EmptyInfo>) {
        let mut store = Store::default();
        let buffer = store.load_buffer("".to_string());

        (TextBoxState::new(buffer), store)
    }

    fn mkboxstr(s: &str) -> (TextBoxState, EditContext, Store<EmptyInfo>) {
        let (mut b, mut store) = mkbox();
        let ctx = EditContext::from(VimState::<EmptyInfo>::default());

        b.set_text(s);
        b.editor_command(&HistoryAction::Checkpoint.into(), &ctx, &mut store)
            .unwrap();

        return (b, ctx, store);
    }

    #[test]
    fn test_scroll_dir1d() {
        let (mut tbox, ctx, mut store) = mkboxstr(
            "1234567890\n\
            abcdefghij\n\
            klmnopqrst\n\
            uvwxyz,.<>\n\
            -_=+[{]}\\|\n\
            !@#$%^&*()\n\
            1234567890\n",
        );

        tbox.set_wrap(false);
        tbox.set_term_info(Rect::new(0, 0, 6, 4));

        // Scroll by terminal cells
        dirscroll!(tbox, MoveDir2D::Down, ScrollSize::Cell, 4.into(), &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(4, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 0));

        dirscroll!(tbox, MoveDir2D::Up, ScrollSize::Cell, 2.into(), &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 0));

        dirscroll!(tbox, MoveDir2D::Right, ScrollSize::Cell, 6.into(), &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 6));
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 6));

        dirscroll!(tbox, MoveDir2D::Left, ScrollSize::Cell, 2.into(), &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 4));
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 6));

        // Scroll by half page
        dirscroll!(tbox, MoveDir2D::Down, ScrollSize::HalfPage, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(4, 4));
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 6));

        dirscroll!(tbox, MoveDir2D::Up, ScrollSize::HalfPage, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 4));
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 6));

        dirscroll!(tbox, MoveDir2D::Right, ScrollSize::HalfPage, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 7));
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 7));

        dirscroll!(tbox, MoveDir2D::Left, ScrollSize::HalfPage, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 4));
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 7));

        // Scroll by page
        dirscroll!(tbox, MoveDir2D::Down, ScrollSize::Page, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(6, 4));
        assert_eq!(tbox.get_cursor(), Cursor::new(6, 7));

        dirscroll!(tbox, MoveDir2D::Up, ScrollSize::Page, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 4));
        assert_eq!(tbox.get_cursor(), Cursor::new(5, 7));

        dirscroll!(tbox, MoveDir2D::Right, ScrollSize::Page, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 9));
        assert_eq!(tbox.get_cursor(), Cursor::new(5, 9));

        dirscroll!(tbox, MoveDir2D::Left, ScrollSize::Page, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 3));
        assert_eq!(tbox.get_cursor(), Cursor::new(5, 8));

        // Cannot scroll cursor and viewport past the end of the line.
        dirscroll!(tbox, MoveDir2D::Right, ScrollSize::Page, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 9));
        assert_eq!(tbox.get_cursor(), Cursor::new(5, 9));

        dirscroll!(tbox, MoveDir2D::Right, ScrollSize::Page, Count::Contextual, &ctx, store);
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 9));
        assert_eq!(tbox.get_cursor(), Cursor::new(5, 9));
    }

    #[test]
    fn test_scroll_cursorpos() {
        let (mut tbox, ctx, mut store) = mkboxstr(
            "1234567890\n\
            abcdefghij\n\
            klmnopqrst\n\
            uvwxyz,.<>\n\
            -_=+[{]}\\|\n\
            !@#$%^&*()\n\
            1234567890\n",
        );

        tbox.set_wrap(false);
        tbox.set_term_info(Rect::new(0, 0, 4, 4));

        // When the cursor is at the top-left corner, these actions are effectively no-ops.
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        cursorpos!(tbox, MovePosition::Beginning, Axis::Vertical, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        cursorpos!(tbox, MovePosition::Middle, Axis::Vertical, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        cursorpos!(tbox, MovePosition::End, Axis::Vertical, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        cursorpos!(tbox, MovePosition::Beginning, Axis::Horizontal, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        cursorpos!(tbox, MovePosition::Middle, Axis::Horizontal, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        cursorpos!(tbox, MovePosition::End, Axis::Horizontal, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        // Move the cursor to the second column of the fifth line, and vertically position cursor.
        let mov = mv!(MoveType::BufferLineOffset, 5);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 0));

        let mov = mv!(MoveType::LineColumnOffset, 2);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 1));

        cursorpos!(tbox, MovePosition::Beginning, Axis::Vertical, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 1));
        assert_eq!(tbox.viewctx.corner, Cursor::new(4, 0));

        cursorpos!(tbox, MovePosition::End, Axis::Vertical, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 1));
        assert_eq!(tbox.viewctx.corner, Cursor::new(1, 0));

        cursorpos!(tbox, MovePosition::Middle, Axis::Vertical, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 1));
        assert_eq!(tbox.viewctx.corner, Cursor::new(3, 0));

        // Move the cursor to the fifth column, and horizontally position cursor.
        let mov = mv!(MoveType::LineColumnOffset, 5);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 4));

        cursorpos!(tbox, MovePosition::Beginning, Axis::Horizontal, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 4));
        assert_eq!(tbox.viewctx.corner, Cursor::new(3, 4));

        cursorpos!(tbox, MovePosition::End, Axis::Horizontal, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 4));
        assert_eq!(tbox.viewctx.corner, Cursor::new(3, 1));

        cursorpos!(tbox, MovePosition::Middle, Axis::Horizontal, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 4));
        assert_eq!(tbox.viewctx.corner, Cursor::new(3, 3));

        // Vertically positioning the cursor after a FirstWord.
        let mov = MoveType::FirstWord(MoveDir1D::Next);
        let act = EditorAction::Edit(EditAction::Motion.into(), mv!(mov, 0));
        tbox.editor_command(&act, &ctx, &mut store).unwrap();
        cursorpos!(tbox, MovePosition::Beginning, Axis::Vertical, &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(4, 0));
    }

    #[test]
    fn test_scroll_linepos() {
        let (mut tbox, ctx, mut store) = mkboxstr(
            "1234567890\n\
            abcdefghij\n\
            klmnopqrst\n\
            uvwxyz,.<>\n\
            -_=+[{]}\\|\n\
            !@#$%^&*()\n\
            1234567890\n",
        );

        tbox.set_wrap(false);
        tbox.set_term_info(Rect::new(0, 0, 4, 4));

        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        // Scroll so that the 3rd line at the top of the screen.
        linepos!(tbox, MovePosition::Beginning, Count::Exact(3), &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(2, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 0));

        // Scroll so that the 7th line is in the middle of the screen.
        linepos!(tbox, MovePosition::Middle, Count::Exact(7), &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(6, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(5, 0));

        // The 1st line cannot be in the middle of the screen.
        linepos!(tbox, MovePosition::Middle, Count::Exact(1), &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        // The 1st line cannot be at the bottom of the screen.
        linepos!(tbox, MovePosition::End, Count::Exact(1), &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        // The 2nd line cannot be at the bottom of the screen.
        linepos!(tbox, MovePosition::End, Count::Exact(2), &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(1, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        // The 3nd line cannot be at the bottom of the screen.
        linepos!(tbox, MovePosition::End, Count::Exact(3), &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(2, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        // The 4th line can be at the bottom of the screen.
        linepos!(tbox, MovePosition::End, Count::Exact(4), &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(3, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));

        // The 5th line can be at the bottom of the screen.
        linepos!(tbox, MovePosition::End, Count::Exact(5), &ctx, store);
        assert_eq!(tbox.get_cursor(), Cursor::new(4, 0));
        assert_eq!(tbox.viewctx.corner, Cursor::new(1, 0));
    }

    #[test]
    fn test_reset_text() {
        let (mut tbox, ctx, mut store) = mkboxstr("foo\nbar\nbaz");

        let mov = mv!(MoveType::BufferLineOffset, 3);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();

        assert_eq!(tbox.get_text(), "foo\nbar\nbaz\n");
        assert_eq!(tbox.get_cursor(), Cursor::new(2, 0));

        assert_eq!(tbox.reset_text(), "foo\nbar\nbaz\n");
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));

        assert_eq!(tbox.get_text(), "\n");
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
    }

    #[test]
    fn test_render_nowrap() {
        let (mut tbox, ctx, mut store) = mkboxstr("foo\nbar\nbaz\nquux 1 2 3 4 5");

        tbox.set_wrap(false);

        let mut buffer = Buffer::empty(Rect::new(0, 0, 10, 10));
        let area = Rect::new(0, 8, 10, 2);

        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.get_term_cursor(), (2, 8).into());

        // Move the cursor to the fourth line, thereby moving corner.
        let mov = mv!(MoveType::BufferLineOffset, 4);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();

        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(3, 0));
        assert_eq!(tbox.get_term_cursor(), (2, 9).into());

        // Move the cursor to the end of the fourth line, again moving corner.
        let mov = mv!(MoveType::LineColumnOffset, 14);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();

        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(tbox.viewctx.corner, Cursor::new(2, 6));
        assert_eq!(tbox.get_cursor(), Cursor::new(3, 13));
        assert_eq!(tbox.get_term_cursor(), (9, 9).into());

        // Now move back to the top-left corner.
        let mov = mv!(MoveType::BufferByteOffset, 0);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();

        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.get_term_cursor(), (2, 8).into());
    }

    #[test]
    fn test_wide_char_cursor() {
        let (mut tbox, ctx, mut store) = mkboxstr("세계를 향한 대화\n");

        let area = Rect::new(0, 0, 20, 20);
        let mut buffer = Buffer::empty(area);

        // Prompt should push everything right by 2 characters.
        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.get_term_cursor(), (2, 0).into());

        // Move the cursor to be over "대", just before the last character.
        let mov = mv!(MoveType::Column(MoveDir1D::Next, false), 7);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();

        // Draw again to update our terminal cursor using oneline().
        TextBox::new().prompt("> ").oneline().render(area, &mut buffer, &mut tbox);

        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 7));
        assert_eq!(tbox.get_term_cursor(), (14, 0).into());
        // Draw again to update our terminal cursor using set_wrap(true).
        tbox.set_wrap(true);
        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 7));
        assert_eq!(tbox.get_term_cursor(), (14, 0).into());

        // Draw again to update our terminal cursor using set_wrap(false).
        tbox.set_wrap(true);
        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 7));
        assert_eq!(tbox.get_term_cursor(), (14, 0).into());
    }

    #[test]
    fn test_wide_char_wrap() {
        let (mut tbox, ctx, mut store) = mkboxstr("세계를 향한대화\n");

        let area = Rect::new(0, 0, 10, 10);
        let mut buffer = Buffer::empty(area);

        let mut expected = buffer.clone();
        expected.set_string(0, 0, "> 세계를 ", Style::new());
        expected.set_string(0, 1, "  향한대화", Style::new());

        // Prompt should push everything right by 2 characters.
        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(buffer, expected);
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 0));
        assert_eq!(tbox.get_term_cursor(), (2, 0).into());
        assert_eq!(tbox.has_lines(5), 3);

        // Move the cursor to be on the " " on the end of the first wrapped line.
        let mov = mv!(MoveType::Column(MoveDir1D::Next, false), 3);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();

        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(buffer, expected);
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 3));
        assert_eq!(tbox.get_term_cursor(), (8, 0).into());

        // Move the cursor right to be over "향".
        let mov = mv!(MoveType::Column(MoveDir1D::Next, false), 1);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();

        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(buffer, expected);
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 4));
        assert_eq!(tbox.get_term_cursor(), (2, 1).into());

        // Move the cursor right to be at the end.
        let mov = mv!(MoveType::Column(MoveDir1D::Next, false), 3);
        let act = EditorAction::Edit(EditAction::Motion.into(), mov);
        tbox.editor_command(&act, &ctx, &mut store).unwrap();

        TextBox::new().prompt("> ").render(area, &mut buffer, &mut tbox);

        assert_eq!(buffer, expected);
        assert_eq!(tbox.viewctx.corner, Cursor::new(0, 0));
        assert_eq!(tbox.get_cursor(), Cursor::new(0, 7));
        assert_eq!(tbox.get_term_cursor(), (8, 1).into());
    }
}