#include <stdlib.h>
#ifdef HAVE_READLINE
-#include <readline/readline.h>
-#include <readline/history.h>
+ #include <readline/readline.h>
+ #include <readline/history.h>
#endif
////////////////////////////////////////////////////////////////////////////////
using namespace std;
-std::string promptCompose();
const std::string getResponse(const std::string & prompt) {
std::string response {
""
};
- // Display prompt, get input.
+ // Display prompt, get input
#ifdef HAVE_READLINE
char * line_read = readline(prompt.c_str());
if (!line_read) {
std::cout << "\n";
response = "<EOF>";
} else {
- // Save history.
+ // Save history
if ( * line_read)
add_history(line_read);
// Command line arguments
Sarge sarge;
- sarge.setArgument("h", "help", "Get help.", false);
- sarge.setArgument("c", "cmd", "Command to execute before entering the shell", true);
- sarge.setDescription("Make a shell from any command");
- sarge.setUsage("gen-shell <options> <command>");
+ sarge.setArgument("a", "after", "Command to execute before leaving the shell", true);
+ sarge.setArgument("b", "before", "Command to execute before entering the shell", true);
+ sarge.setArgument("c", "command", "Command to convert to shell", true);
+ sarge.setArgument("h", "help", "Get help.", false);
+ sarge.setArgument("p", "prompt", "Define a custom prompt", true);
+ sarge.setArgument("", "no-space", "Dont automatically add spaces after custom prompt and command", false);
+ sarge.setDescription("Make a shell from any executable");
+ sarge.setUsage("gen-shell <options>");
if (!sarge.parseArguments(argc, argv)) {
std::cerr << "Couldn't parse arguments..." << std::endl;
return 0;
}
+ bool space;
+ if (sarge.exists("no-space")) {
+ space = false;
+ }
+
+ // Define input command
string arg_cmd;
- sarge.getFlag("cmd", arg_cmd);
- arg_cmd += " ";
+ sarge.getFlag("command", arg_cmd);
+ if ( space )
+ arg_cmd += " ";
+
+ // Define prompt
+ string prompt = "";
+ sarge.getFlag("prompt", prompt);
+
+ if ( prompt == "" )
+ prompt = "% ";
+ if ( space )
+ prompt += " ";
+
+ // Execute before-command
+ string before_command;
+ sarge.getFlag("before", before_command);
+ system (before_command.c_str ());
+
+ // Execute after-command
+ string after_command;
+ sarge.getFlag("after", after_command);
// Main program
while (true) {
- // Compose the prompt.
- auto prompt = promptCompose();
-
- // Display prompt, get input.
+ // Display prompt, get input
auto command = getResponse(prompt);
if (command != "")
{
- // Dispatch command.
+ // Dispatch command
if (command == "<EOF>")
{
+ system (after_command.c_str ());
return 0;
}
else if (command != "")