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 <vector>
  30 #include <string>
  31 #include <cstring>
  32 #include <cstdio>
  33 #include <stdlib.h>
  34 #include <unistd.h>
  35
  36 #ifdef HAVE_READLINE
  37 #include <readline/readline.h>
  38 #include <readline/history.h>
  39 #endif
  40
  41 std::string promptCompose ();
  42
  43 const std::string getResponse (const std::string& prompt)
  44 {
  45   std::string response {""};
  46
  47   // Display prompt, get input.
  48 #ifdef HAVE_READLINE
  49   char *line_read = readline (prompt.c_str ());
  50   if (! line_read)
  51   {
  52     std::cout << "\n";
  53     response = "<EOF>";
  54   }
  55   else
  56   {
  57     // Save history.
  58     if (*line_read)
  59       add_history (line_read);
  60
  61     response = std::string (line_read);
  62     free (line_read);
  63   }
  64 #else
  65   std::cout << prompt;
  66   std::getline (std::cin, response);
  67   if (std::cin.eof () == 1)
  68   {
  69     std::cout << "\n";
  70     response = "<EOF>";
  71   }
  72 #endif
  73
  74   return response;
  75 }
  76
  77 ////////////////////////////////////////////////////////////////////////////////
  78 static int commandLoop (bool autoClear)
  79 {
  80   // Compose the prompt.
  81   auto prompt = promptCompose ();
  82
  83   // Display prompt, get input.
  84   auto command = getResponse (prompt);
  85
  86   // Obey Taskwarrior's rc.tasksh.autoclear.
  87   if (autoClear)
  88     std::cout << "\033[2J\033[0;0H";
  89
  90   int status = 0;
  91   if (! isatty (fileno (stdin)) && command == "")
  92   {
  93     status = -1;
  94   }
  95   else if (command != "")
  96   {
  97     // Dispatch command.
  98     if (command == "<EOF>")                      status = -1;
  99     else if (command != "")
 100     {
 101       command = command;
 102       std::cout << "[" << command << "]\n";
 103       system (command.c_str ());
 104     }
 105   }
 106
 107   return status;
 108 }
 109
 110 ////////////////////////////////////////////////////////////////////////////////
 111 int main (int argc, const char** argv)
 112 {
 113   int status = 0;
 114
 115   // Lightweight version checking that doesn't require initialization or any I/O.
 116   if (argc == 2 && !strcmp (argv[1], "--version"))
 117   {
 118     std::cout << VERSION << "\n";
 119   }
 120   else
 121   {
 122     try
 123     {
 124       bool autoClear = false;
 125       std::string input;
 126       std::string output;
 127       autoClear = (output == "true\n" ||
 128                    output == "1\n"    ||
 129                    output == "y\n"    ||
 130                    output == "yes\n"  ||
 131                    output == "on\n");
 132
 133       while ((status = commandLoop (autoClear)) == 0)
 134         ;
 135     }
 136
 137     catch (const std::string& error)
 138     {
 139       std::cerr << error << "\n";
 140       status = -1;
 141     }
 142
 143     catch (...)
 144     {
 145       std::cerr << "Unknown error." << "\n";
 146       status = -2;
 147     }
 148   }
 149
 150   // Returning -1 drops out of the command loop, but gets translated to 0 here,
 151   // so that there is a clean way to exit.
 152   return status == -1 ? 0 : status;
 153 }
 154
 155 ////////////////////////////////////////////////////////////////////////////////