#include <cmake.h>
#include <iostream>
-#include <vector>
-#include <string>
-#include <cstring>
-#include <cstdio>
+#include <../Sarge/src/sarge.h>
#include <stdlib.h>
-#include <unistd.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);
return response;
}
+////////////////////////////////////////////////////////////////////////////////
+
int main(int argc, char** argv)
{
- int status = 0;
- string root_cmd;
- for (int i = 1; i < argc; ++i) {
- root_cmd +=argv[i];
- }
+ // Command line arguments
+ Sarge sarge;
+
+ 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 1;
+ }
+
+ if (sarge.exists("help")) {
+ sarge.printHelp();
+ return 0;
+ }
- while (status == 0) {
- // Compose the prompt.
- auto prompt = promptCompose();
+ bool space;
+ if (sarge.exists("no-space")) {
+ space = false;
+ }
- // Display prompt, get input.
+ // Define input command
+ string 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) {
+ // Display prompt, get input
auto command = getResponse(prompt);
- int status = 0;
if (command != "")
{
- // Dispatch command.
- if (command == "<EOF>") status = 1;
+ // Dispatch command
+ if (command == "<EOF>")
+ {
+ system (after_command.c_str ());
+ return 0;
+ }
else if (command != "")
{
- if (argc == 0) {
- string whole_command=command;
- std::cout << "[" << command << "]\n";
- system (command.c_str ());
- }
- else {
- string whole_command = root_cmd + " " + command;
- std::cout << "[" << whole_command << "]\n";
- system (whole_command.c_str ());
- }
+ string whole_command = arg_cmd + command;
+ system (whole_command.c_str ());
}
}
- if (status == 1)
- return 0;
}
}