view src/org/tmatesoft/hg/core/Cset.java @ 99:4d7bb95d553b

StatusCommand().add should list copies
author Artem Tikhomirov <tikhomirov.artem@gmail.com>
date Thu, 27 Jan 2011 23:08:51 +0100
parents af1f3b78b918
children a3a2e5deb320
line wrap: on
line source
/*
 * Copyright (c) 2011 TMate Software Ltd
 *  
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * For information on how to redistribute this software under
 * the terms of a license other than GNU General Public License
 * contact TMate Software at support@svnkit.com
 */
package org.tmatesoft.hg.core;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.tmatesoft.hg.core.LogCommand.FileRevision;
import org.tmatesoft.hg.repo.Changeset;
import org.tmatesoft.hg.repo.HgRepository;
import org.tmatesoft.hg.repo.HgStatusCollector;
import org.tmatesoft.hg.util.PathPool;


/**
 * TODO rename to Changeset along with original Changeset moved to .repo and renamed to HgChangeset?
 * Not thread-safe, don't try to read from different threads
 * 
 * @author Artem Tikhomirov
 * @author TMate Software Ltd.
 */
public class Cset implements Cloneable {
	private final HgStatusCollector statusHelper;
	private final PathPool pathHelper;

	//
	private Changeset changeset;
	private Nodeid nodeid;

	//
	private List<FileRevision> modifiedFiles, addedFiles;
	private List<Path> deletedFiles;
	private int revNumber;

	// XXX consider CommandContext with StatusCollector, PathPool etc. Commands optionally get CC through a cons or create new
	// and pass it around
	/*package-local*/Cset(HgStatusCollector statusCollector, PathPool pathPool) {
		statusHelper = statusCollector;
		pathHelper = pathPool;
	}
	
	/*package-local*/
	void init(int localRevNumber, Nodeid nid, Changeset rawChangeset) {
		revNumber = localRevNumber;
		nodeid = nid;
		changeset = rawChangeset;
		modifiedFiles = addedFiles = null;
		deletedFiles = null;
	}
	public int getRevision() {
		return revNumber;
	}
	public Nodeid getNodeid() {
		return nodeid;
	}
	public String getUser() {
		return changeset.user();
	}
	public String getComment() {
		return changeset.comment();
	}
	public String getBranch() {
		return changeset.branch();
	}
	public String getDate() {
		return changeset.dateString();
	}
	public Nodeid getManifestRevision() {
		return changeset.manifest();
	}

	public List<Path> getAffectedFiles() {
		// reports files as recorded in changelog. Note, merge revisions may have no
		// files listed, and thus this method would return empty list, while
		// #getModifiedFiles() would return list with merged file(s) (because it uses status to get 'em, not
		// what #files() gives).
		ArrayList<Path> rv = new ArrayList<Path>(changeset.files().size());
		for (String name : changeset.files()) {
			rv.add(pathHelper.path(name));
		}
		return rv;
	}

	public List<FileRevision> getModifiedFiles() {
		if (modifiedFiles == null) {
			initFileChanges();
		}
		return modifiedFiles;
	}

	public List<FileRevision> getAddedFiles() {
		if (addedFiles == null) {
			initFileChanges();
		}
		return addedFiles;
	}

	public List<Path> getRemovedFiles() {
		if (deletedFiles == null) {
			initFileChanges();
		}
		return deletedFiles;
	}

	@Override
	public Cset clone() {
		try {
			Cset copy = (Cset) super.clone();
			copy.changeset = changeset.clone();
			return copy;
		} catch (CloneNotSupportedException ex) {
			throw new InternalError(ex.toString());
		}
	}

	private /*synchronized*/ void initFileChanges() {
		ArrayList<Path> deleted = new ArrayList<Path>();
		ArrayList<FileRevision> modified = new ArrayList<FileRevision>();
		ArrayList<FileRevision> added = new ArrayList<FileRevision>();
		HgStatusCollector.Record r = new HgStatusCollector.Record();
		statusHelper.change(revNumber, r);
		final HgRepository repo = statusHelper.getRepo();
		for (Path s : r.getModified()) {
			Nodeid nid = r.nodeidAfterChange(s);
			if (nid == null) {
				throw new IllegalArgumentException();
			}
			modified.add(new FileRevision(repo, nid, s));
		}
		for (Path s : r.getAdded()) {
			Nodeid nid = r.nodeidAfterChange(s);
			if (nid == null) {
				throw new IllegalArgumentException();
			}
			added.add(new FileRevision(repo, nid, s));
		}
		for (Path s : r.getRemoved()) {
			deleted.add(s);
		}
		modified.trimToSize();
		added.trimToSize();
		deleted.trimToSize();
		modifiedFiles = Collections.unmodifiableList(modified);
		addedFiles = Collections.unmodifiableList(added);
		deletedFiles = Collections.unmodifiableList(deleted);
	}
}