blob: 91b00a2165253ac2a15dc7cea116db6b2204632f [file] [log] [blame]
/*
* Copyright (C) 2018 CESNET, https://photonics.cesnet.cz/
* Copyright (C) 2018 FIT CVUT, https://fit.cvut.cz/
*
* Written by Václav Kubernát <kubervac@fit.cvut.cz>
*
*/
#include <ostream>
#include "grammars.hpp"
#include "parser.hpp"
#include "parser_context.hpp"
TooManyArgumentsException::~TooManyArgumentsException() = default;
InvalidCommandException::~InvalidCommandException() = default;
Parser::Parser(const std::shared_ptr<const Schema> schema, const std::shared_ptr<const DataQuery> dataQuery)
: m_schema(schema)
, m_dataquery(dataQuery)
{
m_curDir.m_scope = Scope::Absolute;
}
bool Completions::operator==(const Completions& b) const
{
return this->m_completions == b.m_completions && this->m_contextLength == b.m_contextLength;
}
command_ Parser::parseCommand(const std::string& line, std::ostream& errorStream)
{
command_ parsedCommand;
ParserContext ctx(*m_schema, nullptr, m_curDir);
auto it = line.begin();
boost::spirit::x3::error_handler<std::string::const_iterator> errorHandler(it, line.end(), errorStream);
auto grammar =
x3::with<parser_context_tag>(ctx)[
x3::with<x3::error_handler_tag>(std::ref(errorHandler))[command]
];
bool result = x3::phrase_parse(it, line.end(), grammar, space, parsedCommand);
if (!result || it != line.end()) {
throw InvalidCommandException(std::string(it, line.end()) + " this was left of input");
}
return parsedCommand;
}
Completions Parser::completeCommand(const std::string& line, std::ostream& errorStream) const
{
std::set<std::string> completions;
command_ parsedCommand;
ParserContext ctx(*m_schema, m_dataquery, m_curDir);
ctx.m_completing = true;
auto it = line.begin();
boost::spirit::x3::error_handler<std::string::const_iterator> errorHandler(it, line.end(), errorStream);
auto grammar =
x3::with<parser_context_tag>(ctx)[
x3::with<x3::error_handler_tag>(std::ref(errorHandler))[command]
];
x3::phrase_parse(it, line.end(), grammar, space, parsedCommand);
auto completionIterator = ctx.m_completionIterator ? *ctx.m_completionIterator : line.end();
int completionContext = line.end() - completionIterator;
auto filtered = filterByPrefix(ctx.m_suggestions, std::string(completionIterator, line.end()));
if (filtered.size() == 1) {
auto suffix = filtered.begin()->m_whenToAdd == Completion::WhenToAdd::Always
|| filtered.begin()->m_value == std::string{completionIterator, line.end()}
? filtered.begin()->m_suffix
: "";
return {{filtered.begin()->m_value + suffix}, completionContext};
}
std::set<std::string> res;
std::transform(filtered.begin(), filtered.end(), std::inserter(res, res.end()), [](auto it) { return it.m_value; });
return {res, completionContext};
}
void Parser::changeNode(const dataPath_& name)
{
if (name.m_scope == Scope::Absolute) {
m_curDir = name;
} else {
for (const auto& it : name.m_nodes) {
if (it.m_suffix.type() == typeid(nodeup_))
m_curDir.m_nodes.pop_back();
else
m_curDir.m_nodes.push_back(it);
}
}
}
std::string Parser::currentNode() const
{
return pathToDataString(m_curDir, Prefixes::WhenNeeded);
}
struct getSchemaPathVisitor : boost::static_visitor<schemaPath_> {
schemaPath_ operator()(const dataPath_& path) const
{
return dataPathToSchemaPath(path);
}
schemaPath_ operator()(const schemaPath_& path) const
{
return path;
}
};
std::set<std::string> Parser::availableNodes(const boost::optional<boost::variant<boost::variant<dataPath_, schemaPath_>, module_>>& path, const Recursion& option) const
{
auto pathArg = dataPathToSchemaPath(m_curDir);
if (path) {
if (path->type() == typeid(module_)) {
return m_schema->moduleNodes(boost::get<module_>(*path), option);
}
auto schemaPath = boost::apply_visitor(getSchemaPathVisitor(), boost::get<boost::variant<dataPath_, schemaPath_>>(*path));
if (schemaPath.m_scope == Scope::Absolute) {
pathArg = schemaPath;
} else {
pathArg.m_nodes.insert(pathArg.m_nodes.end(), schemaPath.m_nodes.begin(), schemaPath.m_nodes.end());
}
}
return m_schema->childNodes(pathArg, option);
}