Configuration.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 #include "Configuration.h"
  19 #include "Logger.h"
  20 #include "GitRunner.h"
  21 #include "AbstractCommand.h"
  22
  23 #include <QDebug>
  24 #include <QMutexLocker>
  25 #include <QProcess>
  26
  27 #ifndef Q_OS_WIN
  28 #include <unistd.h>
  29 #endif
  30
  31 #define EditedFilesDatabase "vng-editedfiles"
  32
  33 Configuration::Configuration(const QString &section)
  34     : m_repoDir("."),
  35     m_section(section),
  36     m_dirty(true),
  37     m_emptyRepo(false),
  38     m_fetchedBranches(false),
  39     m_editedFilesFetched(false),
  40     m_editMode(StateUnknown)
  41 {
  42 }
  43
  44 bool Configuration::contains(const QString & key) const
  45 {
  46     const_cast<Configuration*> (this)->readConfig();
  47     return m_options.contains(key);
  48 }
  49
  50 QDir Configuration::repository() const
  51 {
  52     const_cast<Configuration*> (this)->readConfig();
  53     return m_repoDir;
  54 }
  55
  56 QDir Configuration::repositoryMetaDir() const
  57 {
  58     const_cast<Configuration*> (this)->readConfig();
  59     return m_repoMetaDataDir;
  60 }
  61
  62 void Configuration::readConfig()
  63 {
  64     if (!m_dirty)
  65         return;
  66     m_dirty = false;
  67     QObject deleterParent;
  68
  69     QDir dir = QDir::current();
  70     do {
  71         QDir git = dir;
  72         if (git.cd(".git")) {
  73             m_repoDir = dir;
  74             m_repoMetaDataDir = git;
  75             QDir refs(git.absoluteFilePath("refs/heads"));
  76             m_emptyRepo = refs.count() == 2; // only '.' and '..'
  77             break;
  78         }
  79         if (!dir.cdUp())
  80             break;
  81     } while(!dir.isRoot());
  82
  83     QString home = QDir::homePath();
  84     QFile *config;
  85     config = new QFile(home + "/.vng/config", &deleterParent);
  86     if (! config->exists())
  87         config = new QFile(home + "/.darcs/defaults", &deleterParent);
  88     if (config->exists()) {
  89         if (! config->open(QIODevice::ReadOnly)) {
  90             Logger::error() << "Failed to open config file, is it readable?\n";
  91             return;
  92         }
  93
  94         char buf[1024];
  95         while(true) {
  96             qint64 lineLength = config->readLine(buf, sizeof(buf));
  97             if (lineLength == -1)
  98                 break;
  99             QString line = QString::fromUtf8(buf, lineLength);
 100             QString option;
 101             if (line.startsWith("ALL "))
 102                 option = line.mid(3).trimmed();
 103             else if (line.length() > m_section.length() && line.startsWith(m_section))
 104                 option = line.mid(m_section.length()).trimmed();
 105             if (! option.isEmpty()) {
 106                 const int index = option.indexOf(' ');
 107                 if (index > 0) {
 108                     m_options.insert(option.left(index).trimmed(), option.mid(index).trimmed());
 109                 }
 110                 else
 111                     m_options.insert(option, QString());
 112             }
 113         }
 114         config->close();
 115     }
 116 }
 117
 118 QString Configuration::optionArgument(const QString &optionName, const QString &defaultValue) const
 119 {
 120     if (m_options.contains(optionName))
 121         return m_options[optionName];
 122     return defaultValue;
 123 }
 124
 125 bool Configuration::colorTerm() const
 126 {
 127 #ifndef Q_OS_WIN
 128     if (isatty(1))
 129         return QString(getenv("TERM")) != QString("dumb") && !Logger::hasNonColorPager();
 130 #endif
 131     return false;
 132 }
 133
 134 bool Configuration::isEmptyRepo() const
 135 {
 136     const_cast<Configuration*> (this)->readConfig();
 137     return m_emptyRepo;
 138 }
 139
 140 QList<Branch> Configuration::allBranches()
 141 {
 142     fetchBranches();
 143     QList<Branch> answer;
 144     answer += m_localBranches;
 145     answer += m_remoteBranches;
 146     return answer;
 147 }
 148
 149 QList<Branch> Configuration::branches()
 150 {
 151     fetchBranches();
 152     return m_localBranches;
 153 }
 154
 155 QList<Branch> Configuration::remoteBranches()
 156 {
 157     fetchBranches();
 158     return m_remoteBranches;
 159 }
 160
 161 void Configuration::fetchBranches()
 162 {
 163     if (m_fetchedBranches)
 164         return;
 165     m_fetchedBranches = true;
 166
 167     QProcess git;
 168     QStringList arguments;
 169     arguments << "ls-remote" << ".";
 170     GitRunner runner(git, arguments);
 171     AbstractCommand::ReturnCodes rc = runner.start(GitRunner::WaitForStandardOutput);
 172     if (rc != AbstractCommand::Ok) {
 173         return;
 174     }
 175     char buf[1024];
 176     while(true) {
 177         qint64 lineLength = Vng::readLine(&git, buf, sizeof(buf));
 178         if (lineLength == -1)
 179             break;
 180         if (lineLength > 46) { // only take stuff that is in the 'refs' dir.
 181             QString name(buf + 46);
 182             name = name.trimmed(); // remove linefeed
 183             Branch branch(name);
 184             if (name.startsWith("remotes"))
 185                 m_remoteBranches.append(branch);
 186             else
 187                 m_localBranches.append(branch);
 188         }
 189     }
 190 }
 191
 192 bool Configuration::useEditMode() const
 193 {
 194     if (m_editMode == StateUnknown) {
 195         TryState editMode;
 196
 197         QProcess git;
 198         QStringList arguments;
 199         arguments << "config" << "--bool" << "vng.useEditMode";
 200         GitRunner runner(git, arguments);
 201         AbstractCommand::ReturnCodes rc = runner.start(GitRunner::WaitForStandardOutput, GitRunner::FailureAccepted);
 202         if (git.exitCode() == 1) // unregistered var means false.
 203             editMode = StateFalse;
 204         else if (rc != AbstractCommand::Ok)
 205             return false;
 206         else {
 207             editMode = StateFalse;
 208             char buf[1024];
 209             while(true) {
 210                 qint64 lineLength = Vng::readLine(&git, buf, sizeof(buf));
 211                 if (lineLength == -1)
 212                     break;
 213                 if (lineLength > 3 && QString(buf).startsWith("true"))
 214                     editMode = StateTrue;
 215             }
 216         }
 217         const_cast<Configuration*> (this)->m_editMode = editMode;
 218     }
 219     return m_editMode == StateTrue;
 220 }
 221
 222 void Configuration::setEditMode(bool enabled)
 223 {
 224     QProcess git;
 225     QStringList arguments;
 226     arguments << "config" << "--replace-all" << "vng.useEditMode" << (enabled ? "true" : "false");
 227     GitRunner runner(git, arguments);
 228     runner.start(GitRunner::WaitUntilFinished);
 229 }
 230
 231 QList<QByteArray> Configuration::editedFiles()
 232 {
 233     if (! useEditMode())
 234         return QList<QByteArray>();
 235     QMutexLocker locker(&m_editedFilesLock);
 236     // TODO do some file-locking
 237     if (!m_editedFilesFetched) {
 238         readConfig();
 239         if (! m_repoMetaDataDir.exists())
 240             return QList<QByteArray> ();
 241         QFile openedFiles(m_repoMetaDataDir.filePath(EditedFilesDatabase));
 242         if (!openedFiles.open(QIODevice::ReadOnly)) {
 243             Logger::error() << "Vng failed to read the database of opened files ('" << openedFiles.fileName() << "`)\n";
 244             return QList<QByteArray>();
 245         }
 246         char buf[1024];
 247         while (true) {
 248             openedFiles.waitForReadyRead(-1);
 249             qint64 len = openedFiles.readLine(buf, sizeof(buf));
 250             if (len == -1)
 251                 break;
 252             if (len == 1 || buf[0] == '#')
 253                 continue;
 254             m_editedFiles << QByteArray(buf, len-1);
 255         }
 256
 257         m_editedFilesFetched = true;
 258     }
 259
 260     return m_editedFiles;
 261 }
 262
 263 void Configuration::addEditedFiles(const QList<QByteArray> &newFileNames)
 264 {
 265     if (newFileNames.size() == 0)
 266         return;
 267     // TODO do some file-locking
 268     QMutexLocker locker(&m_editedFilesLock);
 269     QFile editedFiles(m_repoMetaDataDir.filePath(EditedFilesDatabase));
 270     if (! editedFiles.open(QIODevice::WriteOnly | QIODevice::Append)) {
 271         Logger::error() << "vng-failed: Could not write to the database of edited files\n";
 272         return;
 273     }
 274     // TODO make sure we don't write the same filename to the DB twice.
 275
 276     // remove old stuff
 277     m_editedFilesFetched = false;
 278     m_editedFiles.clear();
 279
 280     foreach(QByteArray ba, newFileNames) {
 281         if (ba.size() == 0)
 282             continue;
 283         editedFiles.write(ba);
 284         if (ba[ba.size()-1] != '\n')
 285             editedFiles.write("\n", 1);
 286     }
 287     editedFiles.close();
 288 }
 289
 290 void Configuration::addEditedFiles(const QStringList &newFileNames)
 291 {
 292     QList<QByteArray> fileNames;
 293     foreach(QString s, newFileNames) {
 294         fileNames << s.toLocal8Bit();
 295     }
 296     addEditedFiles(fileNames);
 297 }
 298
 299 void Configuration::removeEditedFiles(const QStringList &fileNames)
 300 {
 301     QList<QByteArray> files;
 302     foreach(QString s, fileNames) {
 303         files << s.toLocal8Bit();
 304     }
 305     removeEditedFiles(files);
 306 }
 307
 308 void Configuration::removeEditedFiles(const QList<QByteArray> &fileNames)
 309 {
 310     if (fileNames.size() == 0)
 311         return;
 312     // TODO do some file-locking
 313     QList<QByteArray> currentContents = editedFiles();
 314     QMutexLocker locker(&m_editedFilesLock);
 315
 316     QList<QByteArray> currentContent = m_editedFiles;
 317     if (currentContent.isEmpty())
 318         return;
 319
 320     QFile editedFiles(m_repoMetaDataDir.filePath(EditedFilesDatabase));
 321     if (! editedFiles.open(QIODevice::WriteOnly | QIODevice::Truncate)) {
 322         Logger::error() << "vng-failed: Could not write to the database of edited files\n";
 323         return;
 324     }
 325     m_editedFiles.clear();
 326
 327     foreach (QByteArray ba, currentContent) {
 328         bool found = false;
 329         foreach (QByteArray rem, fileNames) {
 330             if (ba.length() == rem.length() && ba.indexOf(rem) == 0) { // thats the file to remove.
 331                 found = true;
 332                 break;
 333             }
 334         }
 335         if (! found) {
 336             m_editedFiles << ba;
 337             editedFiles.write(ba);
 338             if (ba[ba.size()-1] != '\n')
 339                 editedFiles.write("\n", 1);
 340         }
 341     }
 342     editedFiles.close();
 343 }