src/AbstractCommand.cpp

   1 /*
   2  * This file is part of the vng project
   3  * Copyright (C) 2008 Thomas Zander <tzander@trolltech.com>
   4  *
   5  * This program is free software: you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation, either version 3 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #include "AbstractCommand.h"
  20 #include "CommandLineParser.h"
  21 #include "Logger.h"
  22
  23 #include <QDebug>
  24
  25 static const CommandLineOption options[] = {
  26     {"-v, --verbose", "give verbose output"},
  27     {"-q, --quiet", "suppress informational output"},
  28     {"--debug", "print debugging information"},
  29     {"--standard-verbosity", "neither verbose nor quiet output"},
  30     {"--repodir DIRECTORY", "specify the repository directory in which to run"},
  31     {"--dry-run", "don't actually take the action"},
  32     {"--disable", "disable this command"},
  33     {"-h, --help", "shows brief description of command and its arguments"},
  34     {"--use-pager", "use a pager to show the output screen by screen"},
  35     {"--no-pager", "don't use the pager when showing the output"},
  36     CommandLineLastOption
  37 };
  38
  39 AbstractCommand::AbstractCommand(const QString &name)
  40     : m_config(name), m_dryRun(false), m_name(name)
  41 {
  42     CommandLineParser::addOptionDefinitions(options);
  43 }
  44
  45 AbstractCommand::~AbstractCommand()
  46 {
  47 }
  48
  49 AbstractCommand::ReturnCodes AbstractCommand::start()
  50 {
  51     CommandLineParser *args = CommandLineParser::instance();
  52
  53     if (m_config.contains("quiet"))
  54         Logger::setVerbosity(Logger::Quiet);
  55     else if (m_config.contains("verbose"))
  56         Logger::setVerbosity(Logger::Verbose);
  57     else if (m_config.contains("standard-verbosity"))
  58         Logger::setVerbosity(Logger::Chatty);
  59     else if (m_config.contains("debug"))
  60         Logger::setVerbosity(Logger::Debug);
  61
  62     if (args->contains("quiet"))
  63         Logger::setVerbosity(Logger::Quiet);
  64     else if (args->contains("verbose"))
  65         Logger::setVerbosity(Logger::Verbose);
  66     else if (args->contains("standard-verbosity"))
  67         Logger::setVerbosity(Logger::Chatty);
  68     else if (args->contains("debug"))
  69         Logger::setVerbosity(Logger::Debug);
  70
  71     if (args->contains("disable")) {
  72         Logger::warn() << "vng failed: Command `" << m_name << "' disabled with --disable option!" << endl;
  73         return Disabled;
  74     }
  75     if (m_config.contains("disable")) {
  76         Logger::warn() << "vng failed: Command `" << m_name << "' disabled by configuration!" << endl;
  77         return Disabled;
  78     }
  79
  80     if (args->undefinedOptions().count()) {
  81         Logger::error() << "vng failed: ";
  82         bool first = true;
  83         foreach(QString option, args->undefinedOptions()) {
  84             if (! first)
  85                 Logger::error() << "          : ";
  86             Logger::error() << "unrecognized option or missing argument for; `" << option << "'" << endl;
  87             first = false;
  88         }
  89         return InvalidOptions;
  90     }
  91
  92     if (args->contains("help")) {
  93         args->usage(name(), argumentDescription());
  94         QString command = commandDescription();
  95         if (!command.isEmpty()) {
  96             Logger::standardOut() << endl;
  97             Logger::standardOut() << command;
  98         }
  99         return Ok;
 100     }
 101
 102     if (args->parseErrors().count()) {
 103         Logger::error() << "vng failed:  ";
 104         foreach(QString e, args->parseErrors()) {
 105             Logger::error() << e << endl;
 106         }
 107         return InvalidOptions;
 108     }
 109
 110     m_dryRun = args->contains("dry-run") || m_config.contains("dry-run");
 111
 112     if (args->contains("repodir")) {
 113         m_repository = QDir(args->optionArgument("repodir"));
 114         QDir test = m_repository;
 115         if (! test.cd(".git")) {
 116             Logger::error() << "repository dir `" << args->optionArgument("repodir") << "' is not a repository" << endl;
 117             return InvalidOptions;
 118         }
 119     }
 120     else
 121         m_repository = m_config.repository();
 122     return run();
 123 }
 124
 125 QDir AbstractCommand::repository() const
 126 {
 127     return m_repository;
 128 }
 129
 130 bool AbstractCommand::dryRun() const
 131 {
 132     return m_dryRun;
 133 }
 134
 135 QString AbstractCommand::name() const
 136 {
 137     return m_name;
 138 }
 139
 140 QStringList AbstractCommand::rebasedArguments() const
 141 {
 142     return m_rebasedArguments;
 143 }
 144
 145 bool AbstractCommand::checkInRepository() const
 146 {
 147     if (! QDir(m_repository.absolutePath() + "/.git").exists()) {
 148         Logger::error() << "vng failed:  Unable to `" << m_name << "' here\n\n";
 149         Logger::error() << "You need to be in a repository directory to run this command.\n";
 150         return false;
 151     }
 152     return true;
 153 }
 154
 155 bool AbstractCommand::shouldUsePager() const
 156 {
 157     CommandLineParser *args = CommandLineParser::instance();
 158     return (m_config.contains("use-pager") && !args->contains("no-pager")) || args->contains("use-pager");
 159 }
 160
 161 void AbstractCommand::moveToRoot(RebaseOptions options)
 162 {
 163     QString currentDir = QDir::current().absolutePath();
 164     QString repoDir = m_repository.absolutePath();
 165     QDir::setCurrent(repoDir);
 166     if (! m_rebasedArguments.isEmpty())
 167         return;
 168     CommandLineParser *args = CommandLineParser::instance();
 169     if (currentDir != repoDir && currentDir.startsWith(repoDir)) {
 170         QString diff = currentDir.mid(repoDir.length() + 1);
 171         // calcuate rebased paths.
 172         bool first = true;
 173         foreach (QString arg, args->arguments()) {
 174             if (first) {
 175                 first = false;
 176                 continue; // first is the command name
 177             }
 178             QString target;
 179             if (arg == ".") // work around git bug that 'dir/.' doesn't work while 'dir' does work
 180                 target = diff;
 181             else if (arg.startsWith('/')) //absolute path
 182                 target = arg;
 183             else
 184                 target = diff +'/'+ arg;
 185             if (options & CheckFileSystem) {
 186                 QFile rebasedFile(target);
 187                 if (rebasedFile.exists())
 188                     m_rebasedArguments << target;
 189                 else {
 190                     m_rebasedArguments << arg;
 191                     if (options & PrintError) {
 192                         QFile file(arg);
 193                         if (!file.exists())
 194                             Logger::error() << "Error, unknown file " << arg << endl;
 195                     }
 196                 }
 197             }
 198             else
 199                 m_rebasedArguments << target;
 200         }
 201     }
 202     else
 203         m_rebasedArguments = args->arguments().mid(1);
 204 }