Add echo command and unit test for command parser

This commit is contained in:
tangxifan 2020-01-20 20:31:16 -07:00
parent 3ae80a192f
commit 7a5b36fe52
6 changed files with 95 additions and 41 deletions

View File

@ -2,13 +2,13 @@ cmake_minimum_required(VERSION 3.9)
project("libminishell")
#file(GLOB_RECURSE EXEC_SOURCES test/main.cpp)
file(GLOB_RECURSE EXEC_SOURCES test/test_command_parser.cpp)
file(GLOB_RECURSE LIB_SOURCES src/*.cpp)
file(GLOB_RECURSE LIB_HEADERS src/*.h)
files_to_dirs(LIB_HEADERS LIB_INCLUDE_DIRS)
#Remove test executable from library
#list(REMOVE_ITEM LIB_SOURCES ${EXEC_SOURCES})
list(REMOVE_ITEM LIB_SOURCES ${EXEC_SOURCES})
#Create the library
add_library(libminishell STATIC
@ -22,8 +22,8 @@ target_link_libraries(libminishell
libvtrutil)
#Create the test executable
#add_executable(read_arch_openfpga ${EXEC_SOURCES})
#target_link_libraries(read_arch_openfpga libarchopenfpga)
add_executable(test_command_parser ${EXEC_SOURCES})
target_link_libraries(test_command_parser libminishell)
#Supress IPO link warnings if IPO is enabled
#get_target_property(READ_ARCH_USES_IPO read_arch_openfpga INTERPROCEDURAL_OPTIMIZATION)

View File

@ -10,8 +10,8 @@ namespace minishell {
/*********************************************************************
* Public constructors
********************************************************************/
Command::Command(const std::string& name) {
name_ = name;
Command::Command(const char* name) {
name_ = std::string(name);
}
/************************************************************************
@ -48,7 +48,7 @@ std::vector<CommandOptionId> Command::require_value_options() const {
CommandOptionId Command::option(const std::string& name) const {
/* Ensure that the name is unique in the option list */
std::map<std::string, CommandOptionId>::const_iterator name_it = option_name2ids_.find(name);
if (name_it != option_name2ids_.end()) {
if (name_it == option_name2ids_.end()) {
return CommandOptionId::INVALID();
}
return option_name2ids_.at(name);
@ -57,7 +57,7 @@ CommandOptionId Command::option(const std::string& name) const {
CommandOptionId Command::short_option(const std::string& name) const {
/* Ensure that the name is unique in the option list */
std::map<std::string, CommandOptionId>::const_iterator name_it = option_short_name2ids_.find(name);
if (name_it != option_short_name2ids_.end()) {
if (name_it == option_short_name2ids_.end()) {
return CommandOptionId::INVALID();
}
return option_short_name2ids_.at(name);
@ -104,11 +104,11 @@ std::string Command::option_description(const CommandOptionId& option_id) const
***********************************************************************/
/* Add an option without required values */
CommandOptionId Command::add_option(const std::string& name,
CommandOptionId Command::add_option(const char* name,
const bool& option_required,
const std::string& description) {
const char* description) {
/* Ensure that the name is unique in the option list */
std::map<std::string, CommandOptionId>::const_iterator name_it = option_name2ids_.find(name);
std::map<std::string, CommandOptionId>::const_iterator name_it = option_name2ids_.find(std::string(name));
if (name_it != option_name2ids_.end()) {
return CommandOptionId::INVALID();
}
@ -116,40 +116,40 @@ CommandOptionId Command::add_option(const std::string& name,
/* This is a legal name. we can create a new id */
CommandOptionId option = CommandOptionId(option_ids_.size());
option_ids_.push_back(option);
option_names_.push_back(name);
option_names_.push_back(std::string(name));
option_short_names_.emplace_back();
option_required_.push_back(option_required);
option_require_value_types_.push_back(NUM_OPT_VALUE_TYPES);
option_description_.push_back(description);
option_description_.push_back(std::string(description));
/* Register the name and short name in the name2id map */
option_name2ids_[name] = option;
option_name2ids_[std::string(name)] = option;
return option;
}
/* Add a short name to an option */
bool Command::set_option_short_name(const CommandOptionId& option_id,
const std::string& short_name) {
const char* short_name) {
/* Validate the option id */
VTR_ASSERT(true == valid_option_id(option_id));
/* Short name is optional, so do the following only when it is not empty
* Ensure that the short name is unique in the option list
*/
if (true == short_name.empty()) {
if (true == std::string(short_name).empty()) {
return false;
}
std::map<std::string, CommandOptionId>::const_iterator short_name_it = option_short_name2ids_.find(short_name);
std::map<std::string, CommandOptionId>::const_iterator short_name_it = option_short_name2ids_.find(std::string(short_name));
if (short_name_it != option_short_name2ids_.end()) {
return false;
}
option_short_names_[option_id] = short_name;
option_short_names_[option_id] = std::string(short_name);
/* Short name is optional, so register it only when it is not empty */
option_short_name2ids_[short_name] = option_id;
option_short_name2ids_[std::string(short_name)] = option_id;
return true;
}

View File

@ -51,7 +51,7 @@ class Command {
/* Create range */
typedef vtr::Range<command_option_iterator> command_option_range;
public: /* Constructor */
Command(const std::string& name);
Command(const char* name);
public: /* Public accessors */
std::string name() const;
command_option_range options() const;
@ -68,10 +68,10 @@ class Command {
e_option_value_type option_require_value_type(const CommandOptionId& option_id) const;
std::string option_description(const CommandOptionId& option_id) const;
public: /* Public mutators */
CommandOptionId add_option(const std::string& name,
CommandOptionId add_option(const char* name,
const bool& option_required,
const std::string& description);
bool set_option_short_name(const CommandOptionId& option_id, const std::string& short_name);
const char* description);
bool set_option_short_name(const CommandOptionId& option_id, const char* short_name);
void set_option_require_value(const CommandOptionId& option_id,
const e_option_value_type& option_require_value_type);
public: /* Public validators */

View File

@ -14,13 +14,13 @@ namespace minishell {
* This function is mainly used to create help desk for a command
********************************************************************/
void print_command_options(const Command& cmd) {
VTR_LOG("Command '%s' usage:",
VTR_LOG("Command '%s' usage:\n",
cmd.name().c_str());
for (const CommandOptionId& opt : cmd.options()) {
VTR_LOG("%s, %s : %s",
VTR_LOG("--%s, -%s : %s\n",
cmd.option_name(opt).c_str(),
cmd.option_short_name(opt).c_str(),
cmd.option_description(opt));
cmd.option_description(opt).c_str());
}
}
@ -31,23 +31,29 @@ void print_command_options(const Command& cmd) {
********************************************************************/
void print_command_context(const Command& cmd,
const CommandContext& cmd_context) {
VTR_LOG("Confirm selected options when call command '%s':",
VTR_LOG("Confirm selected options when call command '%s':\n",
cmd.name().c_str());
for (const CommandOptionId& opt : cmd.options()) {
if (false == cmd.option_require_value(opt)) {
VTR_LOG("%s, %s : %s",
VTR_LOG("--%s, -%s : %s\n",
cmd.option_name(opt).c_str(),
cmd.option_short_name(opt).c_str(),
cmd_context.option_enable(cmd, opt) ? "on" : "off");
} else {
VTR_ASSERT_SAFE (true == cmd.option_require_value(opt));
VTR_LOG("%s, %s : %s",
/* If the option is disabled, we output an off */
std::string opt_value;
if (false == cmd_context.option_enable(cmd, opt)) {
opt_value = std::string("off");
} else {
opt_value = cmd_context.option_value(cmd, opt);
}
VTR_LOG("--%s, -%s : %s\n",
cmd.option_name(opt).c_str(),
cmd.option_short_name(opt).c_str(),
cmd_context.option_value(cmd, opt).c_str());
opt_value.c_str());
}
}
}
} /* End namespace minshell */

View File

@ -23,7 +23,7 @@ CommandOptionId parse_option(const std::string& argv,
CommandOptionId option_id = cmd.option(argv);
/* Not found, error out */
if (CommandOptionId::INVALID() == option_id) {
VTR_LOG("Detect unknown option '%s'!\n",
VTR_LOG("Detect unknown option '--%s'!\n",
argv.c_str());
}
/* Found, update the CommandContext */
@ -43,7 +43,7 @@ CommandOptionId parse_short_option(const std::string& argv,
CommandOptionId option_id = cmd.short_option(argv);
/* Not found, error out */
if (CommandOptionId::INVALID() == option_id) {
VTR_LOG("Detect unknown option '%s'!\n",
VTR_LOG("Detect unknown short option '-%s'!\n",
argv.c_str());
}
/* Found, update the CommandContext */
@ -63,8 +63,8 @@ bool parse_command(const std::vector<std::string>& argv,
VTR_ASSERT(1 <= argv.size());
/* Validate that the command name matches argv[0] */
if (argv[0] == cmd.name()) {
VTR_LOG("Unexpected command name '%s'!",
if (argv[0] != cmd.name()) {
VTR_LOG("Unexpected command name '%s'!\n",
argv[0]);
return false;
}
@ -73,7 +73,7 @@ bool parse_command(const std::vector<std::string>& argv,
for (size_t iarg = 1; iarg < argv.size(); ++iarg) {
/* Option must start with dash */
if (0 != strncmp("-", argv[iarg].c_str(), 1)) {
VTR_LOG("Invalid option '%s'!",
VTR_LOG("Invalid option '%s'!\n",
argv[iarg].c_str());
return false;
}
@ -136,19 +136,25 @@ bool parse_command(const std::vector<std::string>& argv,
std::vector<CommandOptionId> missing_options = cmd_context.check_required_options(cmd);
if (!missing_options.empty()) {
for (const CommandOptionId& missing_opt : missing_options) {
VTR_LOG("Required option '%s' is missing!",
cmd.option_name(missing_opt));
VTR_LOG("Required option '%s' is missing!\n",
cmd.option_name(missing_opt).c_str());
}
return false;
}
std::vector<CommandOptionId> missing_value_options = cmd_context.check_required_option_values(cmd);
if (!missing_value_options.empty()) {
bool parse_fail = false;
for (const CommandOptionId& missing_opt : missing_value_options) {
VTR_LOG("Require value for option '%s' is missing!",
cmd.option_name(missing_opt));
/* If the missing option is not enabled, we can skip this option */
if (false == cmd_context.option_enable(cmd, missing_opt)) {
continue;
}
return false;
VTR_LOG("Require value for option '%s' is missing!\n",
cmd.option_name(missing_opt).c_str());
parse_fail = true;
}
return !parse_fail;
}
return true;

View File

@ -0,0 +1,42 @@
/********************************************************************
* Test the command parser by pre-defining a simple command
*******************************************************************/
#include "command_parser.h"
#include "command_echo.h"
using namespace minishell;
int main(int argc, char** argv) {
/* Create a new command */
Command cmd("read_openfpga_arch");
/* Add options to the command */
/* Option to specify architecture file */
CommandOptionId opt_arch_file = cmd.add_option("file", true, "Specify the OpenFPGA architecture file");
cmd.set_option_require_value(opt_arch_file, OPT_STRING);
cmd.set_option_short_name(opt_arch_file, "f");
CommandOptionId opt_echo_arch = cmd.add_option("echo", false, "Echo the parsed result to file");
cmd.set_option_require_value(opt_echo_arch, OPT_STRING);
cmd.set_option_short_name(opt_echo_arch, "e");
CommandOptionId opt_help = cmd.add_option("help", false, "Help desk");
cmd.set_option_short_name(opt_help, "h");
/* Parse the option, to avoid issues, we use the command name to replace the argv[0] */
std::vector<std::string> cmd_opts;
cmd_opts.push_back(cmd.name());
for (int iarg = 1; iarg < argc; ++iarg) {
cmd_opts.push_back(std::string(argv[iarg]));
}
CommandContext cmd_context(cmd);
if (false == parse_command(cmd_opts, cmd, cmd_context)) {
/* Echo the command */
print_command_options(cmd);
} else {
/* Let user to confirm selected options */
print_command_context(cmd, cmd_context);
}
return 0;
}