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)
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) {
134 fastImport.setProcessChannelMode(QProcess::ForwardedChannels);
135 fastImport.start("git-fast-import", QStringList());
137 QString outputFile = name;
138 outputFile.replace('/', '_');
139 fastImport.setStandardOutputFile(outputFile, QIODevice::Append);
140 fastImport.start("/bin/cat", QStringList());
145 Repository::Transaction::~Transaction()
149 void Repository::Transaction::setAuthor(const QByteArray &a)
154 void Repository::Transaction::setDateTime(uint dt)
159 void Repository::Transaction::setLog(const QByteArray &l)
164 void Repository::Transaction::deleteFile(const QString &path)
166 deletedFiles.append(path);
169 QIODevice *Repository::Transaction::addFile(const QString &path, int mode, qint64 length)
173 fp.mark = ++lastmark;
176 repository->fastImport.write("blob\nmark :");
177 repository->fastImport.write(QByteArray::number(fp.mark));
178 repository->fastImport.write("\ndata ");
179 repository->fastImport.write(QByteArray::number(length));
180 repository->fastImport.write("\n", 1);
181 repository->fastImport.waitForBytesWritten(0);
184 modifiedFiles.insert(path, fp);
185 return &repository->fastImport;
188 void Repository::Transaction::commit()
190 // create the commit message
191 QByteArray message = log;
192 if (!message.endsWith('\n'))
194 message += "\nsvn path=" + svnprefix + "; revision=" + QByteArray::number(revnum) + "\n";
197 QByteArray branchRef = branch;
198 if (!branchRef.startsWith("refs/"))
199 branchRef.prepend("refs/heads/");
201 QTextStream s(&repository->fastImport);
202 s << "commit " << branchRef << endl;
203 s << "mark :" << revnum << endl;
204 s << "committer " << author << ' ' << datetime << " -0000" << endl;
206 Branch &br = repository->branches[branch];
208 qWarning() << "Branch" << branch << "doesn't exist at revision"
209 << revnum << "-- did you resume from the wrong revision?";
213 s << "data " << message.length() << endl;
216 repository->fastImport.write(message);
217 repository->fastImport.putChar('\n');
219 // write the file deletions
220 if (deletedFiles.contains(""))
221 repository->fastImport.write("deleteall\n");
223 foreach (QString df, deletedFiles)
224 repository->fastImport.write("D " + df.toUtf8() + "\n");
226 // write the file modifications
227 QHash<QString, FileProperties>::ConstIterator it = modifiedFiles.constBegin();
228 for ( ; it != modifiedFiles.constEnd(); ++it) {
229 repository->fastImport.write("M ", 2);
230 repository->fastImport.write(QByteArray::number(it->mode, 8));
231 repository->fastImport.write(" :", 2);
232 repository->fastImport.write(QByteArray::number(it->mark));
233 repository->fastImport.write(" ", 1);
234 repository->fastImport.write(it.key().toUtf8());
235 repository->fastImport.write("\n", 1);
238 repository->fastImport.write("\n");
240 while (repository->fastImport.bytesToWrite() && repository->fastImport.waitForBytesWritten()) {