*/
#include "repository.h"
+#include <QTextStream>
+#include <QDebug>
Repository::Repository(const Rules::Repository &rule)
+ : name(rule.name)
{
foreach (Rules::Repository::Branch branchRule, rule.branches) {
Branch branch;
- branch.branchFrom = branchRule.branchFrom;
branch.isCreated = false;
branches.insert(branchRule.name, branch);
}
- fastImport.setWorkingDirectory(rule.name);
- fastImport.setProcessChannelMode(QProcess::ForwardedChannels);
+ // create the default branch
+ branches["master"].isCreated = true;
+
+ fastImport.setWorkingDirectory(name);
+}
+
+Repository::~Repository()
+{
+ if (fastImport.state() != QProcess::NotRunning) {
+ fastImport.closeWriteChannel();
+ fastImport.waitForFinished();
+ }
+}
+
+void Repository::reloadBranches()
+{
+ QHash<QString, Branch>::Iterator it = branches.begin(),
+ end = branches.end();
+ for ( ; it != end; ++it) {
+ QString branchRef = it.key();
+ if (!branchRef.startsWith("refs/"))
+ branchRef.prepend("refs/heads/");
+
+ bool branchExists;
+ // does this branch already exist?
+ QProcess revParse;
+ revParse.setWorkingDirectory(name);
+ revParse.start("git-rev-parse", QStringList() << "--verify" << branchRef);
+ revParse.waitForFinished();
+
+ if (revParse.exitCode() == 0)
+ branchExists = true;
+ else
+ branchExists = false;
+
+ if (branchExists) {
+ startFastImport();
+ fastImport.write("reset " + branchRef.toUtf8() +
+ "\nfrom " + branchRef.toUtf8() + "^0\n\n");
+ it->isCreated = true;
+ }
+ }
+}
+
+void Repository::createBranch(const QString &branch, int revnum,
+ const QString &branchFrom, int)
+{
+ if (!branches.contains(branch)) {
+ qCritical() << branch << "is not a known branch in repository" << name;
+ exit(1);
+ }
+
+ startFastImport();
+ QByteArray branchRef = branch.toUtf8();
+ if (!branchRef.startsWith("refs/"))
+ branchRef.prepend("refs/heads/");
+
+ Branch &br = branches[branch];
+ if (br.isCreated) {
+ QByteArray backupBranch = branchRef + '_' + QByteArray::number(revnum);
+ qWarning() << branch << "already exists; backing up to" << backupBranch;
+
+ fastImport.write("reset " + backupBranch + "\nfrom " + branchRef + "\n\n");
+ }
+
+ // now create the branch
+ br.isCreated = true;
+ QByteArray branchFromRef = branchFrom.toUtf8();
+ if (!branchFromRef.startsWith("refs/"))
+ branchFromRef.prepend("refs/heads/");
+
+ fastImport.write("reset " + branchRef + "\nfrom " + branchFromRef + "\n\n");
+}
+
+Repository::Transaction *Repository::newTransaction(const QString &branch, const QString &svnprefix,
+ int revnum)
+{
+ if (!branches.contains(branch)) {
+ qCritical() << branch << "is not a known branch in repository" << name;
+ return 0;
+ }
+
+ Transaction *txn = new Transaction;
+ txn->repository = this;
+ txn->branch = branch.toUtf8();
+ txn->svnprefix = svnprefix.toUtf8();
+ txn->datetime = 0;
+ txn->revnum = revnum;
+ txn->lastmark = revnum;
+
+ startFastImport();
+ return txn;
+}
+
+void Repository::startFastImport()
+{
+ if (fastImport.state() == QProcess::NotRunning) {
+ // start the process
+#ifndef DRY_RUN
+ fastImport.setProcessChannelMode(QProcess::ForwardedChannels);
+ fastImport.start("git-fast-import", QStringList());
+#else
+ QString outputFile = name;
+ outputFile.replace('/', '_');
+ fastImport.setStandardOutputFile(outputFile);
+ fastImport.start("/bin/cat", QStringList());
+#endif
+ }
+}
+
+Repository::Transaction::~Transaction()
+{
+}
+
+void Repository::Transaction::setAuthor(const QByteArray &a)
+{
+ author = a;
+}
+
+void Repository::Transaction::setDateTime(uint dt)
+{
+ datetime = dt;
+}
+
+void Repository::Transaction::setLog(const QByteArray &l)
+{
+ log = l;
+}
+
+void Repository::Transaction::deleteFile(const QString &path)
+{
+ deletedFiles.append(path);
+}
+
+QIODevice *Repository::Transaction::addFile(const QString &path, int mode, qint64 length)
+{
+ FileProperties fp;
+ fp.mode = mode;
+ fp.mark = ++lastmark;
+
+#ifndef DRY_RUN
+ repository->fastImport.write("blob\nmark :");
+ repository->fastImport.write(QByteArray::number(fp.mark));
+ repository->fastImport.write("\ndata ");
+ repository->fastImport.write(QByteArray::number(length));
+ repository->fastImport.write("\n", 1);
+ repository->fastImport.waitForBytesWritten(0);
+#endif
+
+ modifiedFiles.insert(path, fp);
+ return &repository->fastImport;
+}
+
+void Repository::Transaction::commit()
+{
+ // create the commit message
+ QByteArray message = log;
+ if (!message.endsWith('\n'))
+ message += '\n';
+ message += "\nsvn path=" + svnprefix + "; revision=" + QByteArray::number(revnum) + "\n";
+
+ {
+ QByteArray branchRef = branch;
+ if (!branchRef.startsWith("refs/"))
+ branchRef.prepend("refs/heads/");
+
+ QTextStream s(&repository->fastImport);
+ s << "commit " << branchRef << endl;
+ s << "mark :" << revnum << endl;
+ s << "committer " << author << ' ' << datetime << " -0000" << endl;
+
+ Branch &br = repository->branches[branch];
+ if (!br.isCreated) {
+ br.isCreated = true;
+ }
+
+ s << "data " << message.length() << endl;
+ }
+
+ repository->fastImport.write(message);
+
+ // write the file deletions
+ foreach (QString df, deletedFiles)
+ repository->fastImport.write("D " + df.toUtf8() + "\n");
+
+ // write the file modifications
+ QHash<QString, FileProperties>::ConstIterator it = modifiedFiles.constBegin();
+ for ( ; it != modifiedFiles.constEnd(); ++it) {
+ repository->fastImport.write("M ", 2);
+ repository->fastImport.write(QByteArray::number(it->mode, 8));
+ repository->fastImport.write(" :", 2);
+ repository->fastImport.write(QByteArray::number(it->mark));
+ repository->fastImport.write(" ", 1);
+ repository->fastImport.write(it.key().toUtf8());
+ repository->fastImport.write("\n", 1);
+ }
+
+ repository->fastImport.write("\n");
+
+ while (repository->fastImport.bytesToWrite() && repository->fastImport.waitForBytesWritten()) {
+ // nothing
+ }
}