#include <QDebug>
Repository::Repository(const Rules::Repository &rule)
- : name(rule.name)
+ : name(rule.name), commitCount(0), processHasStarted(false)
{
foreach (Rules::Repository::Branch branchRule, rule.branches) {
Branch branch;
- branch.isCreated = false;
+ branch.created = 0; // not created
branches.insert(branchRule.name, branch);
}
// create the default branch
- branches["master"].isCreated = true;
+ branches["master"].created = 1;
fastImport.setWorkingDirectory(name);
}
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/");
+ 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();
- 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;
+ branches[branchName].created = 1;
+ fastImport.write("reset refs/heads/" + branchName +
+ "\nfrom refs/heads/" + branchName + "^0\n\n");
}
}
}
const QString &branchFrom, int)
{
if (!branches.contains(branch)) {
- qCritical() << branch << "is not a known branch in repository" << name;
- exit(1);
+ qWarning() << branch << "is not a known branch in repository" << name << endl
+ << "Going to create it automatically";
}
startFastImport();
branchRef.prepend("refs/heads/");
Branch &br = branches[branch];
- if (br.isCreated) {
+ if (br.created && br.created != revnum) {
QByteArray backupBranch = branchRef + '_' + QByteArray::number(revnum);
qWarning() << branch << "already exists; backing up to" << backupBranch;
}
// now create the branch
- br.isCreated = true;
+ 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");
}
int revnum)
{
if (!branches.contains(branch)) {
- qCritical() << branch << "is not a known branch in repository" << name;
- return 0;
+ qWarning() << branch << "is not a known branch in repository" << name << endl
+ << "Going to create it automatically";
}
Transaction *txn = new Transaction;
txn->lastmark = revnum;
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
+ QString outputFile = name;
+ outputFile.replace('/', '_');
+ outputFile.prepend("log-");
+ fastImport.setStandardOutputFile(outputFile, QIODevice::Append);
+ fastImport.setProcessChannelMode(QProcess::MergedChannels);
+
#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->fastImport.write("\ndata ");
repository->fastImport.write(QByteArray::number(length));
repository->fastImport.write("\n", 1);
- repository->fastImport.waitForBytesWritten(0);
#endif
modifiedFiles.insert(path, fp);
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;
+ 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()));
}