src/main.cpp

   1 ////////////////////////////////////////////////////////////////////////////////
   2 //
   3 // Copyright 2006 - 2017, Paul Beckingham, Federico Hernandez.
   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 #include <shared.h>
  36
  37 #ifdef HAVE_READLINE
  38 #include <readline/readline.h>
  39 #include <readline/history.h>
  40 #endif
  41
  42 // TODO These conflict with tw commands. This needs to be resolved.
  43 //      Perhaps an escape, such as '-- help' could invoke local help, or using
  44 //      a 'task' prefix could disambiguate.
  45
  46 // tasksh commands.
  47 int cmdHelp ();
  48 int cmdDiagnostics ();
  49 int cmdReview (const std::vector <std::string>&, bool);
  50 int cmdShell (const std::vector <std::string>&);
  51 std::string promptCompose ();
  52 std::string findTaskwarrior ();
  53
  54 ////////////////////////////////////////////////////////////////////////////////
  55 static void welcome ()
  56 {
  57   std::cout << PACKAGE_STRING << "\n";
  58   cmdHelp ();
  59 }
  60
  61 ////////////////////////////////////////////////////////////////////////////////
  62 const std::string getResponse (const std::string& prompt)
  63 {
  64   std::string response {""};
  65
  66   // Display prompt, get input.
  67 #ifdef HAVE_READLINE
  68   char *line_read = readline (prompt.c_str ());
  69   if (! line_read)
  70   {
  71     std::cout << "\n";
  72     response = "<EOF>";
  73   }
  74   else
  75   {
  76     // Save history.
  77     if (*line_read)
  78       add_history (line_read);
  79
  80     response = std::string (line_read);
  81     free (line_read);
  82   }
  83 #else
  84   std::cout << prompt;
  85   std::getline (std::cin, response);
  86   if (std::cin.eof () == 1)
  87   {
  88     std::cout << "\n";
  89     response = "<EOF>";
  90   }
  91 #endif
  92
  93   return response;
  94 }
  95
  96 ////////////////////////////////////////////////////////////////////////////////
  97 static int commandLoop (bool autoClear)
  98 {
  99   // Compose the prompt.
 100   auto prompt = promptCompose ();
 101
 102   // Display prompt, get input.
 103   auto command = getResponse (prompt);
 104
 105   // Obey Taskwarrior's rc.tasksh.autoclear.
 106   if (autoClear)
 107     std::cout << "\033[2J\033[0;0H";
 108
 109   int status = 0;
 110   if (! isatty (fileno (stdin)) && command == "")
 111   {
 112     status = -1;
 113   }
 114   else if (command != "")
 115   {
 116     auto args = split (command, ' ');
 117
 118     // Dispatch command.
 119          if (args[0] == "<EOF>")                      status = -1;
 120     else if (closeEnough ("exit",        args[0], 3)) status = -1;
 121     else if (closeEnough ("quit",        args[0], 3)) status = -1;
 122     else if (closeEnough ("help",        args[0], 3)) status = cmdHelp ();
 123     else if (closeEnough ("diagnostics", args[0], 3)) status = cmdDiagnostics ();
 124     else if (closeEnough ("review",      args[0], 3)) status = cmdReview (args, autoClear);
 125     else if (closeEnough ("exec",        args[0], 3) ||
 126              args[0][0] == '!')                       status = cmdShell (args);
 127     else if (command != "")
 128     {
 129       command = "task " + command;
 130       std::cout << "[" << command << "]\n";
 131       system (command.c_str ());
 132
 133       // Deliberately ignoreѕ taskwarrior exit status, otherwise empty filters
 134       // cause the shell to terminate.
 135     }
 136   }
 137
 138   return status;
 139 }
 140
 141 ////////////////////////////////////////////////////////////////////////////////
 142 int main (int argc, const char** argv)
 143 {
 144   int status = 0;
 145
 146   // Lightweight version checking that doesn't require initialization or any I/O.
 147   if (argc == 2 && !strcmp (argv[1], "--version"))
 148   {
 149     std::cout << VERSION << "\n";
 150   }
 151   else
 152   {
 153     try
 154     {
 155       // Get the Taskwarrior rc.tasksh.autoclear Boolean setting.
 156       bool autoClear = false;
 157       std::string input;
 158       std::string output;
 159       execute ("task", {"_get", "rc.tasksh.autoclear"}, input, output);
 160       output = lowerCase (output);
 161       autoClear = (output == "true\n" ||
 162                    output == "1\n"    ||
 163                    output == "y\n"    ||
 164                    output == "yes\n"  ||
 165                    output == "on\n");
 166
 167       if (isatty (fileno (stdin)))
 168         welcome ();
 169
 170       while ((status = commandLoop (autoClear)) == 0)
 171         ;
 172     }
 173
 174     catch (const std::string& error)
 175     {
 176       std::cerr << error << "\n";
 177       status = -1;
 178     }
 179
 180     catch (...)
 181     {
 182       std::cerr << "Unknown error." << "\n";
 183       status = -2;
 184     }
 185   }
 186
 187   // Returning -1 drops out of the command loop, but gets translated to 0 here,
 188   // so that there is a clean way to exit.
 189   return status == -1 ? 0 : status;
 190 }
 191
 192 ////////////////////////////////////////////////////////////////////////////////