#include "repository.h"
#include <QTextStream>
+#include <QDebug>
Repository::Repository(const Rules::Repository &rule)
+ : name(rule.name), commitCount(0), processHasStarted(false)
{
foreach (Rules::Repository::Branch branchRule, rule.branches) {
Branch branch;
- branch.branchFrom = branchRule.branchFrom;
- branch.isCreated = false;
+ branch.created = 0; // not created
branches.insert(branchRule.name, branch);
}
- fastImport.setWorkingDirectory(rule.name);
- fastImport.setProcessChannelMode(QProcess::ForwardedChannels);
+ // create the default branch
+ branches["master"].created = 1;
+
+ fastImport.setWorkingDirectory(name);
}
Repository::~Repository()
{
- if (fastImport.state() == QProcess::Running) {
+ if (fastImport.state() != QProcess::NotRunning) {
+ fastImport.write("checkpoint\n");
+ fastImport.waitForBytesWritten(-1);
fastImport.closeWriteChannel();
- fastImport.waitForFinished();
+ if (!fastImport.waitForFinished()) {
+ fastImport.terminate();
+ if (!fastImport.waitForFinished(200))
+ qWarning() << "git-fast-import for repository" << name << "did not die";
+ }
+ }
+}
+
+void Repository::reloadBranches()
+{
+ QProcess revParse;
+ revParse.setWorkingDirectory(name);
+ revParse.start("git", QStringList() << "rev-parse" << "--symbolic" << "--branches");
+ revParse.waitForFinished();
+
+ if (revParse.exitCode() == 0 && revParse.bytesAvailable()) {
+ startFastImport();
+
+ while (revParse.canReadLine()) {
+ QByteArray branchName = revParse.readLine().trimmed();
+
+ branches[branchName].created = 1;
+ fastImport.write("reset refs/heads/" + branchName +
+ "\nfrom refs/heads/" + branchName + "^0\n\n");
+ }
+ }
+}
+
+void Repository::createBranch(const QString &branch, int revnum,
+ const QString &branchFrom, int)
+{
+ if (!branches.contains(branch)) {
+ qWarning() << branch << "is not a known branch in repository" << name << endl
+ << "Going to create it automatically";
+ }
+
+ startFastImport();
+ QByteArray branchRef = branch.toUtf8();
+ if (!branchRef.startsWith("refs/"))
+ branchRef.prepend("refs/heads/");
+
+ Branch &br = branches[branch];
+ if (br.created && br.created != revnum) {
+ 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.created = revnum;
+ QByteArray branchFromRef = branchFrom.toUtf8();
+ if (!branchFromRef.startsWith("refs/"))
+ branchFromRef.prepend("refs/heads/");
+
+ if (!branches.contains(branchFrom) || !branches.value(branchFrom).created) {
+ qCritical() << branch << "in repository" << name
+ << "is branching from branch" << branchFrom
+ << "but the latter doesn't exist. Can't continue.";
+ exit(1);
}
+
+ fastImport.write("reset " + branchRef + "\nfrom " + branchFromRef + "\n\n");
}
Repository::Transaction *Repository::newTransaction(const QString &branch, const QString &svnprefix,
int revnum)
{
- if (!branches.contains(branch))
- return 0;
+ if (!branches.contains(branch)) {
+ qWarning() << branch << "is not a known branch in repository" << name << endl
+ << "Going to create it automatically";
+ }
Transaction *txn = new Transaction;
txn->repository = this;
txn->revnum = revnum;
txn->lastmark = revnum;
- if (fastImport.state() == QProcess::NotRunning)
+ startFastImport();
+ if ((++commitCount % 10000) == 0)
+ // write everything to disk every 10000 commits
+ fastImport.write("checkpoint\n");
+ return txn;
+}
+
+void Repository::startFastImport()
+{
+ if (fastImport.state() == QProcess::NotRunning) {
+ if (processHasStarted)
+ qFatal("git-fast-import has been started once and crashed?");
+ processHasStarted = true;
+
// start the process
- fastImport.start("git-fast-import", QStringList());
+ QString outputFile = name;
+ outputFile.replace('/', '_');
+ outputFile.prepend("log-");
+ fastImport.setStandardOutputFile(outputFile, QIODevice::Append);
+ fastImport.setProcessChannelMode(QProcess::MergedChannels);
- return txn;
+#ifndef DRY_RUN
+ fastImport.start("git-fast-import", QStringList());
+#else
+ fastImport.start("/bin/cat", QStringList());
+#endif
+ }
}
Repository::Transaction::~Transaction()
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;
QByteArray message = log;
if (!message.endsWith('\n'))
message += '\n';
- message += "\nsvn=" + svnprefix + "; revision=" + QByteArray::number(revnum) + "\n";
+ message += "\nsvn path=" + svnprefix + "; revision=" + QByteArray::number(revnum) + "\n";
{
QByteArray branchRef = branch;
- if (!branchRef.startsWith("refs/heads/"))
+ 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;
+ s << "committer " << QString::fromUtf8(author) << ' ' << datetime << " -0000" << endl;
Branch &br = repository->branches[branch];
- if (!br.isCreated) {
- br.isCreated = true;
- s << "from " << br.branchFrom << endl;
+ if (!br.created) {
+ qWarning() << "Branch" << branch << "in repository" << repository->name << "doesn't exist at revision"
+ << revnum << "-- did you resume from the wrong revision?";
+ br.created = revnum;
}
s << "data " << message.length() << endl;
}
repository->fastImport.write(message);
+ repository->fastImport.putChar('\n');
// write the file deletions
- foreach (QString df, deletedFiles)
- repository->fastImport.write("D " + df.toUtf8() + "\n");
+ if (deletedFiles.contains(""))
+ repository->fastImport.write("deleteall\n");
+ else
+ foreach (QString df, deletedFiles)
+ repository->fastImport.write("D " + df.toUtf8() + "\n");
// write the file modifications
QHash<QString, FileProperties>::ConstIterator it = modifiedFiles.constBegin();
repository->fastImport.write("\n");
- while (repository->fastImport.bytesToWrite() && repository->fastImport.waitForBytesWritten()) {
- // nothing
- }
+ while (repository->fastImport.bytesToWrite())
+ if (!repository->fastImport.waitForBytesWritten(-1))
+ qFatal("Failed to write to process: %s", qPrintable(repository->fastImport.errorString()));
}