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 std::string promptCompose();
  41
  42 const std::string getResponse(const std::string & prompt) {
  43   std::string response {
  44     ""
  45   };
  46
  47   // Display prompt, get input.
  48   #ifdef HAVE_READLINE
  49   char * line_read = readline(prompt.c_str());
  50   if (!line_read) {
  51     std::cout << "\n";
  52     response = "<EOF>";
  53   } else {
  54     // Save history.
  55     if ( * line_read)
  56       add_history(line_read);
  57
  58     response = std::string(line_read);
  59     free(line_read);
  60   }
  61   #else
  62   std::cout << prompt;
  63   std::getline(std::cin, response);
  64   if (std::cin.eof() == 1) {
  65     std::cout << "\n";
  66     response = "<EOF>";
  67   }
  68   #endif
  69
  70   return response;
  71 }
  72
  73 ////////////////////////////////////////////////////////////////////////////////
  74
  75 int main(int argc, char** argv)
  76 {
  77
  78   // Command line arguments
  79   Sarge sarge;
  80
  81         sarge.setArgument("h", "help", "Get help.", false);
  82         sarge.setArgument("c", "cmd", "Command to execute before entering the shell", true);
  83         sarge.setDescription("Make a shell from any command");
  84         sarge.setUsage("gen-shell <options> <command>");
  85
  86         if (!sarge.parseArguments(argc, argv)) {
  87                 std::cerr << "Couldn't parse arguments..." << std::endl;
  88                 return 1;
  89         }
  90
  91         if (sarge.exists("help")) {
  92                 sarge.printHelp();
  93     return 0;
  94         }
  95
  96   string arg_cmd;
  97   sarge.getFlag("cmd", arg_cmd);
  98   arg_cmd += " ";
  99
 100   // Main program
 101   while (true) {
 102     // Compose the prompt.
 103     auto prompt = promptCompose();
 104
 105     // Display prompt, get input.
 106     auto command = getResponse(prompt);
 107
 108     if (command != "")
 109     {
 110       // Dispatch command.
 111       if (command == "<EOF>")
 112       {
 113         return 0;
 114       }
 115       else if (command != "")
 116       {
 117         string whole_command = arg_cmd + command;
 118         system (whole_command.c_str ());
 119       }
 120     }
 121   }
 122 }
 123
 124 ////////////////////////////////////////////////////////////////////////////////