108 lines
2.9 KiB
C++
108 lines
2.9 KiB
C++
#include "Cli.hpp"
|
|
|
|
Cli::Cli() {
|
|
add_to_commands_list("help, h", "Help Screen", std::bind(&Cli::print_help, &*this, std::placeholders::_1));
|
|
add_to_commands_list("exit", "Exit Program", std::bind(&Cli::exit_program, &*this, std::placeholders::_1));
|
|
}
|
|
|
|
void Cli::add_to_commands_list(std::string names, std::string desc, Func func) {
|
|
Command cmd = std::make_tuple(names, desc, func);
|
|
_commands.push_back(cmd);
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
//------------------------------------------- Run --------------------------------------------------
|
|
//--------------------------------------------------------------------------------------------------
|
|
|
|
void Cli::run() {
|
|
while(!_exit) { iterate(); }
|
|
}
|
|
|
|
void Cli::iterate() {
|
|
// Ask for input
|
|
std::cout << "\033[1;36mAegis: \033[0m";
|
|
std::string input;
|
|
std::cin >> input;
|
|
|
|
// Get parameters
|
|
std::deque<std::string> parameters;
|
|
std::string line;
|
|
std::getline(std::cin, line);
|
|
std::istringstream iss(line);
|
|
std::string arg;
|
|
while (iss >> arg) {
|
|
parameters.push_back(arg);
|
|
}
|
|
|
|
// Call matching function
|
|
Func todo = get_function(input);
|
|
todo(parameters);
|
|
}
|
|
|
|
Func Cli::get_function(std::string input) {
|
|
for (Command cmd : _commands) {
|
|
// Split command names from _commands
|
|
std::vector<std::string> names;
|
|
boost::split(names, std::get<0>(cmd), boost::is_any_of(", "));
|
|
|
|
// Check if input matches any of the names
|
|
for (std::string name : names) {
|
|
if (name == input) {
|
|
return std::get<2>(cmd);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Default: Basic help command
|
|
Cli cli;
|
|
return std::bind(&Cli::default_help, cli, std::placeholders::_1);
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
//--------------------------------------- Local Methods --------------------------------------------
|
|
//--------------------------------------------------------------------------------------------------
|
|
|
|
void Cli::default_help(Args args) {
|
|
print("Try 'help'");
|
|
}
|
|
|
|
void Cli::exit_program(Args args) {
|
|
_exit = true;
|
|
}
|
|
|
|
void Cli::print_help(Args args) {
|
|
print("Commands:\n");
|
|
if (_commands.size() == 0) {
|
|
print("Expty.");
|
|
return;
|
|
}
|
|
|
|
// determine max length of first elements
|
|
int max_size = 0;
|
|
for (auto& c : _commands) {
|
|
int len = std::get<0>(c).length();
|
|
max_size = (len > max_size) ? len : max_size;
|
|
}
|
|
|
|
// print every available command
|
|
for (auto& c : _commands) {
|
|
int spaces_to_add = max_size - std::get<0>(c).length();
|
|
if (spaces_to_add < 0 || spaces_to_add > 50) { throw "Internal fatal error."; }
|
|
|
|
|
|
std::string command_text;
|
|
command_text += std::get<0>(c);
|
|
for (int i = 0; i < spaces_to_add; i++) { command_text += " "; }
|
|
command_text += " " + std::get<1>(c);
|
|
print(command_text);
|
|
}
|
|
}
|
|
|
|
void Cli::syn(Args args) {
|
|
if (!args.empty()) {
|
|
print("SYN " + args[0]);
|
|
} else {
|
|
print("SYN");
|
|
}
|
|
}
|