Change ParameterParser switch interface, use an Optional<StringView>

Merge has_option and option_value into a single method get_switch.
This commit is contained in:
Maxime Coste 2015-03-14 19:16:46 +00:00
parent 65fd0df041
commit b5854ece43
6 changed files with 77 additions and 116 deletions

View File

@ -138,19 +138,19 @@ void edit(const ParametersParser& parser, Context& context)
buffer = BufferManager::instance().get_buffer_ifp(name); buffer = BufferManager::instance().get_buffer_ifp(name);
if (not buffer) if (not buffer)
{ {
if (parser.has_option("scratch")) if (parser.get_switch("scratch"))
{ {
BufferManager::instance().delete_buffer_if_exists(name); BufferManager::instance().delete_buffer_if_exists(name);
buffer = new Buffer(name, Buffer::Flags::None); buffer = new Buffer(name, Buffer::Flags::None);
} }
else if (parser.has_option("fifo")) else if (auto fifo = parser.get_switch("fifo"))
buffer = open_fifo(name, parser.option_value("fifo"), parser.has_option("scroll")); buffer = open_fifo(name, *fifo, (bool)parser.get_switch("scroll"));
else else
{ {
buffer = create_buffer_from_file(name); buffer = create_buffer_from_file(name);
if (not buffer) if (not buffer)
{ {
if (parser.has_option("existing")) if (parser.get_switch("existing"))
throw runtime_error("unable to open " + name); throw runtime_error("unable to open " + name);
context.print_status({ "new file " + name, get_face("StatusLine") }); context.print_status({ "new file " + name, get_face("StatusLine") });
@ -525,9 +525,9 @@ const CommandDesc add_highlighter_cmd = {
for (; begin != parser.end(); ++begin) for (; begin != parser.end(); ++begin)
highlighter_params.push_back(*begin); highlighter_params.push_back(*begin);
auto& group = (parser.has_option("group")) ? auto group_name = parser.get_switch("group");
get_highlighter(context, parser.option_value("group")) auto& group = group_name ? get_highlighter(context, *group_name)
: context.window().highlighters(); : context.window().highlighters();
auto it = registry.find(name); auto it = registry.find(name);
if (it == registry.end()) if (it == registry.end())
throw runtime_error("No such highlighter factory '" + name + "'"); throw runtime_error("No such highlighter factory '" + name + "'");
@ -602,9 +602,7 @@ const CommandDesc add_hook_cmd = {
CommandManager::instance().execute(command, context, {}, CommandManager::instance().execute(command, context, {},
{ { "hook_param", param.str() } }); { { "hook_param", param.str() } });
}; };
StringView group; auto group = parser.get_switch("group").value_or(StringView{});
if (parser.has_option("group"))
group = parser.option_value("group");
get_scope(parser[0], context).hooks().add_hook(parser[1], group.str(), hook_func); get_scope(parser[0], context).hooks().add_hook(parser[1], group.str(), hook_func);
} }
}; };
@ -650,21 +648,17 @@ void define_command(const ParametersParser& parser, Context& context)
const String& cmd_name = parser[0]; const String& cmd_name = parser[0];
auto& cm = CommandManager::instance(); auto& cm = CommandManager::instance();
if (cm.command_defined(cmd_name) and not parser.has_option("allow-override")) if (cm.command_defined(cmd_name) and not parser.get_switch("allow-override"))
throw runtime_error("command '" + cmd_name + "' already defined"); throw runtime_error("command '" + cmd_name + "' already defined");
CommandFlags flags = CommandFlags::None; CommandFlags flags = CommandFlags::None;
if (parser.has_option("hidden")) if (parser.get_switch("hidden"))
flags = CommandFlags::Hidden; flags = CommandFlags::Hidden;
String docstring;
if (parser.has_option("docstring"))
docstring = parser.option_value("docstring");
const String& commands = parser[1]; const String& commands = parser[1];
Command cmd; Command cmd;
ParameterDesc desc; ParameterDesc desc;
if (parser.has_option("shell-params")) if (parser.get_switch("shell-params"))
{ {
desc = ParameterDesc{ {}, ParameterDesc::Flags::SwitchesAsPositional }; desc = ParameterDesc{ {}, ParameterDesc::Flags::SwitchesAsPositional };
cmd = [=](const ParametersParser& parser, Context& context) { cmd = [=](const ParametersParser& parser, Context& context) {
@ -680,7 +674,7 @@ void define_command(const ParametersParser& parser, Context& context)
} }
CommandCompleter completer; CommandCompleter completer;
if (parser.has_option("file-completion")) if (parser.get_switch("file-completion"))
{ {
completer = [](const Context& context, CompletionFlags flags, completer = [](const Context& context, CompletionFlags flags,
CommandParameters params, CommandParameters params,
@ -693,7 +687,7 @@ void define_command(const ParametersParser& parser, Context& context)
pos_in_token) }; pos_in_token) };
}; };
} }
if (parser.has_option("client-completion")) else if (parser.get_switch("client-completion"))
{ {
completer = [](const Context& context, CompletionFlags flags, completer = [](const Context& context, CompletionFlags flags,
CommandParameters params, CommandParameters params,
@ -705,7 +699,7 @@ void define_command(const ParametersParser& parser, Context& context)
cm.complete_client_name(prefix, pos_in_token) }; cm.complete_client_name(prefix, pos_in_token) };
}; };
} }
if (parser.has_option("buffer-completion")) else if (parser.get_switch("buffer-completion"))
{ {
completer = [](const Context& context, CompletionFlags flags, completer = [](const Context& context, CompletionFlags flags,
CommandParameters params, CommandParameters params,
@ -716,9 +710,9 @@ void define_command(const ParametersParser& parser, Context& context)
complete_buffer_name(prefix, pos_in_token) }; complete_buffer_name(prefix, pos_in_token) };
}; };
} }
else if (parser.has_option("shell-completion")) else if (auto shell_cmd_opt = parser.get_switch("shell-completion"))
{ {
const String& shell_cmd = parser.option_value("shell-completion"); String shell_cmd = shell_cmd_opt->str();
completer = [=](const Context& context, CompletionFlags flags, completer = [=](const Context& context, CompletionFlags flags,
CommandParameters params, CommandParameters params,
size_t token_to_complete, ByteCount pos_in_token) size_t token_to_complete, ByteCount pos_in_token)
@ -735,7 +729,9 @@ void define_command(const ParametersParser& parser, Context& context)
}; };
} }
cm.register_command(cmd_name, cmd, std::move(docstring), desc, flags, CommandHelper{}, completer); auto docstring = parser.get_switch("docstring").value_or(StringView{});
cm.register_command(cmd_name, cmd, docstring.str(), desc, flags, CommandHelper{}, completer);
} }
const CommandDesc define_command_cmd = { const CommandDesc define_command_cmd = {
@ -809,12 +805,11 @@ const CommandDesc echo_cmd = {
[](const ParametersParser& parser, Context& context) [](const ParametersParser& parser, Context& context)
{ {
String message = join(parser, ' ', false); String message = join(parser, ' ', false);
if (parser.has_option("debug")) if (parser.get_switch("debug"))
write_debug(message); write_debug(message);
else else
{ {
auto face = get_face(parser.has_option("color") ? auto face = get_face(parser.get_switch("color").value_or("StatusLine").str());
parser.option_value("color") : "StatusLine");
context.print_status({ std::move(message), face } ); context.print_status({ std::move(message), face } );
} }
} }
@ -951,7 +946,7 @@ const CommandDesc set_option_cmd = {
[](const ParametersParser& parser, Context& context) [](const ParametersParser& parser, Context& context)
{ {
Option& opt = get_scope(parser[0], context).options().get_local_option(parser[1]); Option& opt = get_scope(parser[0], context).options().get_local_option(parser[1]);
if (parser.has_option("add")) if (parser.get_switch("add"))
opt.add_from_string(parser[2]); opt.add_from_string(parser[2]);
else else
opt.set_from_string(parser[2]); opt.set_from_string(parser[2]);
@ -984,13 +979,10 @@ const CommandDesc declare_option_cmd = {
Option* opt = nullptr; Option* opt = nullptr;
OptionFlags flags = OptionFlags::None; OptionFlags flags = OptionFlags::None;
if (parser.has_option("hidden")) if (parser.get_switch("hidden"))
flags = OptionFlags::Hidden; flags = OptionFlags::Hidden;
String docstring; auto docstring = parser.get_switch("docstring").value_or(StringView{}).str();
if (parser.has_option("docstring"))
docstring = parser.option_value("docstring");
OptionsRegistry& reg = GlobalScope::instance().option_registry(); OptionsRegistry& reg = GlobalScope::instance().option_registry();
if (parser[0] == "int") if (parser[0] == "int")
@ -1105,15 +1097,14 @@ void context_wrap(const ParametersParser& parser, Context& context, Func func)
DisableOption<bool> disable_autoshowcompl(context, "autoshowcompl"); DisableOption<bool> disable_autoshowcompl(context, "autoshowcompl");
DisableOption<bool> disable_incsearch(context, "incsearch"); DisableOption<bool> disable_incsearch(context, "incsearch");
const bool disable_hooks = parser.has_option("no-hooks") or const bool disable_hooks = parser.get_switch("no-hooks") or
context.user_hooks_support().is_disabled(); context.user_hooks_support().is_disabled();
const bool disable_keymaps = not parser.has_option("with-maps"); const bool disable_keymaps = not parser.get_switch("with-maps");
ClientManager& cm = ClientManager::instance(); ClientManager& cm = ClientManager::instance();
if (parser.has_option("buffer")) if (auto bufnames = parser.get_switch("buffer"))
{ {
auto names = split(parser.option_value("buffer"), ','); for (auto& name : split(*bufnames, ','))
for (auto& name : names)
{ {
Buffer& buffer = BufferManager::instance().get_buffer(name); Buffer& buffer = BufferManager::instance().get_buffer(name);
InputHandler input_handler{{ buffer, Selection{} }, InputHandler input_handler{{ buffer, Selection{} },
@ -1130,15 +1121,15 @@ void context_wrap(const ParametersParser& parser, Context& context, Func func)
} }
Context* real_context = &context; Context* real_context = &context;
if (parser.has_option("client")) if (auto client_name = parser.get_switch("client"))
real_context = &cm.get_client(parser.option_value("client")).context(); real_context = &cm.get_client(*client_name).context();
else if (parser.has_option("try-client")) else if (auto client_name = parser.get_switch("try-client"))
{ {
if (Client* client = cm.get_client_ifp(parser.option_value("try-client"))) if (Client* client = cm.get_client_ifp(*client_name))
real_context = &client->context(); real_context = &client->context();
} }
if (parser.has_option("draft")) if (parser.get_switch("draft"))
{ {
InputHandler input_handler(real_context->selections(), InputHandler input_handler(real_context->selections(),
Context::Flags::Transient, Context::Flags::Transient,
@ -1153,7 +1144,7 @@ void context_wrap(const ParametersParser& parser, Context& context, Func func)
ScopedDisable hook_disable(c.user_hooks_support(), disable_hooks); ScopedDisable hook_disable(c.user_hooks_support(), disable_hooks);
ScopedDisable keymaps_disable(c.keymaps_support(), disable_keymaps); ScopedDisable keymaps_disable(c.keymaps_support(), disable_keymaps);
if (parser.has_option("itersel")) if (parser.get_switch("itersel"))
{ {
SelectionList sels{real_context->selections()}; SelectionList sels{real_context->selections()};
ScopedEdition edition{c}; ScopedEdition edition{c};
@ -1173,7 +1164,7 @@ void context_wrap(const ParametersParser& parser, Context& context, Func func)
} }
else else
{ {
if (parser.has_option("itersel")) if (parser.get_switch("itersel"))
throw runtime_error("-itersel makes no sense without -draft"); throw runtime_error("-itersel makes no sense without -draft");
ScopedDisable hook_disable(real_context->user_hooks_support(), disable_hooks); ScopedDisable hook_disable(real_context->user_hooks_support(), disable_hooks);
@ -1246,13 +1237,10 @@ const CommandDesc prompt_cmd = {
throw runtime_error("register name should be a single character"); throw runtime_error("register name should be a single character");
const char reg = params[1][0_byte]; const char reg = params[1][0_byte];
const String& command = params[2]; const String& command = params[2];
auto initstr = params.get_switch("init").value_or(StringView{});
String initstr;
if (params.has_option("init"))
initstr = params.option_value("init");
context.input_handler().prompt( context.input_handler().prompt(
params[0], std::move(initstr), get_face("Prompt"), Completer{}, params[0], initstr.str(), get_face("Prompt"), Completer{},
[=](StringView str, PromptEvent event, Context& context) [=](StringView str, PromptEvent event, Context& context)
{ {
if (event != PromptEvent::Validate) if (event != PromptEvent::Validate)
@ -1278,14 +1266,14 @@ const CommandDesc menu_cmd = {
CommandCompleter{}, CommandCompleter{},
[](const ParametersParser& parser, Context& context) [](const ParametersParser& parser, Context& context)
{ {
const bool with_select_cmds = parser.has_option("select-cmds"); const bool with_select_cmds = (bool)parser.get_switch("select-cmds");
const size_t modulo = with_select_cmds ? 3 : 2; const size_t modulo = with_select_cmds ? 3 : 2;
const size_t count = parser.positional_count(); const size_t count = parser.positional_count();
if (count == 0 or (count % modulo) != 0) if (count == 0 or (count % modulo) != 0)
throw wrong_argument_count(); throw wrong_argument_count();
if (count == modulo and parser.has_option("auto-single")) if (count == modulo and parser.get_switch("auto-single"))
{ {
CommandManager::instance().execute(parser[1], context); CommandManager::instance().execute(parser[1], context);
return; return;
@ -1332,28 +1320,26 @@ const CommandDesc info_cmd = {
{ {
InfoStyle style = InfoStyle::Prompt; InfoStyle style = InfoStyle::Prompt;
CharCoord pos; CharCoord pos;
if (parser.has_option("anchor")) if (auto anchor = parser.get_switch("anchor"))
{ {
auto anchor = parser.option_value("anchor"); auto dot = find(*anchor, '.');
auto dot = find(anchor, '.'); if (dot == anchor->end())
if (dot == anchor.end())
throw runtime_error("expected <line>.<column> for anchor"); throw runtime_error("expected <line>.<column> for anchor");
ByteCoord coord{str_to_int({anchor.begin(), dot})-1, ByteCoord coord{str_to_int({anchor->begin(), dot})-1,
str_to_int({dot+1, anchor.end()})-1}; str_to_int({dot+1, anchor->end()})-1};
pos = context.window().display_position(coord); pos = context.window().display_position(coord);
style = InfoStyle::Inline; style = InfoStyle::Inline;
if (parser.has_option("placement")) if (auto placement = parser.get_switch("placement"))
{ {
auto placement = parser.option_value("placement"); if (*placement == "above")
if (placement == "above")
style = InfoStyle::InlineAbove; style = InfoStyle::InlineAbove;
else if (placement == "below") else if (*placement == "below")
style = InfoStyle::InlineBelow; style = InfoStyle::InlineBelow;
else else
throw runtime_error("invalid placement " + placement); throw runtime_error("invalid placement " + *placement);
} }
} }
const String& title = parser.has_option("title") ? parser.option_value("title") : ""; auto title = parser.get_switch("title").value_or(StringView{});
context.ui().info_show(title, parser[0], pos, get_face("Information"), style); context.ui().info_show(title, parser[0], pos, get_face("Information"), style);
} }
} }

View File

@ -1073,12 +1073,9 @@ public:
regions.push_back({ parser[i], {std::move(begin), std::move(end), std::move(recurse)} }); regions.push_back({ parser[i], {std::move(begin), std::move(end), std::move(recurse)} });
} }
String default_group;
if (parser.has_option("default"))
default_group = parser.option_value("default");
return {parser[0], make_unique<RegionsHighlighter>(std::move(regions), auto default_group = parser.get_switch("default").value_or(StringView{}).str();
std::move(default_group))}; return {parser[0], make_unique<RegionsHighlighter>(std::move(regions), default_group)};
} }
catch (RegexError& err) catch (RegexError& err)
{ {

View File

@ -534,56 +534,51 @@ int main(int argc, char* argv[])
ParametersParser parser(params, param_desc); ParametersParser parser(params, param_desc);
if (parser.has_option("p")) if (auto session = parser.get_switch("p"))
{ {
for (auto opt : { "c", "n", "s", "d", "e" }) for (auto opt : { "c", "n", "s", "d", "e" })
{ {
if (parser.has_option(opt)) if (parser.get_switch(opt))
{ {
fprintf(stderr, "error: -%s makes not sense with -p\n", opt); fprintf(stderr, "error: -%s makes not sense with -p\n", opt);
return -1; return -1;
} }
} }
return run_pipe(parser.option_value("p")); return run_pipe(*session);
} }
else if (parser.has_option("f")) else if (auto keys = parser.get_switch("f"))
{ {
Vector<StringView> files; Vector<StringView> files;
for (size_t i = 0; i < parser.positional_count(); ++i) for (size_t i = 0; i < parser.positional_count(); ++i)
files.emplace_back(parser[i]); files.emplace_back(parser[i]);
return run_filter(parser.option_value("f"), files, return run_filter(*keys, files, (bool)parser.get_switch("q"));
parser.has_option("q"));
} }
String init_command; auto init_command = parser.get_switch("e").value_or(StringView{});
if (parser.has_option("e"))
init_command = parser.option_value("e");
if (parser.has_option("c")) if (auto server_session = parser.get_switch("c"))
{ {
for (auto opt : { "n", "s", "d" }) for (auto opt : { "n", "s", "d" })
{ {
if (parser.has_option(opt)) if (parser.get_switch(opt))
{ {
fprintf(stderr, "error: -%s makes not sense with -c\n", opt); fprintf(stderr, "error: -%s makes not sense with -c\n", opt);
return -1; return -1;
} }
} }
return run_client(parser.option_value("c"), init_command); return run_client(*server_session, init_command);
} }
else else
{ {
Vector<StringView> files; Vector<StringView> files;
for (size_t i = 0; i < parser.positional_count(); ++i) for (size_t i = 0; i < parser.positional_count(); ++i)
files.emplace_back(parser[i]); files.emplace_back(parser[i]);
StringView session;
if (parser.has_option("s"))
session = parser.option_value("s");
StringView session = parser.get_switch("s").value_or(StringView{});
return run_server(session, init_command, return run_server(session, init_command,
parser.has_option("n"), (bool)parser.get_switch("n"),
parser.has_option("d"), (bool)parser.get_switch("d"),
files); files);
} }
} }

View File

@ -72,6 +72,9 @@ public:
} }
const T* operator->() const { return const_cast<Optional&>(*this).operator->(); } const T* operator->() const { return const_cast<Optional&>(*this).operator->(); }
template<typename U>
T value_or(U&& fallback) const { return m_valid ? m_value : T{fallback}; }
private: private:
void destruct_ifn() { if (m_valid) m_value.~T(); } void destruct_ifn() { if (m_valid) m_value.~T(); }

View File

@ -46,38 +46,20 @@ ParametersParser::ParametersParser(ParameterList params,
throw wrong_argument_count(); throw wrong_argument_count();
} }
bool ParametersParser::has_option(const String& name) const Optional<StringView> ParametersParser::get_switch(StringView name) const
{ {
kak_assert(m_desc.switches.find(name) != m_desc.switches.end()); auto it = m_desc.switches.find(name);
for (auto& param : m_params) kak_assert(it != m_desc.switches.end());
for (size_t i = 0; i < m_params.size(); ++i)
{ {
const auto& param = m_params[i];
if (param[0_byte] == '-' and param.substr(1_byte) == name) if (param[0_byte] == '-' and param.substr(1_byte) == name)
return true; return it->second.takes_arg ? m_params[i+1] : StringView{};
if (param == "--") if (param == "--")
break; break;
} }
return false; return {};
}
const String& ParametersParser::option_value(const String& name) const
{
#ifdef KAK_DEBUG
auto it = m_desc.switches.find(name);
kak_assert(it != m_desc.switches.end());
kak_assert(it->second.takes_arg);
#endif
for (size_t i = 0; i < m_params.size(); ++i)
{
if (m_params[i][0_byte] == '-' and m_params[i].substr(1_byte) == name)
return m_params[i+1];
if (m_params[i] == "--")
break;
}
static String empty;
return empty;
} }
} }

View File

@ -5,6 +5,7 @@
#include "id_map.hh" #include "id_map.hh"
#include "array_view.hh" #include "array_view.hh"
#include "flags.hh" #include "flags.hh"
#include "optional.hh"
#include "string.hh" #include "string.hh"
namespace Kakoune namespace Kakoune
@ -79,12 +80,9 @@ struct ParametersParser
// boolean option. // boolean option.
ParametersParser(ParameterList params, const ParameterDesc& desc); ParametersParser(ParameterList params, const ParameterDesc& desc);
// check if a named option (either string or boolean) is specified // Return a valid optional if the switch was given, with
bool has_option(const String& name) const; // a non empty StringView value if the switch took an argument.
Optional<StringView> get_switch(StringView name) const;
// get a string option value, returns an empty string if the option
// is not defined
const String& option_value(const String& name) const;
struct iterator : std::iterator<std::forward_iterator_tag, String> struct iterator : std::iterator<std::forward_iterator_tag, String>
{ {