vorto 0.13.0-beta.2

A Vim-flavored modal terminal editor with batteries included: tree-sitter, LSP, fuzzy pickers, vim-surround, multi-cursor, and optional Copilot.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136

//! vim-surround dispatch: `ys{target}{ch}` / `cs{from}{to}` / `ds{ch}`.
//!
//! All three commands collapse to a small set of buffer primitives:
//! resolve a range (via motion or text object), then either wrap it,
//! replace the wrapping pair, or strip it.

use super::resolve_motion_pure;
use crate::action::{Object, Scope, Target};
use crate::app::App;
use crate::app::eval::is_inclusive_motion;
use crate::editor::Cursor;
use crate::effect::Cmd;

pub(super) fn handle_add(app: &mut App, target: Target, ch: char) -> Vec<Cmd> {
    let mut cmds = Vec::new();
    let Some((open, close)) = pair_for(ch) else {
        cmds.push(Cmd::ToastError(format!("no surround pair for `{}`", ch)));
        return cmds;
    };

    let range = match target {
        Target::Motion(m) => {
            let (resolved, last_find_update) = resolve_motion_pure(m.motion, app.last_find);
            if let Some(lf) = last_find_update {
                cmds.push(Cmd::SetLastFind(lf));
            }
            let Some(resolved) = resolved else {
                cmds.push(Cmd::ToastError("no previous find".into()));
                return cmds;
            };
            let start = app.editor.cursor;
            let doc = app.active_doc();
            let end_raw = doc.motion_target(start, resolved, m.count);
            let end = if is_inclusive_motion(resolved) {
                doc.advance_one(end_raw)
            } else {
                end_raw
            };
            Some(order(start, end))
        }
        Target::TextObject { scope, object } => ed_op_ref!(app, text_object_range(scope, object)),
        Target::LineWise => {
            let row = app.editor.cursor.row;
            let line_len = app.active_doc().lines[row].chars().count();
            Some((Cursor { row, col: 0 }, Cursor { row, col: line_len }))
        }
        Target::SearchMatch { .. } => {
            cmds.push(Cmd::ToastError(
                "surround over search match not supported".into(),
            ));
            return cmds;
        }
    };

    let Some((lo, hi)) = range else {
        cmds.push(Cmd::ToastError("no matching object".into()));
        return cmds;
    };
    ed_op!(app, surround_wrap(&open, &close, lo, hi));
    cmds
}

pub(super) fn handle_delete(app: &mut App, ch: char) -> Vec<Cmd> {
    let mut cmds = Vec::new();
    let Some(object) = object_for(ch) else {
        cmds.push(Cmd::ToastError(format!("no surround pair for `{}`", ch)));
        return cmds;
    };
    let Some((lo, hi)) = ed_op_ref!(app, text_object_range(Scope::Around, object)) else {
        cmds.push(Cmd::ToastError("no surrounding pair".into()));
        return cmds;
    };
    ed_op!(app, surround_strip(lo, hi));
    cmds
}

pub(super) fn handle_change(app: &mut App, from: char, to: char) -> Vec<Cmd> {
    let mut cmds = Vec::new();
    let Some(object) = object_for(from) else {
        cmds.push(Cmd::ToastError(format!("no surround pair for `{}`", from)));
        return cmds;
    };
    let Some((new_open, new_close)) = pair_for(to) else {
        cmds.push(Cmd::ToastError(format!("no surround pair for `{}`", to)));
        return cmds;
    };
    let Some((lo, hi)) = ed_op_ref!(app, text_object_range(Scope::Around, object)) else {
        cmds.push(Cmd::ToastError("no surrounding pair".into()));
        return cmds;
    };
    ed_op!(app, surround_replace(lo, hi, &new_open, &new_close));
    cmds
}

/// Map a vim-surround char to the `(open, close)` literal pair to insert.
///
/// Unlike tpope's vim-surround we do *not* distinguish "spaced" vs
/// "tight" by which side of the bracket pair was typed: every variant
/// wraps tightly. Inserting interior spaces is a formatter's job, not
/// surround's. `b` / `B` remain aliases for `)` / `}` for muscle-memory.
fn pair_for(ch: char) -> Option<(String, String)> {
    Some(match ch {
        '(' | ')' | 'b' => ("(".into(), ")".into()),
        '{' | '}' | 'B' => ("{".into(), "}".into()),
        '[' | ']' => ("[".into(), "]".into()),
        '<' | '>' => ("<".into(), ">".into()),
        '"' => ("\"".into(), "\"".into()),
        '\'' => ("'".into(), "'".into()),
        '`' => ("`".into(), "`".into()),
        _ => return None,
    })
}

/// Map a vim-surround char to the text-object kind that locates the
/// existing pair around the cursor. Both opening and closing variants
/// of asymmetric brackets resolve to the same object.
fn object_for(ch: char) -> Option<Object> {
    Some(match ch {
        '(' | ')' | 'b' => Object::Paren,
        '{' | '}' | 'B' => Object::Brace,
        '[' | ']' => Object::Bracket,
        '<' | '>' => Object::AngleBracket,
        '"' => Object::DoubleQuote,
        '\'' => Object::SingleQuote,
        '`' => Object::Backtick,
        _ => return None,
    })
}

fn order(a: Cursor, b: Cursor) -> (Cursor, Cursor) {
    if (a.row, a.col) <= (b.row, b.col) {
        (a, b)
    } else {
        (b, a)
    }
}