123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306 |
- /*
- * Copyright (C) 2009-2010, Google Inc.
- * and other copyright owners as documented in the project's IP log.
- *
- * This program and the accompanying materials are made available
- * under the terms of the Eclipse Distribution License v1.0 which
- * accompanies this distribution, is reproduced below, and is
- * available at http://www.eclipse.org/org/documents/edl-v10.php
- *
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or
- * without modification, are permitted provided that the following
- * conditions are met:
- *
- * - Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * - Redistributions in binary form must reproduce the above
- * copyright notice, this list of conditions and the following
- * disclaimer in the documentation and/or other materials provided
- * with the distribution.
- *
- * - Neither the name of the Eclipse Foundation, Inc. nor the
- * names of its contributors may be used to endorse or promote
- * products derived from this software without specific prior
- * written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
- * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
- * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
- * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
- * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
- * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
- * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
- * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
- * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
-
- package org.eclipse.jgit.pgm.debug;
-
- import java.io.BufferedReader;
- import java.io.File;
- import java.io.FileInputStream;
- import java.io.IOException;
- import java.io.InputStreamReader;
- import java.text.MessageFormat;
- import java.util.ArrayList;
- import java.util.Date;
- import java.util.HashMap;
- import java.util.List;
- import java.util.ListIterator;
- import java.util.Map;
-
- import org.eclipse.jgit.errors.MissingObjectException;
- import org.eclipse.jgit.errors.ObjectWritingException;
- import org.eclipse.jgit.lib.CommitBuilder;
- import org.eclipse.jgit.lib.Constants;
- import org.eclipse.jgit.lib.ObjectId;
- import org.eclipse.jgit.lib.ObjectIdRef;
- import org.eclipse.jgit.lib.ObjectInserter;
- import org.eclipse.jgit.lib.PersonIdent;
- import org.eclipse.jgit.lib.ProgressMonitor;
- import org.eclipse.jgit.lib.Ref;
- import org.eclipse.jgit.lib.RefUpdate;
- import org.eclipse.jgit.lib.RefWriter;
- import org.eclipse.jgit.lib.TextProgressMonitor;
- import org.eclipse.jgit.pgm.CLIText;
- import org.eclipse.jgit.pgm.TextBuiltin;
- import org.eclipse.jgit.revwalk.RevWalk;
- import org.eclipse.jgit.storage.file.LockFile;
- import org.kohsuke.args4j.Argument;
- import org.kohsuke.args4j.Option;
-
- /**
- * Recreates a repository from another one's commit graph.
- * <p>
- * <b>Do not run this on a repository unless you want to destroy it.</b>
- * <p>
- * To create the input files, in the source repository use:
- *
- * <pre>
- * git for-each-ref >in.refs
- * git log --all '--pretty=format:%H %ct %P' >in.dag
- * </pre>
- * <p>
- * Run the rebuild in either an empty repository, or a clone of the source. Any
- * missing commits (which might be the entire graph) will be created. All refs
- * will be modified to match the input exactly, which means some refs may be
- * deleted from the current repository.
- * <p>
- */
- class RebuildCommitGraph extends TextBuiltin {
- private static final String REALLY = "--destroy-this-repository";
-
- @Option(name = REALLY, usage = "usage_approveDestructionOfRepository")
- boolean really;
-
- @Argument(index = 0, required = true, metaVar = "metaVar_refs", usage = "usage_forEachRefOutput")
- File refList;
-
- @Argument(index = 1, required = true, metaVar = "metaVar_refs", usage = "usage_logAllPretty")
- File graph;
-
- private final ProgressMonitor pm = new TextProgressMonitor();
-
- private Map<ObjectId, ObjectId> rewrites = new HashMap<ObjectId, ObjectId>();
-
- @Override
- protected void run() throws Exception {
- if (!really && !db.getAllRefs().isEmpty()) {
- System.err.println(
- MessageFormat.format(CLIText.get().fatalThisProgramWillDestroyTheRepository
- , db.getDirectory().getAbsolutePath(), REALLY));
- throw die(CLIText.get().needApprovalToDestroyCurrentRepository);
- }
- if (!refList.isFile())
- throw die(MessageFormat.format(CLIText.get().noSuchFile, refList.getPath()));
- if (!graph.isFile())
- throw die(MessageFormat.format(CLIText.get().noSuchFile, graph.getPath()));
-
- recreateCommitGraph();
- detachHead();
- deleteAllRefs();
- recreateRefs();
- }
-
- private void recreateCommitGraph() throws IOException {
- final RevWalk rw = new RevWalk(db);
- final Map<ObjectId, ToRewrite> toRewrite = new HashMap<ObjectId, ToRewrite>();
- List<ToRewrite> queue = new ArrayList<ToRewrite>();
- final BufferedReader br = new BufferedReader(new InputStreamReader(
- new FileInputStream(graph), Constants.CHARSET));
- try {
- String line;
- while ((line = br.readLine()) != null) {
- final String[] parts = line.split("[ \t]{1,}");
- final ObjectId oldId = ObjectId.fromString(parts[0]);
- try {
- rw.parseCommit(oldId);
- // We have it already. Don't rewrite it.
- continue;
- } catch (MissingObjectException mue) {
- // Fall through and rewrite it.
- }
-
- final long time = Long.parseLong(parts[1]) * 1000L;
- final ObjectId[] parents = new ObjectId[parts.length - 2];
- for (int i = 0; i < parents.length; i++) {
- parents[i] = ObjectId.fromString(parts[2 + i]);
- }
-
- final ToRewrite t = new ToRewrite(oldId, time, parents);
- toRewrite.put(oldId, t);
- queue.add(t);
- }
- } finally {
- br.close();
- }
-
- pm.beginTask("Rewriting commits", queue.size());
- final ObjectInserter oi = db.newObjectInserter();
- final ObjectId emptyTree = oi.insert(Constants.OBJ_TREE, new byte[] {});
- final PersonIdent me = new PersonIdent("jgit rebuild-commitgraph",
- "rebuild-commitgraph@localhost");
- while (!queue.isEmpty()) {
- final ListIterator<ToRewrite> itr = queue
- .listIterator(queue.size());
- queue = new ArrayList<ToRewrite>();
- REWRITE: while (itr.hasPrevious()) {
- final ToRewrite t = itr.previous();
- final ObjectId[] newParents = new ObjectId[t.oldParents.length];
- for (int k = 0; k < t.oldParents.length; k++) {
- final ToRewrite p = toRewrite.get(t.oldParents[k]);
- if (p != null) {
- if (p.newId == null) {
- // Must defer until after the parent is rewritten.
- queue.add(t);
- continue REWRITE;
- } else {
- newParents[k] = p.newId;
- }
- } else {
- // We have the old parent object. Use it.
- //
- newParents[k] = t.oldParents[k];
- }
- }
-
- final CommitBuilder newc = new CommitBuilder();
- newc.setTreeId(emptyTree);
- newc.setAuthor(new PersonIdent(me, new Date(t.commitTime)));
- newc.setCommitter(newc.getAuthor());
- newc.setParentIds(newParents);
- newc.setMessage("ORIGINAL " + t.oldId.name() + "\n");
- t.newId = oi.insert(newc);
- rewrites.put(t.oldId, t.newId);
- pm.update(1);
- }
- }
- oi.flush();
- oi.release();
- pm.endTask();
- }
-
- private static class ToRewrite {
- final ObjectId oldId;
-
- final long commitTime;
-
- final ObjectId[] oldParents;
-
- ObjectId newId;
-
- ToRewrite(final ObjectId o, final long t, final ObjectId[] p) {
- oldId = o;
- commitTime = t;
- oldParents = p;
- }
- }
-
- private void detachHead() throws IOException {
- final String head = db.getFullBranch();
- final ObjectId id = db.resolve(Constants.HEAD);
- if (!ObjectId.isId(head) && id != null) {
- final LockFile lf;
- lf = new LockFile(new File(db.getDirectory(), Constants.HEAD), db.getFS());
- if (!lf.lock())
- throw new IOException(MessageFormat.format(CLIText.get().cannotLock, Constants.HEAD));
- lf.write(id);
- if (!lf.commit())
- throw new IOException(CLIText.get().cannotDeatchHEAD);
- }
- }
-
- private void deleteAllRefs() throws Exception {
- final RevWalk rw = new RevWalk(db);
- for (final Ref r : db.getAllRefs().values()) {
- if (Constants.HEAD.equals(r.getName()))
- continue;
- final RefUpdate u = db.updateRef(r.getName());
- u.setForceUpdate(true);
- u.delete(rw);
- }
- }
-
- private void recreateRefs() throws Exception {
- final Map<String, Ref> refs = computeNewRefs();
- new RefWriter(refs.values()) {
- @Override
- protected void writeFile(final String name, final byte[] content)
- throws IOException {
- final File file = new File(db.getDirectory(), name);
- final LockFile lck = new LockFile(file, db.getFS());
- if (!lck.lock())
- throw new ObjectWritingException(MessageFormat.format(CLIText.get().cantWrite, file));
- try {
- lck.write(content);
- } catch (IOException ioe) {
- throw new ObjectWritingException(MessageFormat.format(CLIText.get().cantWrite, file));
- }
- if (!lck.commit())
- throw new ObjectWritingException(MessageFormat.format(CLIText.get().cantWrite, file));
- }
- }.writePackedRefs();
- }
-
- private Map<String, Ref> computeNewRefs() throws IOException {
- final RevWalk rw = new RevWalk(db);
- final Map<String, Ref> refs = new HashMap<String, Ref>();
- final BufferedReader br = new BufferedReader(new InputStreamReader(
- new FileInputStream(refList), Constants.CHARSET));
- try {
- String line;
- while ((line = br.readLine()) != null) {
- final String[] parts = line.split("[ \t]{1,}");
- final ObjectId origId = ObjectId.fromString(parts[0]);
- final String type = parts[1];
- final String name = parts[2];
-
- ObjectId id = rewrites.get(origId);
- if (id == null)
- id = origId;
- try {
- rw.parseAny(id);
- } catch (MissingObjectException mue) {
- if (!Constants.TYPE_COMMIT.equals(type)) {
- System.err.println(MessageFormat.format(CLIText.get().skippingObject, type, name));
- continue;
- }
- throw new MissingObjectException(id, type);
- }
- refs.put(name, new ObjectIdRef.Unpeeled(Ref.Storage.PACKED,
- name, id));
- }
- } finally {
- rw.release();
- br.close();
- }
- return refs;
- }
- }
|