Extract insert completion code to insert_completer.{cc,hh}
This commit is contained in:
Maxime Coste
parent
512bfa0c65
commit
f683946681
|
|
@ -1,18 +1,16 @@
|
||||||
#include "input_handler.hh"
|
#include "input_handler.hh"
|
||||||
|
|
||||||
#include "window.hh"
|
|
||||||
#include "utf8.hh"
|
|
||||||
#include "user_interface.hh"
|
|
||||||
#include "buffer_manager.hh"
|
#include "buffer_manager.hh"
|
||||||
#include "register_manager.hh"
|
#include "buffer_utils.hh"
|
||||||
#include "normal.hh"
|
|
||||||
#include "event_manager.hh"
|
|
||||||
#include "client.hh"
|
#include "client.hh"
|
||||||
#include "color_registry.hh"
|
#include "color_registry.hh"
|
||||||
#include "file.hh"
|
#include "event_manager.hh"
|
||||||
#include "word_db.hh"
|
#include "insert_completer.hh"
|
||||||
#include "buffer_utils.hh"
|
#include "normal.hh"
|
||||||
#include "debug.hh"
|
#include "register_manager.hh"
|
||||||
|
#include "user_interface.hh"
|
||||||
|
#include "utf8.hh"
|
||||||
|
#include "window.hh"
|
||||||
|
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
|
|
||||||
|
|
@ -616,359 +614,6 @@ private:
|
||||||
KeyCallback m_callback;
|
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);
|
|
||||||
|
|
||||||
const CharCount tabstop = m_context.options()["tabstop"].get<int>();
|
|
||||||
const CharCount column = get_column(m_context.buffer(), tabstop,
|
|
||||||
m_completions.begin);
|
|
||||||
std::vector<String> menu_entries;
|
|
||||||
for (auto& candidate : m_matching_candidates)
|
|
||||||
menu_entries.push_back(expand_tabs(candidate, tabstop, column));
|
|
||||||
|
|
||||||
m_context.ui().menu_show(menu_entries, 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
|
class Insert : public InputMode
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
|
@ -1003,14 +648,11 @@ public:
|
||||||
if (m_mode == Mode::Complete)
|
if (m_mode == Mode::Complete)
|
||||||
{
|
{
|
||||||
if (key.key == 'f')
|
if (key.key == 'f')
|
||||||
m_completer.try_complete([this](const Buffer& buffer, BufferCoord cursor_pos)
|
m_completer.explicit_file_complete();
|
||||||
{ return m_completer.complete_filename<false>(buffer, cursor_pos); });
|
|
||||||
if (key.key == 'w')
|
if (key.key == 'w')
|
||||||
m_completer.try_complete([this](const Buffer& buffer, BufferCoord cursor_pos)
|
m_completer.explicit_word_complete();
|
||||||
{ return m_completer.complete_word<true>(buffer, cursor_pos); });
|
|
||||||
if (key.key == 'l')
|
if (key.key == 'l')
|
||||||
m_completer.try_complete([this](const Buffer& buffer, BufferCoord cursor_pos)
|
m_completer.explicit_line_complete();
|
||||||
{ return m_completer.complete_line(buffer, cursor_pos); });
|
|
||||||
m_mode = Mode::Default;
|
m_mode = Mode::Default;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
384
src/insert_completer.cc
Normal file
384
src/insert_completer.cc
Normal file
|
|
@ -0,0 +1,384 @@
|
||||||
|
#include "insert_completer.hh"
|
||||||
|
|
||||||
|
#include "buffer_manager.hh"
|
||||||
|
#include "buffer_utils.hh"
|
||||||
|
#include "color_registry.hh"
|
||||||
|
#include "context.hh"
|
||||||
|
#include "debug.hh"
|
||||||
|
#include "display_buffer.hh"
|
||||||
|
#include "file.hh"
|
||||||
|
#include "user_interface.hh"
|
||||||
|
#include "window.hh"
|
||||||
|
#include "word_db.hh"
|
||||||
|
|
||||||
|
namespace Kakoune
|
||||||
|
{
|
||||||
|
|
||||||
|
using StringList = std::vector<String>;
|
||||||
|
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
|
||||||
|
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,
|
||||||
|
OptionManager& options)
|
||||||
|
{
|
||||||
|
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,
|
||||||
|
OptionManager& options, 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() };
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
InsertCompleter::InsertCompleter(const Context& context)
|
||||||
|
: OptionManagerWatcher_AutoRegister(context.options()),
|
||||||
|
m_context(context)
|
||||||
|
{}
|
||||||
|
|
||||||
|
void InsertCompleter::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 InsertCompleter::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 InsertCompleter::reset()
|
||||||
|
{
|
||||||
|
m_completions = InsertCompletion{};
|
||||||
|
if (m_context.has_ui())
|
||||||
|
m_context.ui().menu_hide();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool InsertCompleter::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,
|
||||||
|
options());
|
||||||
|
}))
|
||||||
|
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,
|
||||||
|
options(), 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;
|
||||||
|
}
|
||||||
|
|
||||||
|
void InsertCompleter::menu_show()
|
||||||
|
{
|
||||||
|
if (not m_context.has_ui())
|
||||||
|
return;
|
||||||
|
DisplayCoord menu_pos = m_context.window().display_position(m_completions.begin);
|
||||||
|
|
||||||
|
const CharCount tabstop = m_context.options()["tabstop"].get<int>();
|
||||||
|
const CharCount column = get_column(m_context.buffer(), tabstop,
|
||||||
|
m_completions.begin);
|
||||||
|
std::vector<String> menu_entries;
|
||||||
|
for (auto& candidate : m_matching_candidates)
|
||||||
|
menu_entries.push_back(expand_tabs(candidate, tabstop, column));
|
||||||
|
|
||||||
|
m_context.ui().menu_show(menu_entries, menu_pos,
|
||||||
|
get_color("MenuForeground"),
|
||||||
|
get_color("MenuBackground"),
|
||||||
|
MenuStyle::Inline);
|
||||||
|
m_context.ui().menu_select(m_current_candidate);
|
||||||
|
}
|
||||||
|
|
||||||
|
void InsertCompleter::on_option_changed(const Option& opt)
|
||||||
|
{
|
||||||
|
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();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename CompleteFunc>
|
||||||
|
bool InsertCompleter::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;
|
||||||
|
}
|
||||||
|
|
||||||
|
void InsertCompleter::explicit_file_complete()
|
||||||
|
{
|
||||||
|
try_complete([this](const Buffer& buffer, BufferCoord cursor_pos) {
|
||||||
|
return complete_filename<false>(buffer, cursor_pos, options());
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
void InsertCompleter::explicit_word_complete()
|
||||||
|
{
|
||||||
|
try_complete([this](const Buffer& buffer, BufferCoord cursor_pos) {
|
||||||
|
return complete_word<true>(buffer, cursor_pos);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
void InsertCompleter::explicit_line_complete()
|
||||||
|
{
|
||||||
|
try_complete([this](const Buffer& buffer, BufferCoord cursor_pos) {
|
||||||
|
return complete_line(buffer, cursor_pos);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
53
src/insert_completer.hh
Normal file
53
src/insert_completer.hh
Normal file
|
|
@ -0,0 +1,53 @@
|
||||||
|
#ifndef insert_completer_hh_INCLUDED
|
||||||
|
#define insert_completer_hh_INCLUDED
|
||||||
|
|
||||||
|
#include "buffer.hh"
|
||||||
|
#include "option_manager.hh"
|
||||||
|
|
||||||
|
namespace Kakoune
|
||||||
|
{
|
||||||
|
|
||||||
|
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);
|
||||||
|
InsertCompleter(const InsertCompleter&) = delete;
|
||||||
|
InsertCompleter& operator=(const InsertCompleter&) = delete;
|
||||||
|
|
||||||
|
void select(int offset);
|
||||||
|
void update();
|
||||||
|
void reset();
|
||||||
|
|
||||||
|
void explicit_file_complete();
|
||||||
|
void explicit_word_complete();
|
||||||
|
void explicit_line_complete();
|
||||||
|
|
||||||
|
private:
|
||||||
|
bool setup_ifn();
|
||||||
|
|
||||||
|
template<typename CompleteFunc>
|
||||||
|
bool try_complete(CompleteFunc complete_func);
|
||||||
|
void on_option_changed(const Option& opt) override;
|
||||||
|
|
||||||
|
void menu_show();
|
||||||
|
|
||||||
|
const Context& m_context;
|
||||||
|
InsertCompletion m_completions;
|
||||||
|
CandidateList m_matching_candidates;
|
||||||
|
int m_current_candidate = -1;
|
||||||
|
};
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // insert_completer_hh_INCLUDED
|
||||||
|
|
||||||
Loading…
Reference in New Issue
Block a user