2011-09-07 22:16:56 +04:00
|
|
|
#include "command_manager.hh"
|
|
|
|
|
|
|
|
#include "utils.hh"
|
2011-09-16 13:18:51 +04:00
|
|
|
#include "assert.hh"
|
2012-02-14 01:38:07 +04:00
|
|
|
#include "context.hh"
|
2011-09-16 13:18:51 +04:00
|
|
|
|
2011-09-07 22:16:56 +04:00
|
|
|
#include <algorithm>
|
2012-02-14 01:38:07 +04:00
|
|
|
#include <cstring>
|
|
|
|
#include <sys/types.h>
|
|
|
|
#include <sys/wait.h>
|
2011-09-07 22:16:56 +04:00
|
|
|
|
|
|
|
namespace Kakoune
|
|
|
|
{
|
|
|
|
|
2011-09-16 13:18:51 +04:00
|
|
|
void CommandManager::register_command(const std::string& command_name, Command command,
|
2012-01-15 07:02:08 +04:00
|
|
|
unsigned flags,
|
2011-09-16 13:18:51 +04:00
|
|
|
const CommandCompleter& completer)
|
2011-09-07 22:16:56 +04:00
|
|
|
{
|
2012-01-15 07:02:08 +04:00
|
|
|
m_commands[command_name] = CommandDescriptor { command, flags, completer };
|
2011-09-07 22:16:56 +04:00
|
|
|
}
|
|
|
|
|
2012-02-03 23:14:35 +04:00
|
|
|
void CommandManager::register_commands(const memoryview<std::string>& command_names, Command command,
|
|
|
|
unsigned flags,
|
|
|
|
const CommandCompleter& completer)
|
2011-09-07 22:16:56 +04:00
|
|
|
{
|
|
|
|
for (auto command_name : command_names)
|
2012-01-15 07:02:08 +04:00
|
|
|
register_command(command_name, command, flags, completer);
|
2011-09-07 22:16:56 +04:00
|
|
|
}
|
|
|
|
|
2012-02-14 01:38:07 +04:00
|
|
|
static bool is_blank(char c)
|
|
|
|
{
|
|
|
|
return c == ' ' or c == '\t' or c == '\n';
|
|
|
|
}
|
|
|
|
|
2011-09-14 01:16:48 +04:00
|
|
|
typedef std::vector<std::pair<size_t, size_t>> TokenList;
|
|
|
|
static TokenList split(const std::string& line)
|
2011-09-07 22:16:56 +04:00
|
|
|
{
|
2011-09-14 01:16:48 +04:00
|
|
|
TokenList result;
|
2011-09-07 22:16:56 +04:00
|
|
|
|
|
|
|
size_t pos = 0;
|
2011-11-26 21:20:02 +04:00
|
|
|
while (pos < line.length())
|
2011-09-07 22:16:56 +04:00
|
|
|
{
|
2012-02-14 01:38:07 +04:00
|
|
|
while(is_blank(line[pos]) and pos != line.length())
|
2011-09-07 22:16:56 +04:00
|
|
|
++pos;
|
|
|
|
|
2012-02-14 01:38:07 +04:00
|
|
|
size_t token_start = pos;
|
|
|
|
|
|
|
|
if (line[pos] == '"' or line[pos] == '\'' or line[pos] == '`')
|
2011-11-26 21:20:02 +04:00
|
|
|
{
|
2012-02-14 01:38:07 +04:00
|
|
|
char delimiter = line[pos];
|
2011-11-26 21:20:02 +04:00
|
|
|
++pos;
|
2012-02-14 01:38:07 +04:00
|
|
|
token_start = delimiter == '`' ? pos - 1 : pos;
|
2011-11-26 21:20:02 +04:00
|
|
|
|
2012-02-14 01:38:07 +04:00
|
|
|
while ((line[pos] != delimiter or line[pos-1] == '\\') and
|
|
|
|
pos != line.length())
|
|
|
|
++pos;
|
2011-09-07 22:16:56 +04:00
|
|
|
|
2012-02-14 01:38:07 +04:00
|
|
|
if (delimiter == '`' and line[pos] == '`')
|
|
|
|
++pos;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
while (not is_blank(line[pos]) and pos != line.length() and
|
|
|
|
(line[pos] != ';' or line[pos-1] == '\\'))
|
|
|
|
++pos;
|
2011-09-07 22:16:56 +04:00
|
|
|
|
2012-02-14 01:38:07 +04:00
|
|
|
if (token_start != pos)
|
|
|
|
result.push_back(std::make_pair(token_start, pos));
|
2011-11-26 21:20:02 +04:00
|
|
|
|
2012-01-15 05:37:35 +04:00
|
|
|
if (line[pos] == ';')
|
|
|
|
result.push_back(std::make_pair(pos, pos+1));
|
|
|
|
|
2011-11-26 21:20:02 +04:00
|
|
|
++pos;
|
2011-09-07 22:16:56 +04:00
|
|
|
}
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2011-09-09 22:40:59 +04:00
|
|
|
struct command_not_found : runtime_error
|
2011-09-07 22:16:56 +04:00
|
|
|
{
|
2011-09-09 22:40:59 +04:00
|
|
|
command_not_found(const std::string& command)
|
|
|
|
: runtime_error(command + " : no such command") {}
|
2011-09-07 22:16:56 +04:00
|
|
|
};
|
|
|
|
|
2011-11-26 22:32:57 +04:00
|
|
|
void CommandManager::execute(const std::string& command_line,
|
|
|
|
const Context& context)
|
2011-09-07 22:16:56 +04:00
|
|
|
{
|
2011-09-14 01:16:48 +04:00
|
|
|
TokenList tokens = split(command_line);
|
2011-09-07 22:16:56 +04:00
|
|
|
if (tokens.empty())
|
|
|
|
return;
|
|
|
|
|
2012-02-03 23:14:35 +04:00
|
|
|
std::vector<std::string> params;
|
2012-01-14 18:02:54 +04:00
|
|
|
for (auto it = tokens.begin(); it != tokens.end(); ++it)
|
2011-09-14 01:16:48 +04:00
|
|
|
{
|
|
|
|
params.push_back(command_line.substr(it->first,
|
|
|
|
it->second - it->first));
|
|
|
|
}
|
2011-09-07 22:16:56 +04:00
|
|
|
|
2012-01-14 18:02:54 +04:00
|
|
|
execute(params, context);
|
2011-11-26 22:32:57 +04:00
|
|
|
}
|
|
|
|
|
2012-02-14 01:38:07 +04:00
|
|
|
static void shell_eval(std::vector<std::string>& params,
|
|
|
|
const std::string& cmdline,
|
|
|
|
const Context& context)
|
|
|
|
{
|
|
|
|
int write_pipe[2];
|
|
|
|
int read_pipe[2];
|
|
|
|
|
|
|
|
pipe(write_pipe);
|
|
|
|
pipe(read_pipe);
|
|
|
|
|
|
|
|
if (pid_t pid = fork())
|
|
|
|
{
|
|
|
|
close(write_pipe[0]);
|
|
|
|
close(read_pipe[1]);
|
|
|
|
close(write_pipe[1]);
|
|
|
|
|
|
|
|
std::string output;
|
|
|
|
char buffer[1024];
|
|
|
|
while (size_t size = read(read_pipe[0], buffer, 1024))
|
2012-03-09 01:15:37 +04:00
|
|
|
{
|
|
|
|
if (size == -1)
|
|
|
|
break;
|
2012-02-14 01:38:07 +04:00
|
|
|
output += std::string(buffer, buffer+size);
|
2012-03-09 01:15:37 +04:00
|
|
|
}
|
2012-02-14 01:38:07 +04:00
|
|
|
close(read_pipe[0]);
|
|
|
|
waitpid(pid, NULL, 0);
|
|
|
|
|
|
|
|
TokenList tokens = split(output);
|
|
|
|
|
|
|
|
for (auto it = tokens.begin(); it != tokens.end(); ++it)
|
|
|
|
{
|
|
|
|
params.push_back(output.substr(it->first,
|
|
|
|
it->second - it->first));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
close(write_pipe[1]);
|
|
|
|
close(read_pipe[0]);
|
|
|
|
|
|
|
|
dup2(read_pipe[1], 1);
|
|
|
|
dup2(write_pipe[0], 0);
|
|
|
|
|
2012-03-04 23:42:31 +04:00
|
|
|
if (context.has_buffer())
|
|
|
|
setenv("kak_bufname", context.buffer().name().c_str(), 1);
|
2012-02-14 01:38:07 +04:00
|
|
|
execlp("sh", "sh", "-c", cmdline.c_str(), NULL);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-01-14 18:02:54 +04:00
|
|
|
void CommandManager::execute(const CommandParameters& params,
|
2011-11-26 22:32:57 +04:00
|
|
|
const Context& context)
|
|
|
|
{
|
2012-01-14 18:02:54 +04:00
|
|
|
if (params.empty())
|
|
|
|
return;
|
|
|
|
|
2012-01-15 05:37:35 +04:00
|
|
|
auto begin = params.begin();
|
|
|
|
auto end = begin;
|
|
|
|
while (true)
|
|
|
|
{
|
|
|
|
while (end != params.end() and *end != ";")
|
|
|
|
++end;
|
2011-11-26 22:32:57 +04:00
|
|
|
|
2012-01-15 05:37:35 +04:00
|
|
|
if (end != begin)
|
|
|
|
{
|
2012-02-15 18:19:57 +04:00
|
|
|
std::vector<std::string> expanded_params;
|
2012-01-15 05:37:35 +04:00
|
|
|
auto command_it = m_commands.find(*begin);
|
2012-02-15 18:19:57 +04:00
|
|
|
|
|
|
|
if (command_it == m_commands.end() and
|
|
|
|
begin->front() == '`' and begin->back() == '`')
|
|
|
|
{
|
|
|
|
shell_eval(expanded_params,
|
|
|
|
begin->substr(1, begin->length() - 2),
|
|
|
|
context);
|
|
|
|
if (not expanded_params.empty())
|
|
|
|
{
|
|
|
|
command_it = m_commands.find(expanded_params[0]);
|
|
|
|
expanded_params.erase(expanded_params.begin());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-01-15 05:37:35 +04:00
|
|
|
if (command_it == m_commands.end())
|
|
|
|
throw command_not_found(*begin);
|
|
|
|
|
2012-01-15 07:02:08 +04:00
|
|
|
if (command_it->second.flags & IgnoreSemiColons)
|
|
|
|
end = params.end();
|
|
|
|
|
2012-02-14 01:38:07 +04:00
|
|
|
if (command_it->second.flags & DeferredShellEval)
|
|
|
|
command_it->second.command(CommandParameters(begin + 1, end), context);
|
|
|
|
else
|
|
|
|
{
|
|
|
|
for (auto param = begin+1; param != end; ++param)
|
|
|
|
{
|
|
|
|
if (param->front() == '`' and param->back() == '`')
|
|
|
|
shell_eval(expanded_params,
|
|
|
|
param->substr(1, param->length() - 2),
|
|
|
|
context);
|
|
|
|
else
|
|
|
|
expanded_params.push_back(*param);
|
|
|
|
}
|
|
|
|
command_it->second.command(expanded_params, context);
|
|
|
|
}
|
|
|
|
|
2012-01-15 05:37:35 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
if (end == params.end())
|
|
|
|
break;
|
|
|
|
|
|
|
|
begin = end+1;
|
|
|
|
end = begin;
|
|
|
|
}
|
2011-09-07 22:16:56 +04:00
|
|
|
}
|
|
|
|
|
2011-09-14 01:16:48 +04:00
|
|
|
Completions CommandManager::complete(const std::string& command_line, size_t cursor_pos)
|
|
|
|
{
|
|
|
|
TokenList tokens = split(command_line);
|
|
|
|
|
2011-11-26 23:05:49 +04:00
|
|
|
size_t token_to_complete = tokens.size();
|
2011-09-14 01:16:48 +04:00
|
|
|
for (size_t i = 0; i < tokens.size(); ++i)
|
|
|
|
{
|
2011-10-19 22:50:28 +04:00
|
|
|
if (tokens[i].first <= cursor_pos and tokens[i].second >= cursor_pos)
|
2011-09-14 01:16:48 +04:00
|
|
|
{
|
|
|
|
token_to_complete = i;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-10-19 22:50:28 +04:00
|
|
|
if (token_to_complete == 0 or tokens.empty()) // command name completion
|
2011-09-14 01:16:48 +04:00
|
|
|
{
|
2011-10-19 22:50:28 +04:00
|
|
|
size_t cmd_start = tokens.empty() ? 0 : tokens[0].first;
|
|
|
|
Completions result(cmd_start, cursor_pos);
|
|
|
|
std::string prefix = command_line.substr(cmd_start,
|
|
|
|
cursor_pos - cmd_start);
|
2011-09-14 01:16:48 +04:00
|
|
|
|
|
|
|
for (auto& command : m_commands)
|
|
|
|
{
|
|
|
|
if (command.first.substr(0, prefix.length()) == prefix)
|
|
|
|
result.candidates.push_back(command.first);
|
|
|
|
}
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
2011-09-16 13:18:51 +04:00
|
|
|
|
|
|
|
assert(not tokens.empty());
|
|
|
|
std::string command_name =
|
|
|
|
command_line.substr(tokens[0].first,
|
|
|
|
tokens[0].second - tokens[0].first);
|
|
|
|
|
|
|
|
auto command_it = m_commands.find(command_name);
|
|
|
|
if (command_it == m_commands.end() or not command_it->second.completer)
|
|
|
|
return Completions();
|
|
|
|
|
2012-02-03 23:14:35 +04:00
|
|
|
std::vector<std::string> params;
|
2011-09-16 13:18:51 +04:00
|
|
|
for (auto it = tokens.begin() + 1; it != tokens.end(); ++it)
|
2011-09-14 19:41:56 +04:00
|
|
|
{
|
2011-09-16 13:18:51 +04:00
|
|
|
params.push_back(command_line.substr(it->first,
|
|
|
|
it->second - it->first));
|
2011-09-14 19:41:56 +04:00
|
|
|
}
|
2011-11-26 23:05:49 +04:00
|
|
|
|
|
|
|
size_t start = token_to_complete < tokens.size() ?
|
|
|
|
tokens[token_to_complete].first : cursor_pos;
|
|
|
|
Completions result(start , cursor_pos);
|
|
|
|
size_t cursor_pos_in_token = cursor_pos - start;
|
2011-09-16 13:18:51 +04:00
|
|
|
|
|
|
|
result.candidates = command_it->second.completer(params,
|
|
|
|
token_to_complete - 1,
|
|
|
|
cursor_pos_in_token);
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
CandidateList PerArgumentCommandCompleter::operator()(const CommandParameters& params,
|
|
|
|
size_t token_to_complete,
|
|
|
|
size_t pos_in_token) const
|
|
|
|
{
|
|
|
|
if (token_to_complete >= m_completers.size())
|
|
|
|
return CandidateList();
|
|
|
|
|
|
|
|
// it is possible to try to complete a new argument
|
|
|
|
assert(token_to_complete <= params.size());
|
|
|
|
|
|
|
|
const std::string& argument = token_to_complete < params.size() ?
|
|
|
|
params[token_to_complete] : std::string();
|
|
|
|
return m_completers[token_to_complete](argument, pos_in_token);
|
2011-09-14 01:16:48 +04:00
|
|
|
}
|
|
|
|
|
2011-09-07 22:16:56 +04:00
|
|
|
}
|