2 * Copyright (C) 2007 Thiago Macieira <thiago@kde.org>
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 #include "repository.h"
19 #include <QTextStream>
22 Repository::Repository(const Rules::Repository &rule)
23 : name(rule.name), processHasStarted(false)
25 foreach (Rules::Repository::Branch branchRule, rule.branches) {
27 branch.created = 0; // not created
29 branches.insert(branchRule.name, branch);
32 // create the default branch
33 branches["master"].created = 1;
35 fastImport.setWorkingDirectory(name);
38 Repository::~Repository()
40 if (fastImport.state() != QProcess::NotRunning) {
41 fastImport.closeWriteChannel();
42 fastImport.waitForFinished();
46 void Repository::reloadBranches()
48 QHash<QString, Branch>::Iterator it = branches.begin(),
50 for ( ; it != end; ++it) {
51 QString branchRef = it.key();
52 if (!branchRef.startsWith("refs/"))
53 branchRef.prepend("refs/heads/");
56 // does this branch already exist?
58 revParse.setWorkingDirectory(name);
59 revParse.start("git-rev-parse", QStringList() << "--verify" << branchRef);
60 revParse.waitForFinished();
62 if (revParse.exitCode() == 0)
69 fastImport.write("reset " + branchRef.toUtf8() +
70 "\nfrom " + branchRef.toUtf8() + "^0\n\n");
76 void Repository::createBranch(const QString &branch, int revnum,
77 const QString &branchFrom, int)
79 if (!branches.contains(branch)) {
80 qCritical() << branch << "is not a known branch in repository" << name;
85 QByteArray branchRef = branch.toUtf8();
86 if (!branchRef.startsWith("refs/"))
87 branchRef.prepend("refs/heads/");
89 Branch &br = branches[branch];
90 if (br.created && br.created != revnum) {
91 QByteArray backupBranch = branchRef + '_' + QByteArray::number(revnum);
92 qWarning() << branch << "already exists; backing up to" << backupBranch;
94 fastImport.write("reset " + backupBranch + "\nfrom " + branchRef + "\n\n");
97 // now create the branch
99 QByteArray branchFromRef = branchFrom.toUtf8();
100 if (!branchFromRef.startsWith("refs/"))
101 branchFromRef.prepend("refs/heads/");
103 fastImport.write("reset " + branchRef + "\nfrom " + branchFromRef + "\n\n");
106 Repository::Transaction *Repository::newTransaction(const QString &branch, const QString &svnprefix,
109 if (!branches.contains(branch)) {
110 qCritical() << branch << "is not a known branch in repository" << name;
114 Transaction *txn = new Transaction;
115 txn->repository = this;
116 txn->branch = branch.toUtf8();
117 txn->svnprefix = svnprefix.toUtf8();
119 txn->revnum = revnum;
120 txn->lastmark = revnum;
123 if ((++commitCount % 10000) == 0)
124 // write everything to disk every 10000 commits
125 fastImport.write("checkpoint\n");
129 void Repository::startFastImport()
131 if (fastImport.state() == QProcess::NotRunning) {
132 if (processHasStarted)
133 qFatal("git-fast-import has been started once and crashed?");
134 processHasStarted = true;
137 QString outputFile = name;
138 outputFile.replace('/', '_');
139 outputFile.prepend("log-");
140 fastImport.setStandardOutputFile(outputFile, QIODevice::Append);
143 fastImport.setProcessChannelMode(QProcess::ForwardedChannels);
144 fastImport.start("git-fast-import", QStringList());
146 fastImport.start("/bin/cat", QStringList());
151 Repository::Transaction::~Transaction()
155 void Repository::Transaction::setAuthor(const QByteArray &a)
160 void Repository::Transaction::setDateTime(uint dt)
165 void Repository::Transaction::setLog(const QByteArray &l)
170 void Repository::Transaction::deleteFile(const QString &path)
172 deletedFiles.append(path);
175 QIODevice *Repository::Transaction::addFile(const QString &path, int mode, qint64 length)
179 fp.mark = ++lastmark;
182 repository->fastImport.write("blob\nmark :");
183 repository->fastImport.write(QByteArray::number(fp.mark));
184 repository->fastImport.write("\ndata ");
185 repository->fastImport.write(QByteArray::number(length));
186 repository->fastImport.write("\n", 1);
189 modifiedFiles.insert(path, fp);
190 return &repository->fastImport;
193 void Repository::Transaction::commit()
195 // create the commit message
196 QByteArray message = log;
197 if (!message.endsWith('\n'))
199 message += "\nsvn path=" + svnprefix + "; revision=" + QByteArray::number(revnum) + "\n";
202 QByteArray branchRef = branch;
203 if (!branchRef.startsWith("refs/"))
204 branchRef.prepend("refs/heads/");
206 QTextStream s(&repository->fastImport);
207 s << "commit " << branchRef << endl;
208 s << "mark :" << revnum << endl;
209 s << "committer " << author << ' ' << datetime << " -0000" << endl;
211 Branch &br = repository->branches[branch];
213 qWarning() << "Branch" << branch << "doesn't exist at revision"
214 << revnum << "-- did you resume from the wrong revision?";
218 s << "data " << message.length() << endl;
221 repository->fastImport.write(message);
222 repository->fastImport.putChar('\n');
224 // write the file deletions
225 if (deletedFiles.contains(""))
226 repository->fastImport.write("deleteall\n");
228 foreach (QString df, deletedFiles)
229 repository->fastImport.write("D " + df.toUtf8() + "\n");
231 // write the file modifications
232 QHash<QString, FileProperties>::ConstIterator it = modifiedFiles.constBegin();
233 for ( ; it != modifiedFiles.constEnd(); ++it) {
234 repository->fastImport.write("M ", 2);
235 repository->fastImport.write(QByteArray::number(it->mode, 8));
236 repository->fastImport.write(" :", 2);
237 repository->fastImport.write(QByteArray::number(it->mark));
238 repository->fastImport.write(" ", 1);
239 repository->fastImport.write(it.key().toUtf8());
240 repository->fastImport.write("\n", 1);
243 repository->fastImport.write("\n");
245 while (repository->fastImport.bytesToWrite())
246 if (!repository->fastImport.waitForBytesWritten())
247 qFatal("Failed to write to process: %s", qPrintable(repository->fastImport.errorString()));