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 qWarning() << branch
<< "is not a known branch in repository" << name
<< endl
81 << "Going to create it automatically";
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 if (!branches
.contains(branchFrom
) || !branches
.value(branchFrom
).created
) {
104 qCritical() << branch
<< "in repository" << name
105 << "is branching from branch" << branchFrom
106 << "but the latter doesn't exist. Can't continue.";
110 fastImport
.write("reset " + branchRef
+ "\nfrom " + branchFromRef
+ "\n\n");
113 Repository::Transaction
*Repository::newTransaction(const QString
&branch
, const QString
&svnprefix
,
116 if (!branches
.contains(branch
)) {
117 qCritical() << branch
<< "is not a known branch in repository" << name
;
121 Transaction
*txn
= new Transaction
;
122 txn
->repository
= this;
123 txn
->branch
= branch
.toUtf8();
124 txn
->svnprefix
= svnprefix
.toUtf8();
126 txn
->revnum
= revnum
;
127 txn
->lastmark
= revnum
;
130 if ((++commitCount
% 10000) == 0)
131 // write everything to disk every 10000 commits
132 fastImport
.write("checkpoint\n");
136 void Repository::startFastImport()
138 if (fastImport
.state() == QProcess::NotRunning
) {
139 if (processHasStarted
)
140 qFatal("git-fast-import has been started once and crashed?");
141 processHasStarted
= true;
144 QString outputFile
= name
;
145 outputFile
.replace('/', '_');
146 outputFile
.prepend("log-");
147 fastImport
.setStandardOutputFile(outputFile
, QIODevice::Append
);
148 fastImport
.setProcessChannelMode(QProcess::MergedChannels
);
151 fastImport
.start("git-fast-import", QStringList());
153 fastImport
.start("/bin/cat", QStringList());
158 Repository::Transaction::~Transaction()
162 void Repository::Transaction::setAuthor(const QByteArray
&a
)
167 void Repository::Transaction::setDateTime(uint dt
)
172 void Repository::Transaction::setLog(const QByteArray
&l
)
177 void Repository::Transaction::deleteFile(const QString
&path
)
179 deletedFiles
.append(path
);
182 QIODevice
*Repository::Transaction::addFile(const QString
&path
, int mode
, qint64 length
)
186 fp
.mark
= ++lastmark
;
189 repository
->fastImport
.write("blob\nmark :");
190 repository
->fastImport
.write(QByteArray::number(fp
.mark
));
191 repository
->fastImport
.write("\ndata ");
192 repository
->fastImport
.write(QByteArray::number(length
));
193 repository
->fastImport
.write("\n", 1);
196 modifiedFiles
.insert(path
, fp
);
197 return &repository
->fastImport
;
200 void Repository::Transaction::commit()
202 // create the commit message
203 QByteArray message
= log
;
204 if (!message
.endsWith('\n'))
206 message
+= "\nsvn path=" + svnprefix
+ "; revision=" + QByteArray::number(revnum
) + "\n";
209 QByteArray branchRef
= branch
;
210 if (!branchRef
.startsWith("refs/"))
211 branchRef
.prepend("refs/heads/");
213 QTextStream
s(&repository
->fastImport
);
214 s
<< "commit " << branchRef
<< endl
;
215 s
<< "mark :" << revnum
<< endl
;
216 s
<< "committer " << QString::fromUtf8(author
) << ' ' << datetime
<< " -0000" << endl
;
218 Branch
&br
= repository
->branches
[branch
];
220 qWarning() << "Branch" << branch
<< "in repository" << repository
->name
<< "doesn't exist at revision"
221 << revnum
<< "-- did you resume from the wrong revision?";
225 s
<< "data " << message
.length() << endl
;
228 repository
->fastImport
.write(message
);
229 repository
->fastImport
.putChar('\n');
231 // write the file deletions
232 if (deletedFiles
.contains(""))
233 repository
->fastImport
.write("deleteall\n");
235 foreach (QString df
, deletedFiles
)
236 repository
->fastImport
.write("D " + df
.toUtf8() + "\n");
238 // write the file modifications
239 QHash
<QString
, FileProperties
>::ConstIterator it
= modifiedFiles
.constBegin();
240 for ( ; it
!= modifiedFiles
.constEnd(); ++it
) {
241 repository
->fastImport
.write("M ", 2);
242 repository
->fastImport
.write(QByteArray::number(it
->mode
, 8));
243 repository
->fastImport
.write(" :", 2);
244 repository
->fastImport
.write(QByteArray::number(it
->mark
));
245 repository
->fastImport
.write(" ", 1);
246 repository
->fastImport
.write(it
.key().toUtf8());
247 repository
->fastImport
.write("\n", 1);
250 repository
->fastImport
.write("\n");
252 while (repository
->fastImport
.bytesToWrite())
253 if (!repository
->fastImport
.waitForBytesWritten(-1))
254 qFatal("Failed to write to process: %s", qPrintable(repository
->fastImport
.errorString()));