src/main.cpp

   1 ////////////////////////////////////////////////////////////////////////////////
   2 //
   3 // Copyright 2006 - 2017, Paul Beckingham, Federico Hernandez, 2020 Armaan Bhojwani
   4 //
   5 // Permission is hereby granted, free of charge, to any person obtaining a copy
   6 // of this software and associated documentation files (the "Software"), to deal
   7 // in the Software without restriction, including without limitation the rights
   8 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   9 // copies of the Software, and to permit persons to whom the Software is
  10 // furnished to do so, subject to the following conditions:
  11 //
  12 // The above copyright notice and this permission notice shall be included
  13 // in all copies or substantial portions of the Software.
  14 //
  15 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  16 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  18 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21 // SOFTWARE.
  22 //
  23 // http://www.opensource.org/licenses/mit-license.php
  24 //
  25 ////////////////////////////////////////////////////////////////////////////////
  26
  27 #include <cmake.h>
  28 #include <iostream>
  29 #include <../Sarge/src/sarge.h>
  30 #include <stdlib.h>
  31
  32 #ifdef HAVE_READLINE
  33   #include <readline/readline.h>
  34   #include <readline/history.h>
  35 #endif
  36
  37 ////////////////////////////////////////////////////////////////////////////////
  38
  39 using namespace std;
  40
  41 const std::string getResponse(const std::string & prompt) {
  42   std::string response {
  43     ""
  44   };
  45
  46   // Display prompt, get input
  47   #ifdef HAVE_READLINE
  48   char * line_read = readline(prompt.c_str());
  49   if (!line_read) {
  50     std::cout << "\n";
  51     response = "<EOF>";
  52   } else {
  53     // Save history
  54     if ( * line_read)
  55       add_history(line_read);
  56
  57     response = std::string(line_read);
  58     free(line_read);
  59   }
  60   #else
  61   std::cout << prompt;
  62   std::getline(std::cin, response);
  63   if (std::cin.eof() == 1) {
  64     std::cout << "\n";
  65     response = "<EOF>";
  66   }
  67   #endif
  68
  69   return response;
  70 }
  71
  72 ////////////////////////////////////////////////////////////////////////////////
  73
  74 int main(int argc, char** argv)
  75 {
  76
  77   // Command line arguments
  78   Sarge sarge;
  79
  80         sarge.setArgument("a", "after", "Command to execute before leaving the shell", true);
  81   sarge.setArgument("b", "before", "Command to execute before entering the shell", true);
  82   sarge.setArgument("c", "command", "Command to convert to shell", true);
  83   sarge.setArgument("h", "help", "Get help.", false);
  84   sarge.setArgument("p", "prompt", "Define a custom prompt", true);
  85   sarge.setArgument("", "no-space", "Dont automatically add spaces after custom prompt and command", false);
  86   sarge.setDescription("Make a shell from any executable");
  87   sarge.setUsage("gen-shell <options>");
  88
  89         if (!sarge.parseArguments(argc, argv)) {
  90                 std::cerr << "Couldn't parse arguments..." << std::endl;
  91                 return 1;
  92         }
  93
  94         if (sarge.exists("help")) {
  95                 sarge.printHelp();
  96     return 0;
  97         }
  98
  99   bool space;
 100   if (sarge.exists("no-space")) {
 101     space = false;
 102         }
 103
 104   // Define input command
 105   string arg_cmd;
 106   sarge.getFlag("command", arg_cmd);
 107   if ( space )
 108     arg_cmd += " ";
 109
 110   // Define prompt
 111   string prompt = "";
 112   sarge.getFlag("prompt", prompt);
 113
 114   if ( prompt == "" )
 115     prompt = "% ";
 116   if ( space )
 117     prompt += " ";
 118
 119   // Execute before-command
 120   string before_command;
 121   sarge.getFlag("before", before_command);
 122   system (before_command.c_str ());
 123
 124   // Execute after-command
 125   string after_command;
 126   sarge.getFlag("after", after_command);
 127
 128   // Main program
 129   while (true) {
 130     // Display prompt, get input
 131     auto command = getResponse(prompt);
 132
 133     if (command != "")
 134     {
 135       // Dispatch command
 136       if (command == "<EOF>")
 137       {
 138         system (after_command.c_str ());
 139         return 0;
 140       }
 141       else if (command != "")
 142       {
 143         string whole_command = arg_cmd + command;
 144         system (whole_command.c_str ());
 145       }
 146     }
 147   }
 148 }
 149
 150 ////////////////////////////////////////////////////////////////////////////////