1353 lines
44 KiB
C++
1353 lines
44 KiB
C++
#include "input_handler.hh"
|
|
|
|
#include "window.hh"
|
|
#include "utf8.hh"
|
|
#include "user_interface.hh"
|
|
#include "buffer_manager.hh"
|
|
#include "register_manager.hh"
|
|
#include "normal.hh"
|
|
#include "event_manager.hh"
|
|
#include "client.hh"
|
|
#include "color_registry.hh"
|
|
#include "file.hh"
|
|
#include "word_db.hh"
|
|
#include "debug.hh"
|
|
|
|
#include <unordered_map>
|
|
|
|
namespace Kakoune
|
|
{
|
|
|
|
class InputMode
|
|
{
|
|
public:
|
|
InputMode(InputHandler& input_handler) : m_input_handler(input_handler) {}
|
|
virtual ~InputMode() {}
|
|
InputMode(const InputMode&) = delete;
|
|
InputMode& operator=(const InputMode&) = delete;
|
|
|
|
virtual void on_key(Key key) = 0;
|
|
virtual void on_enabled() {}
|
|
virtual void on_disabled() {}
|
|
Context& context() const { return m_input_handler.context(); }
|
|
|
|
virtual String description() const = 0;
|
|
|
|
virtual KeymapMode keymap_mode() const = 0;
|
|
|
|
using Insertion = InputHandler::Insertion;
|
|
Insertion& last_insert() { return m_input_handler.m_last_insert; }
|
|
|
|
protected:
|
|
void reset_normal_mode();
|
|
private:
|
|
InputHandler& m_input_handler;
|
|
};
|
|
|
|
namespace InputModes
|
|
{
|
|
|
|
static constexpr std::chrono::milliseconds idle_timeout{100};
|
|
static constexpr std::chrono::milliseconds fs_check_timeout{500};
|
|
|
|
class Normal : public InputMode
|
|
{
|
|
public:
|
|
Normal(InputHandler& input_handler)
|
|
: InputMode(input_handler),
|
|
m_idle_timer{Clock::now() + idle_timeout, [this](Timer& timer) {
|
|
context().hooks().run_hook("NormalIdle", "", context());
|
|
}},
|
|
m_fs_check_timer{Clock::now() + fs_check_timeout, [this](Timer& timer) {
|
|
if (not context().has_client())
|
|
return;
|
|
context().client().check_buffer_fs_timestamp();
|
|
timer.set_next_date(Clock::now() + fs_check_timeout);
|
|
}}
|
|
{}
|
|
|
|
void on_enabled() override
|
|
{
|
|
if (not context().has_client())
|
|
return;
|
|
context().client().check_buffer_fs_timestamp();
|
|
m_fs_check_timer.set_next_date(Clock::now() + fs_check_timeout);
|
|
|
|
context().hooks().run_hook("NormalBegin", "", context());
|
|
}
|
|
|
|
void on_disabled() override
|
|
{
|
|
context().hooks().run_hook("NormalEnd", "", context());
|
|
}
|
|
|
|
void on_key(Key key) override
|
|
{
|
|
if (key.modifiers == Key::Modifiers::None and isdigit(key.key))
|
|
m_count = m_count * 10 + key.key - '0';
|
|
else
|
|
{
|
|
auto it = keymap.find(key);
|
|
if (it != keymap.end())
|
|
it->second(context(), m_count);
|
|
m_count = 0;
|
|
}
|
|
context().hooks().run_hook("NormalKey", key_to_str(key), context());
|
|
m_idle_timer.set_next_date(Clock::now() + idle_timeout);
|
|
}
|
|
|
|
String description() const override
|
|
{
|
|
return to_string(context().selections().size()) +
|
|
(m_count != 0 ? " sel; param=" + to_string(m_count) : " sel");
|
|
}
|
|
|
|
KeymapMode keymap_mode() const override { return KeymapMode::Normal; }
|
|
|
|
private:
|
|
int m_count = 0;
|
|
Timer m_idle_timer;
|
|
Timer m_fs_check_timer;
|
|
};
|
|
|
|
class LineEditor
|
|
{
|
|
public:
|
|
void handle_key(Key key)
|
|
{
|
|
if (key == Key::Left or key == ctrl('b'))
|
|
{
|
|
if (m_cursor_pos > 0)
|
|
--m_cursor_pos;
|
|
}
|
|
else if (key == Key::Right or key == ctrl('f'))
|
|
{
|
|
if (m_cursor_pos < m_line.char_length())
|
|
++m_cursor_pos;
|
|
}
|
|
else if (key == Key::Home)
|
|
m_cursor_pos = 0;
|
|
else if (key == Key::End)
|
|
m_cursor_pos = m_line.char_length();
|
|
else if (key == Key::Backspace)
|
|
{
|
|
if (m_cursor_pos != 0)
|
|
{
|
|
m_line = m_line.substr(0, m_cursor_pos - 1)
|
|
+ m_line.substr(m_cursor_pos);
|
|
|
|
--m_cursor_pos;
|
|
}
|
|
}
|
|
else if (key == Key::Erase)
|
|
{
|
|
if (m_cursor_pos != m_line.char_length())
|
|
m_line = m_line.substr(0, m_cursor_pos)
|
|
+ m_line.substr(m_cursor_pos+1);
|
|
}
|
|
else
|
|
{
|
|
m_line = m_line.substr(0, m_cursor_pos) + codepoint_to_str(key.key)
|
|
+ m_line.substr(m_cursor_pos);
|
|
++m_cursor_pos;
|
|
}
|
|
}
|
|
|
|
void insert(const String& str)
|
|
{
|
|
insert_from(m_cursor_pos, str);
|
|
}
|
|
|
|
void insert_from(CharCount start, const String& str)
|
|
{
|
|
kak_assert(start <= m_cursor_pos);
|
|
m_line = m_line.substr(0, start) + str
|
|
+ m_line.substr(m_cursor_pos);
|
|
m_cursor_pos = start + str.char_length();
|
|
}
|
|
|
|
void reset(String line)
|
|
{
|
|
m_line = std::move(line);
|
|
m_cursor_pos = m_line.char_length();
|
|
}
|
|
|
|
const String& line() const { return m_line; }
|
|
CharCount cursor_pos() const { return m_cursor_pos; }
|
|
|
|
DisplayLine build_display_line(CharCount width)
|
|
{
|
|
kak_assert(m_cursor_pos <= m_line.char_length());
|
|
if (m_cursor_pos < m_display_pos)
|
|
m_display_pos = m_cursor_pos;
|
|
if (m_cursor_pos >= m_display_pos + width)
|
|
m_display_pos = m_cursor_pos + 1 - width;
|
|
|
|
if (m_cursor_pos == m_line.char_length())
|
|
return DisplayLine{{ {m_line.substr(m_display_pos, width-1), get_color("StatusLine")},
|
|
{" "_str, get_color("StatusCursor")} }};
|
|
else
|
|
return DisplayLine({ { m_line.substr(m_display_pos, m_cursor_pos - m_display_pos), get_color("StatusLine") },
|
|
{ m_line.substr(m_cursor_pos,1), get_color("StatusCursor") },
|
|
{ m_line.substr(m_cursor_pos+1, width - m_cursor_pos + m_display_pos - 1), get_color("StatusLine") } });
|
|
}
|
|
private:
|
|
CharCount m_cursor_pos = 0;
|
|
CharCount m_display_pos = 0;
|
|
String m_line;
|
|
};
|
|
|
|
class Menu : public InputMode
|
|
{
|
|
public:
|
|
Menu(InputHandler& input_handler, memoryview<String> choices,
|
|
MenuCallback callback)
|
|
: InputMode(input_handler),
|
|
m_callback(callback), m_choices(choices.begin(), choices.end()),
|
|
m_selected(m_choices.begin())
|
|
{
|
|
if (not context().has_ui())
|
|
return;
|
|
DisplayCoord menu_pos{ context().ui().dimensions().line, 0_char };
|
|
context().ui().menu_show(choices, menu_pos, get_color("MenuForeground"),
|
|
get_color("MenuBackground"), MenuStyle::Prompt);
|
|
context().ui().menu_select(0);
|
|
}
|
|
|
|
void on_key(Key key) override
|
|
{
|
|
auto match_filter = [this](const String& str) {
|
|
return boost::regex_match(str.begin(), str.end(), m_filter);
|
|
};
|
|
|
|
if (key == ctrl('m'))
|
|
{
|
|
if (context().has_ui())
|
|
context().ui().menu_hide();
|
|
context().print_status(DisplayLine{});
|
|
reset_normal_mode();
|
|
int selected = m_selected - m_choices.begin();
|
|
m_callback(selected, MenuEvent::Validate, context());
|
|
return;
|
|
}
|
|
else if (key == Key::Escape or key == ctrl('c'))
|
|
{
|
|
if (m_edit_filter)
|
|
{
|
|
m_edit_filter = false;
|
|
m_filter = boost::regex(".*");
|
|
m_filter_editor.reset("");
|
|
context().print_status(DisplayLine{});
|
|
}
|
|
else
|
|
{
|
|
if (context().has_ui())
|
|
context().ui().menu_hide();
|
|
reset_normal_mode();
|
|
int selected = m_selected - m_choices.begin();
|
|
m_callback(selected, MenuEvent::Abort, context());
|
|
}
|
|
}
|
|
else if (key == Key::Down or key == ctrl('i') or
|
|
key == ctrl('n') or key == 'j')
|
|
{
|
|
auto it = std::find_if(m_selected+1, m_choices.end(), match_filter);
|
|
if (it == m_choices.end())
|
|
it = std::find_if(m_choices.begin(), m_selected, match_filter);
|
|
select(it);
|
|
}
|
|
else if (key == Key::Up or key == Key::BackTab or
|
|
key == ctrl('p') or key == 'k')
|
|
{
|
|
ChoiceList::const_reverse_iterator selected(m_selected+1);
|
|
auto it = std::find_if(selected+1, m_choices.rend(), match_filter);
|
|
if (it == m_choices.rend())
|
|
it = std::find_if(m_choices.rbegin(), selected, match_filter);
|
|
select(it.base()-1);
|
|
}
|
|
else if (key == '/' and not m_edit_filter)
|
|
{
|
|
m_edit_filter = true;
|
|
}
|
|
else if (m_edit_filter)
|
|
{
|
|
m_filter_editor.handle_key(key);
|
|
|
|
auto search = ".*" + m_filter_editor.line() + ".*";
|
|
m_filter = boost::regex(search.begin(), search.end());
|
|
auto it = std::find_if(m_selected, m_choices.end(), match_filter);
|
|
if (it == m_choices.end())
|
|
it = std::find_if(m_choices.begin(), m_selected, match_filter);
|
|
select(it);
|
|
}
|
|
|
|
if (m_edit_filter and context().has_ui())
|
|
{
|
|
auto prompt = "filter:"_str;
|
|
auto width = context().ui().dimensions().column - prompt.char_length();
|
|
auto display_line = m_filter_editor.build_display_line(width);
|
|
display_line.insert(display_line.begin(), { prompt, get_color("Prompt") });
|
|
context().print_status(display_line);
|
|
}
|
|
}
|
|
|
|
String description() const override
|
|
{
|
|
return "menu";
|
|
}
|
|
|
|
KeymapMode keymap_mode() const override { return KeymapMode::Menu; }
|
|
|
|
private:
|
|
MenuCallback m_callback;
|
|
|
|
using ChoiceList = std::vector<String>;
|
|
const ChoiceList m_choices;
|
|
ChoiceList::const_iterator m_selected;
|
|
|
|
void select(ChoiceList::const_iterator it)
|
|
{
|
|
m_selected = it;
|
|
int selected = m_selected - m_choices.begin();
|
|
if (context().has_ui())
|
|
context().ui().menu_select(selected);
|
|
m_callback(selected, MenuEvent::Select, context());
|
|
}
|
|
|
|
boost::regex m_filter = boost::regex(".*");
|
|
bool m_edit_filter = false;
|
|
LineEditor m_filter_editor;
|
|
};
|
|
|
|
String common_prefix(memoryview<String> strings)
|
|
{
|
|
String res;
|
|
if (strings.empty())
|
|
return res;
|
|
res = strings[0];
|
|
for (auto& str : strings)
|
|
{
|
|
ByteCount len = std::min(res.length(), str.length());
|
|
ByteCount common_len = 0;
|
|
while (common_len < len and str[common_len] == res[common_len])
|
|
++common_len;
|
|
if (common_len != res.length())
|
|
res = res.substr(0, common_len);
|
|
}
|
|
return res;
|
|
}
|
|
|
|
void history_push(std::vector<String>& history, const String& entry)
|
|
{
|
|
if(entry.empty())
|
|
{
|
|
return;
|
|
}
|
|
std::vector<String>::iterator it;
|
|
while ((it = find(history, entry)) != history.end())
|
|
history.erase(it);
|
|
history.push_back(entry);
|
|
}
|
|
|
|
class Prompt : public InputMode
|
|
{
|
|
public:
|
|
Prompt(InputHandler& input_handler, const String& prompt,
|
|
String initstr, ColorPair colors, Completer completer,
|
|
PromptCallback callback)
|
|
: InputMode(input_handler), m_prompt(prompt), m_prompt_colors(colors),
|
|
m_completer(completer), m_callback(callback)
|
|
{
|
|
m_history_it = ms_history[m_prompt].end();
|
|
if (context().options()["autoshowcompl"].get<bool>())
|
|
refresh_completions(CompletionFlags::Fast);
|
|
m_line_editor.reset(std::move(initstr));
|
|
display();
|
|
}
|
|
|
|
void on_key(Key key) override
|
|
{
|
|
std::vector<String>& history = ms_history[m_prompt];
|
|
const String& line = m_line_editor.line();
|
|
bool showcompl = false;
|
|
|
|
if (m_mode == Mode::InsertReg)
|
|
{
|
|
String reg = RegisterManager::instance()[key.key].values(context())[0];
|
|
m_line_editor.insert(reg);
|
|
m_mode = Mode::Default;
|
|
}
|
|
else if (key == ctrl('m')) // enter
|
|
{
|
|
history_push(history, line);
|
|
context().print_status(DisplayLine{});
|
|
if (context().has_ui())
|
|
context().ui().menu_hide();
|
|
reset_normal_mode();
|
|
// call callback after reset_normal_mode so that callback
|
|
// may change the mode
|
|
m_callback(line, PromptEvent::Validate, context());
|
|
return;
|
|
}
|
|
else if (key == Key::Escape or key == ctrl('c'))
|
|
{
|
|
history_push(history, line);
|
|
context().print_status(DisplayLine{});
|
|
if (context().has_ui())
|
|
context().ui().menu_hide();
|
|
reset_normal_mode();
|
|
m_callback(line, PromptEvent::Abort, context());
|
|
return;
|
|
}
|
|
else if (key == ctrl('r'))
|
|
{
|
|
m_mode = Mode::InsertReg;
|
|
}
|
|
else if (key == Key::Up or key == ctrl('p'))
|
|
{
|
|
if (m_history_it != history.begin())
|
|
{
|
|
if (m_history_it == history.end())
|
|
m_prefix = line;
|
|
auto it = m_history_it;
|
|
// search for the previous history entry matching typed prefix
|
|
do
|
|
{
|
|
--it;
|
|
if (prefix_match(*it, m_prefix))
|
|
{
|
|
m_history_it = it;
|
|
m_line_editor.reset(*it);
|
|
break;
|
|
}
|
|
} while (it != history.begin());
|
|
|
|
clear_completions();
|
|
showcompl = true;
|
|
}
|
|
}
|
|
else if (key == Key::Down or key == ctrl('n')) // next
|
|
{
|
|
if (m_history_it != history.end())
|
|
{
|
|
// search for the next history entry matching typed prefix
|
|
++m_history_it;
|
|
while (m_history_it != history.end() and
|
|
prefix_match(*m_history_it, m_prefix))
|
|
++m_history_it;
|
|
|
|
if (m_history_it != history.end())
|
|
m_line_editor.reset(*m_history_it);
|
|
else
|
|
m_line_editor.reset(m_prefix);
|
|
|
|
clear_completions();
|
|
showcompl = true;
|
|
}
|
|
}
|
|
else if (key == ctrl('i') or key == Key::BackTab) // tab completion
|
|
{
|
|
const bool reverse = (key == Key::BackTab);
|
|
CandidateList& candidates = m_completions.candidates;
|
|
// first try, we need to ask our completer for completions
|
|
if (candidates.empty())
|
|
{
|
|
refresh_completions(CompletionFlags::None);
|
|
|
|
if (candidates.empty())
|
|
return;
|
|
}
|
|
bool did_prefix = false;
|
|
if (m_current_completion == -1 and
|
|
context().options()["complete_prefix"].get<bool>())
|
|
{
|
|
const String& line = m_line_editor.line();
|
|
CandidateList& candidates = m_completions.candidates;
|
|
String prefix = common_prefix(candidates);
|
|
if (m_completions.end - m_completions.start > prefix.length())
|
|
prefix = line.substr(m_completions.start,
|
|
m_completions.end - m_completions.start);
|
|
|
|
if (not prefix.empty())
|
|
{
|
|
auto it = find(candidates, prefix);
|
|
if (it == candidates.end())
|
|
{
|
|
m_current_completion = candidates.size();
|
|
candidates.push_back(prefix);
|
|
}
|
|
else
|
|
m_current_completion = it - candidates.begin();
|
|
|
|
CharCount start = line.char_count_to(m_completions.start);
|
|
did_prefix = prefix != line.substr(start, m_line_editor.cursor_pos() - start);
|
|
}
|
|
}
|
|
if (not did_prefix)
|
|
{
|
|
if (not reverse and ++m_current_completion >= candidates.size())
|
|
m_current_completion = 0;
|
|
else if (reverse and --m_current_completion < 0)
|
|
m_current_completion = candidates.size()-1;
|
|
}
|
|
|
|
const String& completion = candidates[m_current_completion];
|
|
if (context().has_ui())
|
|
context().ui().menu_select(m_current_completion);
|
|
|
|
m_line_editor.insert_from(line.char_count_to(m_completions.start),
|
|
completion);
|
|
|
|
// when we have only one completion candidate, make next tab complete
|
|
// from the new content.
|
|
if (candidates.size() == 1)
|
|
{
|
|
m_current_completion = -1;
|
|
candidates.clear();
|
|
showcompl = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
m_line_editor.handle_key(key);
|
|
clear_completions();
|
|
showcompl = true;
|
|
}
|
|
|
|
if (showcompl and context().options()["autoshowcompl"].get<bool>())
|
|
refresh_completions(CompletionFlags::Fast);
|
|
|
|
display();
|
|
m_callback(line, PromptEvent::Change, context());
|
|
}
|
|
|
|
void set_prompt_colors(ColorPair colors)
|
|
{
|
|
if (colors != m_prompt_colors)
|
|
{
|
|
m_prompt_colors = colors;
|
|
display();
|
|
}
|
|
}
|
|
|
|
String description() const override
|
|
{
|
|
return "prompt";
|
|
}
|
|
|
|
KeymapMode keymap_mode() const override { return KeymapMode::Prompt; }
|
|
|
|
private:
|
|
void refresh_completions(CompletionFlags flags)
|
|
{
|
|
try
|
|
{
|
|
if (not m_completer)
|
|
return;
|
|
const String& line = m_line_editor.line();
|
|
m_completions = m_completer(context(), flags, line,
|
|
line.byte_count_to(m_line_editor.cursor_pos()));
|
|
CandidateList& candidates = m_completions.candidates;
|
|
if (context().has_ui() and not candidates.empty())
|
|
{
|
|
DisplayCoord menu_pos{ context().ui().dimensions().line, 0_char };
|
|
context().ui().menu_show(candidates, menu_pos, get_color("MenuForeground"),
|
|
get_color("MenuBackground"), MenuStyle::Prompt);
|
|
}
|
|
} catch (runtime_error&) {}
|
|
}
|
|
|
|
void clear_completions()
|
|
{
|
|
m_current_completion = -1;
|
|
if (context().has_ui())
|
|
context().ui().menu_hide();
|
|
}
|
|
|
|
void display()
|
|
{
|
|
if (not context().has_ui())
|
|
return;
|
|
|
|
auto width = context().ui().dimensions().column - m_prompt.char_length();
|
|
auto display_line = m_line_editor.build_display_line(width);
|
|
display_line.insert(display_line.begin(), { m_prompt, m_prompt_colors });
|
|
context().print_status(display_line);
|
|
}
|
|
|
|
enum class Mode { Default, InsertReg };
|
|
|
|
PromptCallback m_callback;
|
|
Completer m_completer;
|
|
const String m_prompt;
|
|
ColorPair m_prompt_colors;
|
|
Completions m_completions;
|
|
int m_current_completion = -1;
|
|
String m_prefix;
|
|
LineEditor m_line_editor;
|
|
Mode m_mode = Mode::Default;
|
|
|
|
static std::unordered_map<String, std::vector<String>> ms_history;
|
|
std::vector<String>::iterator m_history_it;
|
|
};
|
|
std::unordered_map<String, std::vector<String>> Prompt::ms_history;
|
|
|
|
class NextKey : public InputMode
|
|
{
|
|
public:
|
|
NextKey(InputHandler& input_handler, KeyCallback callback)
|
|
: InputMode(input_handler), m_callback(callback) {}
|
|
|
|
void on_key(Key key) override
|
|
{
|
|
reset_normal_mode();
|
|
m_callback(key, context());
|
|
}
|
|
|
|
String description() const override
|
|
{
|
|
return "enter key";
|
|
}
|
|
|
|
KeymapMode keymap_mode() const override { return KeymapMode::None; }
|
|
|
|
private:
|
|
KeyCallback m_callback;
|
|
};
|
|
|
|
struct InsertCompletion
|
|
{
|
|
BufferCoord begin;
|
|
BufferCoord end;
|
|
CandidateList candidates;
|
|
size_t timestamp;
|
|
|
|
bool is_valid() const { return not candidates.empty(); }
|
|
};
|
|
|
|
class InsertCompleter : public OptionManagerWatcher_AutoRegister
|
|
{
|
|
public:
|
|
InsertCompleter(const Context& context)
|
|
: OptionManagerWatcher_AutoRegister(context.options()), m_context(context)
|
|
{}
|
|
InsertCompleter(const InsertCompleter&) = delete;
|
|
InsertCompleter& operator=(const InsertCompleter&) = delete;
|
|
|
|
void select(int offset)
|
|
{
|
|
if (not setup_ifn())
|
|
return;
|
|
|
|
auto& buffer = m_context.buffer();
|
|
m_current_candidate = (m_current_candidate + offset) % (int)m_matching_candidates.size();
|
|
if (m_current_candidate < 0)
|
|
m_current_candidate += m_matching_candidates.size();
|
|
const String& candidate = m_matching_candidates[m_current_candidate];
|
|
const auto& cursor_pos = m_context.selections().main().cursor();
|
|
const auto prefix_len = buffer.distance(m_completions.begin, cursor_pos);
|
|
const auto suffix_len = std::max(0_byte, buffer.distance(cursor_pos, m_completions.end));
|
|
const auto buffer_len = buffer.byte_count();
|
|
|
|
auto ref = buffer.string(m_completions.begin, m_completions.end);
|
|
for (auto& sel : m_context.selections())
|
|
{
|
|
auto offset = buffer.offset(sel.cursor());
|
|
auto pos = buffer.iterator_at(sel.cursor());
|
|
if (offset >= prefix_len and offset + suffix_len < buffer_len and
|
|
std::equal(ref.begin(), ref.end(), pos - prefix_len))
|
|
{
|
|
pos = buffer.erase(pos - prefix_len, pos + suffix_len);
|
|
buffer.insert(pos, candidate);
|
|
}
|
|
}
|
|
m_completions.end = cursor_pos;
|
|
m_completions.begin = buffer.advance(m_completions.end, -candidate.length());
|
|
m_completions.timestamp = buffer.timestamp();
|
|
if (m_context.has_ui())
|
|
m_context.ui().menu_select(m_current_candidate);
|
|
|
|
// when we select a match, remove non displayed matches from the candidates
|
|
// which are considered as invalid with the new completion timestamp
|
|
m_completions.candidates.clear();
|
|
std::copy(m_matching_candidates.begin(), m_matching_candidates.end()-1,
|
|
std::back_inserter(m_completions.candidates));
|
|
}
|
|
|
|
void update()
|
|
{
|
|
if (m_completions.is_valid())
|
|
{
|
|
ByteCount longest_completion = 0;
|
|
for (auto& candidate : m_completions.candidates)
|
|
longest_completion = std::max(longest_completion, candidate.length());
|
|
|
|
BufferCoord cursor = m_context.selections().main().cursor();
|
|
BufferCoord compl_beg = m_completions.begin;
|
|
if (cursor.line == compl_beg.line and
|
|
is_in_range(cursor.column - compl_beg.column,
|
|
ByteCount{0}, longest_completion-1))
|
|
{
|
|
String prefix = m_context.buffer().string(compl_beg, cursor);
|
|
|
|
if (m_context.buffer().timestamp() == m_completions.timestamp)
|
|
m_matching_candidates = m_completions.candidates;
|
|
else
|
|
{
|
|
m_matching_candidates.clear();
|
|
for (auto& candidate : m_completions.candidates)
|
|
{
|
|
if (candidate.substr(0, prefix.length()) == prefix)
|
|
m_matching_candidates.push_back(candidate);
|
|
}
|
|
}
|
|
if (not m_matching_candidates.empty())
|
|
{
|
|
m_current_candidate = m_matching_candidates.size();
|
|
m_completions.end = cursor;
|
|
menu_show();
|
|
m_matching_candidates.push_back(prefix);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
reset();
|
|
setup_ifn();
|
|
}
|
|
|
|
void reset()
|
|
{
|
|
m_completions = InsertCompletion{};
|
|
if (m_context.has_ui())
|
|
m_context.ui().menu_hide();
|
|
}
|
|
|
|
template<typename CompleteFunc>
|
|
bool try_complete(CompleteFunc complete_func)
|
|
{
|
|
auto& buffer = m_context.buffer();
|
|
BufferCoord cursor_pos = m_context.selections().main().cursor();
|
|
try
|
|
{
|
|
m_completions = complete_func(buffer, cursor_pos);
|
|
}
|
|
catch (runtime_error& e)
|
|
{
|
|
write_debug("error while trying to run completer: "_str + e.what());
|
|
return false;
|
|
}
|
|
if (not m_completions.is_valid())
|
|
return false;
|
|
|
|
kak_assert(cursor_pos >= m_completions.begin);
|
|
m_matching_candidates = m_completions.candidates;
|
|
m_current_candidate = m_matching_candidates.size();
|
|
menu_show();
|
|
m_matching_candidates.push_back(buffer.string(m_completions.begin, m_completions.end));
|
|
return true;
|
|
}
|
|
using StringList = std::vector<String>;
|
|
|
|
static WordDB& get_word_db(const Buffer& buffer)
|
|
{
|
|
static const ValueId word_db_id = ValueId::get_free_id();
|
|
Value& cache_val = buffer.values()[word_db_id];
|
|
if (not cache_val)
|
|
cache_val = Value(WordDB{buffer});
|
|
return cache_val.as<WordDB>();
|
|
}
|
|
|
|
template<bool other_buffers>
|
|
InsertCompletion complete_word(const Buffer& buffer, BufferCoord cursor_pos)
|
|
{
|
|
auto pos = buffer.iterator_at(cursor_pos);
|
|
if (pos == buffer.begin() or not is_word(*utf8::previous(pos)))
|
|
return {};
|
|
|
|
auto end = buffer.iterator_at(cursor_pos);
|
|
auto begin = end-1;
|
|
while (begin != buffer.begin() and is_word(*begin))
|
|
--begin;
|
|
if (not is_word(*begin))
|
|
++begin;
|
|
|
|
String prefix{begin, end};
|
|
|
|
while (end != buffer.end() and is_word(*end))
|
|
++end;
|
|
|
|
String current_word{begin, end};
|
|
|
|
auto& word_db = get_word_db(buffer);
|
|
std::unordered_set<String> matches;
|
|
auto bufmatches = word_db.find_prefix(prefix);
|
|
matches.insert(bufmatches.begin(), bufmatches.end());
|
|
|
|
if (word_db.get_word_occurences(current_word) <= 1)
|
|
matches.erase(current_word);
|
|
|
|
if (other_buffers)
|
|
{
|
|
for (const auto& buf : BufferManager::instance())
|
|
{
|
|
if (buf.get() == &buffer)
|
|
continue;
|
|
bufmatches = get_word_db(*buf).find_prefix(prefix);
|
|
matches.insert(bufmatches.begin(), bufmatches.end());
|
|
}
|
|
}
|
|
matches.erase(prefix);
|
|
CandidateList result;
|
|
std::copy(matches.begin(), matches.end(),
|
|
inserter(result, result.begin()));
|
|
std::sort(result.begin(), result.end());
|
|
return { begin.coord(), end.coord(), std::move(result), buffer.timestamp() };
|
|
}
|
|
|
|
template<bool require_slash>
|
|
InsertCompletion complete_filename(const Buffer& buffer, BufferCoord cursor_pos)
|
|
{
|
|
auto pos = buffer.iterator_at(cursor_pos);
|
|
auto begin = pos;
|
|
|
|
auto is_filename = [](char c)
|
|
{
|
|
return isalnum(c) or c == '/' or c == '.' or c == '_' or c == '-';
|
|
};
|
|
while (begin != buffer.begin() and is_filename(*(begin-1)))
|
|
--begin;
|
|
|
|
if (begin == pos)
|
|
return {};
|
|
|
|
String prefix{begin, pos};
|
|
if (require_slash and not contains(prefix, '/'))
|
|
return {};
|
|
|
|
StringList res;
|
|
if (prefix.front() == '/')
|
|
res = Kakoune::complete_filename(prefix, Regex{});
|
|
else
|
|
{
|
|
for (auto dir : options()["path"].get<StringList>())
|
|
{
|
|
if (not dir.empty() and dir.back() != '/')
|
|
dir += '/';
|
|
for (auto& filename : Kakoune::complete_filename(dir + prefix, Regex{}))
|
|
res.push_back(filename.substr(dir.length()));
|
|
}
|
|
}
|
|
if (res.empty())
|
|
return {};
|
|
return { begin.coord(), pos.coord(), std::move(res), buffer.timestamp() };
|
|
}
|
|
|
|
InsertCompletion complete_option(const Buffer& buffer, BufferCoord cursor_pos, const String& option_name)
|
|
{
|
|
const StringList& opt = options()[option_name].get<StringList>();;
|
|
if (opt.empty())
|
|
return {};
|
|
|
|
auto& desc = opt[0];
|
|
static const Regex re(R"((\d+)\.(\d+)(?:\+(\d+))?@(\d+))");
|
|
boost::smatch match;
|
|
if (boost::regex_match(desc.begin(), desc.end(), match, re))
|
|
{
|
|
BufferCoord coord{ str_to_int(match[1].str()) - 1, str_to_int(match[2].str()) - 1 };
|
|
if (not buffer.is_valid(coord))
|
|
return {};
|
|
auto end = coord;
|
|
if (match[3].matched)
|
|
{
|
|
ByteCount len = str_to_int(match[3].str());
|
|
end = buffer.advance(coord, len);
|
|
}
|
|
size_t timestamp = (size_t)str_to_int(match[4].str());
|
|
|
|
ByteCount longest_completion = 0;
|
|
for (auto it = opt.begin() + 1; it != opt.end(); ++it)
|
|
longest_completion = std::max(longest_completion, it->length());
|
|
|
|
if (timestamp == buffer.timestamp() and
|
|
cursor_pos.line == coord.line and cursor_pos.column >= coord.column and
|
|
buffer.distance(coord, cursor_pos) < longest_completion)
|
|
return { coord, end, { opt.begin() + 1, opt.end() }, timestamp };
|
|
}
|
|
return {};
|
|
}
|
|
|
|
InsertCompletion complete_line(const Buffer& buffer, BufferCoord cursor_pos)
|
|
{
|
|
String prefix = buffer[cursor_pos.line].substr(0_byte, cursor_pos.column);
|
|
StringList res;
|
|
for (LineCount l = 0_line; l < buffer.line_count(); ++l)
|
|
{
|
|
if (l == cursor_pos.line)
|
|
continue;
|
|
ByteCount len = buffer[l].length();
|
|
if (len > cursor_pos.column and std::equal(prefix.begin(), prefix.end(), buffer[l].begin()))
|
|
res.push_back(buffer[l].substr(0_byte, len-1));
|
|
}
|
|
if (res.empty())
|
|
return {};
|
|
std::sort(res.begin(), res.end());
|
|
res.erase(std::unique(res.begin(), res.end()), res.end());
|
|
return { cursor_pos.line, cursor_pos, std::move(res), buffer.timestamp() };
|
|
}
|
|
|
|
private:
|
|
void on_option_changed(const Option& opt) override
|
|
{
|
|
auto& completers = options()["completers"].get<StringList>();
|
|
StringList option_names;
|
|
for (auto& completer : completers)
|
|
{
|
|
if (completer.substr(0_byte, 7_byte) == "option=")
|
|
option_names.emplace_back(completer.substr(7_byte));
|
|
}
|
|
if (contains(option_names, opt.name()))
|
|
{
|
|
reset();
|
|
setup_ifn();
|
|
}
|
|
}
|
|
|
|
void menu_show()
|
|
{
|
|
if (not m_context.has_ui())
|
|
return;
|
|
DisplayCoord menu_pos = m_context.window().display_position(m_completions.begin);
|
|
m_context.ui().menu_show(m_matching_candidates, menu_pos,
|
|
get_color("MenuForeground"),
|
|
get_color("MenuBackground"),
|
|
MenuStyle::Inline);
|
|
m_context.ui().menu_select(m_current_candidate);
|
|
}
|
|
|
|
bool setup_ifn()
|
|
{
|
|
using namespace std::placeholders;
|
|
if (not m_completions.is_valid())
|
|
{
|
|
auto& completers = options()["completers"].get<StringList>();
|
|
for (auto& completer : completers)
|
|
{
|
|
if (completer == "filename" and
|
|
try_complete([this](const Buffer& buffer, BufferCoord cursor_pos)
|
|
{ return complete_filename<true>(buffer, cursor_pos); }))
|
|
return true;
|
|
if (completer.substr(0_byte, 7_byte) == "option=" and
|
|
try_complete([&, this](const Buffer& buffer, BufferCoord cursor_pos)
|
|
{ return complete_option(buffer, cursor_pos, completer.substr(7_byte)); }))
|
|
return true;
|
|
if (completer == "word=buffer" and
|
|
try_complete([this](const Buffer& buffer, BufferCoord cursor_pos)
|
|
{ return complete_word<false>(buffer, cursor_pos); }))
|
|
return true;
|
|
if (completer == "word=all" and
|
|
try_complete([this](const Buffer& buffer, BufferCoord cursor_pos)
|
|
{ return complete_word<true>(buffer, cursor_pos); }))
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
const Context& m_context;
|
|
InsertCompletion m_completions;
|
|
CandidateList m_matching_candidates;
|
|
int m_current_candidate = -1;
|
|
};
|
|
|
|
class Insert : public InputMode
|
|
{
|
|
public:
|
|
Insert(InputHandler& input_handler, InsertMode mode)
|
|
: InputMode(input_handler),
|
|
m_insert_mode(mode),
|
|
m_edition(context()),
|
|
m_completer(context()),
|
|
m_idle_timer{Clock::now() + idle_timeout,
|
|
[this](Timer& timer) {
|
|
context().hooks().run_hook("InsertIdle", "", context());
|
|
m_completer.update();
|
|
}}
|
|
{
|
|
last_insert().first = mode;
|
|
last_insert().second.clear();
|
|
context().hooks().run_hook("InsertBegin", "", context());
|
|
prepare(m_insert_mode);
|
|
}
|
|
|
|
void on_key(Key key) override
|
|
{
|
|
auto& buffer = context().buffer();
|
|
last_insert().second.push_back(key);
|
|
if (m_mode == Mode::InsertReg)
|
|
{
|
|
if (key.modifiers == Key::Modifiers::None)
|
|
insert(RegisterManager::instance()[key.key].values(context()));
|
|
m_mode = Mode::Default;
|
|
return;
|
|
}
|
|
if (m_mode == Mode::Complete)
|
|
{
|
|
if (key.key == 'f')
|
|
m_completer.try_complete([this](const Buffer& buffer, BufferCoord cursor_pos)
|
|
{ return m_completer.complete_filename<false>(buffer, cursor_pos); });
|
|
if (key.key == 'w')
|
|
m_completer.try_complete([this](const Buffer& buffer, BufferCoord cursor_pos)
|
|
{ return m_completer.complete_word<true>(buffer, cursor_pos); });
|
|
if (key.key == 'l')
|
|
m_completer.try_complete([this](const Buffer& buffer, BufferCoord cursor_pos)
|
|
{ return m_completer.complete_line(buffer, cursor_pos); });
|
|
m_mode = Mode::Default;
|
|
return;
|
|
}
|
|
|
|
bool update_completions = true;
|
|
bool moved = false;
|
|
if (key == Key::Escape or key == ctrl('c'))
|
|
{
|
|
context().hooks().run_hook("InsertEnd", "", context());
|
|
m_completer.reset();
|
|
reset_normal_mode();
|
|
}
|
|
else if (key == Key::Backspace)
|
|
{
|
|
for (auto& sel : context().selections())
|
|
{
|
|
if (sel.cursor() == BufferCoord{0,0})
|
|
continue;
|
|
auto pos = buffer.iterator_at(sel.cursor());
|
|
buffer.erase(utf8::previous(pos), pos);
|
|
}
|
|
}
|
|
else if (key == Key::Erase)
|
|
{
|
|
for (auto& sel : context().selections())
|
|
{
|
|
auto pos = buffer.iterator_at(sel.cursor());
|
|
buffer.erase(pos, utf8::next(pos));
|
|
}
|
|
}
|
|
else if (key == Key::Left)
|
|
{
|
|
move(-1_char);
|
|
moved = true;
|
|
}
|
|
else if (key == Key::Right)
|
|
{
|
|
move(1_char);
|
|
moved = true;
|
|
}
|
|
else if (key == Key::Up)
|
|
{
|
|
move(-1_line);
|
|
moved = true;
|
|
}
|
|
else if (key == Key::Down)
|
|
{
|
|
move(1_line);
|
|
moved = true;
|
|
}
|
|
else if (key.modifiers == Key::Modifiers::None)
|
|
insert(key.key);
|
|
else if (key == ctrl('r'))
|
|
m_mode = Mode::InsertReg;
|
|
else if ( key == ctrl('m'))
|
|
insert('\n');
|
|
else if ( key == ctrl('i'))
|
|
insert('\t');
|
|
else if ( key == ctrl('n'))
|
|
{
|
|
m_completer.select(1);
|
|
update_completions = false;
|
|
}
|
|
else if ( key == ctrl('p'))
|
|
{
|
|
m_completer.select(-1);
|
|
update_completions = false;
|
|
}
|
|
else if ( key == ctrl('x'))
|
|
m_mode = Mode::Complete;
|
|
else if ( key == ctrl('u'))
|
|
context().buffer().commit_undo_group();
|
|
|
|
context().hooks().run_hook("InsertKey", key_to_str(key), context());
|
|
|
|
if (update_completions)
|
|
m_idle_timer.set_next_date(Clock::now() + idle_timeout);
|
|
if (moved)
|
|
context().hooks().run_hook("InsertMove", key_to_str(key), context());
|
|
}
|
|
|
|
String description() const override
|
|
{
|
|
return "insert";
|
|
}
|
|
|
|
KeymapMode keymap_mode() const override { return KeymapMode::Insert; }
|
|
|
|
private:
|
|
template<typename Type>
|
|
void move(Type offset)
|
|
{
|
|
auto& selections = context().selections();
|
|
for (auto& sel : selections)
|
|
{
|
|
auto cursor = context().has_window() ? context().window().offset_coord(sel.cursor(), offset)
|
|
: context().buffer().offset_coord(sel.cursor(), offset);
|
|
sel.anchor() = sel.cursor() = cursor;
|
|
}
|
|
selections.sort_and_merge_overlapping();
|
|
}
|
|
|
|
void insert(memoryview<String> strings)
|
|
{
|
|
auto& buffer = context().buffer();
|
|
auto& selections = context().selections();
|
|
for (size_t i = 0; i < selections.size(); ++i)
|
|
{
|
|
size_t index = std::min(i, strings.size()-1);
|
|
buffer.insert(buffer.iterator_at(selections[i].cursor()),
|
|
strings[index]);
|
|
}
|
|
}
|
|
|
|
void insert(Codepoint key)
|
|
{
|
|
auto str = codepoint_to_str(key);
|
|
auto& buffer = context().buffer();
|
|
for (auto& sel : context().selections())
|
|
buffer.insert(buffer.iterator_at(sel.cursor()), str);
|
|
context().hooks().run_hook("InsertChar", str, context());
|
|
}
|
|
|
|
void prepare(InsertMode mode)
|
|
{
|
|
SelectionList& selections = context().selections();
|
|
Buffer& buffer = context().buffer();
|
|
|
|
for (auto& sel : selections)
|
|
{
|
|
BufferCoord anchor, cursor;
|
|
switch (mode)
|
|
{
|
|
case InsertMode::Insert:
|
|
anchor = sel.max();
|
|
cursor = sel.min();
|
|
break;
|
|
case InsertMode::Replace:
|
|
anchor = cursor = Kakoune::erase(buffer, sel).coord();
|
|
break;
|
|
case InsertMode::Append:
|
|
anchor = sel.min();
|
|
cursor = sel.max();
|
|
// special case for end of lines, append to current line instead
|
|
if (cursor.column != buffer[cursor.line].length() - 1)
|
|
cursor = buffer.char_next(cursor);
|
|
break;
|
|
|
|
case InsertMode::OpenLineBelow:
|
|
case InsertMode::AppendAtLineEnd:
|
|
anchor = cursor = BufferCoord{sel.max().line, buffer[sel.max().line].length() - 1};
|
|
break;
|
|
|
|
case InsertMode::OpenLineAbove:
|
|
case InsertMode::InsertAtLineBegin:
|
|
anchor = sel.min().line;
|
|
if (mode == InsertMode::OpenLineAbove)
|
|
anchor = buffer.char_prev(anchor);
|
|
else
|
|
{
|
|
auto anchor_non_blank = buffer.iterator_at(anchor);
|
|
while (*anchor_non_blank == ' ' or *anchor_non_blank == '\t')
|
|
++anchor_non_blank;
|
|
if (*anchor_non_blank != '\n')
|
|
anchor = anchor_non_blank.coord();
|
|
}
|
|
cursor = anchor;
|
|
break;
|
|
case InsertMode::InsertAtNextLineBegin:
|
|
kak_assert(false); // not implemented
|
|
break;
|
|
}
|
|
if (buffer.is_end(anchor))
|
|
anchor = buffer.char_prev(anchor);
|
|
if (buffer.is_end(cursor))
|
|
cursor = buffer.char_prev(cursor);
|
|
sel.anchor() = anchor;
|
|
sel.cursor() = cursor;
|
|
}
|
|
if (mode == InsertMode::OpenLineBelow or mode == InsertMode::OpenLineAbove)
|
|
{
|
|
insert('\n');
|
|
if (mode == InsertMode::OpenLineAbove)
|
|
{
|
|
for (auto& sel : selections)
|
|
{
|
|
// special case, the --first line above did nothing, so we need to compensate now
|
|
if (sel.anchor() == buffer.char_next({0,0}))
|
|
sel.anchor() = sel.cursor() = BufferCoord{0,0};
|
|
}
|
|
}
|
|
}
|
|
selections.sort_and_merge_overlapping();
|
|
selections.check_invariant();
|
|
buffer.check_invariant();
|
|
}
|
|
|
|
void on_disabled() override
|
|
{
|
|
for (auto& sel : context().selections())
|
|
{
|
|
if (m_insert_mode == InsertMode::Append and sel.cursor().column > 0)
|
|
sel.cursor() = context().buffer().char_prev(sel.cursor());
|
|
avoid_eol(context().buffer(), sel);
|
|
}
|
|
}
|
|
|
|
enum class Mode { Default, Complete, InsertReg };
|
|
Mode m_mode = Mode::Default;
|
|
InsertMode m_insert_mode;
|
|
ScopedEdition m_edition;
|
|
InsertCompleter m_completer;
|
|
Timer m_idle_timer;
|
|
};
|
|
|
|
}
|
|
|
|
void InputMode::reset_normal_mode()
|
|
{
|
|
m_input_handler.reset_normal_mode();
|
|
}
|
|
|
|
InputHandler::InputHandler(Buffer& buffer, SelectionList selections, String name)
|
|
: m_mode(new InputModes::Normal(*this)),
|
|
m_context(*this, buffer, std::move(selections), std::move(name))
|
|
{
|
|
}
|
|
|
|
InputHandler::~InputHandler()
|
|
{}
|
|
|
|
void InputHandler::change_input_mode(InputMode* new_mode)
|
|
{
|
|
m_mode->on_disabled();
|
|
m_mode_trash.emplace_back(std::move(m_mode));
|
|
m_mode.reset(new_mode);
|
|
new_mode->on_enabled();
|
|
}
|
|
|
|
void InputHandler::insert(InsertMode mode)
|
|
{
|
|
change_input_mode(new InputModes::Insert(*this, mode));
|
|
}
|
|
|
|
void InputHandler::repeat_last_insert()
|
|
{
|
|
if (m_last_insert.second.empty())
|
|
return;
|
|
|
|
std::vector<Key> keys;
|
|
swap(keys, m_last_insert.second);
|
|
// context.last_insert will be refilled by the new Insert
|
|
// this is very inefficient.
|
|
change_input_mode(new InputModes::Insert(*this, m_last_insert.first));
|
|
for (auto& key : keys)
|
|
m_mode->on_key(key);
|
|
kak_assert(dynamic_cast<InputModes::Normal*>(m_mode.get()) != nullptr);
|
|
}
|
|
|
|
void InputHandler::prompt(const String& prompt, String initstr,
|
|
ColorPair prompt_colors, Completer completer,
|
|
PromptCallback callback)
|
|
{
|
|
change_input_mode(new InputModes::Prompt(*this, prompt, initstr,
|
|
prompt_colors, completer,
|
|
callback));
|
|
}
|
|
|
|
void InputHandler::set_prompt_colors(ColorPair prompt_colors)
|
|
{
|
|
InputModes::Prompt* prompt = dynamic_cast<InputModes::Prompt*>(m_mode.get());
|
|
if (prompt)
|
|
prompt->set_prompt_colors(prompt_colors);
|
|
}
|
|
|
|
void InputHandler::menu(memoryview<String> choices,
|
|
MenuCallback callback)
|
|
{
|
|
change_input_mode(new InputModes::Menu(*this, choices, callback));
|
|
}
|
|
|
|
void InputHandler::on_next_key(KeyCallback callback)
|
|
{
|
|
change_input_mode(new InputModes::NextKey(*this, callback));
|
|
}
|
|
|
|
static bool is_valid(Key key)
|
|
{
|
|
return key != Key::Invalid and key.key <= 0x10FFFF;
|
|
}
|
|
|
|
void InputHandler::handle_key(Key key)
|
|
{
|
|
if (is_valid(key))
|
|
{
|
|
const bool was_recording = is_recording();
|
|
|
|
auto keymap_mode = m_mode->keymap_mode();
|
|
KeymapManager& keymaps = m_context.keymaps();
|
|
if (keymaps.is_mapped(key, keymap_mode))
|
|
{
|
|
for (auto& k : keymaps.get_mapping(key, keymap_mode))
|
|
m_mode->on_key(k);
|
|
}
|
|
else
|
|
m_mode->on_key(key);
|
|
|
|
// do not record the key that made us enter or leave recording mode.
|
|
if (was_recording and is_recording())
|
|
m_recorded_keys += key_to_str(key);
|
|
}
|
|
}
|
|
|
|
void InputHandler::start_recording(char reg)
|
|
{
|
|
kak_assert(m_recording_reg == 0);
|
|
m_recorded_keys = "";
|
|
m_recording_reg = reg;
|
|
}
|
|
|
|
bool InputHandler::is_recording() const
|
|
{
|
|
return m_recording_reg != 0;
|
|
}
|
|
|
|
void InputHandler::stop_recording()
|
|
{
|
|
kak_assert(m_recording_reg != 0);
|
|
RegisterManager::instance()[m_recording_reg] = memoryview<String>(m_recorded_keys);
|
|
m_recording_reg = 0;
|
|
}
|
|
|
|
void InputHandler::reset_normal_mode()
|
|
{
|
|
change_input_mode(new InputModes::Normal(*this));
|
|
}
|
|
|
|
String InputHandler::mode_string() const
|
|
{
|
|
return m_mode->description();
|
|
}
|
|
|
|
void InputHandler::clear_mode_trash()
|
|
{
|
|
m_mode_trash.clear();
|
|
}
|
|
|
|
}
|